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 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* fFitTrackArrayElectron; 
    TClonesArray* fFitTrackArrayMuon;     
    TClonesArray* fFitTrackArrayPion;     
    TClonesArray* fFitTrackArrayKaon;     
    TClonesArray* fFitTrackArrayProton;   

    TString fTrackInBranchName;      
    TString fTrackOutBranchName;     

    TString fMvdBranchName;           
    TString fCentralTrackerBranchName;

    TString fFitWithHypo;

    PndRecoKalmanFit *fFitter;

    Bool_t fPersistence;           
    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];                                
    PndGeoSttPar *fSttParameters;  
    PndGeoFtsPar *fFtsParameters;  

    TDatabasePDG *pdg;             










    ClassDef(PndRecoMultiKalmanTask,1);

};

#endif