KalmanFitterRefTrack.h

/* Copyright 2013, Ludwig-Maximilians Universität München,
   Authors: Tobias Schlüter & Johannes Rauch
 
   This file is part of GENFIT.
 
   GENFIT is free software: you can redistribute it and/or modify
   it under the terms of the GNU Lesser General Public License as published
   by the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.
 
   GENFIT is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU Lesser General Public License for more details.
 
   You should have received a copy of the GNU Lesser General Public License
   along with GENFIT.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef genfit_KalmanFitterRefTrack_h
#define genfit_KalmanFitterRefTrack_h
 
#include "AbsKalmanFitter.h"
 
 
namespace genfit {
 
class KalmanFitterInfo;
class TrackPoint;
 
class KalmanFitterRefTrack : public AbsKalmanFitter {
 public:
  KalmanFitterRefTrack(unsigned int maxIterations = 4, double deltaPval = 1e-3, double blowUpFactor = 1e3,
           bool squareRootFormalism = false)
    : AbsKalmanFitter(maxIterations, deltaPval, blowUpFactor), refitAll_(false), deltaChi2Ref_(1),
      squareRootFormalism_(squareRootFormalism)
  {}
 
  virtual ~KalmanFitterRefTrack() {}
 
  TrackPoint* fitTrack(Track* tr, const AbsTrackRep* rep, double& chi2, double& ndf, int direction);
 
  void processTrackWithRep(Track* tr, const AbsTrackRep* rep, bool resortHits = false) override;
 
  bool prepareTrack(Track* tr, const AbsTrackRep* rep, bool setSortingParams, int& nFailedHits);
 
  void setRefitAll(bool refit = true) {refitAll_ = refit;}
 
  void setDeltaChi2Ref(double dChi2) {deltaChi2Ref_ = dChi2;}
 
 private:
  void processTrackPoint(KalmanFitterInfo* fi, const KalmanFitterInfo* prevFi, const TrackPoint* tp, double& chi2, double& ndf, int direction);
  void processTrackPointSqrt(KalmanFitterInfo* fi, const KalmanFitterInfo* prevFi, const TrackPoint* tp, double& chi2, double& ndf, int direction);
 
  bool removeOutdated(Track* tr, const AbsTrackRep* rep, int& notChangedUntil, int& notChangedFrom);
 
  void removeForwardBackwardInfo(Track* tr, const AbsTrackRep* rep, int notChangedUntil, int notChangedFrom) const;
 
  bool refitAll_; // always refit all points or only if reference states have changed
  double deltaChi2Ref_; // reference track update cut
 
  // aux variables for prepareTrack
  TMatrixD FTransportMatrix_; 
  TMatrixD BTransportMatrix_; 
  TMatrixDSym FNoiseMatrix_; 
  TMatrixDSym BNoiseMatrix_; 
  TVectorD forwardDeltaState_; 
  TVectorD backwardDeltaState_; 
 
  // aux variables for processTrackPoint
  TVectorD p_; 
  TMatrixDSym C_; 
  TMatrixDSym covSumInv_; 
  TMatrixDSym Rinv_; 
  TVectorD res_; 
 
  // aux variables for removeOutdated
  TVectorD resM_; 
 
  bool squareRootFormalism_;
 
 public:
  ClassDefOverride(KalmanFitterRefTrack, 1)
 
};
 
}  /* End of namespace genfit */
#endif //genfit_KalmanFitterRefTrack_h