GFKalman.h

Go to the documentation of this file.

/* Copyright 2008-2010, Technische Universitaet Muenchen,
   Authors: Christian Hoeppner & Sebastian Neubert

   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 GFKALMAN_H
#define GFKALMAN_H

#include <map>
#include <iostream>

#include "TMatrixT.h"

class GFAbsRecoHit;
class GFAbsTrackRep;
class GFTrack;

class GFKalman {
 public:
  // friend class KalmanTester; // gives the Tester access to private methods

  // Constructors/Destructors ---------
  GFKalman();
  ~GFKalman();

  // Operators
  inline void operator()(GFTrack *track) { processTrack(track); }

  inline void operator()(std::pair<int, GFTrack *> tr) { processTrack(tr.second); }

  // Operations ----------------------

  void setLazy(Int_t) { std::cerr << "Using outdates setLazy method of class GFKalman:" << std::endl; }

  void setNumIterations(Int_t i) { fNumIt = i; }

  void processTrack(GFTrack *trk);

  void fittingPass(GFTrack *, int dir); // continues track from lastHitInFit

  double getChi2Hit(GFAbsRecoHit *, GFAbsTrackRep *);

  void setInitialDirection(int d) { fInitialDirection = d; }

  void setBlowUpFactor(double f) { fBlowUpFactor = f; }

  // Private Methods -----------------
 private:
  void processHit(GFTrack *, int, int, int);

  void switchDirection(GFTrack *trk); // switches the direction of propagation for all reps

  TMatrixT<double> calcGain(const TMatrixT<double> &cov, const TMatrixT<double> &HitCov, const TMatrixT<double> &H);

  double chi2Increment(const TMatrixT<double> &r, const TMatrixT<double> &H, const TMatrixT<double> &cov, const TMatrixT<double> &V);

  void blowUpCovs(GFTrack *trk);

  int fInitialDirection;
  Int_t fNumIt;
  double fBlowUpFactor;
  bool fSmooth;
};

#endif