RootCore Packages: /cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/TauCorrUncert/TauCorrUncert/TauSF.h Source File

00001 #ifndef TAUSF_H
00002 #define TAUSF_H
00003 
00004 #include "CommonSF.h"
00005 
00006 namespace TauCorrUncert {
00007 
00008   typedef enum {
00009     IDNONE,
00010     BDTLOOSE,
00011     BDTMEDIUM,
00012     BDTTIGHT,
00013     BDTFAIL,
00014     BDTOTHER,
00015     LLHLOOSE,
00016     LLHMEDIUM,
00017     LLHTIGHT,
00018     LLHFAIL
00019   } ID;
00020 
00021   typedef enum {
00022     NONE,
00023     LOOSE,
00024     MEDIUM,
00025     TIGHT,
00026     OTHER
00027   } EVETO;
00028 
00029   typedef enum {
00030     OLRNONE,
00031     LOOSEPP,
00032     MEDIUMPP,
00033     TIGHTPP,
00034     OLROTHER
00035   } OLR;
00036 
00037 
00038   class TauSF : public CommonSF
00039   {
00040 
00041   public:
00042 
00043     TauSF(const std::string& sharepath,
00044       bool debug=false);
00045 
00046     ~TauSF()
00047     {
00048       delete m_mEVeto;
00049       delete m_mEVeto_3P;
00050     }
00051 
00052     double GetIDSF(ID level,
00053            double eta,
00054            int prongness,
00055            double pT=30);
00056 
00057     double GetIDSysUnc(ID level,
00058               double eta,
00059               int prongness,
00060               double pT=30);
00061 
00062     double GetIDStatUnc(ID level,
00063             double eta,
00064             int prongness,
00065             double pT=30);
00066 
00067     double GetExclIDSF(ID level,
00068                double eta,
00069                int prongness,
00070                double pT=30);
00071 
00072     double GetExclIDSysUnc(ID level,
00073                double eta,
00074                int prongness,
00075                double pT=30);
00076 
00077     double GetExclIDStatUnc(ID level,
00078                 double eta,
00079                 int prongness,
00080                 double pT=30);
00081 
00082     double GetEVetoSF(float trackEta,
00083               ID level,
00084               EVETO evLevel,
00085               OLR olr);
00086 
00087     double GetEVetoSFUnc(float trackEta,
00088              ID level,
00089              EVETO evLevel,
00090              OLR olr,
00091              int direction);
00092 
00093     double GetEVetoSF(float trackEta,
00094               int nTrack,
00095               ID level,
00096               EVETO evLevel,
00097               OLR olr);
00098 
00099     double GetEVetoSFUnc(float trackEta,
00100              int nTrack,
00101              ID level,
00102              EVETO evLevel,
00103              OLR olr,
00104              int direction);
00105     
00106     void SwitchOnPtBinnedIDSF(bool bSwitch = true);
00107     void SwitchOnEtaBinnedIDSF(bool bSwitch = true);
00108     void SwitchOnHighPtUncert(bool bSwitch = true);
00109 
00110   private:
00111     std::string ConvertEtaToString(const float& fEta);
00112     std::string ConvertIDToString(ID level);
00113     std::string ConvertEVetoToString(EVETO level);
00114     std::string ConvertOLRToString(OLR level);
00115     void checkTrackEtaValidity(float& trackEta);
00116     float GetIDValue(const std::string& sWorkingPoint, const float& fPt);
00117 
00118     SFMAP* m_mEVeto;
00119     SFMAP* m_mEVeto_3P;
00120     bool m_bUseHighPtUncert;
00121     std::string m_sIDpt;
00122     bool m_bUseInclusiveEta;
00123     ClassDef(TauSF, 1)
00124   };
00125 } // namespace TauCorrUncert
00126 
00127 #endif // TAUSF_H
00128 
00129 /*
00130 #ifndef TAUSF_H
00131 #define TAUSF_H
00132 
00133 #include <string>
00134 #include <vector>
00135 #include <map>
00136 #include <assert.h>
00137 #include <iostream>
00138 #include <typeinfo>
00139 
00140 #include "TROOT.h"
00141 #include "TClass.h"
00142 #include "TFile.h"
00143 #include "TGraph.h"
00144 #include "TH1F.h"
00145 #include "TH1D.h"
00146 #include "TObject.h"
00147 #include "TLorentzVector.h"
00148 #include "TKey.h"
00149 #include "TParameter.h"
00150 
00151 namespace TauCorrUncert {
00152 
00153   typedef enum {
00154     IDNONE,
00155     BDTLOOSE,
00156     BDTMEDIUM,
00157     BDTTIGHT,
00158     BDTFAIL,
00159     BDTOTHER,
00160     LLHLOOSE,
00161     LLHMEDIUM,
00162     LLHTIGHT,
00163     LLHFAIL
00164   } ID;
00165 
00166   typedef enum {
00167     NONE,
00168     LOOSE,
00169     MEDIUM,
00170     TIGHT,
00171     OTHER
00172   } EVETO;
00173 
00174   typedef enum {
00175     OLRNONE,
00176     LOOSEPP,
00177     MEDIUMPP,
00178     TIGHTPP,
00179     OLROTHER
00180   } OLR;
00181 
00182 
00183   class TauSF : public TObject
00184   {
00185 
00186   public:
00187 
00188     TauSF(const std::string& sharepath,
00189       bool debug=false);
00190 
00191     ~TauSF()
00192     {
00193       delete m_mSF;
00194       delete m_mEVeto;
00195       delete m_mEVeto_3P;
00196     }
00197 
00198     double GetIDSF(ID level,
00199            double eta,
00200            int prongness,
00201            double pT=30);
00202 
00203     double GetIDSysUnc(ID level,
00204               double eta,
00205               int prongness,
00206               double pT=30);
00207 
00208     double GetIDStatUnc(ID level,
00209             double eta,
00210             int prongness,
00211             double pT=30);
00212 
00213     double GetExclIDSF(ID level,
00214                double eta,
00215                int prongness,
00216                double pT=30);
00217 
00218     double GetExclIDSysUnc(ID level,
00219                double eta,
00220                int prongness,
00221                double pT=30);
00222 
00223     double GetExclIDStatUnc(ID level,
00224                 double eta,
00225                 int prongness,
00226                 double pT=30);
00227 
00228     double GetEVetoSF(float trackEta,
00229               ID level,
00230               EVETO evLevel,
00231               OLR olr);
00232 
00233     double GetEVetoSFUnc(float trackEta,
00234              ID level,
00235              EVETO evLevel,
00236              OLR olr,
00237              int direction);
00238 
00239     double GetEVetoSF(float trackEta,
00240               int nTrack,
00241               ID level,
00242               EVETO evLevel,
00243               OLR olr);
00244 
00245     double GetEVetoSFUnc(float trackEta,
00246              int nTrack,
00247              ID level,
00248              EVETO evLevel,
00249              OLR olr,
00250              int direction);
00251     
00252     void SwitchOnPtBinnedIDSF(bool bSwitch = true);
00253     void SwitchOnEtaBinnedIDSF(bool bSwitch = true);
00254     void SwitchOnHighPtUncert(bool bSwitch = true);
00255 
00256   private:
00257 
00258     std::string ConvertProngToString(int prongness);
00259     std::string ConvertIDToString(ID level);
00260     std::string ConvertEVetoToString(EVETO level);
00261     std::string ConvertOLRToString(OLR level);
00262     std::string ConvertEtaToString(const float& fEta);
00263     void checkTrackEtaValidity(float& trackEta);
00264     float GetIDValue(const std::string& sWorkingPoint, const float& fPt);
00265 
00266     typedef std::map<std::string, TH1F*> SFMAP;
00267     SFMAP* m_mSF;
00268     SFMAP* m_mEVeto;
00269     SFMAP* m_mEVeto_3P;
00270     bool m_bUseHighPtUncert;
00271     std::string m_sIDpt;
00272     bool m_bUseInclusiveEta;
00273     bool m_debug;
00274     ClassDef(TauSF, 1)
00275 
00276     template<class T>
00277     void ReadInputs(TFile* f, std::map<std::string, T>* map)
00278     {
00279       if(!f || !f->IsOpen())
00280     {
00281       Fatal("ReadInputs", "Could not open %s", f->GetName());
00282       return;
00283     }
00284       assert(f->IsOpen());
00285       TKey *key;
00286       TIter next(f->GetListOfKeys());
00287       while ((key = (TKey*)next()))
00288     {
00289       TClass *cl = gROOT->GetClass(key->GetClassName());
00290       if (!cl->InheritsFrom("TH1"))
00291         continue;
00292       T obj = (T)key->ReadObj();
00293       obj->SetDirectory(0);
00294       (*map)[std::string(obj->GetName())] = obj;
00295     }
00296       Info("ReadInputs", "data loaded from %s", f->GetName());
00297     }
00298   };
00299 } // namespace TauCorrUncert
00300 
00301 #endif // TAUSF_H
00302 */