#include <TrackListCombiner.h>

Inheritance diagram for TrackListCombiner:

Classes
struct	Tags

Public Member Functions
virtual void	produce (edm::StreamID, edm::Event &ev, const edm::EventSetup &es) const override

	TrackListCombiner (const edm::ParameterSet &ps)

	~TrackListCombiner ()

Public Member Functions inherited from edm::global::EDProducer<>
	EDProducer ()=default

Public Member Functions inherited from edm::global::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

Private Attributes
std::vector< Tags >	trackProducers

Additional Inherited Members
Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::global::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 14 of file TrackListCombiner.h.

Constructor & Destructor Documentation

TrackListCombiner::TrackListCombiner ( const edm::ParameterSet & ps )

explicit

Definition at line 16 of file TrackListCombiner.cc.

References edm::ParameterSet::getParameter(), parseEventContent::prod, AlCaHLTBitMon_QueryRunRegistry::string, and preDuplicateMergingDisplacedTracks_cfi::trackProducers.

 {
   for(const std::string& prod: ps.getParameter<vector<string> >("trackProducers")) {
     trackProducers.emplace_back(consumes<vector<Trajectory>>(prod),
                                 consumes<TrajTrackAssociationCollection>(prod));
   }
 
   produces<reco::TrackCollection>();
   produces<reco::TrackExtraCollection>();
   produces<TrackingRecHitCollection>();
   produces<vector<Trajectory> >();
   produces<TrajTrackAssociationCollection>();
 }

TrackListCombiner::~TrackListCombiner ( )

Definition at line 31 of file TrackListCombiner.cc.

32 {

33 }

Member Function Documentation

void TrackListCombiner::produce	(	edm::StreamID	,
		edm::Event &	ev,
		const edm::EventSetup &	es
	)		const

overridevirtual

Definition at line 36 of file TrackListCombiner.cc.

References patPFMETCorrections_cff::algo, edm::AssociationMap< Tag >::begin(), edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, reco::TrackBase::detachedTripletStep, edm::AssociationMap< Tag >::end(), edm::Event::getByToken(), hfClusterShapes_cfi::hits, mps_fire::i, diffTreeTool::index, reco::Track::innerDetId(), reco::Track::innerMomentum(), reco::Track::innerOk(), reco::Track::innerPosition(), reco::Track::innerStateCovariance(), edm::Ref< C, T, F >::key(), tablePrinter::labels, LogTrace, reco::TrackBase::lowPtTripletStep, edm::ProductLabels::module, eostools::move(), nTracks(), reco::Track::outerDetId(), reco::Track::outerMomentum(), reco::Track::outerOk(), reco::Track::outerPosition(), reco::Track::outerStateCovariance(), reco::TrackBase::pixelPairStep, edm::Event::put(), reco::Track::recHit(), reco::Track::recHitsSize(), reco::Track::seedDirection(), reco::Track::seedRef(), reco::TrackBase::setAlgorithm(), edm::AssociationMap< Tag >::size(), findElectronsInSiStrips_cfi::trackProducer, preDuplicateMergingDisplacedTracks_cfi::trackProducers, and reco::TrackBase::undefAlgorithm.

 {
   auto recoTracks = std::make_unique<reco::TrackCollection>();
   auto recoTrackExtras = std::make_unique<reco::TrackExtraCollection>();
   auto recoHits = std::make_unique<TrackingRecHitCollection>();
   auto recoTrajectories = std::make_unique<vector<Trajectory>>();
   auto recoTrajTrackMap = std::make_unique<TrajTrackAssociationCollection>();
 
   LogTrace("MinBiasTracking")
     << "[TrackListCombiner]";
 
   // Go through all track producers
   int i = 1;
   for(auto trackProducer = trackProducers.begin();
                            trackProducer!= trackProducers.end();
                            trackProducer++, i++)
   {
     reco::TrackBase::TrackAlgorithm algo;
     switch(i) 
     {
       case 1:  algo = reco::TrackBase::lowPtTripletStep; break;
       case 2:  algo = reco::TrackBase::pixelPairStep; break;
       case 3:  algo = reco::TrackBase::detachedTripletStep; break;
       default: algo = reco::TrackBase::undefAlgorithm;
     }
 
     edm::Handle<vector<Trajectory> > theTrajectoryCollection;
     edm::Handle<TrajTrackAssociationCollection> theAssoMap;  
 
     ev.getByToken(trackProducer->trajectory, theTrajectoryCollection);
     ev.getByToken(trackProducer->assoMap, theAssoMap);
 
 #ifdef EDM_ML_DEBUG
     edm::EDConsumerBase::Labels labels;
     labelsForToken(trackProducer->trajectory, labels);
 
     LogTrace("MinBiasTracking")
       << " [TrackListCombiner] " << labels.module
       << " : " << theAssoMap->size();
 #endif
 
     
     // The track collection iterators
     TrajTrackAssociationCollection::const_iterator anAssociation;  
     TrajTrackAssociationCollection::const_iterator lastAssociation;
     anAssociation = theAssoMap->begin();
     lastAssociation = theAssoMap->end();
 
     // Build the map of correspondance between reco tracks and sim tracks
     for ( ; anAssociation != lastAssociation; ++anAssociation )
     { 
       edm::Ref<vector<Trajectory> > aTrajectoryRef = anAssociation->key;
       reco::TrackRef aTrackRef = anAssociation->val;
       
       // A copy of the track
       reco::Track aRecoTrack(*aTrackRef);
 
       // Set algorithm
       aRecoTrack.setAlgorithm(algo);
 
       recoTracks->push_back(aRecoTrack);      
 
       // A copy of the hits
       unsigned nh = aRecoTrack.recHitsSize();
       for(unsigned ih=0; ih<nh; ++ih)
       {
         TrackingRecHit *hit = aRecoTrack.recHit(ih)->clone();
         recoHits->push_back(hit);
       }
       
       // A copy of the trajectories
       recoTrajectories->push_back(*aTrajectoryRef);
       
     }
   }
 
   LogTrace("MinBiasTracking")
     << " [TrackListCombiner] allTracks : " << recoTracks->size()
                                     << "|" << recoTrajectories->size();
 
   // Save the tracking recHits
   edm::OrphanHandle<TrackingRecHitCollection> theRecoHits = ev.put(std::move(recoHits));
   
   edm::RefProd<TrackingRecHitCollection> theRecoHitsProd(theRecoHits);
   // Create the track extras and add the references to the rechits
   unsigned hits = 0;
   unsigned nTracks = recoTracks->size();
   recoTrackExtras->reserve(nTracks); // To save some time at push_back
   for(unsigned index = 0; index < nTracks; ++index )
   { 
     reco::Track& aTrack = recoTracks->at(index);
     reco::TrackExtra aTrackExtra(aTrack.outerPosition(),
                                  aTrack.outerMomentum(),
                                  aTrack.outerOk(),
                                  aTrack.innerPosition(),
                                  aTrack.innerMomentum(),
                                  aTrack.innerOk(),
                                  aTrack.outerStateCovariance(),
                                  aTrack.outerDetId(),
                                  aTrack.innerStateCovariance(),
                                  aTrack.innerDetId(),
                                  aTrack.seedDirection(),
                                  aTrack.seedRef());
     
     unsigned nHits = aTrack.recHitsSize();
     aTrackExtra.setHits(theRecoHitsProd,hits,nHits);
     hits +=nHits;
 
     recoTrackExtras->push_back(aTrackExtra);
   }
   
   // Save the track extras
   edm::OrphanHandle<reco::TrackExtraCollection> theRecoTrackExtras =
     ev.put(std::move(recoTrackExtras));
   
   // Add the reference to the track extra in the tracks
   for(unsigned index = 0; index<nTracks; ++index)
   { 
     const reco::TrackExtraRef theTrackExtraRef(theRecoTrackExtras,index);
     (recoTracks->at(index)).setExtra(theTrackExtraRef);
   }
   
   // Save the tracks
   edm::OrphanHandle<reco::TrackCollection> theRecoTracks = ev.put(std::move(recoTracks));
   
   // Save the trajectories
   edm::OrphanHandle<vector<Trajectory> > theRecoTrajectories =
     ev.put(std::move(recoTrajectories));
   
   // Create and set the trajectory/track association map 
   for(unsigned index = 0; index<nTracks; ++index)
   { 
     edm::Ref<vector<Trajectory> > trajRef( theRecoTrajectories, index );
     edm::Ref<reco::TrackCollection>    tkRef( theRecoTracks, index );
     recoTrajTrackMap->insert(trajRef,tkRef);
   }
   
   // Save the association map
   ev.put(std::move(recoTrajTrackMap));
 }

Member Data Documentation

std::vector<Tags> TrackListCombiner::trackProducers

private

Definition at line 28 of file TrackListCombiner.h.

Classes

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation