#include <HadronOriginDecorator.h>

Inheritance diagram for DerivationFramework::HadronOriginDecorator:

Collaboration diagram for DerivationFramework::HadronOriginDecorator:

Public Member Functions
	HadronOriginDecorator (const std::string &t, const std::string &n, const IInterface *p)

	~HadronOriginDecorator ()

StatusCode	initialize ()

StatusCode	finalize ()

virtual StatusCode	addBranches () const

Private Attributes
std::string	m_TruthEventName

ToolHandle< DerivationFramework::HadronOriginClassifier >	m_Tool

Detailed Description

Definition at line 25 of file HadronOriginDecorator.h.

Constructor & Destructor Documentation

◆ HadronOriginDecorator()

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

Definition at line 11 of file HadronOriginDecorator.cxx.

                                                                                                          :
     base_class(t,n,p),
     m_Tool("")
   {
  
     declareProperty("ToolName",m_Tool);
     declareProperty("TruthEventName",m_TruthEventName="TruthParticles");
   }

◆ ~HadronOriginDecorator()

DerivationFramework::HadronOriginDecorator::~HadronOriginDecorator ( )

Definition at line 20 of file HadronOriginDecorator.cxx.

20 {}

Member Function Documentation

◆ addBranches()

StatusCode DerivationFramework::HadronOriginDecorator::addBranches ( ) const

virtual

Definition at line 37 of file HadronOriginDecorator.cxx.

                                                      {
     const xAOD::TruthParticleContainer* xTruthParticleContainer = nullptr;
     if (evtStore()->retrieve(xTruthParticleContainer,m_TruthEventName).isFailure()) {
       ATH_MSG_WARNING("could not retrieve TruthParticleContainer " <<m_TruthEventName);
       return StatusCode::FAILURE;
     }
  
     std::map<const xAOD::TruthParticle*, DerivationFramework::HadronOriginClassifier::HF_id>  hadronMap=m_Tool->GetOriginMap();
  
     for(xAOD::TruthParticleContainer::const_iterator PItr = xTruthParticleContainer->begin(); PItr!=xTruthParticleContainer->end(); ++PItr){
       int flavortype=6;
       if(hadronMap.find((*PItr))!=hadronMap.end()){
         flavortype= static_cast<int>(hadronMap[(*PItr)]);
       }
       SG::AuxElement::Decorator< int > decoration("TopHadronOriginFlag");
       decoration(**PItr) = flavortype;
     }
  
     return StatusCode::SUCCESS;
   }

◆ finalize()

StatusCode DerivationFramework::HadronOriginDecorator::finalize ( )

Definition at line 33 of file HadronOriginDecorator.cxx.

                                             {
     return StatusCode::SUCCESS;
   }

◆ initialize()

StatusCode DerivationFramework::HadronOriginDecorator::initialize ( )

Definition at line 22 of file HadronOriginDecorator.cxx.

                                               {
     ATH_MSG_INFO("Initialize " );
  
     if(m_Tool.retrieve().isFailure()){
       ATH_MSG_ERROR("unable to retrieve the tool " <<m_Tool);
       return StatusCode::FAILURE;
     }
  
     return StatusCode::SUCCESS;
   }

Member Data Documentation

◆ m_Tool

ToolHandle<DerivationFramework::HadronOriginClassifier> DerivationFramework::HadronOriginDecorator::m_Tool

private

Definition at line 41 of file HadronOriginDecorator.h.

◆ m_TruthEventName

std::string DerivationFramework::HadronOriginDecorator::m_TruthEventName