|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#ifndef TRACKINFO_TPTORECOTRACK_H
#define TRACKINFO_TPTORECOTRACK_H
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingVertex.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingVertexContainer.h"
#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/Math/interface/Point3D.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalVector.h"
#include "TMath.h"
//#include <vector>
class TPtoRecoTrack {
public:
TPtoRecoTrack();
~TPtoRecoTrack();
void SetTrackingParticle(TrackingParticleRef tp) { trackingParticle_ = tp; }
void SetRecoTrack_AlgoA(reco::TrackBaseRef track) { recoTrack_AlgoA_ = track; }
void SetRecoTrack_AlgoB(reco::TrackBaseRef track) { recoTrack_AlgoB_ = track; }
void SetShared_AlgoA(const float &mA) { sharedA_ = mA; }
void SetShared_AlgoB(const float &mB) { sharedB_ = mB; }
void SetRecoVertex_AlgoA(reco::VertexRef vertex) { recoVertex_AlgoA_ = vertex; }
void SetRecoVertex_AlgoB(reco::VertexRef vertex) { recoVertex_AlgoB_ = vertex; }
void SetBeamSpot(const math::XYZPoint &bs) { beamSpot_ = bs; }
// Interogation Functions
reco::Track RTA() const { return recoTrack_AlgoA_.isNonnull() ? *recoTrack_AlgoA_ : reco::Track(); }
reco::Track RTB() const { return recoTrack_AlgoB_.isNonnull() ? *recoTrack_AlgoB_ : reco::Track(); }
TrackingParticle TP() const { return trackingParticle_.isNonnull() ? *trackingParticle_ : TrackingParticle(); }
reco::Vertex RVA() const { return recoVertex_AlgoA_.isNonnull() ? *recoVertex_AlgoA_ : reco::Vertex(); }
reco::Vertex RVB() const { return recoVertex_AlgoB_.isNonnull() ? *recoVertex_AlgoB_ : reco::Vertex(); }
math::XYZPoint BeamSpot() const { return beamSpot_; }
bool matched() const { return matchedA() && matchedB(); }
bool matchedA() const { return trackingParticle_.isNonnull() && recoTrack_AlgoA_.isNonnull(); }
bool matchedB() const { return trackingParticle_.isNonnull() && recoTrack_AlgoB_.isNonnull(); }
bool matchedAnotB() const { return matchedA() && !matchedB(); }
bool matchedBnotA() const { return matchedB() && !matchedA(); }
bool hasRVA() const {
return recoVertex_AlgoA_.isNonnull() && fabs(recoVertex_AlgoA_->position().Mag2()) > 0.0;
} // position is ROOT::MATH::Cartesian3D<double>
bool hasRVB() const {
return recoVertex_AlgoB_.isNonnull() && fabs(recoVertex_AlgoB_->position().Mag2()) > 0.0;
} // position is ROOT::MATH::Cartesian3D<double>
bool hasRV() const { return hasRVA() && hasRVB(); }
bool hasPCA() const { return s_pca().mag() < 9999.0; }
bool allmatchedA() const { return matchedA() && hasRVA(); }
bool allmatchedB() const { return matchedB() && hasRVA(); }
bool allmatched() const { return matched() && hasRV(); }
float GetSharedA() const { return sharedA_; }
float GetSharedB() const { return sharedB_; }
// These members for reco d0 and dz are the TIP and LIP w.r.t the reconstructed BeamSpot (as opposed to (0,0,0) ).
double rA_dxy() const { return RTA().dxy(BeamSpot()); }
double rB_dxy() const { return RTB().dxy(BeamSpot()); }
double rA_dsz() const { return RTA().dsz(BeamSpot()); }
double rB_dsz() const { return RTB().dsz(BeamSpot()); }
double rA_d0() const { return -1.0 * rA_dxy(); }
double rB_d0() const { return -1.0 * rB_dxy(); }
double rA_dz() const { return RTA().dz(BeamSpot()); }
double rB_dz() const { return RTB().dz(BeamSpot()); }
// These members for reco d0 and dz are the TIP and LIP w.r.t the reconstructed vertex (as opposed to (0,0,0) ).
double rA_d02() const { return -1.0 * RTA().dxy(RVA().position()); }
double rA_dz2() const { return RTA().dz(RVA().position()); }
double rB_d02() const { return -1.0 * RTB().dxy(RVB().position()); }
double rB_dz2() const { return RTB().dz(RVB().position()); }
// These members for sim d0 and dz are not included in the TrackingParticle class and must be included seperately.
void SetTrackingParticleMomentumPCA(const GlobalVector &p) { simMomPCA_ = p; }
void SetTrackingParticlePCA(const GlobalPoint &v) { simPCA_ = v; }
GlobalVector s_p() const { return simMomPCA_; }
GlobalPoint s_pca() const { return simPCA_; }
GlobalPoint s_v() const {
return GlobalPoint(s_pca().x() - BeamSpot().x(), s_pca().y() - BeamSpot().y(), s_pca().z() - BeamSpot().z());
}
double s_qoverp() const { return TP().charge() / s_p().mag(); }
double s_theta() const { return s_p().theta(); }
double s_lambda() const { return M_PI / 2 - s_p().theta(); }
double s_phi() const { return s_p().phi(); }
double s_eta() const { return -1.0 * log(tan(0.5 * s_p().theta())); }
double s_dxy() const { return (-s_v().x() * s_p().y() + s_v().y() * s_p().x()) / s_p().perp(); }
double s_dsz() const {
return s_v().z() * s_p().perp() / s_p().mag() -
((s_v().x() * s_p().x() + s_v().y() * s_p().y()) / s_p().perp()) * s_p().z() / s_p().mag();
}
double s_d0() const { return -1.0 * s_dxy(); }
double s_dz() const {
return s_v().z() - (s_v().x() * s_p().x() + s_v().y() * s_p().y()) / s_p().perp() * s_p().z() / s_p().perp();
}
// Short cut methods to get TP truth info
TrackingParticle TPMother(unsigned short i) const;
TrackingParticle TPMother() const { return numTPMothers() == 1 ? TPMother(0) : TrackingParticle(); }
int numTPSourceTracks() const { return TP().parentVertex()->nSourceTracks(); }
int numTPMothers() const;
bool hasTPMother() const { return numTPMothers() > 0; }
protected:
reco::TrackBaseRef recoTrack_AlgoA_;
reco::VertexRef recoVertex_AlgoA_;
reco::TrackBaseRef recoTrack_AlgoB_;
reco::VertexRef recoVertex_AlgoB_;
TrackingParticleRef trackingParticle_;
GlobalVector simMomPCA_; // Momentum at point of closest approach to the beamspot of the trackingParticle.
GlobalPoint simPCA_; // Point of closest approach to the BeamSpot of the TrackingParticle.
math::XYZPoint beamSpot_; // I use type XYZPoint to faciliate the use of the recoTrack memeber dxy(XYZPoint).
float sharedA_; // Number of shared hits with track A
float sharedB_; // Number of shared hits with track B
};
#endif // TRACKINFO_TPTORECOTRACK_H
|