#include <TrackCandidateProducer.h>

Inheritance diagram for TrackCandidateProducer:

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

	TrackCandidateProducer (const edm::ParameterSet &conf)

virtual	~TrackCandidateProducer ()

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

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

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

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

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

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

virtual	~ProducerBase ()

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

	EDConsumerBase ()

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

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

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

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

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

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Attributes
unsigned int	maxNumberOfCrossedLayers

unsigned int	minNumberOfCrossedLayers

std::string	propagatorLabel

edm::EDGetTokenT < SiTrackerGSMatchedRecHit2DCollection >	recHitToken

bool	rejectOverlaps

edm::EDGetTokenT< edm::View < TrajectorySeed > >	seedToken

edm::InputTag	simTracks_

edm::EDGetTokenT < edm::SimTrackContainer >	simTrackToken

edm::EDGetTokenT < edm::SimVertexContainer >	simVertexToken

bool	splitHits

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T...>	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T...>	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT < LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	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::stream::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 36 of file TrackCandidateProducer.h.

Constructor & Destructor Documentation

TrackCandidateProducer::TrackCandidateProducer ( const edm::ParameterSet & conf )

explicit

Definition at line 43 of file TrackCandidateProducer.cc.

References edm::ParameterSet::getParameter(), minNumberOfCrossedLayers, propagatorLabel, recHitToken, rejectOverlaps, seedToken, simTrackToken, simVertexToken, splitHits, and AlCaHLTBitMon_QueryRunRegistry::string.

 {  
   // products
   produces<TrackCandidateCollection>();
   
   // general parameters
   minNumberOfCrossedLayers = conf.getParameter<unsigned int>("MinNumberOfCrossedLayers");
   rejectOverlaps = conf.getParameter<bool>("OverlapCleaning");
   splitHits = conf.getParameter<bool>("SplitHits");
 
   // input tags, labels, tokens
   edm::InputTag simTrackLabel = conf.getParameter<edm::InputTag>("simTracks");
   simVertexToken = consumes<edm::SimVertexContainer>(simTrackLabel);
   simTrackToken = consumes<edm::SimTrackContainer>(simTrackLabel);
 
   edm::InputTag seedLabel = conf.getParameter<edm::InputTag>("src");
   seedToken = consumes<edm::View<TrajectorySeed> >(seedLabel);
 
   edm::InputTag recHitLabel = conf.getParameter<edm::InputTag>("recHits");
   recHitToken = consumes<SiTrackerGSMatchedRecHit2DCollection>(recHitLabel);
   
   propagatorLabel = conf.getParameter<std::string>("propagator");
 }

virtual TrackCandidateProducer::~TrackCandidateProducer ( )

inlinevirtual

Definition at line 42 of file TrackCandidateProducer.h.

42 {;}

Member Function Documentation

void TrackCandidateProducer::produce	(	edm::Event &	e,
		const edm::EventSetup &	es
	)

overridevirtual

Implements edm::stream::EDProducerBase.

Definition at line 68 of file TrackCandidateProducer.cc.

References clone(), TrajectorySeed::direction(), edm::OwnVector< T, P >::front(), TrackingRecHit::geographicalId(), edm::EventSetup::get(), edm::Event::getByToken(), cmsHarvester::index, TrajectorySeedHitCandidate::isOnTheSameLayer(), TrajectoryStateOnSurface::isValid(), TrajectorySeedHitCandidate::layerNumber(), TrajectorySeedHitCandidate::localError(), LogDebug, HLT_25ns14e33_v1_cff::magneticField, minNumberOfCrossedLayers, TrajectorySeed::nHits(), oppositeToMomentum, convertSQLitetoXML_cfg::output, trajectoryStateTransform::persistentState(), position, edm::ESHandle< class >::product(), HLT_25ns14e33_v1_cff::propagator, propagatorLabel, edm::OwnVector< T, P >::push_back(), edm::Event::put(), DetId::rawId(), TrajectorySeed::recHits(), HLT_50ns_5e33_v3_cff::recHits, recHitToken, rejectOverlaps, fileCollector::seed, seedToken, simTrackToken, simVertexToken, splitHits, TrajectorySeedHitCandidate::subDetId(), and GeomDet::surface().

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), and HTMLExport.HTMLExportStatic::export().

                                                                     {        
 
   // get services
   edm::ESHandle<MagneticField>          magneticField;
   es.get<IdealMagneticFieldRecord>().get(magneticField);
 
   edm::ESHandle<TrackerGeometry>        trackerGeometry;
   es.get<TrackerDigiGeometryRecord>().get(trackerGeometry);
 
   edm::ESHandle<TrackerTopology>        trackerTopology;
   es.get<TrackerTopologyRcd>().get(trackerTopology);
 
   edm::ESHandle<Propagator>             propagator;
   es.get<TrackingComponentsRecord>().get(propagatorLabel,propagator);
     
   // get products
   edm::Handle<edm::View<TrajectorySeed> > seeds;
   e.getByToken(seedToken,seeds);
 
   edm::Handle<SiTrackerGSMatchedRecHit2DCollection> recHits;
   e.getByToken(recHitToken, recHits);
 
   edm::Handle<edm::SimVertexContainer> simVertices;
   e.getByToken(simVertexToken,simVertices);
 
   edm::Handle<edm::SimTrackContainer> simTracks;
   e.getByToken(simTrackToken,simTracks);
   
   // output collection
   std::auto_ptr<TrackCandidateCollection> output(new TrackCandidateCollection);    
 
   // loop over the seeds
   for (unsigned seednr = 0; seednr < seeds->size(); ++seednr){
     
     const BasicTrajectorySeed seed = seeds->at(seednr);
     if(seed.nHits()==0){
       edm::LogError("TrackCandidateProducer") << "empty trajectory seed in TrajectorySeedCollection" << std::endl;
       return;
     }
 
     // Get all the rechits associated to this track
     int simTrackId =  ((const SiTrackerGSMatchedRecHit2D*) (&*(seed.recHits().first)))->simtrackId();
     SiTrackerGSMatchedRecHit2DCollection::range recHitRange = recHits->get(simTrackId);
     SiTrackerGSMatchedRecHit2DCollection::const_iterator recHitIter = recHitRange.first;
     SiTrackerGSMatchedRecHit2DCollection::const_iterator recHitEnd  = recHitRange.second;
 
     // Count number of crossed layers, apply overlap rejection
     std::vector<TrajectorySeedHitCandidate> recHitCandidates;
     TrajectorySeedHitCandidate recHitCandidate;
     unsigned numberOfCrossedLayers = 0;      
     for ( ; recHitIter != recHitEnd; ++recHitIter) {
       recHitCandidate = TrajectorySeedHitCandidate(&(*recHitIter),trackerGeometry.product(),trackerTopology.product());
       if ( recHitCandidates.size() == 0 || !recHitCandidate.isOnTheSameLayer(recHitCandidates.back()) ) {
     ++numberOfCrossedLayers;
       }
 
       if( recHitCandidates.size() == 0 ||                                                // add the first seeding hit in any case
       !rejectOverlaps ||                                                             // without overlap rejection:   add each hit
       recHitCandidate.subDetId()    != recHitCandidates.back().subDetId() ||         // with overlap rejection:      only add if hits are not on the same layer
       recHitCandidate.layerNumber() != recHitCandidates.back().layerNumber() ){
     recHitCandidates.push_back(recHitCandidate);
       }
       else if ( recHitCandidate.localError() < recHitCandidates.back().localError() ){
     recHitCandidates.back() = recHitCandidate;
       }
     }
     if ( numberOfCrossedLayers < minNumberOfCrossedLayers ) {
       continue;
     }
 
     // Convert TrajectorySeedHitCandidate to TrackingRecHit and split hits
     edm::OwnVector<TrackingRecHit> trackRecHits;
     for ( unsigned index = 0; index<recHitCandidates.size(); ++index ) {
       if(splitHits && recHitCandidates[index].matchedHit()->isMatched()){
     trackRecHits.push_back(recHitCandidates[index].matchedHit()->firstHit()->clone());
     trackRecHits.push_back(recHitCandidates[index].matchedHit()->secondHit()->clone());
       }
       else {
     trackRecHits.push_back(recHitCandidates[index].hit()->clone());
       }
     }
     // reverse order if needed
     // when is this relevant? perhaps for the cases when track finding goes backwards?
     if (seed.direction()==oppositeToMomentum){
       LogDebug("FastTracking")<<"reversing the order of the hits";
       std::reverse(recHitCandidates.begin(),recHitCandidates.end());
     }
     
     // initial track candidate parameters parameters
     int vertexIndex = simTracks->at(simTrackId).vertIndex();
     GlobalPoint  position(simVertices->at(vertexIndex).position().x(),
               simVertices->at(vertexIndex).position().y(),
               simVertices->at(vertexIndex).position().z());
     GlobalVector momentum( simTracks->at(simTrackId).momentum().x() , 
                simTracks->at(simTrackId).momentum().y() , 
                simTracks->at(simTrackId).momentum().z() );
     float        charge   = simTracks->at(simTrackId).charge();
     GlobalTrajectoryParameters initialParams(position,momentum,(int)charge,magneticField.product());
     AlgebraicSymMatrix55 errorMatrix= AlgebraicMatrixID();    
     CurvilinearTrajectoryError initialError(errorMatrix);
     FreeTrajectoryState initialFTS(initialParams, initialError);      
 
     // create track candidate
     const GeomDet* initialLayer = trackerGeometry->idToDet(trackRecHits.front().geographicalId());
     const TrajectoryStateOnSurface initialTSOS = propagator->propagate(initialFTS,initialLayer->surface()) ;
     if (!initialTSOS.isValid()) continue; 
     PTrajectoryStateOnDet PTSOD = trajectoryStateTransform::persistentState(initialTSOS,trackRecHits.front().geographicalId().rawId()); 
     TrackCandidate newTrackCandidate(trackRecHits,seed,PTSOD,edm::RefToBase<TrajectorySeed>(seeds,seednr));
 
     // add track candidate to output collection
     output->push_back(newTrackCandidate);
     
   }
   
   // Save the track candidates
   e.put(output);
 }