DerivationFramework::TruthClassificationDecorator Class Reference

#include <TruthClassificationDecorator.h>

Inheritance diagram for DerivationFramework::TruthClassificationDecorator:

Collaboration diagram for DerivationFramework::TruthClassificationDecorator:

Public Member Functions
virtual StatusCode	initialize () override final
virtual StatusCode	finalize () override final
virtual StatusCode	addBranches (const EventContext &ctx) const override final

Private Attributes
std::atomic< unsigned int >	m_ntotpart {}
SG::ReadHandleKey< xAOD::TruthParticleContainer >	m_particlesKey {this, "ParticlesKey", "TruthParticles", "ReadHandleKey for input TruthParticleContainer"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_originDecoratorKey {this, "classifierParticleOrigin", m_particlesKey, "classifierParticleOrigin", "Particle origin decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_typeDecoratorKey {this, "classifierParticleType", m_particlesKey, "classifierParticleType", "Particle type decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_outcomeDecoratorKey {this, "classifierParticleOutCome", m_particlesKey, "classifierParticleOutCome", "Particle outcome decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_classificationDecoratorKey {this, "Classification", m_particlesKey, "Classification", "Classification code decorator"}
PublicToolHandle< IMCTruthClassifier >	m_classifier {this, "MCTruthClassifier", "MCTruthClassifier/MCTruthClassifier"}

Detailed Description

Definition at line 21 of file TruthClassificationDecorator.h.

Member Function Documentation

addBranches()

StatusCode DerivationFramework::TruthClassificationDecorator::addBranches ( const EventContext & ctx ) const

finaloverridevirtual

Definition at line 44 of file TruthClassificationDecorator.cxx.

{
 
    // Event context for multi-threading
 
    // Retrieve truth collections
    SG::ReadHandle<xAOD::TruthParticleContainer> truthParticles(m_particlesKey,ctx);
    if (!truthParticles.isValid()) {
        ATH_MSG_ERROR("Couldn't retrieve TruthParticle collection with name " << m_particlesKey);
        return StatusCode::FAILURE;
    }
 
    unsigned int nParticles = truthParticles->size();
    m_ntotpart += nParticles;
 
    // Set up decorators
    SG::WriteDecorHandle<xAOD::TruthParticleContainer, unsigned int> originDecorator(m_originDecoratorKey, ctx);
    SG::WriteDecorHandle<xAOD::TruthParticleContainer, unsigned int> typeDecorator(m_typeDecoratorKey, ctx);
    SG::WriteDecorHandle<xAOD::TruthParticleContainer, unsigned int> outcomeDecorator(m_outcomeDecoratorKey, ctx);
    SG::WriteDecorHandle<xAOD::TruthParticleContainer, unsigned int> classificationDecorator(m_classificationDecoratorKey, ctx);
 
    for (unsigned int i=0; i<nParticles; ++i) {
        MCTruthPartClassifier::Info info;
        auto classification = m_classifier->particleTruthClassifier((*truthParticles)[i], &info);
        unsigned int particleOutCome = info.particleOutCome;
        unsigned int result = (unsigned int)( (*truthParticles)[i] ? std::get<0>(MCTruthPartClassifier::defOrigOfParticle((*truthParticles)[i])) : 0 );
        unsigned int particleType = classification.first;
        unsigned int particleOrigin = classification.second;
        typeDecorator(*((*truthParticles)[i])) = particleType;
        originDecorator(*((*truthParticles)[i])) = particleOrigin;
        outcomeDecorator(*((*truthParticles)[i])) = particleOutCome;
 
        classificationDecorator(*((*truthParticles)[i])) = result;
    }
 
    return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::TruthClassificationDecorator::finalize ( )

finaloverridevirtual

Definition at line 36 of file TruthClassificationDecorator.cxx.

{
    ATH_MSG_VERBOSE("finalize() ...");
    ATH_MSG_INFO("Processed and decorated "<< m_ntotpart <<" truth particles");
    return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::TruthClassificationDecorator::initialize ( )

finaloverridevirtual

Definition at line 19 of file TruthClassificationDecorator.cxx.

{
    ATH_MSG_VERBOSE("initialize() ...");
    ATH_CHECK(m_classifier.retrieve());
 
    ATH_CHECK( m_particlesKey.initialize() );
    ATH_MSG_INFO("Decorating " << m_particlesKey.key() << " with classification information");
 
    // Decorators
    ATH_CHECK(m_originDecoratorKey.initialize());
    ATH_CHECK(m_typeDecoratorKey.initialize());
    ATH_CHECK(m_outcomeDecoratorKey.initialize());
    ATH_CHECK(m_classificationDecoratorKey.initialize());
 
    return StatusCode::SUCCESS;
}

Member Data Documentation

m_classificationDecoratorKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthClassificationDecorator::m_classificationDecoratorKey {this, "Classification", m_particlesKey, "Classification", "Classification code decorator"}

private

Definition at line 41 of file TruthClassificationDecorator.h.

{this, "Classification", m_particlesKey, "Classification", "Classification code decorator"};

m_classifier

PublicToolHandle<IMCTruthClassifier> DerivationFramework::TruthClassificationDecorator::m_classifier {this, "MCTruthClassifier", "MCTruthClassifier/MCTruthClassifier"}

private

Definition at line 43 of file TruthClassificationDecorator.h.

{this, "MCTruthClassifier", "MCTruthClassifier/MCTruthClassifier"};

m_ntotpart

std::atomic<unsigned int> DerivationFramework::TruthClassificationDecorator::m_ntotpart {}

mutableprivate

Definition at line 31 of file TruthClassificationDecorator.h.

{};

m_originDecoratorKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthClassificationDecorator::m_originDecoratorKey {this, "classifierParticleOrigin", m_particlesKey, "classifierParticleOrigin", "Particle origin decoration"}

private

Definition at line 35 of file TruthClassificationDecorator.h.

{this, "classifierParticleOrigin", m_particlesKey, "classifierParticleOrigin", "Particle origin decoration"};

m_outcomeDecoratorKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthClassificationDecorator::m_outcomeDecoratorKey {this, "classifierParticleOutCome", m_particlesKey, "classifierParticleOutCome", "Particle outcome decoration"}

private

Definition at line 39 of file TruthClassificationDecorator.h.

{this, "classifierParticleOutCome", m_particlesKey, "classifierParticleOutCome", "Particle outcome decoration"};

m_particlesKey

SG::ReadHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthClassificationDecorator::m_particlesKey {this, "ParticlesKey", "TruthParticles", "ReadHandleKey for input TruthParticleContainer"}

private

Definition at line 32 of file TruthClassificationDecorator.h.

{this, "ParticlesKey", "TruthParticles", "ReadHandleKey for input TruthParticleContainer"};

m_typeDecoratorKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthClassificationDecorator::m_typeDecoratorKey {this, "classifierParticleType", m_particlesKey, "classifierParticleType", "Particle type decoration"}

private

Definition at line 37 of file TruthClassificationDecorator.h.

{this, "classifierParticleType", m_particlesKey, "classifierParticleType", "Particle type decoration"};

---

The documentation for this class was generated from the following files:

• TruthClassificationDecorator.h
• TruthClassificationDecorator.cxx