21 |
|
#include "MitAna/DataTree/interface/TriggerObjectCol.h" |
22 |
|
#include "MitCommon/MathTools/interface/MathUtils.h" |
23 |
|
#include "MitPhysics/Utils/interface/VertexMVA.h" |
24 |
+ |
#include "MitAna/DataTree/interface/PFCandidateCol.h" |
25 |
|
|
26 |
+ |
#include "TMVA/Reader.h" |
27 |
|
|
28 |
|
namespace mithep { |
29 |
+ |
typedef std::vector<double> VertexZarray; |
30 |
+ |
|
31 |
|
class VertexTools { |
32 |
< |
public: |
33 |
< |
VertexTools(); |
34 |
< |
|
35 |
< |
static const Vertex* BestVtx(const VertexCol *c, const VertexMVA *mva); |
36 |
< |
|
32 |
> |
public: |
33 |
> |
|
34 |
> |
static double NewMass(const Photon* ph1, const Photon* ph2, const BaseVertex* vert); |
35 |
> |
|
36 |
> |
static VertexZarray ExtractZarray(const VertexCol* vcol, float zmin=-30, float zmax = 30, |
37 |
> |
const BaseVertex *fBeamSpot = NULL); |
38 |
> |
static VertexZarray ExtractZarray(float zmin=-30, float zmax=30, float step=0.05); |
39 |
> |
|
40 |
> |
static const Vertex* BestVtx(const PFCandidateCol *fPFJets, const VertexCol *c, |
41 |
> |
const BaseVertex *fBeamSpot, FourVector diboso); |
42 |
> |
static double BestVtx(const PFCandidateCol *fPFJets, VertexZarray zcol, |
43 |
> |
const BaseVertex *fBeamSpot, FourVector diboso); |
44 |
> |
static double Prob(const PFCandidateCol *fPFJets, double zpos, |
45 |
> |
const BaseVertex *fBeamSpot, FourVector diboso); |
46 |
> |
|
47 |
> |
static double VertexWidth(const Vertex*, const BaseVertex* ); |
48 |
> |
|
49 |
> |
static VertexTools* instance(const char* str){ |
50 |
> |
if(meobject == NULL){ |
51 |
> |
meobject = new VertexTools(str); |
52 |
> |
} |
53 |
> |
return meobject; |
54 |
> |
} |
55 |
> |
|
56 |
> |
Float_t tmvar1, tmvar2, tmvar3, tmvar4, tmvar5, tmvar6; |
57 |
> |
TMVA::Reader* reader; |
58 |
> |
|
59 |
> |
private: |
60 |
> |
static VertexTools *meobject; |
61 |
> |
|
62 |
> |
VertexTools(); |
63 |
> |
VertexTools(const char* str); |
64 |
> |
TString relname; |
65 |
> |
|
66 |
|
ClassDef(VertexTools, 0) // Muon tools |
67 |
< |
}; |
67 |
> |
}; |
68 |
|
} |
69 |
|
|
70 |
|
#endif |