PndRecoKalmanTask.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 Task
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Sebastian Neubert    TUM            (original author)
//      Stefano Spataro, UNI Torino
//
//-----------------------------------------------------------

#ifndef PNDRECOKALMANTASK_HH
#define PNDRECOKALMANTASK_HH

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

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

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

class PndRecoKalmanTask : public PndPersistencyTask {
 public:
  // Constructors/Destructors ---------
  PndRecoKalmanTask(const char *name = "Genfit", Int_t iVerbose = 0);
  ~PndRecoKalmanTask();

  // Operators

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

  // Modifiers -----------------------
  void SetTrackInBranchName(const TString &name) { fTrackInBranchName = name; }
  //  void SetTrackInIDBranchName(const TString& name) { fTrackInIDBranchName = name;}
  void SetTrackOutBranchName(const TString &name) { fTrackOutBranchName = name; }
  void SetMvdBranchName(const TString &name) { fMvdBranchName = name; }
  void SetCentralTrackerBranchName(const TString &name) { fCentralTrackerBranchName = name; }
  void SetPersistence(Bool_t opt = kTRUE) { fPersistence = opt; }
  void SetGeane(Bool_t opt = kTRUE) { fUseGeane = opt; }
  void SetIdealHyp(Bool_t opt = kTRUE) { fIdealHyp = opt; }
  void SetDaf(Bool_t opt = kTRUE) { fDaf = opt; }
  void SetPropagateToIP(Bool_t opt = kTRUE) { fPropagateToIP = opt; }
  void SetPropagateDistance(Float_t opt = -1.) { fPropagateDistance = opt; }
  void SetPerpPlane(Bool_t opt = kTRUE) { fPerpPlane = opt; }
  void SetNumIterations(Int_t num) { fNumIt = num; }
  void SetTrackRep(Short_t num) { fTrackRep = num; }
  void SetParticleHypo(TString s);
  void SetParticleHypo(Int_t h);
  void SetBusyCut(Int_t b) { fBusyCut = b; }

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

 protected:
  // Private Data Members ------------
  TClonesArray *fTrackArray; 
  //  TClonesArray* fTrackIDArray;    //! Input TCA for PndTrackID
  TClonesArray *fMCTrackArray;  
  TClonesArray *fFitTrackArray; 

  TString fTrackInBranchName; 
  //  TString fTrackInIDBranchName;    //! Name of the input TCA
  TString fTrackOutBranchName; 

  TString fMvdBranchName;            
  TString fCentralTrackerBranchName; 

  PndRecoKalmanFit *fFitter;
  PndRecoDafFit *fDafFitter;
  TDatabasePDG *pdg; 

  Bool_t fPersistence; 

  Bool_t fUseGeane;           
  Bool_t fSmoothing;          
  Bool_t fIdealHyp;           
  Bool_t fDaf;                
  Bool_t fPropagateToIP;      
  Float_t fPropagateDistance; 
  Bool_t fPerpPlane;          
  Short_t fTrackRep;          
  Int_t fNumIt;               
  Int_t fPDGHyp;              
  Int_t fBusyCut;             

  ClassDef(PndRecoKalmanTask, 1);
};

#endif