CkfTrajectoryBuilder.h

Go to the documentation of this file.

#ifndef CkfTrajectoryBuilder_H
#define CkfTrajectoryBuilder_H

#include <vector>

class Propagator;
class TrajectoryStateUpdator;
class MeasurementEstimator;
class TrajectorySeed;
class TrajectoryStateOnSurface;
class TrajectoryFilter;

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"

#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"

#include "RecoTracker/CkfPattern/interface/BaseCkfTrajectoryBuilder.h"

#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "TrackingTools/PatternTools/interface/TempTrajectory.h"
#include "TrackingTools/PatternTools/interface/TrajectoryMeasurement.h"

#include <map>
#include <boost/unordered_map.hpp>

class TransientTrackingRecHitBuilder;
class TrajectoryFilter;

class CkfTrajectoryBuilder :public BaseCkfTrajectoryBuilder {

public:

  typedef std::vector<Trajectory>         TrajectoryContainer;
  typedef std::vector<TempTrajectory>     TempTrajectoryContainer;

  CkfTrajectoryBuilder(const edm::ParameterSet&              conf,
               const TrajectoryStateUpdator*         updator,
               const Propagator*                     propagatorAlong,
               const Propagator*                     propagatorOpposite,
               const Chi2MeasurementEstimatorBase*   estimator,
               const TransientTrackingRecHitBuilder* recHitBuilder,
               const TrajectoryFilter*               filter);

  ~CkfTrajectoryBuilder() {}

  // Return a clone of this, with the data pointer set
  virtual CkfTrajectoryBuilder * clone(const MeasurementTrackerEvent *data) const {
    CkfTrajectoryBuilder *ret = new CkfTrajectoryBuilder(*this);
    ret->setData(data);
    return ret;
  }
  
  virtual TrajectoryContainer trajectories(const TrajectorySeed& seed) const;
  virtual void trajectories(const TrajectorySeed& seed, TrajectoryContainer &ret) const;

  // new interface returning the start Trajectory...
  TempTrajectory buildTrajectories (const TrajectorySeed&,
                    TrajectoryContainer &ret,
                    const TrajectoryFilter*) const;
  
  
  void  rebuildTrajectories(TempTrajectory const& startingTraj, const TrajectorySeed&,
                TrajectoryContainer& result) const {}

  //  virtual void setEvent(const edm::Event& event) const;



 protected:
  int theMaxCand;               
  float theLostHitPenalty;      
  bool theIntermediateCleaning; 
  bool theAlwaysUseInvalidHits;


 protected:
  virtual void findCompatibleMeasurements(const TrajectorySeed&seed, const TempTrajectory& traj, std::vector<TrajectoryMeasurement> & result) const;

  void limitedCandidates(const TrajectorySeed&seed, TempTrajectory& startingTraj, TrajectoryContainer& result) const;
  void limitedCandidates(const boost::shared_ptr<const TrajectorySeed> & sharedSeed, TempTrajectoryContainer &candidates, TrajectoryContainer& result) const;
  
  void updateTrajectory( TempTrajectory& traj, const TM& tm) const;

  /*  
      //not mature for integration.  
      bool theSharedSeedCheck;
      std::string theUniqueName;
      void rememberSeedAndTrajectories(const TrajectorySeed& seed,TrajectoryContainer &result) const;
      bool seedAlreadyUsed(const TrajectorySeed& seed,TempTrajectoryContainer &candidates) const;
      bool sharedSeed(const TrajectorySeed& seed1,const TrajectorySeed& seed2) const;
      //  mutable TempTrajectoryContainer theCachedTrajectories;
      typedef boost::unordered_multimap<uint32_t,TempTrajectory> SharedTrajectory;
      mutable SharedTrajectory theCachedTrajectories;
  */
};

#endif