CMSSW_4_4_3_patch1/src/TrackingTools/GsfTracking/interface/GsfTrajectoryFitter.h

Go to the documentation of this file.

#ifndef GsfTrajectoryFitter_H_
#define GsfTrajectoryFitter_H_

#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "TrackingTools/PatternTools/interface/TrajectoryStateUpdator.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "TrackingTools/TrackFitters/interface/TrajectoryFitter.h"
#include "TrackingTools/PatternTools/interface/TrajectoryMeasurement.h"
#include "TrackingTools/PatternTools/interface/MeasurementEstimator.h"
#include "TrackingTools/DetLayers/interface/DetLayerGeometry.h"

#include <vector>

class MultiTrajectoryStateMerger;

class GsfTrajectoryFitter : public TrajectoryFitter {

private:
  typedef TrajectoryStateOnSurface TSOS;
  typedef TrajectoryMeasurement TM;

public:

  GsfTrajectoryFitter(const Propagator& aPropagator,
                      const TrajectoryStateUpdator& aUpdator,
                      const MeasurementEstimator& aEstimator,
                      const MultiTrajectoryStateMerger& aMerger,
                      const DetLayerGeometry* detLayerGeometry=0);
  
  virtual ~GsfTrajectoryFitter(); 
  
  virtual std::vector<Trajectory> fit(const Trajectory& aTraj) const;
  virtual std::vector<Trajectory> fit(const TrajectorySeed& aSeed,
                                      const RecHitContainer& hits) const;
  virtual std::vector<Trajectory> fit(const TrajectorySeed& aSeed,
                                      const RecHitContainer& hits, 
                                      const TSOS& firstPredTsos) const;

  const Propagator* propagator() const {return thePropagator;}
  const TrajectoryStateUpdator* updator() const {return theUpdator;}
  const MeasurementEstimator* estimator() const {return theEstimator;}
  const MultiTrajectoryStateMerger* merger() const {return theMerger;}
  
  virtual GsfTrajectoryFitter* clone() const
  {
    return new GsfTrajectoryFitter(*thePropagator,*theUpdator,*theEstimator,*theMerger,theGeometry);
  }

private:
  const Propagator* thePropagator;
  const TrajectoryStateUpdator* theUpdator;
  const MeasurementEstimator* theEstimator;
  const MultiTrajectoryStateMerger* theMerger;
  const DetLayerGeometry dummyGeometry;
  const DetLayerGeometry* theGeometry;

  bool theTiming;
};

#endif //TR_GsfTrajectoryFitter_H_