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 "larreco/Genfit/GFDetPlane.h"
#include <iostream>

#include "RtypesCore.h"
#include "TMatrixT.h"

namespace genf {

  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 /* flag */)
    {
      std::cerr << "Using outdates setLazy method of class GFKalman:" << std::endl;
    }

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

    void processTrack(GFTrack*);

    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; }
    void setMomLow(Double_t f) { fMomLow = f; }
    void setMomHigh(Double_t f) { fMomHigh = f; }
    void setMaxUpdate(Double_t f) { fMaxUpdate = f; }
    void setErrorScaleSTh(Double_t f) { fErrScaleSTh = f; }
    void setErrorScaleMTh(Double_t f) { fErrScaleMTh = f; }

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

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

    TMatrixT<Double_t> calcGain(const TMatrixT<Double_t>& cov,
                                const TMatrixT<Double_t>& HitCov,
                                const TMatrixT<Double_t>& H);
    TMatrixT<Double_t> calcCov7x7(const TMatrixT<Double_t>& cov, const genf::GFDetPlane& plane);

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

    void blowUpCovs(GFTrack* trk);
    void blowUpCovsDiag(GFTrack* trk);

    int fInitialDirection;
    Int_t fNumIt;
    double fBlowUpFactor;

    Double_t fMomLow;
    Double_t fMomHigh;
    Double_t fMaxUpdate;
    Double_t fErrScaleSTh; // simulated theta error scale
    Double_t fErrScaleMTh; // measured theta error scale
    bool fGENfPRINT;
    //TH1D* fUpdate;
    //TH1D* fIhitvUpdate;
  };

} // namespace genf
#endif