| 1 |
khahn |
1.1 |
#include <TFile.h> // file handle class
|
| 2 |
|
|
#include <TString.h>
|
| 3 |
|
|
#include <TTree.h> // class to access ntuples
|
| 4 |
|
|
#include <TChain.h> // class to access ntuples
|
| 5 |
|
|
#include <iostream> // standard I/O
|
| 6 |
|
|
#include <iomanip> // functions to format standard I/O
|
| 7 |
|
|
#include <fstream> // functions for file I/O
|
| 8 |
|
|
#include <sstream> // class for parsing strings
|
| 9 |
|
|
#include <math.h>
|
| 10 |
|
|
#include <cassert>
|
| 11 |
|
|
#include <map>
|
| 12 |
|
|
|
| 13 |
|
|
#include "TMVA/Reader.h"
|
| 14 |
|
|
using namespace std;
|
| 15 |
|
|
|
| 16 |
|
|
class bdtiface
|
| 17 |
|
|
{
|
| 18 |
|
|
public:
|
| 19 |
|
|
bdtiface(TString weightDir="./data", TString cutfile="./data/bdtcuts.txt");
|
| 20 |
|
|
void readBDTCuts(TString file);
|
| 21 |
|
|
TString getBinName(double pt, double eta);
|
| 22 |
|
|
double getBDTCut(double pt, double eta, TString tightness);
|
| 23 |
|
|
Bool_t passBDT(Double_t pt, Double_t eta, TString tightness);
|
| 24 |
|
|
|
| 25 |
|
|
void setSigmaIEtaIEta (double x) { bdtvarEleSigmaIEtaIEta = x; };
|
| 26 |
|
|
void setDEtaIn (double x) { bdtvarEleDEtaIn = x; };
|
| 27 |
|
|
void setDPhiIn (double x) { bdtvarEleDPhiIn = x; };
|
| 28 |
|
|
void setD0 (double x) { bdtvarEleD0 = x; };
|
| 29 |
|
|
void setFBrem (double x) { bdtvarEleFBrem = x; };
|
| 30 |
|
|
void setEOverP (double x) { bdtvarEleEOverP = x; };
|
| 31 |
|
|
void setESeedClusterOverPout (double x) { bdtvarEleESeedClusterOverPout = x; };
|
| 32 |
|
|
void setSigmaIPhiIPhi (double x) { bdtvarEleSigmaIPhiIPhi = x; };
|
| 33 |
|
|
void setNBrem (double x) { bdtvarEleNBrem = x; };
|
| 34 |
|
|
void setOneOverEMinusOneOverP (double x) { bdtvarEleOneOverEMinusOneOverP = x; };
|
| 35 |
|
|
void setESeedClusterOverPIn (double x) { bdtvarEleESeedClusterOverPIn = x; };
|
| 36 |
|
|
|
| 37 |
|
|
private:
|
| 38 |
|
|
map<TString,TMVA::Reader*> readers;
|
| 39 |
|
|
vector<TString> binnamev;
|
| 40 |
|
|
map<TString,double> loosecuts,mediumcuts,tightcuts;
|
| 41 |
|
|
|
| 42 |
|
|
Float_t bdtvarEleSigmaIEtaIEta;
|
| 43 |
|
|
Float_t bdtvarEleDEtaIn;
|
| 44 |
|
|
Float_t bdtvarEleDPhiIn;
|
| 45 |
|
|
Float_t bdtvarEleD0;
|
| 46 |
|
|
Float_t bdtvarEleFBrem;
|
| 47 |
|
|
Float_t bdtvarEleEOverP;
|
| 48 |
|
|
Float_t bdtvarEleESeedClusterOverPout;
|
| 49 |
|
|
Float_t bdtvarEleSigmaIPhiIPhi;
|
| 50 |
|
|
Float_t bdtvarEleNBrem;
|
| 51 |
|
|
Float_t bdtvarEleOneOverEMinusOneOverP;
|
| 52 |
|
|
Float_t bdtvarEleESeedClusterOverPIn;
|
| 53 |
|
|
|
| 54 |
|
|
Float_t bdtvarEleIP3d;
|
| 55 |
|
|
Float_t bdtvarEleIP3dSig;
|
| 56 |
|
|
Float_t bdtvarStandardLikelihood;
|
| 57 |
|
|
Float_t bdtvarCicPass;
|
| 58 |
|
|
Float_t bdtvarPt;
|
| 59 |
|
|
Float_t bdtvarEta;
|
| 60 |
|
|
Float_t bdtvarSCeta;
|
| 61 |
|
|
Float_t bdtvarPFIso;
|
| 62 |
|
|
Float_t bdtvarIsConv;
|
| 63 |
|
|
Float_t bdtvarWeight;
|
| 64 |
|
|
Float_t bdtvarNExpHits;
|
| 65 |
|
|
|
| 66 |
|
|
};
|
