#include <CkfTrackCandidateMakerBase.h>

Inheritance diagram for cms::CkfTrackCandidateMakerBase:

Public Member Functions
virtual void	beginRunBase (edm::Run &, edm::EventSetup const &es)

	CkfTrackCandidateMakerBase (const edm::ParameterSet &conf)

virtual void	produceBase (edm::Event &e, const edm::EventSetup &es)

virtual	~CkfTrackCandidateMakerBase ()

Protected Member Functions
virtual void	countSeedsDebugger ()

virtual void	deleteAssocDebugger ()

virtual TrajectorySeedCollection::const_iterator	lastSeed (TrajectorySeedCollection const &theSeedColl)

virtual void	printHitsDebugger (edm::Event &e)

Protected Attributes
bool	cleanTrajectoryAfterInOut

edm::ParameterSet	conf_

bool	doSeedingRegionRebuilding

unsigned int	maxSeedsBeforeCleaning_

bool	reverseTrajectories

edm::ESHandle < GeometricSearchTracker >	theGeomSearchTracker

TransientInitialStateEstimator *	theInitialState

edm::ESHandle< MagneticField >	theMagField

unsigned int	theMaxNSeeds

const NavigationSchool *	theNavigationSchool

std::string	theNavigationSchoolName

RedundantSeedCleaner *	theSeedCleaner

edm::InputTag	theSeedLabel

bool	theTrackCandidateOutput

const TrajectoryBuilder *	theTrajectoryBuilder

std::string	theTrajectoryBuilderName

const TrajectoryCleaner *	theTrajectoryCleaner

std::string	theTrajectoryCleanerName

bool	theTrajectoryOutput

bool	useSplitting

Private Member Functions
void	setEventSetup (const edm::EventSetup &es)
	Initialize EventSetup objects at each event. More...

Detailed Description

Definition at line 27 of file CkfTrackCandidateMakerBase.h.

Constructor & Destructor Documentation

cms::CkfTrackCandidateMakerBase::CkfTrackCandidateMakerBase ( const edm::ParameterSet & conf )

explicit

Definition at line 43 of file CkfTrackCandidateMakerBase.cc.

References edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), maxSeedsBeforeCleaning_, and theSeedLabel.

                                                                                     : 
 
     conf_(conf),
     theTrackCandidateOutput(true),
     theTrajectoryOutput(false),
     useSplitting(conf.getParameter<bool>("useHitsSplitting")),
     doSeedingRegionRebuilding(conf.getParameter<bool>("doSeedingRegionRebuilding")),
     cleanTrajectoryAfterInOut(conf.getParameter<bool>("cleanTrajectoryAfterInOut")),
     reverseTrajectories(conf.existsAs<bool>("reverseTrajectories") && conf.getParameter<bool>("reverseTrajectories")),
     theMaxNSeeds(conf.getParameter<unsigned int>("maxNSeeds")),
     theTrajectoryBuilderName(conf.getParameter<std::string>("TrajectoryBuilder")), 
     theTrajectoryBuilder(0),
     theTrajectoryCleanerName(conf.getParameter<std::string>("TrajectoryCleaner")), 
     theTrajectoryCleaner(0),
     theInitialState(0),
     theNavigationSchoolName(conf.getParameter<std::string>("NavigationSchool")),
     theNavigationSchool(0),
     theSeedCleaner(0),
     maxSeedsBeforeCleaning_(0)
   {  
     //produces<TrackCandidateCollection>();  
     // old configuration totally descoped.
     //    if (!conf.exists("src"))
     //      theSeedLabel = InputTag(conf_.getParameter<std::string>("SeedProducer"),conf_.getParameter<std::string>("SeedLabel"));
     //    else
       theSeedLabel= conf.getParameter<edm::InputTag>("src");
       if ( conf.exists("maxSeedsBeforeCleaning") ) 
        maxSeedsBeforeCleaning_=conf.getParameter<unsigned int>("maxSeedsBeforeCleaning");
 
   }

cms::CkfTrackCandidateMakerBase::~CkfTrackCandidateMakerBase ( )

virtual

Definition at line 76 of file CkfTrackCandidateMakerBase.cc.

References theInitialState, and theSeedCleaner.

                                                           {
     delete theInitialState;  
     if (theSeedCleaner) delete theSeedCleaner;
   }  

Member Function Documentation

void cms::CkfTrackCandidateMakerBase::beginRunBase	(	edm::Run &	r,
		edm::EventSetup const &	es
	)

virtual

Definition at line 81 of file CkfTrackCandidateMakerBase.cc.

References conf_, edm::hlt::Exception, edm::ParameterSet::existsAs(), edm::ParameterSet::getParameter(), and theSeedCleaner.

Referenced by cms::CkfDebugTrackCandidateMaker::beginRun(), cms::CkfTrackCandidateMaker::beginRun(), and cms::CkfTrajectoryMaker::beginRun().

   {
     std::string cleaner = conf_.getParameter<std::string>("RedundantSeedCleaner");
     if (cleaner == "SeedCleanerByHitPosition") {
         theSeedCleaner = new SeedCleanerByHitPosition();
     } else if (cleaner == "SeedCleanerBySharedInput") {
         theSeedCleaner = new SeedCleanerBySharedInput();
     } else if (cleaner == "CachingSeedCleanerByHitPosition") {
         theSeedCleaner = new CachingSeedCleanerByHitPosition();
     } else if (cleaner == "CachingSeedCleanerBySharedInput") {
       int numHitsForSeedCleaner = conf_.existsAs<int>("numHitsForSeedCleaner") ? 
     conf_.getParameter<int>("numHitsForSeedCleaner") : 4;
       int onlyPixelHits = conf_.existsAs<bool>("onlyPixelHitsForSeedCleaner") ? 
     conf_.getParameter<bool>("onlyPixelHitsForSeedCleaner") : false;
       theSeedCleaner = new CachingSeedCleanerBySharedInput(numHitsForSeedCleaner,onlyPixelHits);
     } else if (cleaner == "none") {
         theSeedCleaner = 0;
     } else {
         throw cms::Exception("RedundantSeedCleaner not found", cleaner);
     }
   }

virtual void cms::CkfTrackCandidateMakerBase::countSeedsDebugger ( )

inlineprotectedvirtual

Reimplemented in cms::CkfDebugTrackCandidateMaker.

Definition at line 73 of file CkfTrackCandidateMakerBase.h.

Referenced by produceBase().

73 {;}

virtual void cms::CkfTrackCandidateMakerBase::deleteAssocDebugger ( )

inlineprotectedvirtual

Reimplemented in cms::CkfDebugTrackCandidateMaker.

Definition at line 74 of file CkfTrackCandidateMakerBase.h.

Referenced by produceBase().

74 {;}

virtual TrajectorySeedCollection::const_iterator cms::CkfTrackCandidateMakerBase::lastSeed ( TrajectorySeedCollection const & theSeedColl )

inlineprotectedvirtual

Definition at line 71 of file CkfTrackCandidateMakerBase.h.

71 {return theSeedColl.end();}

virtual void cms::CkfTrackCandidateMakerBase::printHitsDebugger ( edm::Event & e )

inlineprotectedvirtual

Reimplemented in cms::CkfDebugTrackCandidateMaker.

Definition at line 72 of file CkfTrackCandidateMakerBase.h.

Referenced by produceBase().

72 {;}

void cms::CkfTrackCandidateMakerBase::produceBase	(	edm::Event &	e,
		const edm::EventSetup &	es
	)

virtual

Definition at line 134 of file CkfTrackCandidateMakerBase.cc.

References RedundantSeedCleaner::add(), alongMomentum, TrajectoryCleaner::clean(), cleanTrajectoryAfterInOut, countSeedsDebugger(), deleteAssocDebugger(), RedundantSeedCleaner::done(), doSeedingRegionRebuilding, PrintoutHelper::dumpCandidates(), edm::EventSetup::get(), edm::Event::getByLabel(), RedundantSeedCleaner::good(), RedundantSeedCleaner::init(), TransientInitialStateEstimator::innerState(), edm::detail::isnan(), TrajectoryStateOnSurface::isValid(), Trajectory::isValid(), j, LogDebug, LogTrace, maxSeedsBeforeCleaning_, oppositeToMomentum, convertSQLitetoXML_cfg::output, trajectoryStateTransform::persistentState(), printHitsDebugger(), Trajectory::push(), edm::OwnVector< T, P >::push_back(), edm::Event::put(), DetId::rawId(), TrajectoryBuilder::rebuildSeedingRegion(), PrintoutHelper::regressionTest(), edm::OwnVector< T, P >::reserve(), reverseTrajectories, TrajectoryBuilder::setEvent(), setEventSetup(), Trajectory::setNLoops(), Trajectory::setSeedRef(), evf::utils::state, theInitialState, theMaxNSeeds, theNavigationSchool, theSeedCleaner, theSeedLabel, theTrackCandidateOutput, theTrajectoryBuilder, theTrajectoryCleaner, theTrajectoryOutput, patCandidatesForDimuonsSequences_cff::tracker, TrajectoryBuilder::trajectories(), TrajectoryBuilder::unset(), and useSplitting.

Referenced by cms::CkfDebugTrackCandidateMaker::produce(), cms::CkfTrackCandidateMaker::produce(), and cms::CkfTrajectoryMaker::produce().

   { 
     // getting objects from the EventSetup
     setEventSetup( es ); 
 
     // set the correct navigation
     NavigationSetter setter( *theNavigationSchool);
     
     // propagator
     edm::ESHandle<Propagator> thePropagator;
     es.get<TrackingComponentsRecord>().get("AnyDirectionAnalyticalPropagator",
                        thePropagator);
 
     // method for Debugging
     printHitsDebugger(e);
 
     // Step A: set Event for the TrajectoryBuilder
     theTrajectoryBuilder->setEvent(e);        
     
     // Step B: Retrieve seeds
     
     edm::Handle<View<TrajectorySeed> > collseed;
     e.getByLabel(theSeedLabel, collseed);
     
     // Step C: Create empty output collection
     std::auto_ptr<TrackCandidateCollection> output(new TrackCandidateCollection);    
     std::auto_ptr<std::vector<Trajectory> > outputT (new std::vector<Trajectory>());
 
     if ( (*collseed).size()>theMaxNSeeds ) {
       LogError("TooManySeeds")<<"Exceeded maximum numeber of seeds! theMaxNSeeds="<<theMaxNSeeds<<" nSeed="<<(*collseed).size();
       if (theTrackCandidateOutput){ e.put(output);}
       if (theTrajectoryOutput){e.put(outputT);}
       theTrajectoryBuilder->unset();
       return;
     }
     
     // Step D: Invoke the building algorithm
     if ((*collseed).size()>0){
 
       unsigned int lastCleanResult=0;
        vector<Trajectory> rawResult;
        rawResult.reserve(collseed->size() * 4);
 
       if (theSeedCleaner) theSeedCleaner->init( &rawResult );
       
       // method for debugging
       countSeedsDebugger();
 
       vector<Trajectory> theTmpTrajectories;
 
       // Loop over seeds
       size_t collseed_size = collseed->size(); 
       for (size_t j = 0; j < collseed_size; j++){
        
     // Check if seed hits already used by another track
     if (theSeedCleaner && !theSeedCleaner->good( &((*collseed)[j])) ) {
           LogDebug("CkfTrackCandidateMakerBase")<<" Seed cleaning kills seed "<<j;
           continue; 
         }
 
     // Build trajectory from seed outwards
         theTmpTrajectories.clear();
     theTrajectoryBuilder->trajectories( (*collseed)[j], theTmpTrajectories );
     
        
     LogDebug("CkfPattern") << "======== In-out trajectory building found " << theTmpTrajectories.size()
                         << " trajectories from seed " << j << " ========"<<endl
                    <<PrintoutHelper::dumpCandidates(theTmpTrajectories);
     
         if (cleanTrajectoryAfterInOut) {
 
       // Select the best trajectory from this seed (declare others invalid)
       theTrajectoryCleaner->clean(theTmpTrajectories);
 
       LogDebug("CkfPattern") << "======== In-out trajectory cleaning gave the following valid trajectories from seed " 
                                  << j << " ========"<<endl
                  << PrintoutHelper::dumpCandidates(theTmpTrajectories);
         }
 
     // Optionally continue building trajectory back through 
     // seed and if possible further inwards.
     
     if (doSeedingRegionRebuilding) {
       theTrajectoryBuilder->rebuildSeedingRegion((*collseed)[j],theTmpTrajectories);      
 
       LogDebug("CkfPattern") << "======== Out-in trajectory building found " << theTmpTrajectories.size()
                           << " valid/invalid trajectories from seed " << j << " ========"<<endl
                  <<PrintoutHelper::dumpCandidates(theTmpTrajectories);
         }
     
 
         // Select the best trajectory from this seed (after seed region rebuilding, can be more than one)
     theTrajectoryCleaner->clean(theTmpTrajectories);
 
         LogDebug("CkfPattern") << "======== Trajectory cleaning gave the following valid trajectories from seed " 
                                << j << " ========"<<endl
                    <<PrintoutHelper::dumpCandidates(theTmpTrajectories);
 
     for(vector<Trajectory>::iterator it=theTmpTrajectories.begin();
         it!=theTmpTrajectories.end(); it++){
       if( it->isValid() ) {
         it->setSeedRef(collseed->refAt(j));
         // Store trajectory
         rawResult.push_back(*it);
         // Tell seed cleaner which hits this trajectory used.
             //TO BE FIXED: this cut should be configurable via cfi file
             if (theSeedCleaner && it->foundHits()>3) theSeedCleaner->add( & (*it) );
             //if (theSeedCleaner ) theSeedCleaner->add( & (*it) );
       }
     }
 
         theTmpTrajectories.clear();
         
     LogDebug("CkfPattern") << "rawResult trajectories found so far = " << rawResult.size();
 
     if ( maxSeedsBeforeCleaning_ >0 && rawResult.size() > maxSeedsBeforeCleaning_+lastCleanResult) {
           theTrajectoryCleaner->clean(rawResult);
           rawResult.erase(std::remove_if(rawResult.begin(),rawResult.end(),
                      std::not1(std::mem_fun_ref(&Trajectory::isValid))),
               rawResult.end());
           lastCleanResult=rawResult.size();
         }
 
       }
       // end of loop over seeds
       
       if (theSeedCleaner) theSeedCleaner->done();
       
       // Step E: Clean the results to avoid duplicate tracks
       // Rejected ones just flagged as invalid.
       theTrajectoryCleaner->clean(rawResult);
 
       LogDebug("CkfPattern") << "======== Final cleaning of entire event found " << rawResult.size() 
                              << " valid/invalid trajectories ======="<<endl
                  <<PrintoutHelper::dumpCandidates(rawResult);
 
       LogDebug("CkfPattern") << "removing invalid trajectories.";
 
       vector<Trajectory> & unsmoothedResult(rawResult);
       unsmoothedResult.erase(std::remove_if(unsmoothedResult.begin(),unsmoothedResult.end(),
                         std::not1(std::mem_fun_ref(&Trajectory::isValid))),
                  unsmoothedResult.end());
       
       // If requested, reverse the trajectories creating a new 1-hit seed on the last measurement of the track
       if (reverseTrajectories) {
         vector<Trajectory> reversed; 
         reversed.reserve(unsmoothedResult.size());
         for (vector<Trajectory>::const_iterator it = unsmoothedResult.begin(), ed = unsmoothedResult.end(); it != ed; ++it) {
           // reverse the trajectory only if it has valid hit on the last measurement (should happen)
           if (it->lastMeasurement().updatedState().isValid() && 
               it->lastMeasurement().recHit().get() != 0     &&
               it->lastMeasurement().recHit()->isValid()) {
             // I can't use reverse in place, because I want to change the seed
             // 1) reverse propagation direction
             PropagationDirection direction = it->direction();
             if (direction == alongMomentum)           direction = oppositeToMomentum;
             else if (direction == oppositeToMomentum) direction = alongMomentum;
             // 2) make a seed
             TrajectoryStateOnSurface initState = it->lastMeasurement().updatedState();
             DetId                    initDetId = it->lastMeasurement().recHit()->geographicalId();
             PTrajectoryStateOnDet state = trajectoryStateTransform::persistentState( initState, initDetId.rawId());
             TrajectorySeed::recHitContainer hits; 
             hits.push_back(*it->lastMeasurement().recHit()->hit());
             boost::shared_ptr<const TrajectorySeed> seed(new TrajectorySeed(state, hits, direction));
             // 3) make a trajectory
             Trajectory trajectory(seed, direction);
         trajectory.setNLoops(it->nLoops());
             trajectory.setSeedRef(it->seedRef());
             // 4) push states in reversed order
             const Trajectory::DataContainer &meas = it->measurements();
             for (Trajectory::DataContainer::const_reverse_iterator itmeas = meas.rbegin(), endmeas = meas.rend(); itmeas != endmeas; ++itmeas) {
               trajectory.push(*itmeas);
             } 
             reversed.push_back(trajectory);
           } else {
             edm::LogWarning("CkfPattern_InvalidLastMeasurement") << "Last measurement of the trajectory is invalid, cannot reverse it";
             reversed.push_back(*it);
           }     
         }
         unsmoothedResult.swap(reversed);
       }
 
       //      for (vector<Trajectory>::const_iterator itraw = rawResult.begin();
       //       itraw != rawResult.end(); itraw++) {
       //if((*itraw).isValid()) unsmoothedResult.push_back( *itraw);
       //}
 
       //analyseCleanedTrajectories(unsmoothedResult);
       
       if (theTrackCandidateOutput){
     // Step F: Convert to TrackCandidates
        output->reserve(unsmoothedResult.size());
        for (vector<Trajectory>::const_iterator it = unsmoothedResult.begin();
         it != unsmoothedResult.end(); ++it) {
     
      Trajectory::RecHitContainer thits;
      //it->recHitsV(thits);
      LogDebug("CkfPattern") << "retrieving "<<(useSplitting?"splitted":"un-splitted")<<" hits from trajectory";
      it->recHitsV(thits,useSplitting);
      OwnVector<TrackingRecHit> recHits;
      recHits.reserve(thits.size());
      LogDebug("CkfPattern") << "cloning hits into new collection.";
      for (Trajectory::RecHitContainer::const_iterator hitIt = thits.begin();
           hitIt != thits.end(); ++hitIt) {
        recHits.push_back( (**hitIt).hit()->clone());
      }
 
      LogDebug("CkfPattern") << "getting initial state.";
      const bool doBackFit = !doSeedingRegionRebuilding && !reverseTrajectories;
      std::pair<TrajectoryStateOnSurface, const GeomDet*> initState = 
        theInitialState->innerState( *it , doBackFit);
 
      // temporary protection againt invalid initial states
      if (! initState.first.isValid() || initState.second == 0 || std::isnan(initState.first.globalPosition().x())) {
        //cout << "invalid innerState, will not make TrackCandidate" << endl;
        continue;
      }
      
      PTrajectoryStateOnDet state;
      if(useSplitting && (initState.second != thits.front()->det()) && thits.front()->det() ){    
        LogDebug("CkfPattern") << "propagating to hit front in case of splitting.";
        TrajectoryStateOnSurface propagated = thePropagator->propagate(initState.first,thits.front()->det()->surface());
        if (!propagated.isValid()) continue;
        state = trajectoryStateTransform::persistentState(propagated,
                                       thits.front()->det()->geographicalId().rawId());
      }
      else state = trajectoryStateTransform::persistentState( initState.first,
                                     initState.second->geographicalId().rawId());
      LogDebug("CkfPattern") << "pushing a TrackCandidate.";
      output->push_back(TrackCandidate(recHits,it->seed(),state,it->seedRef(),it->nLoops() ) );
        }
       }//output trackcandidates
 
       edm::ESHandle<TrackerGeometry> tracker;
       es.get<TrackerDigiGeometryRecord>().get(tracker);            
       LogTrace("CkfPattern|TrackingRegressionTest") << "========== CkfTrackCandidateMaker Info =========="
                             << "number of Seed: " << collseed->size()<<endl
                                 <<PrintoutHelper::regressionTest(*tracker,unsmoothedResult);
 
       
      
       if (theTrajectoryOutput){ outputT->swap(unsmoothedResult);}
 
     }// end of ((*collseed).size()>0)
     
     // method for debugging
     deleteAssocDebugger();
 
     // Step G: write output to file
     if (theTrackCandidateOutput){ e.put(output);}
     if (theTrajectoryOutput){e.put(outputT);}
     
     //reset the MT.
     theTrajectoryBuilder->unset();
   }

void cms::CkfTrackCandidateMakerBase::setEventSetup ( const edm::EventSetup & es )

private

Initialize EventSetup objects at each event.

Definition at line 103 of file CkfTrackCandidateMakerBase.cc.

References conf_, edm::EventSetup::get(), edm::ParameterSet::getParameter(), edm::ESHandle< class >::product(), TransientInitialStateEstimator::setEventSetup(), theGeomSearchTracker, theInitialState, theMagField, theNavigationSchool, theNavigationSchoolName, theTrajectoryBuilder, theTrajectoryBuilderName, theTrajectoryCleaner, and theTrajectoryCleanerName.

Referenced by produceBase().

                                                                           {
 
     //services
     es.get<TrackerRecoGeometryRecord>().get( theGeomSearchTracker );
     es.get<IdealMagneticFieldRecord>().get( theMagField );
 
     if (!theInitialState){
       // constructor uses the EventSetup, it must be in the setEventSetup were it has a proper value.
       // get nested parameter set for the TransientInitialStateEstimator
       ParameterSet tise_params = conf_.getParameter<ParameterSet>("TransientInitialStateEstimatorParameters") ;
       theInitialState          = new TransientInitialStateEstimator( es,tise_params);
     }
 
     theInitialState->setEventSetup( es );
 
     edm::ESHandle<TrajectoryCleaner> trajectoryCleanerH;
     es.get<TrajectoryCleaner::Record>().get(theTrajectoryCleanerName, trajectoryCleanerH);
     theTrajectoryCleaner= trajectoryCleanerH.product();
 
     edm::ESHandle<NavigationSchool> navigationSchoolH;
     es.get<NavigationSchoolRecord>().get(theNavigationSchoolName, navigationSchoolH);
     theNavigationSchool = navigationSchoolH.product();
 
     // set the TrajectoryBuilder
     edm::ESHandle<TrajectoryBuilder> theTrajectoryBuilderHandle;
     es.get<CkfComponentsRecord>().get(theTrajectoryBuilderName,theTrajectoryBuilderHandle);
     theTrajectoryBuilder = theTrajectoryBuilderHandle.product();    
        
   }