PndRecoKalmanFit2.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Kalman Filter for single tracks
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Stefano Spataro, UNI Torino      //
//      modified by Elisabetta Prencipe 19/05/2014
//-----------------------------------------------------------

#ifndef PNDRECOKALMANFIT2_HH
#define PNDRECOKALMANFIT2_HH

// Base Class Headers ----------------
#include "TNamed.h"
#include "TString.h"

// Collaborating Class Headers ------
#include "FairGeanePro.h"
#include "PndTrack.h"
#include "KalmanFitter.h"
#include "AbsMeasurement.h"
#include "MeasurementFactory.h"
// Collaborating Class Declarations --

class PndRecoKalmanFit2 : public TNamed {
 public:
  // Constructors/Destructors ---------
  PndRecoKalmanFit2();
  ~PndRecoKalmanFit2();

  // Modifiers -----------------------
  void SetGeane(Bool_t opt = kTRUE) { fUseGeane = opt; }
  void SetPropagateToIP(Bool_t opt = kTRUE) { fPropagateToIP = opt; }
  void SetPropagateDistance(Float_t opt = -1.f) { fPropagateDistance = opt; }
  void SetPerpPlane(Bool_t opt = kTRUE) { fPerpPlane = opt; }
  void SetNumIterations(Int_t num) { fNumIt = num; }
  void SetVerbose(Int_t verb) { fVerbose = verb; }
  void SetMvdBranchName(const TString &name) { fMvdBranchName = name; }
  void SetCentralTrackerBranchName(const TString &name) { fCentralTrackerBranchName = name; }
  // Operations ----------------------
  Bool_t Init();
  PndTrack *Fit(PndTrack *tBefore, Int_t PDG);

  genfit::MeasurementFactory<genfit::AbsMeasurement> *GetRecoHitFactory() { return fTheRecoHitFactory; };

 private:
  // Private Data Members ------------

  genfit::MeasurementFactory<genfit::AbsMeasurement> *fTheRecoHitFactory;
  genfit::KalmanFitter fGenFitter;

  FairGeanePro *fPro; 

  TString fMvdBranchName;            
  TString fCentralTrackerBranchName; 

  Bool_t fUseGeane;           
  Bool_t fPropagateToIP;      
  Float_t fPropagateDistance; 
  Bool_t fPerpPlane;          
  Int_t fNumIt;               
  Int_t fVerbose;             

  ClassDef(PndRecoKalmanFit2, 0);
};

#endif