TruthParticleCnvTool Class Reference

#include <TruthParticleCnvTool.h>

Inheritance diagram for TruthParticleCnvTool:

Collaboration diagram for TruthParticleCnvTool:

Public Member Functions
	TruthParticleCnvTool (const std::string &type, const std::string &name, const IInterface *parent)
	Constructor with parameters: More...
virtual	~TruthParticleCnvTool ()
	Destructor: More...
virtual StatusCode	initialize () override
	Athena Algorithm's Hooks. More...
virtual StatusCode	execute () const override
virtual StatusCode	execute (const EventContext &ctx) const override
StatusCode	convert (const McEventCollection *mcEvts, const unsigned int genEvtIndex, TruthParticleContainer *mcParts, const ITruthParticleVisitor *visitor) const override
	Converts a McEventCollection into an TruthParticleContainer (ie: converts it into an AOD compliant collection). More...
void	setDataType (const int type)
	Inline methods: More...
void	setVxCandidateName (const std::string &name)

Protected Types
typedef ToolHandle< ITruthIsolationTool >	IsolTool_t

Protected Attributes
BooleanProperty	m_doEtIsolation
	Switch to compute or not the Et-isolations for TruthParticle (and their underlying HepMC::GenParticle). More...
IsolTool_t	m_isolationTool
	Pointer to the ITruthIsolationTool to be able to retrieve the previously computed transverse energy isolations. More...

Private Attributes
ParticleDataType::DataType	m_dataType
	Type of data we are dealing with (Full/Fast/Truth/...) More...
int	m_dataType_int
std::string	m_vxCandidatesName
SG::ReadHandleKey< McEventCollection >	m_mcEventsReadHandleKey {this,"McEvents","GEN_AOD","ReadHandleKey for the input McEventCollection, from which the TruthParticles will be made from"}
	ReadHandleKey for the McEventCollection the TruthParticles will be made from. More...
SG::WriteHandleKey< TruthParticleContainer >	m_mcPartsOutputWriteHandleKey {this,"TruthParticlesOutput","SpclMC","WriteHandleKey for the output truth particles (built from the McEventCollection)"}
	Output TruthParticle WriteHandleKey (built from the McEventCollection) More...
const HepPDT::ParticleDataTable *	m_pdt
	Particle Property service. More...
PileUpClassification::PileuType_t	m_selectSignalType
	Type of truth particle we want to create. More...
int	m_selectSignalTypeProp
	Type of truth particle we want to create (property to be set by jobO) More...

Detailed Description

AlgTool that converts, on the fly, on demand, each GenParticle to a TruthParticle and builds the TruthParticleContainer. So the GEN_AOD and TruthParticleContainer are identical in contents, only the interfaces are different: the TruthParticle class has the same interface as other AOD particles. Hence the analysis tools can operate on all the AOD particles (ie: including Mc Truth). When the AOD is created, the TruthParticleContainer is empty. When the user requests the TruthParticleContainer in his/her analysis code, a custom converter is triggered where by the GEN_AOD is convertered into the TruthParticleContainer using the TruthParticleCnvTool.

In pile-up events, the many GenEvent are grouped into 5 TruthParticleContainer according to the classification of : https://twiki.cern.ch/twiki/bin/viewauth/Atlas/PileupDigitization#Arrangement_of_Truth_Information

Definition at line 59 of file TruthParticleCnvTool.h.

Member Typedef Documentation

IsolTool_t

typedef ToolHandle<ITruthIsolationTool> TruthParticleCnvTool::IsolTool_t

protected

Definition at line 149 of file TruthParticleCnvTool.h.

Constructor & Destructor Documentation

TruthParticleCnvTool()

TruthParticleCnvTool::TruthParticleCnvTool	(	const std::string &	type,
		const std::string &	name,
		const IInterface *	parent
	)

Constructor with parameters:

Public methods:

Constructors

Definition at line 48 of file TruthParticleCnvTool.cxx.

                                                   : 
  base_class( type, name, parent ),
  m_dataType         ( ParticleDataType::True ),
  m_vxCandidatesName ( ),
  m_pdt              ( nullptr ),
  m_selectSignalTypeProp ( 0 )
{
  //
  // Property declaration
  // 
  
  declareProperty( "DataType", 
           m_dataType_int = 4,
           "Type of data we are dealing with (Full/Fast/Truth...)" );
 
  declareProperty( "TruthIsolationTool",
           m_isolationTool = IsolTool_t( "TruthIsolationTool" ),
           "Pointer to the TruthIsolationTool to be able to compute "
           "transverse energy isolations for various isolation cones cuts. "
           "See McParticleEvent/TruthParticleParameters.h for cone cuts." );
 
 
  declareProperty( "DoEtIsolations",
           m_doEtIsolation = false,
           "Switch to compute or not the Et-isolations for TruthParticle "
           "(and their underlying @c HepMC::GenParticle).\n"
           "Default is to not compute these Et-isolations (and save CPU)." );
 
  setDataType( m_dataType_int );
 
  declareProperty( "SelectSignalType",
           m_selectSignalTypeProp,
           "Switch to select different type of signal\n"
           " hard scatter               0\n"
           " hard scatter plus minbias  1\n" 
           " only minbias              -1\n" 
           "Default is hard scatter only." );
 
  declareInterface<ITruthParticleCnvTool>(this);
}

~TruthParticleCnvTool()

TruthParticleCnvTool::~TruthParticleCnvTool ( )

virtual

Destructor:

Destructor.

Definition at line 93 of file TruthParticleCnvTool.cxx.

{ 
  ATH_MSG_DEBUG("Calling destructor");
}

Member Function Documentation

convert()

StatusCode TruthParticleCnvTool::convert ( const McEventCollection *  mcEvts, const unsigned int  genEvtIndex, TruthParticleContainer *  mcParts, const ITruthParticleVisitor *  visitor  ) const

override

Converts a McEventCollection into an TruthParticleContainer (ie: converts it into an AOD compliant collection).

Const methods:

@in mcEvts the McEventCollection holding the HepMC::GenEvent we want to convert into a TruthParticleContainer @in genEvtIndex the index to the HepMC::GenEvent to be converted @out mcParts a valid pointer to a TruthParticleContainer which will be filled with adaptors to HepMC::GenParticles.

Get GenEvent from McEventCollection

Create a map to enhance access between GenParticles and TruthParticles

Definition at line 224 of file TruthParticleCnvTool.cxx.

{
  ATH_MSG_DEBUG("Converting McEventCollection to TruthParticleContainer");
 
  if ( nullptr == m_pdt ) {
    ATH_MSG_ERROR("Could not convert McEventCollection into "\
          "TruthParticleContainer if NO ParticleDataTable is "\
          "available !!");
    return StatusCode::FAILURE;
  }
  
  if ( nullptr == mcCollection ) {
    ATH_MSG_WARNING("Null pointer to McEventCollection !");
    return StatusCode::RECOVERABLE;
  }
 
  if ( mcCollection->size() <= genEventIndex ) {
    ATH_MSG_WARNING("McEventCollection size: " << mcCollection->size() 
            << endmsg
            << "Requested element nbr : " << genEventIndex << " !!");
    return StatusCode::RECOVERABLE;
  }
 
  IProxyDict* sg = SG::CurrentEventStore::store();
 
  ATH_MSG_DEBUG("Retrieve the GenEvent from given McEventCollection");
  const HepMC::GenEvent * evt = (*mcCollection)[genEventIndex];
  container->setGenEvent( mcCollection, genEventIndex, sg );
 
  // reserve enough space for the container so we don't have to relocate it
  container->reserve( container->size() + evt->particles_size() );
 
  TruthParticleContainer::Map_t bcToMcPart = container->m_particles;
 
  
#ifdef HEPMC3
  // Process particles in barcode order.
  auto bcmapatt = evt->attribute<HepMC::GenEventBarcodes>("barcodes");
  if (!bcmapatt) ATH_MSG_ERROR("TruthParticleCnvTool.cxx: Event does not contain barcodes attribute"); 
  std::map<int, HepMC3::ConstGenParticlePtr> bcmap = bcmapatt->barcode_to_particle_map();
  for (const auto &[bc,hepMcPart]: bcmap) {
#else
  for (auto hepMcPart: *evt) {
    int bc = HepMC::barcode(hepMcPart);
#endif
 
    TruthParticle * mcPart = new TruthParticle( hepMcPart, container );
    container->push_back( mcPart );
    
    if ( visitor ) {
      visitor->visit( mcPart );
    }
 
    mcPart->setCharge( MC::charge( int(mcPart->pdgId())) );
    mcPart->setGenEventIndex( genEventIndex);
 
    if ( hepMcPart != mcPart->genParticle() ) {
      ATH_MSG_ERROR("TruthParticle is not wrapping the GenParticle : " 
            << hepMcPart << " !!");
    }
    HepMcParticleLink mcLink( bc, genEventIndex, HepMcParticleLink::IS_POSITION, HepMcParticleLink::IS_BARCODE, sg ); // FIXME barcode-based
    bcToMcPart[ mcLink.compress() ] = mcPart;
 
  }//> end loop over particles
 
  // at this point the whole GenEvent has been proxied.
  // so we can setup its VectorMap
  container->setParticles( bcToMcPart );
 
  // connect the TruthParticleContainer to the container of TruthEtIsolations
  // if it exists and if we are asked for
  if ( m_doEtIsolation.value() ) {
    const std::string& etIsolName 
      = m_isolationTool->etIsolationsName( container->genEventName() );
    if ( etIsolName.empty() ) {
      ATH_MSG_WARNING("Could not retrieve the name of the "
              "TruthEtIsolations container (requested: [" 
              << container->genEventName() << "])");
      return StatusCode::RECOVERABLE;
    }
 
    const TruthEtIsolationsContainer* etIsols = nullptr;
    if ( !evtStore()->retrieve( etIsols, etIsolName ).isSuccess() ) {
      ATH_MSG_WARNING("Could not retrieve the TruthEtIsolations container at ["
              << etIsolName << "] !!");
      return StatusCode::RECOVERABLE;
    }
 
    // index of HepMC::GenEvent within the McEventCollection is the same
    // than the one of the TruthEtIsolations within the TruthEtIsolationsCont.
    container->setEtIsolations( etIsols, genEventIndex );
  }
 
  return StatusCode::SUCCESS;
}

execute() [1/2]

StatusCode TruthParticleCnvTool::execute ( ) const

overridevirtual

Definition at line 146 of file TruthParticleCnvTool.cxx.

{
  return execute (Gaudi::Hive::currentContext());
}

execute() [2/2]

StatusCode TruthParticleCnvTool::execute ( const EventContext &  ctx ) const

overridevirtual

Definition at line 152 of file TruthParticleCnvTool.cxx.

{  
  ATH_MSG_DEBUG("Executing " << name() << "...");
 
  //Setup WriteHandle, and record new TruthParticleContainer
  SG::WriteHandle<TruthParticleContainer> mcPartsOutputWriteHandle(m_mcPartsOutputWriteHandleKey, ctx);
  ATH_CHECK(mcPartsOutputWriteHandle.record(std::make_unique<TruthParticleContainer>()));  
  
  if (!mcPartsOutputWriteHandle.isValid()){
    ATH_MSG_ERROR("Invalid WriteHandle for TruthParticleContainer with key: " << m_mcPartsOutputWriteHandleKey.key());
    return StatusCode::FAILURE;
  }
 
  //Setup ReadHandle for input McEventCollection
  SG::ReadHandle<McEventCollection> mcEventsReadHandle(m_mcEventsReadHandleKey, ctx);
 
  if (!mcEventsReadHandle.isValid()){
     ATH_MSG_WARNING("Invalid ReadHandle for McEventCollection with key ["
             << m_mcEventsReadHandleKey.key() << "] !!"
             << endmsg
             << "TruthParticleContainer [" << m_mcPartsOutputWriteHandleKey.key()
             << "] will be EMPTY !!");
     return StatusCode::RECOVERABLE;
  }
  
  ATH_MSG_DEBUG(" Found McEventCollection of size = "<< mcEventsReadHandle->size() );
  // do the actual convertion (it is merely an interface adaptation now...)
  //  selection for particles from minbias copied from the JetsFromTruthTool
 
  std::size_t genEventIndex = 0;
  const ITruthParticleVisitor* dummyVisitor = nullptr;
  bool all_good = true;
 
  McEventCollection::const_iterator fEvt = mcEventsReadHandle->begin();
  McEventCollection::const_iterator lEvt = mcEventsReadHandle->end();
  PileUpClassification::findEventIterators(m_selectSignalType, fEvt,lEvt);
 
  ATH_MSG_DEBUG(" Found McEventCollection iterators : "<< (fEvt-mcEventsReadHandle->begin()) << " to "<< (lEvt-mcEventsReadHandle->begin()) );
 
  for(McEventCollection::const_iterator it=fEvt ; it != lEvt; ++it){
    const HepMC::GenEvent* evt = *it;
    // there are holes in a McEventCollection when pile-up is enabled
    // so deal with it
    if (nullptr == evt) {
      continue;
    }
    genEventIndex = (it - mcEventsReadHandle->begin());
    ATH_MSG_DEBUG(" adding event id="<< HepMC::signal_process_id(evt)<<"  genEventIndex="<< genEventIndex );
 
    if( HepMC::signal_process_id(evt) == 0 ) continue;
    if (!this->convert( mcEventsReadHandle.ptr(), genEventIndex, mcPartsOutputWriteHandle.ptr(), dummyVisitor ).isSuccess()) {
      ATH_MSG_DEBUG("Failed to convert an event...");
      all_good = false;
    }
  }
 
  // VERY IMPORTANT ! Reset the index to the first GenEvent included
  // in this TruthParticleContainer. This will be used when read back from disk.
  mcPartsOutputWriteHandle->setGenEvent( mcEventsReadHandle.ptr(), (fEvt - mcEventsReadHandle->begin() ) );
 
  
  return all_good 
    ? StatusCode::SUCCESS
    : StatusCode::RECOVERABLE;
}

initialize()

StatusCode TruthParticleCnvTool::initialize ( )

overridevirtual

Athena Algorithm's Hooks.

Definition at line 100 of file TruthParticleCnvTool.cxx.

{
  ATH_MSG_INFO("Initializing " << name() << "...");
 
  // Get the Particle Properties Service
  ServiceHandle<IPartPropSvc> partPropSvc("PartPropSvc", name());
  if ( !partPropSvc.retrieve().isSuccess() ) {
    ATH_MSG_ERROR(" Could not initialize Particle Properties Service");
    return StatusCode::FAILURE;
  }      
 
  m_pdt = partPropSvc->PDT();
  if ( nullptr == m_pdt ) {
    ATH_MSG_ERROR("Could not retrieve HepPDT::ParticleDataTable from "\
          "ParticleProperties Service !!");
    return StatusCode::FAILURE;
  }
 
  // retrieve the TruthIsolation tool only if asked for.
  if ( m_doEtIsolation.value() ) {
    ATH_CHECK(m_isolationTool.retrieve());
  }
  else {
    m_isolationTool.disable();
  }
 
  //initialize DataHandleKeys
  ATH_CHECK(m_mcPartsOutputWriteHandleKey.initialize());
  ATH_CHECK(m_mcEventsReadHandleKey.initialize());
 
  // for compatibility with earlier version we accept <0 values :
  if(m_selectSignalTypeProp<0) m_selectSignalType = PileUpClassification::ALLMINBIAS;
  else m_selectSignalType = (PileUpClassification::PileuType_t) m_selectSignalTypeProp;
  
 
  ATH_MSG_INFO
    (" DoEtIsolations:       [" << std::boolalpha 
     << m_doEtIsolation.value() << "]" << endmsg
     << " McEvents:             [" << m_mcEventsReadHandleKey.key() << "]" << endmsg
     << " TruthParticlesOutput: [" << m_mcPartsOutputWriteHandleKey.key() << "]"
     << endmsg
     << " SelectSignalType:     [" << m_selectSignalType << "]");
 
  return StatusCode::SUCCESS;
}

setDataType()

void TruthParticleCnvTool::setDataType ( const int  type )

inline

Inline methods:

Definition at line 161 of file TruthParticleCnvTool.h.

{ 
  m_dataType = ParticleDataType::DataType(type); 
}

setVxCandidateName()

void TruthParticleCnvTool::setVxCandidateName ( const std::string &  name )

inline

Definition at line 166 of file TruthParticleCnvTool.h.

{ 
  m_vxCandidatesName = name; 
}

Member Data Documentation

m_dataType

ParticleDataType::DataType TruthParticleCnvTool::m_dataType

private

Type of data we are dealing with (Full/Fast/Truth/...)

Definition at line 110 of file TruthParticleCnvTool.h.

m_dataType_int

int TruthParticleCnvTool::m_dataType_int

private

Definition at line 111 of file TruthParticleCnvTool.h.

m_doEtIsolation

BooleanProperty TruthParticleCnvTool::m_doEtIsolation

protected

Switch to compute or not the Et-isolations for TruthParticle (and their underlying HepMC::GenParticle).

Default is to not compute these Et-isolations (and save CPU)

Definition at line 147 of file TruthParticleCnvTool.h.

m_isolationTool

IsolTool_t TruthParticleCnvTool::m_isolationTool

protected

Pointer to the ITruthIsolationTool to be able to retrieve the previously computed transverse energy isolations.

Definition at line 154 of file TruthParticleCnvTool.h.

m_mcEventsReadHandleKey

SG::ReadHandleKey<McEventCollection> TruthParticleCnvTool::m_mcEventsReadHandleKey {this,"McEvents","GEN_AOD","ReadHandleKey for the input McEventCollection, from which the TruthParticles will be made from"}

private

ReadHandleKey for the McEventCollection the TruthParticles will be made from.

Definition at line 117 of file TruthParticleCnvTool.h.

m_mcPartsOutputWriteHandleKey

SG::WriteHandleKey<TruthParticleContainer> TruthParticleCnvTool::m_mcPartsOutputWriteHandleKey {this,"TruthParticlesOutput","SpclMC","WriteHandleKey for the output truth particles (built from the McEventCollection)"}

private

Output TruthParticle WriteHandleKey (built from the McEventCollection)

Definition at line 121 of file TruthParticleCnvTool.h.

m_pdt

const HepPDT::ParticleDataTable* TruthParticleCnvTool::m_pdt

private

Particle Property service.

Definition at line 125 of file TruthParticleCnvTool.h.

m_selectSignalType

PileUpClassification::PileuType_t TruthParticleCnvTool::m_selectSignalType

private

Type of truth particle we want to create.

Definition at line 129 of file TruthParticleCnvTool.h.

m_selectSignalTypeProp

int TruthParticleCnvTool::m_selectSignalTypeProp

private

Type of truth particle we want to create (property to be set by jobO)

Definition at line 133 of file TruthParticleCnvTool.h.

m_vxCandidatesName

std::string TruthParticleCnvTool::m_vxCandidatesName

private

Definition at line 113 of file TruthParticleCnvTool.h.

---

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

• TruthParticleCnvTool.h
• TruthParticleCnvTool.cxx