#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 ()
Private Attributes
edm::InputTag	seedProducer
const PixelTrackFilter *	theFilter
const PixelFitter *	theFitter
TrackingRegionProducer *	theRegionProducer

Detailed Description

Definition at line 20 of file PixelTracksProducer.h.

Constructor & Destructor Documentation

PixelTracksProducer::PixelTracksProducer ( const edm::ParameterSet & conf ) [explicit]

Definition at line 33 of file PixelTracksProducer.cc.

References reco_skim_cfg_mod::filterName, reco::get(), edm::ParameterSet::getParameter(), seedProducer, 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);
  
  const edm::ParameterSet& fitterPSet = conf.getParameter<edm::ParameterSet>("FitterPSet");
  std::string fitterName = fitterPSet.getParameter<std::string>("ComponentName");
  theFitter = PixelFitterFactory::get()->create( fitterName, fitterPSet);
  
  const edm::ParameterSet& filterPSet = conf.getParameter<edm::ParameterSet>("FilterPSet");
  std::string filterName = filterPSet.getParameter<std::string>("ComponentName");
  theFilter = PixelTrackFilterFactory::get()->create( filterName, filterPSet);
  
  // The name of the seed producer
  seedProducer = conf.getParameter<edm::InputTag>("SeedProducer");

}

PixelTracksProducer::~PixelTracksProducer ( )

Definition at line 62 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 73 of file PixelTracksProducer.cc.

References reco::TrackExtraBase::add(), 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.

                                                                 {        
  
  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];
  }

}