1 |
bendavid |
1.1 |
//--------------------------------------------------------------------------------------------------
|
2 |
|
|
// $Id $
|
3 |
|
|
//
|
4 |
|
|
// EGEnergyCorrector
|
5 |
|
|
//
|
6 |
|
|
// Helper Class for photon Identification decisions.
|
7 |
|
|
//
|
8 |
|
|
// Authors: J.Bendavid
|
9 |
|
|
//--------------------------------------------------------------------------------------------------
|
10 |
|
|
|
11 |
|
|
#ifndef MITPHYSICS_UTILS_EGEnergyCorrector_H
|
12 |
|
|
#define MITPHYSICS_UTILS_EGEnergyCorrector_H
|
13 |
|
|
|
14 |
|
|
#include "MitAna/DataTree/interface/Photon.h"
|
15 |
|
|
#include "MitAna/DataTree/interface/PhotonCol.h"
|
16 |
|
|
#include "MitAna/DataTree/interface/TrackCol.h"
|
17 |
|
|
#include "MitAna/DataTree/interface/Electron.h"
|
18 |
|
|
#include "MitAna/DataTree/interface/ElectronCol.h"
|
19 |
|
|
#include "MitAna/DataTree/interface/BaseVertex.h"
|
20 |
|
|
#include "MitAna/DataTree/interface/DecayParticleCol.h"
|
21 |
|
|
#include "MitAna/DataTree/interface/VertexCol.h"
|
22 |
|
|
#include "MitAna/DataTree/interface/BeamSpotCol.h"
|
23 |
|
|
#include "MitAna/DataTree/interface/TriggerObjectCol.h"
|
24 |
|
|
#include "MitAna/DataTree/interface/MCParticleCol.h"
|
25 |
|
|
#include "MitAna/DataTree/interface/SuperCluster.h"
|
26 |
|
|
#include "MitAna/DataTree/interface/SuperClusterCol.h"
|
27 |
|
|
#include "MitCommon/MathTools/interface/MathUtils.h"
|
28 |
|
|
#include "MitPhysics/Utils/interface/PhotonFix.h"
|
29 |
|
|
#include "MitPhysics/Utils/interface/PhotonTools.h"
|
30 |
|
|
|
31 |
|
|
|
32 |
|
|
class TRandom3;
|
33 |
bendavid |
1.2 |
class GBRForest;
|
34 |
|
|
// namespace TMVA {
|
35 |
|
|
// class Reader;
|
36 |
|
|
// }
|
37 |
|
|
|
38 |
bendavid |
1.1 |
|
39 |
|
|
namespace mithep {
|
40 |
|
|
class EGEnergyCorrector {
|
41 |
|
|
public:
|
42 |
|
|
EGEnergyCorrector();
|
43 |
|
|
~EGEnergyCorrector();
|
44 |
|
|
|
45 |
bendavid |
1.5 |
void Initialize(TString phfixstring, TString phfixfile, TString regweights);
|
46 |
bendavid |
1.1 |
Bool_t IsInitialized() const { return fIsInitialized; }
|
47 |
|
|
|
48 |
bendavid |
1.7 |
void CorrectEnergyWithError(Photon *p, const VertexCol *vtxs = 0, Double_t rho = 0., UInt_t version=1, Bool_t applyRescale = kFALSE);
|
49 |
bendavid |
1.1 |
std::pair<double,double> CorrectedEnergyWithError(const Photon *p);
|
50 |
bendavid |
1.4 |
std::pair<double,double> CorrectedEnergyWithErrorV2(const Photon *p, const VertexCol *vtxs);
|
51 |
bendavid |
1.7 |
std::pair<double,double> CorrectedEnergyWithErrorV3(const Photon *p, const VertexCol *vtxs, Double_t rho, Bool_t applyRescale = kFALSE);
|
52 |
bendavid |
1.4 |
|
53 |
bendavid |
1.1 |
|
54 |
|
|
protected:
|
55 |
|
|
PhotonFix fPhFix;
|
56 |
bendavid |
1.2 |
GBRForest *fReadereb;
|
57 |
|
|
GBRForest *fReaderebvariance;
|
58 |
|
|
GBRForest *fReaderee;
|
59 |
|
|
GBRForest *fReadereevariance;
|
60 |
bendavid |
1.1 |
|
61 |
|
|
TString fMethodname;
|
62 |
|
|
|
63 |
|
|
Bool_t fIsInitialized;
|
64 |
|
|
Bool_t fIsMC;
|
65 |
bendavid |
1.2 |
|
66 |
|
|
Float_t *fVals;
|
67 |
bendavid |
1.1 |
|
68 |
|
|
ClassDef(EGEnergyCorrector, 0) // Muon tools
|
69 |
|
|
};
|
70 |
|
|
}
|
71 |
|
|
|
72 |
|
|
#endif
|