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/DetLayers/interface/MeasurementEstimator.h"
#include "TrackingTools/DetLayers/interface/DetLayerGeometry.h"
 
#include <vector>
 
class MultiTrajectoryStateMerger;
 
class GsfTrajectoryFitter final : 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 = nullptr);
 
  ~GsfTrajectoryFitter() override;
 
  Trajectory fitOne(const Trajectory& t, fitType type) const override;
  Trajectory fitOne(const TrajectorySeed& aSeed,
                    const RecHitContainer& hits,
                    const TrajectoryStateOnSurface& firstPredTsos,
                    fitType type) const override;
  Trajectory fitOne(const TrajectorySeed& aSeed, const RecHitContainer& hits, fitType type) const override;
 
  const Propagator* propagator() const { return thePropagator; }
  const TrajectoryStateUpdator* updator() const { return theUpdator; }
  const MeasurementEstimator* estimator() const { return theEstimator; }
  const MultiTrajectoryStateMerger* merger() const { return theMerger; }
 
  std::unique_ptr<TrajectoryFitter> clone() const override {
    return std::unique_ptr<TrajectoryFitter>(
        new GsfTrajectoryFitter(*thePropagator, *theUpdator, *theEstimator, *theMerger, theGeometry));
  }
 
  void setHitCloner(TkCloner const* hc) override { theHitCloner = hc; }
 
private:
  const Propagator* thePropagator;
  const TrajectoryStateUpdator* theUpdator;
  const MeasurementEstimator* theEstimator;
  const MultiTrajectoryStateMerger* theMerger;
  TkCloner const* theHitCloner = nullptr;
  const DetLayerGeometry dummyGeometry;
  const DetLayerGeometry* theGeometry;
 
  bool theTiming;
};
 
#endif  //TR_GsfTrajectoryFitter_H_