KalmanFitterRefTrack.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file 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