/data/refman/pasoursint/CMSSW_4_1_8_patch13/src/RecoTracker/TrackProducer/interface/DAFTrackProducerAlgorithm.h

Go to the documentation of this file.

#ifndef DAFTrackProducerAlgorithm_h
#define DAFTrackProducerAlgorithm_h

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHit.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"

class MagneticField;
class TrackingGeometry;
class TrajectoryFitter;
class Trajectory;
class TrajectoryStateOnSurface;
class TransientTrackingRecHitBuilder;
class MultiRecHitCollector;
class SiTrackerMultiRecHitUpdator;
namespace reco{
        class Track;
}

class DAFTrackProducerAlgorithm {
   
   typedef std::pair<Trajectory*, std::pair<reco::Track*,PropagationDirection> > AlgoProduct;
   typedef std::vector< AlgoProduct >  AlgoProductCollection;
  
 public:

  DAFTrackProducerAlgorithm(const edm::ParameterSet& pset):conf_(pset){}

  ~DAFTrackProducerAlgorithm() {}
  
  void runWithCandidate(const TrackingGeometry *, 
                        const MagneticField *, 
                        //const TrackCandidateCollection&,
                        const std::vector<Trajectory>&,
                        const TrajectoryFitter *,
                        const TransientTrackingRecHitBuilder*,
                        const MultiRecHitCollector* measurementTracker,
                        const SiTrackerMultiRecHitUpdator*,
                        const reco::BeamSpot&,
                        AlgoProductCollection &) const;

 private:
  bool buildTrack(const std::vector<Trajectory>&,
                  AlgoProductCollection& algoResults,
                  float,
                  const reco::BeamSpot&) const;

  void fit(const std::pair<TransientTrackingRecHit::RecHitContainer, TrajectoryStateOnSurface>& hits,
           const TrajectoryFitter * theFitter,
           std::vector<Trajectory>& vtraj) const;

  //calculates the ndof according to the DAF prescription
  float calculateNdof(const std::vector<Trajectory>& vtraj) const;

  //creates MultiRecHits out of a KF trajectory         
  std::pair<TransientTrackingRecHit::RecHitContainer, TrajectoryStateOnSurface>
  collectHits(const std::vector<Trajectory>& vtraj,
              const MultiRecHitCollector* measurementCollector) const;

  //updates the hits with the specified annealing factor
  std::pair<TransientTrackingRecHit::RecHitContainer, TrajectoryStateOnSurface>
  updateHits(const std::vector<Trajectory>& vtraj,
             const SiTrackerMultiRecHitUpdator* updator,
             double annealing) const; 

  //removes from the trajectory isolated hits with very low weight
  void filter(const TrajectoryFitter* fitter, 
              std::vector<Trajectory>& input, 
              int minhits, std::vector<Trajectory>& output) const;

  edm::ParameterSet conf_;
};

#endif