|
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 Alignment_OfflineValidation_TrackerTrackerValidationVariables_h
#define Alignment_OfflineValidation_TrackerTrackerValidationVariables_h
// system include files
#include <vector>
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/Utilities/interface/ESGetToken.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "TrackingTools/PatternTools/interface/TrajTrackAssociation.h"
class Trajectory;
namespace edm {
class ConsumesCollector;
class Event;
class EventSetup;
class ParameterSet;
class ParameterSetDescription;
} // namespace edm
class TrackerValidationVariables {
public:
struct AVHitStruct {
AVHitStruct()
: resX(-999.),
resY(-999.),
resErrX(-999.),
resErrY(-999.),
resXprime(-999.),
resXatTrkY(-999.),
resXprimeErr(-999.),
resYprime(-999.),
resYprimeErr(-999.),
phi(-999.),
eta(-999.),
inside(false),
localX(-999.),
localY(-999.),
localXnorm(-999.),
localYnorm(-999.),
localAlpha(-999.),
localBeta(-999.),
rawDetId(0),
isOnEdgePixel(false),
isOtherBadPixel(false) {}
float resX;
float resY;
float resErrX;
float resErrY;
float resXprime;
float resXatTrkY;
float resXprimeErr;
float resYprime;
float resYprimeErr;
float phi;
float eta;
bool inside;
float localX;
float localY;
float localXnorm;
float localYnorm;
float localAlpha;
float localBeta;
uint32_t rawDetId;
bool isOnEdgePixel;
bool isOtherBadPixel;
};
struct AVTrackStruct {
AVTrackStruct()
: p(0.),
pt(0.),
ptError(0.),
px(0.),
py(0.),
pz(0.),
eta(0.),
phi(0.),
kappa(0.),
chi2(0.),
chi2Prob(0.),
normchi2(0),
d0(-999.),
dz(-999.),
charge(-999),
numberOfValidHits(0),
numberOfLostHits(0) {}
float p;
float pt;
float ptError;
float px;
float py;
float pz;
float eta;
float phi;
float kappa;
float chi2;
float chi2Prob;
float normchi2;
float d0;
float dz;
int charge;
int numberOfValidHits;
int numberOfLostHits;
std::vector<AVHitStruct> hits;
};
TrackerValidationVariables(const edm::ParameterSet& config, edm::ConsumesCollector&& iC);
~TrackerValidationVariables();
void fillHitQuantities(const Trajectory* trajectory, std::vector<AVHitStruct>& v_avhitout);
void fillHitQuantities(reco::Track const& track, std::vector<AVHitStruct>& v_avhitout);
void fillTrackQuantities(const edm::Event&, const edm::EventSetup&, std::vector<AVTrackStruct>& v_avtrackout);
// all Tracks are passed to the trackFilter first, and only processed if it returns true.
void fillTrackQuantities(const edm::Event& event,
const edm::EventSetup& eventSetup,
std::function<bool(const reco::Track&)> trackFilter,
std::vector<AVTrackStruct>& v_avtrackout);
static void fillPSetDescription(edm::ParameterSetDescription& descriptions);
private:
edm::EDGetTokenT<std::vector<Trajectory>> trajCollectionToken_;
edm::EDGetTokenT<std::vector<reco::Track>> tracksToken_;
edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magneticFieldToken_;
};
#endif
|