CbmTrdSetTracksPidLike.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                    CbmTrdSetTracksPidLike header file          -----
// -----                  Created 25/02/2007  by F. Uhlig              -----
// -----                  Updated 31/08/2016  by J. Book               -----
// -------------------------------------------------------------------------
 
 
#ifndef CBMTRDSETTRACKSPIDLIKE_H
#define CBMTRDSETTRACKSPIDLIKE_H 1
 
#include "FairTask.h"
 
class TClonesArray;
 
class CbmTrdSetTracksPidLike : public FairTask {
 
public:
  CbmTrdSetTracksPidLike();
 
 
  CbmTrdSetTracksPidLike(const char* name, const char* title = "FairTask");
 
 
  virtual ~CbmTrdSetTracksPidLike();
 
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* opt);
 
 
  virtual void Finish();
 
 
  virtual void SetParContainers();
 
 
  Int_t GetNofTracks() const { return fNofTracks; };
 
  void SetInputFileName(const char* file) { fFileName = file; }
  void SetUseMCInfo(Bool_t use = kTRUE) { fMCinput = use; }
  void SetUseMomDependence(Bool_t use = kTRUE) { fMomDep = use; }
 
private:
  CbmTrdSetTracksPidLike& operator=(const CbmTrdSetTracksPidLike&);
  CbmTrdSetTracksPidLike(const CbmTrdSetTracksPidLike&);
 
  TString fFileName = "";      // input file
  Bool_t fMCinput   = kFALSE;  // use MC information for input histograms
  Bool_t fMomDep    = kTRUE;   // use momentum dependence for input histograms
  TClonesArray* fTrackArray       = NULL; 
  TClonesArray* fTrdHitArray      = NULL; 
  TClonesArray* fglobalTrackArray = NULL; 
  TObjArray* fHistdEdx = NULL;  // Prob. of dEdx for 5 particle species
  Int_t fNofTracks     = 0;     // Number of tracks successfully fitted
 
  static const Int_t fgkNParts = 5;  // numer of different particle species
  enum EParticleType {
    kElectron = 0,
    kPion     = 1,
    kKaon     = 2,
    kProton   = 3,
    kMuon     = 4,
  };
 
  Bool_t ReadData();
 
  Double_t GetProbability(Int_t iType, Double_t mom, Double_t dedx) const;
 
 
  ClassDef(CbmTrdSetTracksPidLike, 1);
};
 
#endif