#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.

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.

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

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.

46{this,"PRD_MultiTruthCollections", {}, "PRD multi truth collection names this builder is working on"};

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

Public Member Functions

Protected Member Functions

Protected Attributes