PndRecoKalmanTask.h

Go to the documentation of this file.

//-----------------------------------------------------------
// 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