CbmTrdSetTracksPidLike.h

Go to the documentation of this file.

/* Copyright (C) 2019 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Florian Uhlig [committer] */
 
// -------------------------------------------------------------------------
// -----                    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 "CbmTrdParSetGas.h"
#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       = nullptr; 
  TClonesArray* fTrdHitArray      = nullptr; 
  TClonesArray* fglobalTrackArray = nullptr; 
  TObjArray* fHistdEdx = nullptr;  // Prob. of dEdx for 5 particle species
  Int_t fNofTracks     = 0;        // Number of tracks successfully fitted
 
  CbmTrdParSetGas* fGasPar = nullptr;  // Gas parameters, path to input reference spectra located here
 
  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, 2);
};
 
#endif