de/d3d/PndRecoKalmanTask_8h_source.html

//-----------------------------------------------------------

// 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 "FairTask.h"


// Collaborating Class Headers -------

#include "TString.h"

#include "PndRecoKalmanFit.h"

#include "PndRecoDafFit.h"


// Collaborating Class Declarations --

class TClonesArray;

class GFRecoHitFactory;


class PndRecoKalmanTask : public FairTask {

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 SetPerpPlane(Bool_t opt = kTRUE)            { fPerpPlane = opt;           }

  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);


  void StoreTrackParameterization( void )          { mStoreTrackParameterization = true; };


protected:


  // Private Data Members ------------

  TClonesArray* fTrackArray;

  TClonesArray* fTrackIDArray;

  TClonesArray* fMCTrackArray;

  TClonesArray* fFitTrackArray;


  TString fTrackInBranchName;

  TString fTrackInIDBranchName;

  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;

  Bool_t fPerpPlane;

  Short_t fTrackRep;

  Int_t fNumIt;

  Int_t fPDGHyp;

  Int_t fBusyCut;


  // May want to instruct Kalman filter to record track parameterization (state

  // vector) at the hit locations;

  bool mStoreTrackParameterization;


  ClassDef(PndRecoKalmanTask,2);


};


#endif