25 |
|
|
26 |
|
#include "TMVA/Reader.h" |
27 |
|
|
28 |
+ |
namespace TMVA {//MVA |
29 |
+ |
class Reader; |
30 |
+ |
} |
31 |
+ |
|
32 |
|
namespace mithep { |
33 |
|
typedef std::vector<double> VertexZarray; |
34 |
|
typedef std::vector<const Track*> TrackArray; |
35 |
|
|
36 |
|
class VertexTools { |
37 |
|
public: |
38 |
< |
|
38 |
> |
|
39 |
> |
VertexTools(); |
40 |
> |
|
41 |
|
static double NewMass(const Photon* ph1, const Photon* ph2, const BaseVertex* vert); |
42 |
|
|
43 |
|
static VertexZarray ExtractZarray(const VertexCol* vcol, float zmin=-30, float zmax = 30, |
46 |
|
|
47 |
|
static const Vertex* BestVtx(const PFCandidateCol *fPFJets, const VertexCol *c, |
48 |
|
const BaseVertex *fBeamSpot, FourVector diboso); |
49 |
+ |
|
50 |
+ |
|
51 |
|
static double BestVtx(const PFCandidateCol *fPFJets, VertexZarray zcol, |
52 |
|
const BaseVertex *fBeamSpot, FourVector diboso); |
53 |
+ |
|
54 |
|
static double Prob(const PFCandidateCol *fPFJets, double zpos, |
55 |
|
const BaseVertex *fBeamSpot, FourVector diboso); |
56 |
|
|
61 |
|
|
62 |
|
static VertexTools* instance(const char* str){ |
63 |
|
if(meobject == NULL){ |
64 |
< |
meobject = new VertexTools(str); |
64 |
> |
meobject = new VertexTools(); |
65 |
> |
meobject->InitM(str); |
66 |
|
} |
67 |
|
return meobject; |
68 |
< |
} |
68 |
> |
} |
69 |
|
|
70 |
< |
Float_t tmvar1, tmvar2, tmvar3, tmvar4, tmvar5, tmvar6; |
71 |
< |
TMVA::Reader* reader; |
70 |
> |
// ---------------------------------------------------------- |
71 |
> |
// Methods (added by Fabian) on the EPS BaseLine Analysis |
72 |
> |
const Vertex* findVtxBasicRanking(const Photon* ph1, |
73 |
> |
const Photon* ph2, |
74 |
> |
const BaseVertex* bsp, |
75 |
> |
const VertexCol* vtcs, |
76 |
> |
const DecayParticleCol* conv, Bool_t useMva, Double_t &vtxProb); |
77 |
> |
// ---------------------------------------------------------- |
78 |
> |
|
79 |
> |
|
80 |
> |
static std::pair<double,double> VtxZFromConversion(const Photon *p, const DecayParticle *c, const BaseVertex *bsp); |
81 |
> |
|
82 |
> |
void InitM(const char* str); |
83 |
> |
void InitP(int version = 1); |
84 |
> |
|
85 |
> |
Bool_t IsInitMvaM() const { return fIsInitMvaM; } |
86 |
> |
Bool_t IsInitMvaP() const { return fIsInitMvaP; } |
87 |
> |
|
88 |
> |
static Double_t DeltaMassVtx(Double_t xp1, Double_t yp1, Double_t zp1, |
89 |
> |
Double_t xp2, Double_t yp2, Double_t zp2, |
90 |
> |
Double_t xv, Double_t yv, Double_t zv, |
91 |
> |
Double_t dz); |
92 |
|
|
93 |
|
private: |
94 |
+ |
|
95 |
+ |
double VtxMvaP(float ptbal, float ptasym, float logsumpt2, float limPullToConv, float nConv) const; |
96 |
+ |
|
97 |
|
static VertexTools *meobject; |
98 |
+ |
|
99 |
|
|
66 |
– |
VertexTools(); |
67 |
– |
VertexTools(const char* str); |
100 |
|
TString relname; |
101 |
+ |
|
102 |
|
TrackArray excluded; |
103 |
|
|
104 |
+ |
Bool_t fIsInitMvaM; |
105 |
+ |
Bool_t fIsInitMvaP; |
106 |
|
|
107 |
+ |
Float_t tmvar1, tmvar2, tmvar3, tmvar4, tmvar5, tmvar6; |
108 |
+ |
TMVA::Reader* reader; |
109 |
+ |
|
110 |
+ |
|
111 |
+ |
TMVA::Reader *readervtx; |
112 |
+ |
TMVA::Reader *readerevt; |
113 |
+ |
mutable Float_t fMvaPVars[5]; |
114 |
+ |
mutable Float_t fMvaPEvtVars[8]; |
115 |
+ |
|
116 |
+ |
TString fVtxProbMvaName; |
117 |
+ |
|
118 |
|
ClassDef(VertexTools, 0) // Muon tools |
119 |
|
}; |
120 |
|
} |