Lfluk.h

Go to the documentation of this file.

#ifndef LFLUK_HH
#define LFLUK_HH 

#include <map>
#include <vector>
#include <iostream>
#include <fstream>
#include <cmath>
#include "Rtypes.h"
#include "TMVA/TSpline1.h"

class TFile;
class TH2F;
class TH2D;
class TH2;
class TF1;

class Registry;

class Lfluk {

public:

   Lfluk(/*const char* dir=""*/);
   virtual ~Lfluk();

   void SetConfig(const std::string param, const std::string skew, const std::string na49fit, const double wmin, const double wmax);

   void Config(const Registry &reg);

   void Init();

   void SetParameters(std::vector<double> par);
      
   const std::vector<double>& GetParameters() {return fPar;};
   
   double GetWeight(int ptype, double pt, double pz);

   double GetMeanPT(int ptype) {return fMeanPT[GetParticleEnum(ptype)];};
   double GetReweightedMeanPT(int ptype) {return fWeightedMeanPT[GetParticleEnum(ptype)];};
   double GetPTshift(int ptype, double pz_low=0.,double pz_up=120.);
   double GetNFrac(int ptype, double pz_low=0., double pz_up=120., double pt_low=0., double pt_up=1.);
   TH2F* GetReweightedPTXF(int ptype) {return fWeightedPTXF[GetParticleEnum(ptype)];};
   TH2F* GetPTXF(int ptype) {return fPTXF[GetParticleEnum(ptype)];};
   TH2D* GetWeightHistogram(int ptype) {return fWeightHist[GetParticleEnum(ptype)];};
   TH2* GetWeightHist(const std::string &ptype);
      
   
   TH1* GetFlukKPiRatio() {return fFlukKPiRat;};
   TH1* GetPiPlusReWProj() {return fPiPlusReWProj;};
   TH1* GetKPlusf05Proj()  {return fKPlusf05Proj;};
   TH1* GetKPlusf05PipReWRatio() {return fKPlusf05PipReWRatio;};

private:
   
   typedef enum EParticleType 
   {
      kPiPlus  = 8,
      kPiMinus = 9,
      kKPlus   = 11,
      kKMinus  = 12,
      kK0L     = 10,
      kUnknown = 0
   } ParticleType_t;

   double CalcWeight(int ptype, double pt, double pz);

   double GetSKZPWeight(const Lfluk::ParticleType_t particle, const double pT, const double xF);
   double GetSKPCWeight(const Lfluk::ParticleType_t particle, const double pT, const double xF);
   double GetQuadWeight(const Lfluk::ParticleType_t particle, const double pT, const double xF);
   double GetMeanPTWeight(const Lfluk::ParticleType_t particle, const double pT, const double xF);
   
   void SkewSKZP(const double xF, const double A, const double B, const double C, long double &Ap, long double &Bp, long double &Cp);
   void SkewSKPC(const double xF, const double A, const double B, const double C,
             const double D, const double E, double &Ap, double &Bp, double &Cp,
             double &Dp, double &Ep);
   
   double GetSKZPA(const Lfluk::ParticleType_t particle, double xf) const;
   double GetSKZPB(const Lfluk::ParticleType_t particle, double xf) const;
   double GetSKZPC(const Lfluk::ParticleType_t particle, double xf) const;

   double GetSKPCA(const Lfluk::ParticleType_t particle, double xf) const;
   double GetSKPCB(const Lfluk::ParticleType_t particle, double xf) const;
   double GetSKPCC(const Lfluk::ParticleType_t particle, double xf) const;
   double GetSKPCD(const Lfluk::ParticleType_t particle, double xf) const;
   double GetSKPCE(const Lfluk::ParticleType_t particle, double xf) const;

   std::string GetParticleName(Lfluk::ParticleType_t ptype); 

   Lfluk::ParticleType_t GetParticleEnum(const std::string &ptype);
   Lfluk::ParticleType_t GetParticleEnum(int ptype);

   void RecalculateWeights();

   
private:

   std::string fParam;
   std::string fSkew;
   std::string fNA49Fit;
   std::string fPath;

   double fMinW;
   double fMaxW;
   double fKPiRatScale;

   bool fWghtFromHist;
   bool fUseFlukPi;
   
   std::vector<double> fPar;
   std::vector<ParticleType_t> fPlist;
   
   std::map<Lfluk::ParticleType_t, TH2F* > fPTXF;
   std::map<Lfluk::ParticleType_t, TH2F* > fWeightedPTXF;
   std::map<Lfluk::ParticleType_t, TH2D* > fWeightHist;
   std::map<Lfluk::ParticleType_t, double > fMeanPT;
   std::map<Lfluk::ParticleType_t, double > fN;
   std::map<Lfluk::ParticleType_t, double > fNWeighted;
   std::map<Lfluk::ParticleType_t, double > fWeightedMeanPT;
   std::map<Lfluk::ParticleType_t, int > fNBinsY,fNBinsX;
   
   //TMVA::TSpline1* fNA49pirat;
   TF1* fNA49pirat;
   std::vector<double> fPiRatio;
   std::vector<double> fFluka05KPiRatio;
   std::vector<double> fFluka05PiRatio;
   std::vector<double> fKPiRatio;
   
   bool isFirst;
   
   TH1 *fFlukKPiRat;
   TH1 *fFlukPiRat;
   TH1 *fPiPlusReWProj;
   TH1 *fKPlusf05Proj;
   TH1 *fKPlusf05PipReWRatio;
   
   
 ClassDef(Lfluk,0)
};

#endif

---