CbmHRGModel.h

Go to the documentation of this file.

/*
 *====================================================================
 *
 *  CBM Hadron Resonance Gas
 *  
 *  Authors: V.Vovchenko
 *
 *  e-mail : 
 *
 *====================================================================
 *
 *  Statistical hadron-resonance gas model calculations
 *
 *====================================================================
 */
 
#ifndef _CbmHRGModel_h_
#define _CbmHRGModel_h_
 
#include "CbmBilinearSplineFunction.h"
#include "CbmModelBase.h"
#include "KFQA/CbmKFTrErrMCPoints.h"
#include "TMath.h"
#include "ThermalModelFit.h"
#include <algorithm>
#include <map>
#include <vector>
 
class TClonesArray;
class CbmVertex;
class TDirectory;
class TH1F;
class TH2F;
class TGraphErrors;
class TSpline3;
class FairRootManager;
 
namespace HRGModelNamespace {
 
  struct AcceptanceFunction {
    Double_t dy, dpt;
    std::vector<Double_t> ys, pts, probs;
    BilinearSplineFunction sfunc;
    void setSpline() { sfunc.setData(ys, pts, probs); }
    Double_t getAcceptance(const Double_t& y, const Double_t& pt) const;
    AcceptanceFunction() : dy(), dpt(), ys(), pts(), probs(), sfunc() {}
  };
 
  struct ReconstructionEfficiencyFunction {
    Double_t p1, p2, p3;
    ReconstructionEfficiencyFunction(double p1_ = 1.,
                                     double p2_ = 0.,
                                     double p3_ = 1.)
      : p1(p1_), p2(p2_), p3(p3_) {}
    Double_t f(double p) { return p1 - p2 * TMath::Exp(-p * p / 2. / p3 / p3); }
  };
}  // namespace HRGModelNamespace
 
using namespace HRGModelNamespace;
 
class ThermalModelBase;
 
class CbmHRGModel : public CbmModelBase {
public:
  //static const int p_sz = 8;
  CbmHRGModel(
    Int_t recoLevel =
      -1,  // -1 - MC Tracks, 0 - Reconstructible MC Tracks, 1 - Reco Tracks with MC Data, 2 - Reco Tracks with Reco Momentum and MC Primary, 3 - Reco Tracks with Reco Momentum and Reco Primary
    //Int_t usePID = 1, // 1 - MC, 2 - TOF
    //Int_t trackNumber = 1,
    Int_t iVerbose                  = 1,
    TString Mode                    = "MC",
    Int_t EventStats                = 1,
    KFParticleTopoReconstructor* tr = 0,
    Bool_t useWidth                 = 0,
    Bool_t useStatistics            = 0,
    Double_t rad                    = 0.);
  ~CbmHRGModel();
 
  virtual void ReInit(FairRootManager* fManger);
  virtual void Init();
  virtual void Exec();
  virtual void Finish();
 
  void AddRatio(int pdgid1,
                int pdgid2,
                /*TString name1="", TString name2="",*/ double SystError = 0.);
 
  ClassDef(CbmHRGModel, 1);
 
private:
  void ReadAcceptanceFunction(HRGModelNamespace::AcceptanceFunction& func,
                              TString filename);
 
  Double_t RecEfficiency(Double_t p);
  bool checkIfReconstructable(CbmKFTrErrMCPoints* inTrack);  // 0 - No, 1 - Yes
 
  void CalculateMultiplicitiesInEvent(std::vector<int>& Mult,
                                      int RecoLevel,
                                      bool UpdateGlobal = 0);
 
  ThermalModelFitParameters GetThermalParameters(const std::vector<int>& Mults);
 
  Float_t ekin;
  Float_t ycm;
  Bool_t fUpdate;
  Int_t fusePID;
  Int_t fRecoLevel;
  Int_t fTrackNumber;  // 0 - test calculation, only Pions, 1 - UrQMD Au+Au
  Int_t fEventStats;
  Int_t events;
 
  Bool_t fUseWidth, fUseStatistics;
  double fRadius;
 
 
  TString fModeName;
  TString outfileName;
 
  TDirectory* histodir;
 
  TClonesArray* flistMCTracks;
 
  //Int_t events;
 
  static const int nHisto1D =
    15;  //T,errT,muB,errmuB,muS,errmuS,muQ,errmuQ,nB,energy,entropy,pressure,energy per particle,etaovers,chi2fit
  int IndexTemperature, IndexErrorTemperature, IndexMuB, IndexErrorMuB,
    IndexMuS, IndexErrorMuS, IndexMuQ, IndexErrorMuQ, IndexnB, IndexEnergy,
    IndexEntropy, IndexPressure, IndexEoverN, IndexEtaoverS, IndexChi2Fit;
 
  TH1F* histo1D[nHisto1D];
 
  static const int nHisto2D = 3;  //T-muB,T-nB,T-energy
  int IndexTmuB, IndexTnB, IndexTE;
 
  TH2F* histo2D[nHisto2D];
 
  TGraphErrors* grTmuB;
  TH1F *pullT, *pullmuB;
  std::vector<double> Ts, muB, errT, errmuB;
 
  std::map<int, int> PDGtoIndex;
  std::vector<int> ParticlePDGsTrack;
  std::vector<TString> ParticleNames;
  std::vector<std::pair<int, int>> RatiosToFit;
  std::vector<double> SystErrors;
  std::vector<int> MultGlobal, MultLocal;
 
  HRGModelNamespace::AcceptanceFunction AcceptanceSTS, AcceptanceSTSTOF;
 
  CbmHRGModel(const CbmHRGModel&);
  CbmHRGModel& operator=(const CbmHRGModel&);
 
  ThermalParticleSystem TPS;
 
  ThermalModelBase* model;
};
 
#endif