| 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, |
| 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 |
|
} |
| 69 |
|
|
| 70 |
|
// ---------------------------------------------------------- |
| 71 |
|
// Methods (added by Fabian) on the EPS BaseLine Analysis |
| 72 |
< |
static const Vertex* findVtxBasicRanking(const Photon* ph1, |
| 72 |
> |
const Vertex* findVtxBasicRanking(const Photon* ph1, |
| 73 |
|
const Photon* ph2, |
| 74 |
|
const BaseVertex* bsp, |
| 75 |
|
const VertexCol* vtcs, |
| 76 |
< |
const DecayParticleCol* conv = NULL); |
| 76 |
> |
const DecayParticleCol* conv, Bool_t useMva, Double_t &vtxProb); |
| 77 |
|
// ---------------------------------------------------------- |
| 78 |
|
|
| 79 |
< |
Float_t tmvar1, tmvar2, tmvar3, tmvar4, tmvar5, tmvar6; |
| 80 |
< |
TMVA::Reader* reader; |
| 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 |
< |
VertexTools(); |
| 79 |
< |
VertexTools(const char* str); |
| 99 |
> |
|
| 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 |
|
} |
