#include <TruthGuidedProtoTrackCreatorTool.h>

Inheritance diagram for ActsTrk::TruthGuidedProtoTrackCreatorTool:

Collaboration diagram for ActsTrk::TruthGuidedProtoTrackCreatorTool:

Public Member Functions
	TruthGuidedProtoTrackCreatorTool (const std::string &type, const std::string &name, const IInterface *parent)

virtual	~TruthGuidedProtoTrackCreatorTool ()=default

virtual StatusCode	initialize () override

virtual StatusCode	findProtoTracks (const EventContext &ctx, const xAOD::PixelClusterContainer &pixelContainer, const xAOD::StripClusterContainer &stripContainer, std::vector< ActsTrk::ProtoTrack > &foundProtoTracks) const override final
	EF-style pattern recognition to create prototracks. More...

Protected Member Functions
std::unique_ptr< Acts::BoundTrackParameters >	makeDummyParams (const HepMC::ConstGenParticlePtr &truthParticle) const
	creates a random, dummy set of parameters Warning: This is not a real parameter estimate. More...

Protected Attributes
SG::ReadHandleKeyArray< PRD_MultiTruthCollection >	m_prdMultiTruthCollectionNames {this,"PRD_MultiTruthCollections", {}, "PRD multi truth collection names this builder is working on"}
	Truth track collection. More...

Detailed Description

Definition at line 16 of file TruthGuidedProtoTrackCreatorTool.h.

Constructor & Destructor Documentation

◆ TruthGuidedProtoTrackCreatorTool()

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

Definition at line 11 of file TruthGuidedProtoTrackCreatorTool.cxx.

14 : base_class(type,name,parent)

15 {}

◆ ~TruthGuidedProtoTrackCreatorTool()

virtual ActsTrk::TruthGuidedProtoTrackCreatorTool::~TruthGuidedProtoTrackCreatorTool ( )

virtualdefault

Member Function Documentation

◆ findProtoTracks()

StatusCode ActsTrk::TruthGuidedProtoTrackCreatorTool::findProtoTracks	(	const EventContext &	ctx,
		const xAOD::PixelClusterContainer &	pixelContainer,
		const xAOD::StripClusterContainer &	stripContainer,
		std::vector< ActsTrk::ProtoTrack > &	foundProtoTracks
	)		const

finaloverridevirtual

EF-style pattern recognition to create prototracks.

Parameters

ctx	Event context
pixelContainer	pixel cluster
stripContainer	sct cluster
foundProtoTracks	vector to hold the found proto tracks - will be populated by the method. Method will not discard existing content

Definition at line 23 of file TruthGuidedProtoTrackCreatorTool.cxx.

                                                                                               {
  
     // Read the PRD information
     std::vector<const PRD_MultiTruthCollection*> prdMultiTruthCollections;        
     prdMultiTruthCollections.reserve(m_prdMultiTruthCollectionNames.size());
     // load the PRD collections from SG
     for(const auto& pmtCollNameIter:m_prdMultiTruthCollectionNames)
     {
      // try to retrieve the PRD multi truth collection
      SG::ReadHandle<PRD_MultiTruthCollection> curColl (pmtCollNameIter, ctx);
      if (!curColl.isValid())
      {
        ATH_MSG_WARNING("Could not retrieve " << pmtCollNameIter << ". Ignoring ... ");
      }
      else
      {
        ATH_MSG_INFO("Added " << pmtCollNameIter << " to collection list for truth track creation.");
        prdMultiTruthCollections.push_back(curColl.cptr());
      }
    }
  
     // create the map for the ine
     std::map<Identifier, HepMC::ConstGenParticlePtr> identToHepMCMap;
     for (auto & PRD_truthCollec: prdMultiTruthCollections )
     {
         // loop over the map and get the identifier, GenParticle relation
         PRD_MultiTruthCollection::const_iterator prdMtCIter  = PRD_truthCollec->begin();
         PRD_MultiTruthCollection::const_iterator prdMtCIterE = PRD_truthCollec->end();
         for ( ; prdMtCIter != prdMtCIterE; ++ prdMtCIter ){
  
             // check if entry exists and if   
 #ifdef HEPMC3
             HepMC::ConstGenParticlePtr curGenP       = (*prdMtCIter).second.scptr();
 #else
 //AV Looks like an implicit conversion
             HepMC::ConstGenParticlePtr curGenP       = (*prdMtCIter).second;
 #endif
             Identifier                curIdentifier = (*prdMtCIter).first;
  
             // Min pT cut
             if ( curGenP->momentum().perp() < 500. ) continue;
  
  
  
             identToHepMCMap[curIdentifier] = curGenP;
         }
     }
  
     // Now loop over the pixel and strip container and make collectiong
     std::map<HepMC::ConstGenParticlePtr, std::vector<ActsTrk::ATLASUncalibSourceLink>> trackCollections;
  
     for(const auto cluster: pixelContainer)
     {
         // Get the idetifier list for the RDOs
         auto identifierList = cluster->rdoList();
  
         // Loop and push back the cluster in the corresponding trith particle
         for(auto& id: identifierList)
         {
             // Found a match, so push it into the track collection
             if(identToHepMCMap.find(id) != identToHepMCMap.end())
             {
                 auto truthParticle = identToHepMCMap.at(id);
                 trackCollections[truthParticle].emplace_back(makeATLASUncalibSourceLink(&pixelContainer, cluster, ctx));
             }
         }
     }
  
     for(const auto cluster: stripContainer)
     {
         // Get the idetifier list for the RDOs
         auto identifierList = cluster->rdoList();
  
         // Loop and push back the cluster in the corresponding trith particle
         for(auto& id: identifierList)
         {
             // Found a match, so push it into the track collection
             if(identToHepMCMap.find(id) != identToHepMCMap.end())
             {
                 auto truthParticle = identToHepMCMap.at(id);
                 trackCollections[truthParticle].emplace_back(makeATLASUncalibSourceLink(&stripContainer, cluster, ctx));
             }
         }
     }
  
     for(const auto& var: trackCollections)
     {
         // Skip if we find less than 3 clusters per truth track
         if(var.second.size() < 3) continue;
  
         // Make the intput perigee
         auto inputPerigee = makeDummyParams(var.first);
         ATH_MSG_INFO("Found " << var.second.size() << " clusters for truth partcle "<<var.first);
         foundProtoTracks.push_back({var.second,std::move(inputPerigee)});
     }
  
     // and add to the list (will only make one prototrack per event for now)
  
     return StatusCode::SUCCESS;
 }

◆ initialize()

StatusCode ActsTrk::TruthGuidedProtoTrackCreatorTool::initialize ( )

overridevirtual

Definition at line 17 of file TruthGuidedProtoTrackCreatorTool.cxx.

 {
     ATH_CHECK( m_prdMultiTruthCollectionNames.initialize() );
     return StatusCode::SUCCESS;
 }

◆ makeDummyParams()

std::unique_ptr< Acts::BoundTrackParameters > ActsTrk::TruthGuidedProtoTrackCreatorTool::makeDummyParams ( const HepMC::ConstGenParticlePtr & truthParticle ) const

protected

creates a random, dummy set of parameters Warning: This is not a real parameter estimate.

Should only serve as a placeholder. Use with care

Parameters

truthParticle Input truth particle to get the parameters from

Returns: a set of puesdo-dummy params

Definition at line 129 of file TruthGuidedProtoTrackCreatorTool.cxx.

                                                                                                             {
  
   using namespace Acts::UnitLiterals;
   std::shared_ptr<const Acts::Surface> actsSurface = Acts::Surface::makeShared<Acts::PerigeeSurface>(
         Acts::Vector3(0., 0., 0.));
   Acts::BoundVector params;
  
  
   // No, this is not a physically correct parameter estimate! 
   // We just want a placeholder to point in roughly the expected direction... 
   // A real track finder would do something more reasonable here. 
   params << 0., 0.,
         truthParticle->momentum().phi(), truthParticle->momentum().theta(),
         static_cast<float>(::charge(truthParticle)) / (truthParticle->momentum().e()), 0.;
  
  
   // Covariance - let's be honest and say we have no clue ;-) 
   Acts::BoundSquareMatrix cov = Acts::BoundSquareMatrix::Identity();
   cov *= 100000; 
  
   // some ACTS paperwork 
   Trk::ParticleHypothesis hypothesis = Trk::pion;
   float mass = Trk::ParticleMasses::mass[hypothesis] * Acts::UnitConstants::MeV;
   Acts::PdgParticle absPdg = Acts::makeAbsolutePdgParticle(Acts::ePionPlus);
   Acts::ParticleHypothesis actsHypothesis{
     absPdg, mass, Acts::AnyCharge{static_cast<float>(::charge(truthParticle))}};
  
   return std::make_unique<Acts::BoundTrackParameters>(actsSurface, params,
                                     cov, actsHypothesis);
  
 }

Member Data Documentation

◆ m_prdMultiTruthCollectionNames

SG::ReadHandleKeyArray<PRD_MultiTruthCollection> ActsTrk::TruthGuidedProtoTrackCreatorTool::m_prdMultiTruthCollectionNames {this,"PRD_MultiTruthCollections", {}, "PRD multi truth collection names this builder is working on"}

protected

Truth track collection.

Definition at line 46 of file TruthGuidedProtoTrackCreatorTool.h.

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

Public Member Functions

Protected Member Functions

Protected Attributes