29 |
|
Vertex(const ThreeVector &pos) : |
30 |
|
BaseVertex(pos), fChi2(0), fNdof(0), fAdaptiveNdof(0), fNTracks(0) {} |
31 |
|
|
32 |
< |
void AddTrack(const Track *t) { fTracks.Add(t); } |
32 |
> |
void AddTrack(const Track *t, Double32_t wgt = -1) { fTracks.Add(t); fTrkWeights.Add(wgt); } |
33 |
|
Double_t Chi2() const { return fChi2; } |
34 |
|
Int_t Compare(const TObject *o) const; |
35 |
|
Bool_t HasTrack(const Track *t) const { return fTracks.HasObject(t); } |
45 |
|
void SetNdof(Double_t nDof) { fAdaptiveNdof = nDof; } |
46 |
|
void SetNTracksFit(UInt_t n) { fNTracks = n; } |
47 |
|
const Track *Trk(UInt_t i) const { return fTracks.At(i); } |
48 |
+ |
Double32_t TrackWeight(const Track *t) const; |
49 |
|
|
50 |
|
protected: |
51 |
|
Double32_t fChi2; //[0,0,12]chi squared of conversion vertex fit |
53 |
|
UShort_t fNdof; //number of degrees of freedom of conversion vertex fit |
54 |
|
Double32_t fAdaptiveNdof; //number of degrees of freedom of vertex fit (can be non-integer for weighted components) |
55 |
|
UShort_t fNTracks; //number of tracks used for the fit |
56 |
+ |
FArrDouble32 fTrkWeights; //track weights |
57 |
|
RefArray<Track> fTracks; //tracks associated with the PV |
58 |
|
|
59 |
< |
ClassDef(Vertex, 3) // Vertex class |
59 |
> |
ClassDef(Vertex, 4) // Vertex class |
60 |
|
}; |
61 |
|
} |
62 |
|
|
87 |
|
|
88 |
|
return 0; |
89 |
|
} |
90 |
+ |
|
91 |
+ |
//-------------------------------------------------------------------------------------------------- |
92 |
+ |
inline Double_t mithep::Vertex::TrackWeight(const Track *t) const |
93 |
+ |
{ |
94 |
+ |
for(UInt_t i = 0; i < fTracks.Entries(); i++) |
95 |
+ |
{ |
96 |
+ |
if(t == fTracks.At(i)) |
97 |
+ |
return fTrkWeights.At(i); |
98 |
+ |
} |
99 |
+ |
|
100 |
+ |
return -1; |
101 |
+ |
} |
102 |
|
#endif |