PndRecoMultiKalmanTask.h

Go to the documentation of this file.

//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Kalman Filter Task for multiple particle hypothesis
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Sebastian Neubert    TUM            (original author)
//      Stefano Spatarot, UNI Torino
//
//-----------------------------------------------------------

#ifndef PNDRECOMULTIKALMANTASK_HH
#define PNDRECOMULTIKALMANTASK_HH

// Base Class Headers ----------------
#include "PndPersistencyTask.h"

// Collaborating Class Headers -------
#include "TString.h"
#include "PndRecoKalmanFit.h"
#include "PndGeoSttPar.h"
#include "PndGeoFtsPar.h"

// Collaborating Class Declarations --
class TClonesArray;
class GFRecoHitFactory;

class PndRecoMultiKalmanTask : public PndPersistencyTask {
 public:
  // Constructors/Destructors ---------
  PndRecoMultiKalmanTask(const char *name = "Genfit", Int_t iVerbose = 0, TString fithypo = "electron;muon;pion;kaon;proton");
  ~PndRecoMultiKalmanTask();

  // Operators

  // Accessors -----------------------

  // Modifiers -----------------------
  void SetTrackInBranchName(const TString &name) { fTrackInBranchName = name; }
  void SetTrackOutBranchName(const TString &name) { fTrackOutBranchName = name; }
  void SetMvdBranchName(const TString &name) { fMvdBranchName = name; }
  void SetCentralTrackerBranchName(const TString &name) { fCentralTrackerBranchName = name; }
  void SetGeane(Bool_t opt = kTRUE) { fUseGeane = opt; }
  void SetNumIterations(Int_t num) { fNumIt = num; }
  void SetFitHypotheses(const TString &name) { fFitWithHypo = name; }
  void SetPropagateToIP(Bool_t opt = kTRUE) { fPropagateToIP = opt; }
  void SetBusyCut(Int_t b) { fBusyCut = b; }
  void SetTrackRep(Short_t num) { fTrackRep = num; }

  // Operations ----------------------
  virtual InitStatus Init();
  virtual void Exec(Option_t *opt);

  void SetParContainers();

 private:
  // Private Data Members ------------
  TClonesArray *fTrackArray;        
  TClonesArray *fFitTrackArrays[5]; 

  TString fTrackInBranchName;  
  TString fTrackOutBranchName; 

  TString fMvdBranchName;            
  TString fCentralTrackerBranchName; 

  TString fFitWithHypo;

  PndRecoKalmanFit *fFitter;

  Bool_t fUseGeane;             
  Bool_t fIdealHyp;             
  Bool_t fPropagateToIP;        
  Float_t fPropagateDistance;   
  Bool_t fPerpPlane;            
  Short_t fTrackRep;            
  Int_t fNumIt;                 
  Int_t fBusyCut;               
  Bool_t fSmoothing;            
  Bool_t fHypoFlag[5];          
  int fPDGs[5];                 
  PndGeoSttPar *fSttParameters; 
  PndGeoFtsPar *fFtsParameters; 

  TDatabasePDG *pdg; 

  ClassDef(PndRecoMultiKalmanTask, 1);
};

#endif