#include <PixelTracksProducer.h>

Inheritance diagram for PixelTracksProducer:

Public Member Functions
	PixelTracksProducer (const edm::ParameterSet &conf)

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

	~PixelTracksProducer ()

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
	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
	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

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Attributes
edm::InputTag	seedProducer

const PixelTrackFilter *	theFilter

const PixelFitter *	theFitter

TrackingRegionProducer *	theRegionProducer

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

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

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

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

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 20 of file PixelTracksProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 34 of file PixelTracksProducer.cc.

References edm::EDConsumerBase::consumesCollector(), reco_skim_cfg_mod::filterName, reco::get(), edm::ParameterSet::getParameter(), seedProducer, AlCaHLTBitMon_QueryRunRegistry::string, theFilter, theFitter, and theRegionProducer.

                                                                     : 
   theFitter(0), 
   theFilter(0), 
   theRegionProducer(0)
 {  
 
   produces<reco::TrackCollection>();
   produces<TrackingRecHitCollection>();
   produces<reco::TrackExtraCollection>();
 
   const edm::ParameterSet& regfactoryPSet = conf.getParameter<edm::ParameterSet>("RegionFactoryPSet");
   std::string regfactoryName = regfactoryPSet.getParameter<std::string>("ComponentName");
   theRegionProducer = TrackingRegionProducerFactory::get()->create(regfactoryName,
     regfactoryPSet, consumesCollector());
   
   const edm::ParameterSet& fitterPSet = conf.getParameter<edm::ParameterSet>("FitterPSet");
   std::string fitterName = fitterPSet.getParameter<std::string>("ComponentName");
   theFitter = PixelFitterFactory::get()->create( fitterName, fitterPSet);
   
   edm::ConsumesCollector iC = consumesCollector();
   const edm::ParameterSet& filterPSet = conf.getParameter<edm::ParameterSet>("FilterPSet");
   std::string filterName = filterPSet.getParameter<std::string>("ComponentName");
   theFilter = PixelTrackFilterFactory::get()->create( filterName, filterPSet, iC);
   
   // The name of the seed producer
   seedProducer = conf.getParameter<edm::InputTag>("SeedProducer");
 
 }

PixelTracksProducer::~PixelTracksProducer ( )

Definition at line 65 of file PixelTracksProducer.cc.

References theFilter, theFitter, and theRegionProducer.

                                           {
 
   delete theFilter;
   delete theFitter;
   delete theRegionProducer;
 
 } 

Member Function Documentation

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

virtual

Implements edm::EDProducer.

Definition at line 76 of file PixelTracksProducer.cc.

References reco::TrackExtraBase::add(), statics::clone, edm::Event::getByLabel(), i, gen::k, edm::Event::put(), TrackingRegionProducer::regions(), PixelFitter::run(), seedProducer, reco::TrackBase::setHitPattern(), theFilter, theFitter, theRegionProducer, and testEve_cfg::tracks.

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

                                                                  {        
   
   std::auto_ptr<reco::TrackCollection> tracks(new reco::TrackCollection);    
   std::auto_ptr<TrackingRecHitCollection> recHits(new TrackingRecHitCollection);
   std::auto_ptr<reco::TrackExtraCollection> trackExtras(new reco::TrackExtraCollection);
   typedef std::vector<const TrackingRecHit *> RecHits;
   
   TracksWithRecHits pixeltracks;
   TracksWithRecHits cleanedTracks;
   
   edm::Handle<TrajectorySeedCollection> theSeeds;
   e.getByLabel(seedProducer,theSeeds);
 
   // No seed -> output an empty track collection
   if(theSeeds->size() == 0) {
     e.put(tracks);
     e.put(recHits);
     e.put(trackExtras);
     return;
   }
   
   //only one region Global, but it is called at every event...
   //maybe there is a smarter way to set it only once
   //NEED TO FIX
   typedef std::vector<TrackingRegion* > Regions;
   typedef Regions::const_iterator IR;
   Regions regions = theRegionProducer->regions(e,es);
   for (IR ir=regions.begin(), irEnd=regions.end(); ir < irEnd; ++ir) {
     const TrackingRegion & region = **ir;
     
     // Loop over the seeds
     TrajectorySeedCollection::const_iterator aSeed = theSeeds->begin();
     TrajectorySeedCollection::const_iterator lastSeed = theSeeds->end();
     for ( ; aSeed!=lastSeed; ++aSeed ) { 
       
       // Find the first hit and last hit of the Seed
       TrajectorySeed::range theSeedingRecHitRange = aSeed->recHits();
       edm::OwnVector<TrackingRecHit>::const_iterator aSeedingRecHit = theSeedingRecHitRange.first;
       edm::OwnVector<TrackingRecHit>::const_iterator theLastSeedingRecHit = theSeedingRecHitRange.second;
 
       // Loop over the rechits
       std::vector<const TrackingRecHit*> TripletHits(3,static_cast<const TrackingRecHit*>(0));
       for ( unsigned i=0; aSeedingRecHit!=theLastSeedingRecHit; ++i,++aSeedingRecHit )  
     TripletHits[i] = &(*aSeedingRecHit);
       
       // fitting the triplet
       reco::Track* track = theFitter->run(es, TripletHits, region);
       
       // decide if track should be skipped according to filter 
       if ( ! (*theFilter)(track) ) { 
     delete track; 
     continue; 
       }
       
       // add tracks 
       pixeltracks.push_back(TrackWithRecHits(track, TripletHits));
       
     }
   }
   
   int cc=0;
   int nTracks = pixeltracks.size();
   for (int i = 0; i < nTracks; ++i) {
 
     reco::Track* track   =  pixeltracks.at(i).first;
     const RecHits & hits = pixeltracks.at(i).second;
     
     for (unsigned int k = 0; k < hits.size(); k++) {
       TrackingRecHit *hit = (hits.at(k))->clone();
       track->setHitPattern(*hit, k);
       recHits->push_back(hit);
     }
 
     tracks->push_back(*track);
     delete track;
     
   }
   
   edm::OrphanHandle <TrackingRecHitCollection> ohRH = e.put( recHits );
   
   for (int k = 0; k < nTracks; ++k) {
 
     // reco::TrackExtra* theTrackExtra = new reco::TrackExtra();
     reco::TrackExtra theTrackExtra;
     
     //fill the TrackExtra with TrackingRecHitRef
     // unsigned int nHits = tracks->at(k).numberOfValidHits();
     unsigned nHits = 3; // We are dealing with triplets!
     for(unsigned int i = 0; i < nHits; ++i) {
       theTrackExtra.add(TrackingRecHitRef(ohRH,cc++));
       //theTrackExtra->add(TrackingRecHitRef(ohRH,cc));
       //cc++;
     }
     
     trackExtras->push_back(theTrackExtra);
     //trackExtras->push_back(*theTrackExtra);
     //delete theTrackExtra;
   }
   
   edm::OrphanHandle<reco::TrackExtraCollection> ohTE = e.put(trackExtras);
   
   for (int k = 0; k < nTracks; k++) {
 
     const reco::TrackExtraRef theTrackExtraRef(ohTE,k);
     (tracks->at(k)).setExtra(theTrackExtraRef);
 
   }
   
   e.put(tracks);
   
   // Avoid a memory leak !
   unsigned nRegions = regions.size();
   for ( unsigned iRegions=0; iRegions<nRegions; ++iRegions ) {
     delete regions[iRegions];
   }
 
 }