KFFittingSmoother.h

Go to the documentation of this file.

#ifndef CD_KFFittingSmoother_H_
#define CD_KFFittingSmoother_H_

#include "TrackingTools/PatternTools/interface/TrajectorySmoother.h"
#include "TrackingTools/TrackFitters/interface/TrajectoryFitter.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "TrackingTools/TrackFitters/interface/TrajectoryStateWithArbitraryError.h"

class KFFittingSmoother GCC11_FINAL : public TrajectoryFitter {

public:
KFFittingSmoother(const TrajectoryFitter& aFitter,
                  const TrajectorySmoother& aSmoother,
                  double estimateCut = -1,
                  double logPixelProbabilityCut = -16.0,
                  int minNumberOfHits = 5,
                  bool rejectTracks = false,
                  bool BreakTrajWith2ConsecutiveMissing = false,
                  bool NoInvalidHitsBeginEnd = false) :
  theFitter(aFitter.clone()),
      theSmoother(aSmoother.clone()),
      theEstimateCut(estimateCut),

      // ggiurgiu@fnal.gov
      theLogPixelProbabilityCut( logPixelProbabilityCut ),

      theMinNumberOfHits(minNumberOfHits),
      rejectTracksFlag(rejectTracks),
      breakTrajWith2ConsecutiveMissing(BreakTrajWith2ConsecutiveMissing),
      noInvalidHitsBeginEnd(NoInvalidHitsBeginEnd) {}

  virtual ~KFFittingSmoother() {}

  Trajectory fitOne(const Trajectory& t, fitType type) const;
  Trajectory fitOne(const TrajectorySeed& aSeed,
                    const RecHitContainer& hits,
                    const TrajectoryStateOnSurface& firstPredTsos, fitType type) const;
  Trajectory fitOne(const TrajectorySeed& aSeed,
            const RecHitContainer& hits, fitType type) const;

  const TrajectoryFitter* fitter() const {return theFitter.get();}
  const TrajectorySmoother* smoother() const {return theSmoother.get();}

  std::unique_ptr<TrajectoryFitter> clone() const override {
    return std::unique_ptr<TrajectoryFitter>(
        new KFFittingSmoother(*theFitter,
                              *theSmoother,
                              theEstimateCut,theLogPixelProbabilityCut,
                              theMinNumberOfHits,rejectTracksFlag,
                              breakTrajWith2ConsecutiveMissing,noInvalidHitsBeginEnd));
  }


  virtual void setHitCloner(TkCloner const * hc) {
        theFitter->setHitCloner(hc);
        theSmoother->setHitCloner(hc);
  }


private:

  Trajectory smoothingStep(Trajectory const & fitted) const {return theSmoother->trajectory(fitted);}

private:

  const std::unique_ptr<TrajectoryFitter> theFitter;
  const std::unique_ptr<TrajectorySmoother> theSmoother;
  double theEstimateCut;

  double theLogPixelProbabilityCut; // ggiurgiu@fnal.gov

  int theMinNumberOfHits;
  bool rejectTracksFlag;
  bool breakTrajWith2ConsecutiveMissing;
  bool noInvalidHitsBeginEnd;

  TrajectoryStateWithArbitraryError   tsosWithError;

  bool checkForNans(const Trajectory &theTraj) const;

};

#endif //CD_KFFittingSmoother_H_