KFSplittingFitter.h

Go to the documentation of this file.

#ifndef CD_KFSplittingFitter_H_
#define CD_KFSplittingFitter_H_

#include "TrackingTools/TrackFitters/interface/KFTrajectoryFitter.h"
#include "TrackingTools/TrackFitters/interface/RecHitSplitter.h"

class KFSplittingFitter final : public TrajectoryFitter {

private:

  typedef RecHitSplitter::RecHitContainer        RecHitContainer;

  typedef TrajectoryStateOnSurface TSOS;
  typedef FreeTrajectoryState FTS;
  typedef TrajectoryMeasurement TM;

public:

  KFSplittingFitter(const Propagator& aPropagator,
                    const TrajectoryStateUpdator& aUpdator,
                    const MeasurementEstimator& aEstimator) :
    fitter(aPropagator, aUpdator, aEstimator) {}


  KFSplittingFitter(const Propagator* aPropagator,
            const TrajectoryStateUpdator* aUpdator,
            const MeasurementEstimator* aEstimator) :
    fitter(aPropagator, aUpdator, aEstimator) {}

    virtual std::unique_ptr<TrajectoryFitter> clone() const override {
      return std::unique_ptr<TrajectoryFitter>(
          new KFSplittingFitter(fitter.propagator(),
                                fitter.updator(),
                                fitter.estimator()));
  }

  Trajectory fitOne(const Trajectory& aTraj,
            fitType type) const;
  Trajectory fitOne(const TrajectorySeed& aSeed,
            const RecHitContainer& hits,
            fitType type) const;
 Trajectory fitOne(const TrajectorySeed& aSeed,
            const RecHitContainer& hits,
            const TSOS& firstPredTsos,
            fitType type) const;

  virtual void setHitCloner(TkCloner const * hc) {
        fitter.setHitCloner(hc);
  }

 private :

 void sorter(const RecHitContainer& hits, PropagationDirection dir, RecHitContainer & result) const;

 KFTrajectoryFitter fitter;

};

#endif //CD_KFSplittingFitter_H_