DerivationFramework::TruthClassificationDecorator Class Reference

#include <TruthClassificationDecorator.h>

Inheritance diagram for DerivationFramework::TruthClassificationDecorator:

Collaboration diagram for DerivationFramework::TruthClassificationDecorator:

Public Member Functions
	TruthClassificationDecorator (const std::string &t, const std::string &n, const IInterface *p)
	~TruthClassificationDecorator ()
StatusCode	initialize ()
StatusCode	finalize ()
virtual StatusCode	addBranches () const

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_linkDecoratorKey {this, "originalTruthParticle", m_particlesKey, "originalTruthParticle", "Link to the original truth particle"}
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"}
ToolHandle< IMCTruthClassifier >	m_classifier

Detailed Description

Definition at line 21 of file TruthClassificationDecorator.h.

Constructor & Destructor Documentation

TruthClassificationDecorator()

DerivationFramework::TruthClassificationDecorator::TruthClassificationDecorator	(	const std::string &	t,
		const std::string &	n,
		const IInterface *	p
	)

Definition at line 19 of file TruthClassificationDecorator.cxx.

                                                                                        :
base_class(t,n,p),
m_ntotpart(0),
m_classifier("MCTruthClassifier/MCTruthClassifier")
{
    declareProperty("MCTruthClassifier", m_classifier);
}

~TruthClassificationDecorator()

DerivationFramework::TruthClassificationDecorator::~TruthClassificationDecorator

(

)

Definition at line 30 of file TruthClassificationDecorator.cxx.

                                                                               {
}

Member Function Documentation

addBranches()

StatusCode DerivationFramework::TruthClassificationDecorator::addBranches ( ) const

virtual

Definition at line 60 of file TruthClassificationDecorator.cxx.

{
    
    // Event context for multi-threading
    const EventContext& ctx = Gaudi::Hive::currentContext();
 
    // 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, ElementLink<xAOD::TruthParticleContainer> > linkDecorator(m_linkDecoratorKey, ctx);
    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

(

)

Definition at line 52 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

(

)

Definition at line 34 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_linkDecoratorKey.initialize());    
    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 42 of file TruthClassificationDecorator.h.

m_classifier

ToolHandle<IMCTruthClassifier> DerivationFramework::TruthClassificationDecorator::m_classifier

private

Definition at line 44 of file TruthClassificationDecorator.h.

m_linkDecoratorKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthClassificationDecorator::m_linkDecoratorKey {this, "originalTruthParticle", m_particlesKey, "originalTruthParticle", "Link to the original truth particle"}

private

Definition at line 34 of file TruthClassificationDecorator.h.

m_ntotpart

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

mutableprivate

Definition at line 30 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 36 of file TruthClassificationDecorator.h.

m_outcomeDecoratorKey

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

private

Definition at line 40 of file TruthClassificationDecorator.h.

m_particlesKey

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

private

Definition at line 31 of file TruthClassificationDecorator.h.

m_typeDecoratorKey

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

private

Definition at line 38 of file TruthClassificationDecorator.h.

---

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

• TruthClassificationDecorator.h
• TruthClassificationDecorator.cxx