BmnPid.h

Go to the documentation of this file.

#ifndef BMNPID_H
#define BMNPID_H
 
#include "BmnGlobalTrack.h"
 
#include "FairTask.h"
#include "FairRunAna.h"
 
#include "TClonesArray.h"
#include "TDatabasePDG.h"
#include "TParticlePDG.h"
#include "TString.h"
#include "TMath.h"
#include "BmnEnums.h"
 
#include "vector"
using namespace std;
 
class BmnPid : public FairTask
{
  public:
    // Constructors/Destructors ---------
    BmnPid(Int_t power = 1);
 
    virtual ~BmnPid();
 
    virtual InitStatus Init();
    virtual void Exec(Option_t* opt);
    virtual void Finish();
 
    void SetVector();   
 
  private:
    Int_t EnumToPdg(PidParticles part);
    
    void NormalizeVector(vector<Double_t>& vec);
    Double_t GetSum(const vector<Double_t>& vec);   
    
    Double_t EstimateProbability(Double_t p, Double_t beta, Double_t mass, Int_t power);
    
    // Private Data Members ------------
    TString fGlobalTracksBranchName; 
    TDatabasePDG* db = nullptr;
 
    Int_t fModelPower; // number of using identification model
                       // 1 - first order distance model
                       // 2 - second order distance model
                       // 3 - third order distance model
                       // ...
                            
    Int_t fEventNo; // event counter
    
    TClonesArray* fGlobalTracksArray;   
 
  ClassDef(BmnPid, 1);
};
 
#endif /* BMNPID_H */