KalmanFitterRefTrack.h

Go to the documentation of this file.

/* 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)
    : AbsKalmanFitter(maxIterations, deltaPval, blowUpFactor), refitAll_(false), deltaChi2Ref_(1)
  {
  }

  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);

  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);

  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_; 

 public:
  ClassDef(KalmanFitterRefTrack, 1)
};

} /* End of namespace genfit */
#endif // genfit_KalmanFitterRefTrack_h