12 |
|
#include "CondFormats/JetMETObjects/interface/FactorizedJetCorrector.h" |
13 |
|
#include <vector> |
14 |
|
#include "MitAna/DataTree/interface/PFTau.h" |
15 |
+ |
#include "CondFormats/EgammaObjects/interface/GBRForest.h" |
16 |
|
|
17 |
|
class TRandom3; |
18 |
|
namespace TMVA { |
48 |
|
TauIsoMVA(); |
49 |
|
~TauIsoMVA(); |
50 |
|
void Initialize (TString iWeightFile); |
51 |
+ |
void InitializeGBR (TString iWeightFile); |
52 |
|
double MVAValue (const PFTau *iTau,double iRho); |
53 |
|
IsoRings computeIsoRings(const PFTau *iTau); |
54 |
|
|
55 |
|
protected: |
56 |
+ |
bool fGBR; |
57 |
+ |
GBRForest *fGBRReader; |
58 |
|
TMVA::Reader *fReader; |
59 |
< |
ClassDef(TauIsoMVA,0) |
59 |
> |
ClassDef(TauIsoMVA,1) |
60 |
|
}; |
61 |
|
} |
62 |
|
#endif |