R3BNeulandMultiplicityCheat.cxx

Go to the documentation of this file.

#include "R3BNeulandMultiplicityCheat.h"
#include "FairRootManager.h"
 
R3BNeulandMultiplicityCheat::R3BNeulandMultiplicityCheat(std::string_view input, std::string_view output)
    : FairTask("R3BNeulandMultiplicityCheat")
    , fPrimaryHits(input)
    , fMultiplicity(std::make_unique<R3BNeulandMultiplicity>())
    , multiplicity_ptr_{ fMultiplicity.get() }
    , fOutputName(output)
{
}
 
InitStatus R3BNeulandMultiplicityCheat::Init()
{
    // Input
    fPrimaryHits.init();
 
    // Output
    auto* ioman = FairRootManager::Instance();
    if (ioman == nullptr)
    {
        throw R3B::runtime_error("TCAInputConnector: No FairRootManager");
    }
    ioman->RegisterAny(fOutputName.c_str(), multiplicity_ptr_, true);
 
    return kSUCCESS;
}
 
void R3BNeulandMultiplicityCheat::Exec(Option_t*)
{
    fMultiplicity->m.fill(0.);
    fMultiplicity->m[fPrimaryHits.size()] = 1.;
}
 
ClassImp(R3BNeulandMultiplicityCheat)