JiveXML::TruthTrackRetriever Class Reference

Retrieves the McEventCollection or the TrackRecordCollection in simulated cosmics. More...

#include <TruthTrackRetriever.h>

Inheritance diagram for JiveXML::TruthTrackRetriever:

Collaboration diagram for JiveXML::TruthTrackRetriever:

Public Member Functions
	TruthTrackRetriever (const std::string &type, const std::string &name, const IInterface *parent)
	Standard Constructor. More...
virtual StatusCode	retrieve (ToolHandle< IFormatTool > &FormatTool)
	Retrieve all the data. More...
virtual std::string	dataTypeName () const
	Return the name of the data type. More...
StatusCode	initialize ()
	Default AthAlgTool methods. More...

Private Attributes
const std::string	m_typeName
	The data type that is generated by this retriever. More...
std::string	m_McEvtCollName
	Storegate key for the McEventCollection (different between RDO/ESD and AOD) More...
double	m_MinPtCut
	Minimum pT for a particle to get accepted. More...
double	m_MinRhoCut
	Minium radius of the end-vertex for the particle to get accepted. More...

Detailed Description

Retrieves the McEventCollection or the TrackRecordCollection in simulated cosmics.

• Properties
  • StoreGateKey = TruthEvent : Storegate key for the McEventCollection (different between RDO/ESD and AOD)
  • UnstableMinPtCut = 100*MeV Minimum pT for a particle to get accepted.
  • UnstableMinRhoCut = 40*mm Minium radius of the end-vertex for the particle to get accepted.
• Retrieved Data
  • code : the PDG ID of the particle
  • id : the particle barcode
  • pt : transverse momentum
  • eta, phi : \(\eta\) and \(\phi\) of the momentum vector
  • rhoVertex,phiVertex,zVertex : position of the production vertex in \(\rho\), \(\phi\) and \(z\)
  • rhoEndVertex,phiEndVertex,zEndVertex : position of the end-vertex in \(\rho\), \(\phi\) and \(z\)

Definition at line 41 of file TruthTrackRetriever.h.

Constructor & Destructor Documentation

TruthTrackRetriever()

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

Standard Constructor.

This is the standard AthAlgTool constructor.

Parameters

type	AlgTool type name
name	AlgTool instance name
parent	AlgTools parent owning this tool

Definition at line 25 of file TruthTrackRetriever.cxx.

                                                                                                              :
    base_class(type,name,parent),
    m_typeName("STr")
  {
    declareProperty("StoreGateKey", m_McEvtCollName = "TruthEvent", "Name of the McEventCollection");
    declareProperty("UnstableMinPtCut",     m_MinPtCut = 100*Gaudi::Units::MeV, "Minimum pT for an unstable particle to get accepted");
    declareProperty("UnstableMinRhoCut",    m_MinRhoCut = 40*Gaudi::Units::mm, "Minium radius of the end-vertex for unstable particle to get accepted");
 
  }

Member Function Documentation

dataTypeName()

virtual std::string JiveXML::TruthTrackRetriever::dataTypeName ( ) const

inlinevirtual

Return the name of the data type.

Definition at line 52 of file TruthTrackRetriever.h.

{ return m_typeName; };

initialize()

StatusCode JiveXML::TruthTrackRetriever::initialize

(

)

Default AthAlgTool methods.

Initialize before event loop.

Definition at line 38 of file TruthTrackRetriever.cxx.

                                            {
    //Nothing to do here
    return StatusCode::SUCCESS;
  }

retrieve()

StatusCode JiveXML::TruthTrackRetriever::retrieve ( ToolHandle< IFormatTool > &  FormatTool )

virtual

Retrieve all the data.

Loop over all true particles and get their basic parameters.

Parameters

FormatTool

the tool that will create formated output from the DataMap

Definition at line 47 of file TruthTrackRetriever.cxx.

                                                                              {
 
    //be verbose
    if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG) << "Retrieving " << dataTypeName() << endmsg; 
 
    //Retrieve the collection
    const McEventCollection* McEvtColl = NULL;
    if ( !evtStore()->contains<McEventCollection>( m_McEvtCollName )){ 
      if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG) << "Could not find McEventCollection " << m_McEvtCollName << endmsg;
      return StatusCode::SUCCESS;
    }
    if( evtStore()->retrieve(McEvtColl, m_McEvtCollName).isFailure() ){
      if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG) << "Could not retrieve McEventCollection " << m_McEvtCollName << endmsg;
      return StatusCode::SUCCESS;
    }
    
    // Calculate the size
    long NParticles=0;
    McEventCollection::const_iterator McEvtCollItr = McEvtColl->begin(); 
    for ( ; McEvtCollItr != McEvtColl->end(); ++McEvtCollItr)
      NParticles +=  (*McEvtCollItr)->particles_size();
 
    //Show in verbose mode
    if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG) <<  "Total number of particles in McEventCollection \""
                                            << m_McEvtCollName << "\" is " << NParticles << endmsg;
 
    //Reserve space for the output data
    DataVect pt; pt.reserve(NParticles);
    DataVect phi; phi.reserve(NParticles);
    DataVect eta; eta.reserve(NParticles);
    DataVect rhoVertex; rhoVertex.reserve(NParticles);
    DataVect phiVertex; phiVertex.reserve(NParticles);
    DataVect zVertex; zVertex.reserve(NParticles);
    DataVect code; code.reserve(NParticles);
    DataVect id; id.reserve(NParticles);
    DataVect rhoEndVertex; rhoEndVertex.reserve(NParticles);
    DataVect phiEndVertex; phiEndVertex.reserve(NParticles);
    DataVect zEndVertex; zEndVertex.reserve(NParticles);
 
      
    //Now loop events and retrieve
    for ( McEvtCollItr = McEvtColl->begin(); McEvtCollItr != McEvtColl->end(); ++McEvtCollItr){
 
      //Loop over particles in the event
#ifdef HEPMC3
      const auto &barcodes = (*McEvtCollItr)->attribute<HepMC::GenEventBarcodes> ("barcodes");
      std::map<int,int> id_to_barcode_map;
      if (barcodes) id_to_barcode_map = barcodes->id_to_barcode_map();
#endif      
      for (const auto& particle:  *(*McEvtCollItr) ) {
        
        //Additional cuts for decaying particles
        if ( particle->end_vertex() ) {
          //Reject particles that fail the pt cut
          if ( particle->momentum().perp() < m_MinPtCut) continue ; 
          //Reject particles that fail the minimum end-vertex cut
          if (particle->end_vertex()->position().perp() < m_MinRhoCut ) continue ;
        }
 
        //Get basic parameters (eta, phi, pt, ...)
        pt.push_back(DataType(particle->momentum().perp()/Gaudi::Units::GeV));
        float thePhi = particle->momentum().phi();
        phi.push_back(DataType( (thePhi<0) ? thePhi+=2*M_PI : thePhi ));
        eta.push_back(DataType( particle->momentum().pseudoRapidity() ));
        code.push_back(DataType( particle->pdg_id() ));
#ifdef HEPMC3
        id.push_back(DataType( id_to_barcode_map.at(particle->id() )));
#else
        id.push_back(DataType( HepMC::barcode(*particle) ));
#endif
 
        // Get the vertex information
        const auto& vertexprod =  particle->production_vertex();
        if (vertexprod) {
          const auto& pos=vertexprod->position();
          rhoVertex.push_back(DataType( std::sqrt(pos.x()*pos.x()+pos.y()*pos.y()+pos.z()*pos.z())*Gaudi::Units::mm/Gaudi::Units::cm ));
          float vtxPhi = pos.phi();
          phiVertex.push_back(DataType( (vtxPhi<0)? vtxPhi+=2*M_PI : vtxPhi ));
          zVertex.push_back(DataType( pos.z()*Gaudi::Units::mm/Gaudi::Units::cm )); 
        } else {
          rhoVertex.push_back(DataType( 0. ));
          phiVertex.push_back(DataType( 0. ));
          zVertex.push_back(DataType( 0. )); 
        }
        //Do the same for the end vertex
        const auto& vertexend =  particle->end_vertex();
        if ( vertexend ) {
         const auto& pos=vertexend->position();
         rhoEndVertex.push_back(DataType(std::sqrt(pos.x()*pos.x()+pos.y()*pos.y()+pos.z()*pos.z())*Gaudi::Units::mm/Gaudi::Units::cm));
         float vtxPhi = pos.phi();
         phiEndVertex.push_back(DataType( (vtxPhi<0)? vtxPhi+=2*M_PI : vtxPhi ));
         zEndVertex.push_back(DataType(pos.z()*Gaudi::Units::mm/Gaudi::Units::cm)); 
        } else {
         rhoEndVertex.push_back(DataType( 0. ));
         phiEndVertex.push_back(DataType( 0. ));
         zEndVertex.push_back(DataType( 0. )); 
        }
      }
    }
 
    DataMap myDataMap;
    myDataMap["pt"] = pt;
    myDataMap["phi"] = phi;
    myDataMap["eta"] = eta;
    myDataMap["code"] = code;
    myDataMap["id"] = id;
    myDataMap["rhoVertex"] = rhoVertex;
    myDataMap["phiVertex"] = phiVertex;
    myDataMap["zVertex"] = zVertex;
    myDataMap["rhoEndVertex"] = rhoEndVertex;
    myDataMap["phiEndVertex"] = phiEndVertex;
    myDataMap["zEndVertex"] = zEndVertex;
 
    //Be verbose
    if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG) << dataTypeName() << ": "<< pt.size() << endmsg;
 
    //forward data to formating tool
    return FormatTool->AddToEvent(dataTypeName(), m_McEvtCollName, &myDataMap);
  }

Member Data Documentation

m_McEvtCollName

std::string JiveXML::TruthTrackRetriever::m_McEvtCollName

private

Storegate key for the McEventCollection (different between RDO/ESD and AOD)

Definition at line 62 of file TruthTrackRetriever.h.

m_MinPtCut

double JiveXML::TruthTrackRetriever::m_MinPtCut

private

Minimum pT for a particle to get accepted.

Definition at line 64 of file TruthTrackRetriever.h.

m_MinRhoCut

double JiveXML::TruthTrackRetriever::m_MinRhoCut

private

Minium radius of the end-vertex for the particle to get accepted.

Definition at line 66 of file TruthTrackRetriever.h.

m_typeName

const std::string JiveXML::TruthTrackRetriever::m_typeName

private

The data type that is generated by this retriever.

Definition at line 59 of file TruthTrackRetriever.h.

---

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

• TruthTrackRetriever.h
• TruthTrackRetriever.cxx