ConversionTrackFinder.h

Go to the documentation of this file.

#ifndef RecoEGAMMA_ConversionTrack_ConversionTrackFinder_h
#define RecoEGAMMA_ConversionTrack_ConversionTrackFinder_h

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"
#include "TrackingTools/PatternTools/interface/TrajectoryMeasurement.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
//
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"
//
#include "RecoEgamma/EgammaPhotonAlgos/interface/ConversionSeedFinder.h"

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

// C/C++ headers
#include <string>
#include <vector>

class TransientInitialStateEstimator;
class ConversionTrackFinder {

 public:
  
  ConversionTrackFinder( const edm::ParameterSet& config, const BaseCkfTrajectoryBuilder *trajectoryBuilder);
                       
  
  virtual ~ConversionTrackFinder();
 
  
  virtual std::vector<Trajectory> tracks(const TrajectorySeedCollection& seeds , TrackCandidateCollection &candidate) const =0;

  void setEventSetup( const edm::EventSetup& es ) ; 

 protected: 
  
  const MagneticField* theMF_;

  std::string theMeasurementTrackerName_;
  const MeasurementTracker*     theMeasurementTracker_;
  const BaseCkfTrajectoryBuilder*  theCkfTrajectoryBuilder_;

  std::unique_ptr<TransientInitialStateEstimator> theInitialState_;
  const TrackerGeometry* theTrackerGeom_;
  KFUpdator*                          theUpdator_;

  edm::ESHandle<Propagator> thePropagator_;

  bool useSplitHits_;

  struct ExtractNumOfHits {
    typedef int result_type;
    result_type operator()(const Trajectory& t) const {return t.foundHits();}
    result_type operator()(const Trajectory* t) const {return t->foundHits();}
  };


  struct ExtractChi2 {
    typedef float result_type;
    result_type operator()(const Trajectory& t) const {return t.chiSquared();}
    result_type operator()(const Trajectory* t) const {return t->chiSquared();}
  };

};

#endif