3 |
|
// |
4 |
|
// Vertex |
5 |
|
// |
6 |
< |
// Vertex class implemented as holding a 3d vector as a point in space with fit information |
6 |
> |
// Vertex class implemented as holding a 3d vector as a point in space with fit information. |
7 |
|
// |
8 |
|
// Authors: J.Bendavid |
9 |
|
//-------------------------------------------------------------------------------------------------- |
10 |
|
|
11 |
< |
#ifndef DATATREE_VERTEX_H |
12 |
< |
#define DATATREE_VERTEX_H |
11 |
> |
#ifndef MITANA_DATATREE_VERTEX_H |
12 |
> |
#define MITANA_DATATREE_VERTEX_H |
13 |
|
|
14 |
|
#include <TMath.h> |
15 |
< |
#include "MitAna/DataTree/interface/DataObject.h" |
15 |
> |
#include "MitAna/DataTree/interface/BaseVertex.h" |
16 |
|
|
17 |
|
namespace mithep |
18 |
|
{ |
19 |
< |
class Vertex : public DataObject |
19 |
> |
class Vertex : public BaseVertex |
20 |
|
{ |
21 |
|
public: |
22 |
< |
Vertex() : |
23 |
< |
fPosition(0,0,0), |
24 |
< |
fXErr(0.0), |
25 |
< |
fYErr(0.0), |
26 |
< |
fZErr(0.0), |
27 |
< |
fChi2(0.0), |
28 |
< |
fNdof(0), |
29 |
< |
fNTracks(0) {} |
30 |
< |
Vertex(Double_t x, Double_t y, Double_t z) : |
31 |
< |
fPosition(x,y,z), |
32 |
< |
fXErr(0.0), |
33 |
< |
fYErr(0.0), |
34 |
< |
fZErr(0.0), |
35 |
< |
fChi2(0.0), |
36 |
< |
fNdof(0), |
37 |
< |
fNTracks(0) {} |
22 |
> |
Vertex() : fChi2(0), fNdof(0), fNTracks(0) {} |
23 |
> |
Vertex(Double_t x, Double_t y, Double_t z) : |
24 |
> |
BaseVertex(x,y,z), fChi2(0), fNdof(0), fNTracks(0) {} |
25 |
|
Vertex(Double_t x, Double_t y, Double_t z, Double_t xErr, Double_t yErr, Double_t zErr) : |
26 |
< |
fPosition(x,y,z), |
27 |
< |
fXErr(xErr), |
28 |
< |
fYErr(yErr), |
42 |
< |
fZErr(zErr), |
43 |
< |
fChi2(0.0), |
44 |
< |
fNdof(0), |
45 |
< |
fNTracks(0) {} |
46 |
< |
Vertex(const ThreeVector &pos) : |
47 |
< |
fPosition(pos), |
48 |
< |
fXErr(0.0), |
49 |
< |
fYErr(0.0), |
50 |
< |
fZErr(0.0), |
51 |
< |
fChi2(0.0), |
52 |
< |
fNdof(0), |
53 |
< |
fNTracks(0) {} |
54 |
< |
~Vertex() {} |
26 |
> |
BaseVertex(x,y,z,xErr,yErr,zErr), fChi2(0), fNdof(0), fNTracks(0) {} |
27 |
> |
Vertex(const ThreeVector &pos) : |
28 |
> |
BaseVertex(pos), fChi2(0), fNdof(0), fNTracks(0) {} |
29 |
|
|
30 |
< |
Double_t Phi() const { return fPosition.Phi(); } |
31 |
< |
Double_t Rho() const { return fPosition.Rho(); } |
32 |
< |
const ThreeVector &Position() const { return fPosition; } |
33 |
< |
void SetPosition(const ThreeVector &pos) { fPosition = pos; } |
60 |
< |
void SetPosition(Double_t x, Double_t y, Double_t z) { fPosition.SetXYZ(x,y,z); } |
61 |
< |
Double_t X() const { return fPosition.X(); } |
62 |
< |
Double_t Y() const { return fPosition.Y(); } |
63 |
< |
Double_t Z() const { return fPosition.Z(); } |
64 |
< |
Double_t Chi2() const { return fChi2; } |
65 |
< |
UInt_t Ndof() const { return fNdof; } |
66 |
< |
Double_t XErr() const { return fXErr; } |
67 |
< |
Double_t YErr() const { return fYErr; } |
68 |
< |
Double_t ZErr() const { return fZErr; } |
30 |
> |
Double_t Chi2() const { return fChi2; } |
31 |
> |
UShort_t Ndof() const { return fNdof; } |
32 |
> |
UInt_t NTracks() const { return fNTracks; } |
33 |
> |
EObjType ObjType() const { return kVertex; } |
34 |
|
Double_t Prob() const { return TMath::Prob(fChi2,fNdof); } |
35 |
< |
void SetChi2(Double_t chi2) { fChi2 = chi2; } |
36 |
< |
void SetErrors(Double_t xErr, Double_t yErr, Double_t zErr); |
37 |
< |
void SetNdof(UInt_t nDof) { fNdof = nDof; } |
73 |
< |
UInt_t NTracks() const { return fNTracks; } |
74 |
< |
void SetNTracks(UInt_t ntrks) { fNTracks = ntrks; } |
35 |
> |
void SetChi2(Double_t chi2) { fChi2 = chi2; } |
36 |
> |
void SetNdof(UShort_t nDof) { fNdof = nDof; } |
37 |
> |
void SetNTracks(UShort_t ntrks) { fNTracks = ntrks; } |
38 |
|
|
39 |
|
protected: |
40 |
< |
ThreeVector fPosition; //point in space |
41 |
< |
Double32_t fXErr; //error in x |
42 |
< |
Double32_t fYErr; //error in y |
80 |
< |
Double32_t fZErr; //error in z |
81 |
< |
Double32_t fChi2; //chi squared of conversion vertex fit |
82 |
< |
UInt_t fNdof; //number of degrees of freedom of conversion vertex fit |
83 |
< |
UInt_t fNTracks; //number of tracks used for the fit |
40 |
> |
Double32_t fChi2; //[0,0,12]chi squared of conversion vertex fit |
41 |
> |
UShort_t fNdof; //number of degrees of freedom of conversion vertex fit |
42 |
> |
UShort_t fNTracks; //number of tracks used for the fit |
43 |
|
|
44 |
|
ClassDef(Vertex, 1) // Vertex class |
45 |
|
}; |
46 |
|
} |
88 |
– |
|
89 |
– |
//-------------------------------------------------------------------------------------------------- |
90 |
– |
inline void mithep::Vertex::SetErrors(Double_t xErr, Double_t yErr, Double_t zErr) |
91 |
– |
{ |
92 |
– |
// set uncertainties on vertex position |
93 |
– |
|
94 |
– |
fXErr = xErr; |
95 |
– |
fYErr = yErr; |
96 |
– |
fZErr = zErr; |
97 |
– |
} |
47 |
|
#endif |