LZfluk.h

Go to the documentation of this file.

#ifndef LZFLUK_HH
#define LZFLUK_HH 
/***************************************************************************
 *       
 *   This function reweights pions and kaons depending on their pT & xF
 *
 *   Several functional forms are available which were used for:
 *   minos-1650 (SKZP and TV from PRL 2006 era), 
 *   minos-1945 (Boston 16 parameter fit)
 *   minos-2965 & minos-4166 
 *
 **************************************************************************/ 
#include <map>
#include <vector>
#include <iostream>
#include <fstream>
#include "Rtypes.h"
#include <cmath>

class TFile;
class TH2F;
class TH2D;

class Registry;

class LZfluk {

public:
   
   LZfluk(/*const char* dir=""*/);
   virtual ~LZfluk();

   void Config(const Registry &reg);

   void Init();
   
   void SetParameters(std::vector<double> par) {fPar=par; RecalculateWeights();};
   const std::vector<double>& GetParameters() {return fPar;};
   
   double GetWeight(int ptype, double pt, double pz);
   double CalcWeight(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)];};
   
   inline void UseParameterization(const char *which){
      std::string temp=which;
      if (temp=="SKZP") whichParameterization=0;
      else whichParameterization=1;
   };
   
private:
   typedef enum EParticleType 
   {
      kPiPlus  = 8,
      kPiMinus = 9,
      kKPlus   = 11,
      kKMinus  = 12,
      kK0L     = 10,
      kUnknown = 0
   } ParticleType_t;
 
   std::string GetParticleName(LZfluk::ParticleType_t ptype); 
   LZfluk::ParticleType_t GetParticleEnum(int ptype);
   void RecalculateWeights();
   
   int whichParameterization;
   
   std::vector<double> fPar;
   std::vector<ParticleType_t> fPlist;
   
   std::map<LZfluk::ParticleType_t, TH2F* > fPTXF;
   std::map<LZfluk::ParticleType_t, TH2F* > fWeightedPTXF;
   std::map<LZfluk::ParticleType_t, TH2D* > fWeightHist;
   std::map<LZfluk::ParticleType_t, double > fMeanPT;
   std::map<LZfluk::ParticleType_t, double > fN;
   std::map<LZfluk::ParticleType_t, double > fNWeighted;
   std::map<LZfluk::ParticleType_t, double > fWeightedMeanPT;
   std::map<LZfluk::ParticleType_t, int > fNBinsY,fNBinsX;

   

   bool fWghtFromHist;
   
   ClassDef(LZfluk,0)
      };

#endif

---