51 |
|
#include <SimDataFormats/Track/interface/SimTrackContainer.h> |
52 |
|
// BeamSpot |
53 |
|
#include "DataFormats/BeamSpot/interface/BeamSpot.h" |
54 |
– |
// Trigger |
55 |
– |
#include "CondFormats/L1TObjects/interface/L1GtTriggerMenu.h" |
56 |
– |
#include "CondFormats/DataRecord/interface/L1GtTriggerMenuRcd.h" |
57 |
– |
#include "CondFormats/L1TObjects/interface/L1GtPrescaleFactors.h" |
58 |
– |
#include "CondFormats/DataRecord/interface/L1GtPrescaleFactorsAlgoTrigRcd.h" |
59 |
– |
#include "CondFormats/DataRecord/interface/L1GtPrescaleFactorsTechTrigRcd.h" |
60 |
– |
#include "CondFormats/L1TObjects/interface/L1GtTriggerMask.h" |
61 |
– |
#include "CondFormats/DataRecord/interface/L1GtTriggerMaskAlgoTrigRcd.h" |
62 |
– |
#include "CondFormats/DataRecord/interface/L1GtTriggerMaskTechTrigRcd.h" |
63 |
– |
#include "CondFormats/DataRecord/interface/L1GtTriggerMaskVetoAlgoTrigRcd.h" |
64 |
– |
#include "CondFormats/DataRecord/interface/L1GtTriggerMaskVetoTechTrigRcd.h" |
65 |
– |
#include "DataFormats/L1GlobalTrigger/interface/L1GlobalTriggerReadoutSetupFwd.h" |
66 |
– |
#include "DataFormats/L1GlobalTrigger/interface/L1GlobalTriggerReadoutSetup.h" |
67 |
– |
#include "DataFormats/L1GlobalTrigger/interface/L1GlobalTriggerReadoutRecord.h" |
68 |
– |
|
54 |
|
|
55 |
|
//root |
56 |
|
#include <TROOT.h> |
80 |
|
realData_ = iConfig.getUntrackedParameter<bool>("realData",false); |
81 |
|
analyze_ = iConfig.getUntrackedParameter<bool>("analyzeOnTheFly",false); |
82 |
|
saventuple_ = iConfig.getUntrackedParameter<bool>("saventuple",false); |
83 |
< |
outputfilename_ = iConfig.getUntrackedParameter<string>("OutputFileName"); |
83 |
> |
outputfilename_ = iConfig.getUntrackedParameter<string>("OutputFileName"); |
84 |
> |
histoFileName_ = iConfig.getUntrackedParameter<std::string> ("histoFileName"); |
85 |
|
ntrkdiffcut_ = iConfig.getUntrackedParameter<double>("ntrkdiffcut"); |
86 |
|
nTrkMin_ = iConfig.getUntrackedParameter<int>("nTrkMin"); |
87 |
|
nTrkMax_ = iConfig.getUntrackedParameter<int>("nTrkMax"); |
88 |
|
zsigncut_ = iConfig.getUntrackedParameter<int>("zsigncut"); |
89 |
+ |
fHighPuritycut_ = iConfig.getUntrackedParameter<double>("fHighPuritycut"); |
90 |
+ |
useHWTrk_ = iConfig.getUntrackedParameter<bool>("useHWTrk",false); |
91 |
|
bsSrc = iConfig.getParameter< edm::InputTag >("beamSpot"); |
104 |
– |
applyEvtClean_ = iConfig.getUntrackedParameter<bool>("applyEvtClean",false); |
105 |
– |
histoFileName_ = iConfig.getUntrackedParameter<std::string> ("histoFileName"); |
92 |
|
|
93 |
|
//now do what ever initialization is needed |
94 |
|
pvinfo.clear(); |
95 |
|
|
110 |
– |
|
96 |
|
// Specify the data mode vector |
97 |
|
if(realData_) datamode.push_back(0); |
98 |
|
else { |
112 |
|
|
113 |
|
// pvtxtree_ analyzing the pvtxs ootb |
114 |
|
pvtxtree_ = new TTree("pvtxtree","pvtxtree"); |
130 |
– |
//pvtx using all tracks |
131 |
– |
pvtxtree_->Branch("nrecPV",&nrecPV_,"nrecPV/I"); |
132 |
– |
pvtxtree_->Branch("nTrkPV",&nTrkPV_,"nTrkPV[nrecPV]/I"); |
133 |
– |
pvtxtree_->Branch("sumptsq",&sumptsq_,"sumptsq[nrecPV]/D"); |
134 |
– |
pvtxtree_->Branch("isValid",&isValid_,"isValid/I"); |
135 |
– |
pvtxtree_->Branch("isFake",&isFake_,"isFake/I"); |
136 |
– |
pvtxtree_->Branch("recx",&recx_,"recx[nrecPV]/D"); |
137 |
– |
pvtxtree_->Branch("recy",&recy_,"recy[nrecPV]/D"); |
138 |
– |
pvtxtree_->Branch("recz",&recz_,"recz[nrecPV]/D"); |
139 |
– |
pvtxtree_->Branch("recx_err",&recx_err_,"recx_err[nrecPV]/D"); |
140 |
– |
pvtxtree_->Branch("recy_err",&recy_err_,"recy_err[nrecPV]/D"); |
141 |
– |
pvtxtree_->Branch("recz_err",&recz_err_,"recz_err[nrecPV]/D"); |
115 |
|
|
116 |
+ |
// Event information for the data |
117 |
+ |
pvtxtree_->Branch("glob_runno",&glob_runno_,"glob_runno/I"); |
118 |
+ |
pvtxtree_->Branch("glob_evtno",&glob_evtno_,"glob_evtno/I"); |
119 |
+ |
pvtxtree_->Branch("glob_ls",&glob_ls_,"glob_ls/I"); |
120 |
+ |
pvtxtree_->Branch("glob_bx",&glob_bx_,"glob_bx/I"); |
121 |
+ |
|
122 |
+ |
|
123 |
+ |
pvtxtree_->Branch("nrecPV",&nrecPV_,"nrecPV/I"); |
124 |
+ |
pvtxtree_->Branch("nrecPV1_spl",&nrecPV1_spl_,"nrecPV1_spl/I"); |
125 |
+ |
pvtxtree_->Branch("nrecPV2_spl",&nrecPV2_spl_,"nrecPV2_spl/I"); |
126 |
+ |
// Event level information |
127 |
|
pvtxtree_->Branch("min_zsep",&min_zsep_,"min_zsep/D"); |
128 |
|
pvtxtree_->Branch("min_ntrksep",&min_ntrksep_,"min_ntrksep/D"); |
129 |
|
pvtxtree_->Branch("min_sumpt2sep",&min_sumpt2sep_,"min_sumpt2sep/D"); |
130 |
|
|
131 |
< |
//pvtx using splittracks1 |
132 |
< |
pvtxtree_->Branch("nrecPV1_spl",&nrecPV1_spl_,"nrecPV1_spl/I"); |
133 |
< |
pvtxtree_->Branch("nTrkPV1_spl",&nTrkPV1_spl_,"nTrkPV1_spl[nrecPV1_spl]/I"); |
134 |
< |
pvtxtree_->Branch("sumptsq1_spl",&sumptsq1_spl_,"sumptsq1_spl[nrecPV1_spl]/D"); |
151 |
< |
pvtxtree_->Branch("isValid1_spl",&isValid1_spl_,"isValid1_spl/I"); |
152 |
< |
pvtxtree_->Branch("isFake1_spl",&isFake1_spl_,"isFake1_spl/I"); |
153 |
< |
pvtxtree_->Branch("recx1_spl",&recx1_spl_,"recx1_spl[nrecPV1_spl]/D"); |
154 |
< |
pvtxtree_->Branch("recy1_spl",&recy1_spl_,"recy1_spl[nrecPV1_spl]/D"); |
155 |
< |
pvtxtree_->Branch("recz1_spl",&recz1_spl_,"recz1_spl[nrecPV1_spl]/D"); |
156 |
< |
pvtxtree_->Branch("recx1_err_spl",&recx1_err_spl_,"recx1_err_spl[nrecPV1_spl]/D"); |
157 |
< |
pvtxtree_->Branch("recy1_err_spl",&recy1_err_spl_,"recy1_err_spl[nrecPV1_spl]/D"); |
158 |
< |
pvtxtree_->Branch("recz1_err_spl",&recz1_err_spl_,"recz1_err_spl[nrecPV1_spl]/D"); |
159 |
< |
|
160 |
< |
//pvtx using splittracks2 |
161 |
< |
pvtxtree_->Branch("nrecPV2_spl",&nrecPV2_spl_,"nrecPV2_spl/I"); |
162 |
< |
pvtxtree_->Branch("nTrkPV2_spl",&nTrkPV2_spl_,"nTrkPV2_spl[nrecPV2_spl]/I"); |
163 |
< |
pvtxtree_->Branch("sumptsq2_spl",&sumptsq2_spl_,"sumptsq2_spl[nrecPV2_spl]/D"); |
164 |
< |
pvtxtree_->Branch("isValid2_spl",&isValid2_spl_,"isValid2_spl/I"); |
165 |
< |
pvtxtree_->Branch("isFake2_spl",&isFake2_spl_,"isFake2_spl/I"); |
166 |
< |
pvtxtree_->Branch("recx2_spl",&recx2_spl_,"recx2_spl[nrecPV2_spl]/D"); |
167 |
< |
pvtxtree_->Branch("recy2_spl",&recy2_spl_,"recy2_spl[nrecPV2_spl]/D"); |
168 |
< |
pvtxtree_->Branch("recz2_spl",&recz2_spl_,"recz2_spl[nrecPV2_spl]/D"); |
169 |
< |
pvtxtree_->Branch("recx2_err_spl",&recx2_err_spl_,"recx2_err_spl[nrecPV2_spl]/D"); |
170 |
< |
pvtxtree_->Branch("recy2_err_spl",&recy2_err_spl_,"recy2_err_spl[nrecPV2_spl]/D"); |
171 |
< |
pvtxtree_->Branch("recz2_err_spl",&recz2_err_spl_,"recz2_err_spl[nrecPV2_spl]/D"); |
172 |
< |
|
173 |
< |
//pixeVertices |
174 |
< |
pvtxtree_->Branch("nrecPV_pxlpvtx",&nrecPV_pxlpvtx_,"nrecPV_pxlpvtx/I"); |
175 |
< |
pvtxtree_->Branch("nTrkPV_pxlpvtx",&nTrkPV_pxlpvtx_,"nTrkPV_pxlpvtx[nrecPV_pxlpvtx]/I"); |
176 |
< |
pvtxtree_->Branch("sumptsq_pxlpvtx",&sumptsq_pxlpvtx_,"sumptsq_pxlpvtx[nrecPV_pxlpvtx]/D"); |
177 |
< |
pvtxtree_->Branch("isValid_pxlpvtx",&isValid_pxlpvtx_,"isValid_pxlpvtx/I"); |
178 |
< |
pvtxtree_->Branch("isFake_pxlpvtx",&isFake_pxlpvtx_,"isFake_pxlpvtx/I"); |
179 |
< |
pvtxtree_->Branch("recx_pxlpvtx",&recx_pxlpvtx_,"recx_pxlpvtx[nrecPV_pxlpvtx]/D"); |
180 |
< |
pvtxtree_->Branch("recy_pxlpvtx",&recy_pxlpvtx_,"recy_pxlpvtx[nrecPV_pxlpvtx]/D"); |
181 |
< |
pvtxtree_->Branch("recz_pxlpvtx",&recz_pxlpvtx_,"recz_pxlpvtx[nrecPV_pxlpvtx]/D"); |
182 |
< |
pvtxtree_->Branch("recx_err_pxlpvtx",&recx_err_pxlpvtx_,"recx_err_pxlpvtx[nrecPV_pxlpvtx]/D"); |
183 |
< |
pvtxtree_->Branch("recy_err_pxlpvtx",&recy_err_pxlpvtx_,"recy_err_pxlpvtx[nrecPV_pxlpvtx]/D"); |
184 |
< |
pvtxtree_->Branch("recz_err_pxlpvtx",&recz_err_pxlpvtx_,"recz_err_pxlpvtx[nrecPV_pxlpvtx]/D"); |
131 |
> |
//Fill the variables in the twovtx pair (recvtx1, recvtx2) |
132 |
> |
// Information related to the analyzing the two-vertex method |
133 |
> |
|
134 |
> |
|
135 |
|
|
186 |
– |
//Fill the variables in the twovtx pair (recvtx1, recvtx2) |
136 |
|
pvtxtree_->Branch("nrecPV_twovtx",&nrecPV_twovtx_,"nrecPV_twovtx/I"); |
188 |
– |
pvtxtree_->Branch("nTrkPV1_spl_twovtx",&nTrkPV1_spl_twovtx_,"nTrkPV1_spl_twovtx[nrecPV_twovtx]/I"); |
189 |
– |
pvtxtree_->Branch("nTrkPV2_spl_twovtx",&nTrkPV2_spl_twovtx_,"nTrkPV2_spl_twovtx[nrecPV_twovtx]/I"); |
190 |
– |
pvtxtree_->Branch("sumptsq1_spl_twovtx",&sumptsq1_spl_twovtx_,"sumptsq1_spl_twovtx[nrecPV_twovtx]/D"); |
191 |
– |
pvtxtree_->Branch("sumptsq2_spl_twovtx",&sumptsq2_spl_twovtx_,"sumptsq2_spl_twovtx[nrecPV_twovtx]/D"); |
137 |
|
pvtxtree_->Branch("nTrkPV_twovtx",&nTrkPV_twovtx_,"nTrkPV_twovtx[nrecPV_twovtx]/I"); |
138 |
|
pvtxtree_->Branch("deltax_twovtx",&deltax_twovtx_,"deltax_twovtx[nrecPV_twovtx]/D"); |
139 |
|
pvtxtree_->Branch("deltay_twovtx",&deltay_twovtx_,"deltay_twovtx[nrecPV_twovtx]/D"); |
144 |
|
pvtxtree_->Branch("pullx_twovtx",&pullx_twovtx_,"pullx_twovtx[nrecPV_twovtx]/D"); |
145 |
|
pvtxtree_->Branch("pully_twovtx",&pully_twovtx_,"pully_twovtx[nrecPV_twovtx]/D"); |
146 |
|
pvtxtree_->Branch("pullz_twovtx",&pullz_twovtx_,"pullz_twovtx[nrecPV_twovtx]/D"); |
147 |
+ |
|
148 |
+ |
// Information for the splitVertexColl1 |
149 |
+ |
pvtxtree_->Branch("nTrkPV1_spl_twovtx",&nTrkPV1_spl_twovtx_,"nTrkPV1_spl_twovtx[nrecPV_twovtx]/I"); |
150 |
+ |
pvtxtree_->Branch("ndofPV1_spl_twovtx",&ndofPV1_spl_twovtx_,"ndofPV1_spl_twovtx[nrecPV_twovtx]/D"); |
151 |
+ |
pvtxtree_->Branch("normchi2PV1_spl_twovtx",&normchi2PV1_spl_twovtx_,"normchi2PV1_spl_twovtx[nrecPV_twovtx]/D"); |
152 |
+ |
pvtxtree_->Branch("avgPtPV1_spl_twovtx",&avgPtPV1_spl_twovtx_,"avgPtPV1_spl_twovtx[nrecPV_twovtx]/D"); |
153 |
+ |
pvtxtree_->Branch("errx1_spl_twovtx",&errx1_spl_twovtx_,"errx1_spl_twovtx[nrecPV_twovtx]/D"); |
154 |
+ |
pvtxtree_->Branch("erry1_spl_twovtx",&erry1_spl_twovtx_,"erry1_spl_twovtx[nrecPV_twovtx]/D"); |
155 |
+ |
pvtxtree_->Branch("errz1_spl_twovtx",&errz1_spl_twovtx_,"errz1_spl_twovtx[nrecPV_twovtx]/D"); |
156 |
+ |
|
157 |
+ |
|
158 |
+ |
// Information for the splitVertexColl2 |
159 |
+ |
pvtxtree_->Branch("nTrkPV2_spl_twovtx",&nTrkPV2_spl_twovtx_,"nTrkPV2_spl_twovtx[nrecPV_twovtx]/I"); |
160 |
+ |
pvtxtree_->Branch("ndofPV2_spl_twovtx",&ndofPV2_spl_twovtx_,"ndofPV2_spl_twovtx[nrecPV_twovtx]/D"); |
161 |
+ |
pvtxtree_->Branch("normchi2PV2_spl_twovtx",&normchi2PV2_spl_twovtx_,"normchi2PV2_spl_twovtx[nrecPV_twovtx]/D"); |
162 |
+ |
pvtxtree_->Branch("avgPtPV2_spl_twovtx",&avgPtPV2_spl_twovtx_,"avgPtPV2_spl_twovtx[nrecPV_twovtx]/D"); |
163 |
+ |
pvtxtree_->Branch("errx2_spl_twovtx",&errx2_spl_twovtx_,"errx2_spl_twovtx[nrecPV_twovtx]/D"); |
164 |
+ |
pvtxtree_->Branch("erry2_spl_twovtx",&erry2_spl_twovtx_,"erry2_spl_twovtx[nrecPV_twovtx]/D"); |
165 |
+ |
pvtxtree_->Branch("errz2_spl_twovtx",&errz2_spl_twovtx_,"errz2_spl_twovtx[nrecPV_twovtx]/D"); |
166 |
+ |
|
167 |
|
|
168 |
|
// MC variables |
169 |
|
if(!realData_) { |
181 |
|
pvtxtree_->Branch("deltaz_mct",&deltaz_mct_,"deltaz_mct[nrecPV_mct]/D"); |
182 |
|
pvtxtree_->Branch("pullx_mct",&pullx_mct_,"pullx_mct[nrecPV_mct]/D"); |
183 |
|
pvtxtree_->Branch("pully_mct",&pully_mct_,"pully_mct[nrecPV_mct]/D"); |
184 |
< |
pvtxtree_->Branch("pullz_mct",&pullz_mct_,"pullz_mct[nrecPV_mct]/D"); |
184 |
> |
pvtxtree_->Branch("pullz_mct",&pullz_mct_,"pullz_mct[nrecPV_mct]/D"); |
185 |
> |
pvtxtree_->Branch("errx_mct",&errx_mct_,"errx_mct[nrecPV_mct]/D"); |
186 |
> |
pvtxtree_->Branch("erry_mct",&erry_mct_,"erry_mct[nrecPV_mct]/D"); |
187 |
> |
pvtxtree_->Branch("errz_mct",&errz_mct_,"errz_mct[nrecPV_mct]/D"); |
188 |
> |
pvtxtree_->Branch("nTrkPV_mct",&nTrkPV_mct_,"nTrkPV_mct[nrecPV_mct]/I"); |
189 |
> |
pvtxtree_->Branch("ndofPV_mct",&ndofPV_mct_,"ndofPV_mct[nrecPV_mct]/D"); |
190 |
> |
pvtxtree_->Branch("normchi2PV_mct",&normchi2PV_mct_,"normchi2PV_mct[nrecPV_mct]/D"); |
191 |
> |
pvtxtree_->Branch("avgPtPV_mct",&avgPtPV_mct_,"avgPtPV_mct[nrecPV_mct]/D"); |
192 |
> |
|
193 |
|
// For pvtxs with splittracks1 |
194 |
|
pvtxtree_->Branch("nrecPV_spl1_mct",&nrecPV_spl1_mct_,"nrecPV_spl1_mct/I"); |
195 |
|
pvtxtree_->Branch("deltax_spl1_mct",&deltax_spl1_mct_,"deltax_spl1_mct[nrecPV_spl1_mct]/D"); |
198 |
|
pvtxtree_->Branch("pullx_spl1_mct",&pullx_spl1_mct_,"pullx_spl1_mct[nrecPV_spl1_mct]/D"); |
199 |
|
pvtxtree_->Branch("pully_spl1_mct",&pully_spl1_mct_,"pully_spl1_mct[nrecPV_spl1_mct]/D"); |
200 |
|
pvtxtree_->Branch("pullz_spl1_mct",&pullz_spl1_mct_,"pullz_spl1_mct[nrecPV_spl1_mct]/D"); |
201 |
+ |
pvtxtree_->Branch("errx_spl1_mct",&errx_spl1_mct_,"errx_spl1_mct[nrecPV_spl1_mct]/D"); |
202 |
+ |
pvtxtree_->Branch("erry_spl1_mct",&erry_spl1_mct_,"erry_spl1_mct[nrecPV_spl1_mct]/D"); |
203 |
+ |
pvtxtree_->Branch("errz_spl1_mct",&errz_spl1_mct_,"errz_spl1_mct[nrecPV_spl1_mct]/D"); |
204 |
+ |
pvtxtree_->Branch("nTrkPV_spl1_mct",&nTrkPV_spl1_mct_,"nTrkPV_spl1_mct[nrecPV_spl1_mct]/I"); |
205 |
+ |
pvtxtree_->Branch("ndofPV_spl1_mct",&ndofPV_spl1_mct_,"ndofPV_spl1_mct[nrecPV_spl1_mct]/D"); |
206 |
+ |
pvtxtree_->Branch("normchi2PV_spl1_mct",&normchi2PV_spl1_mct_,"normchi2PV_spl1_mct[nrecPV_spl1_mct]/D"); |
207 |
+ |
pvtxtree_->Branch("avgPtPV_spl1_mct",&avgPtPV_spl1_mct_,"avgPtPV_spl1_mct[nrecPV_spl1_mct]/D"); |
208 |
+ |
|
209 |
|
// For pvtxs with splittracks1 |
210 |
|
pvtxtree_->Branch("nrecPV_spl2_mct",&nrecPV_spl2_mct_,"nrecPV_spl2_mct/I"); |
211 |
|
pvtxtree_->Branch("deltax_spl2_mct",&deltax_spl2_mct_,"deltax_spl2_mct[nrecPV_spl2_mct]/D"); |
213 |
|
pvtxtree_->Branch("deltaz_spl2_mct",&deltaz_spl2_mct_,"deltaz_spl2_mct[nrecPV_spl2_mct]/D"); |
214 |
|
pvtxtree_->Branch("pullx_spl2_mct",&pullx_spl2_mct_,"pullx_spl2_mct[nrecPV_spl2_mct]/D"); |
215 |
|
pvtxtree_->Branch("pully_spl2_mct",&pully_spl2_mct_,"pully_spl2_mct[nrecPV_spl2_mct]/D"); |
216 |
< |
pvtxtree_->Branch("pullz_spl2_mct",&pullz_spl2_mct_,"pullz_spl2_mct[nrecPV_spl2_mct]/D"); |
216 |
> |
pvtxtree_->Branch("pullz_spl2_mct",&pullz_spl2_mct_,"pullz_spl2_mct[nrecPV_spl2_mct]/D"); |
217 |
> |
pvtxtree_->Branch("errx_spl2_mct",&errx_spl2_mct_,"errx_spl2_mct[nrecPV_spl2_mct]/D"); |
218 |
> |
pvtxtree_->Branch("erry_spl2_mct",&erry_spl2_mct_,"erry_spl2_mct[nrecPV_spl2_mct]/D"); |
219 |
> |
pvtxtree_->Branch("errz_spl2_mct",&errz_spl2_mct_,"errz_spl2_mct[nrecPV_spl2_mct]/D"); |
220 |
> |
pvtxtree_->Branch("nTrkPV_spl2_mct",&nTrkPV_spl2_mct_,"nTrkPV_spl2_mct[nrecPV_spl2_mct]/I"); |
221 |
> |
pvtxtree_->Branch("ndofPV_spl2_mct",&ndofPV_spl2_mct_,"ndofPV_spl2_mct[nrecPV_spl2_mct]/D"); |
222 |
> |
pvtxtree_->Branch("normchi2PV_spl2_mct",&normchi2PV_spl2_mct_,"normchi2PV_spl2_mct[nrecPV_spl2_mct]/D"); |
223 |
> |
pvtxtree_->Branch("avgPtPV_spl2_mct",&avgPtPV_spl2_mct_,"avgPtPV_spl2_mct[nrecPV_spl2_mct]/D"); |
224 |
> |
|
225 |
|
} |
226 |
|
} |
227 |
|
|
245 |
|
|
246 |
|
// Book Histograms: |
247 |
|
h_pvtrk = new PVHistograms(); |
259 |
– |
h_pixvtx = new PVHistograms(); |
248 |
|
h_misc = new PVHistograms(); |
249 |
|
h_summary = new PVHistograms(); |
250 |
|
h_others = new PVHistograms(); |
257 |
|
h_pvtrk->Init("pvTrk", ss.str(),"spl"); |
258 |
|
} |
259 |
|
} |
272 |
– |
h_pixvtx->Init("pixVtx"); |
260 |
|
h_misc->Init("misc"); |
261 |
|
|
262 |
|
// Book MC only plots |
305 |
|
theFile->cd(); |
306 |
|
theFile->cd("Summary"); |
307 |
|
h_pvtrk->Save(); |
321 |
– |
h_pixvtx->Save(); |
308 |
|
h_misc->Save(); |
309 |
|
h_summary->Save(); |
310 |
|
if (!realData_) |
342 |
|
gStyle->SetTitleSize(0.055, ""); // size for pad title; default is 0.02 |
343 |
|
gStyle->SetLabelSize(0.03, "XYZ"); // size for axis labels; default is 0.04 |
344 |
|
gStyle->SetStatFontSize(0.08); // size for stat. box |
345 |
< |
gStyle->SetTitleFont(32, "XYZ"); // times-bold-italic font (p. 153) for axes |
346 |
< |
gStyle->SetTitleFont(32, ""); // same for pad title |
347 |
< |
gStyle->SetLabelFont(32, "XYZ"); // same for axis labels |
348 |
< |
gStyle->SetStatFont(32); // same for stat. box |
345 |
> |
gStyle->SetTitleFont(42, "XYZ"); // times-bold-italic font (p. 153) for axes |
346 |
> |
gStyle->SetTitleFont(42, ""); // same for pad title |
347 |
> |
gStyle->SetLabelFont(42, "XYZ"); // same for axis labels |
348 |
> |
gStyle->SetStatFont(42); // same for stat. box |
349 |
|
gStyle->SetLabelOffset(0.006, "Y"); // default is 0.005 |
350 |
|
// |
351 |
|
return; |
532 |
|
using namespace edm; |
533 |
|
using namespace reco; |
534 |
|
|
535 |
+ |
//======================================================================== |
536 |
+ |
// Step 0: Prepare root variables and get information from the Event |
537 |
+ |
//======================================================================== |
538 |
+ |
|
539 |
|
edm::LogInfo("Debug")<<"[PVStudy]"<<endl; |
550 |
– |
//======================================================= |
540 |
|
// Initialize Root-tuple variables if needed |
541 |
< |
//======================================================= |
542 |
< |
//if(saventuple_) |
543 |
< |
SetVarToZero(); |
555 |
< |
|
556 |
< |
//======================================================= |
557 |
< |
// Track accessors |
558 |
< |
//======================================================= |
559 |
< |
//trackCollection |
541 |
> |
SetVarToZero(); |
542 |
> |
|
543 |
> |
// ====== TrackCollection |
544 |
|
static const reco::TrackCollection s_empty_trackColl; |
545 |
|
const reco::TrackCollection *trackColl = &s_empty_trackColl; |
546 |
|
edm::Handle<reco::TrackCollection> trackCollectionHandle; |
550 |
|
} else { |
551 |
|
edm::LogInfo("Debug") << "[PVStudy] trackCollection cannot be found -> using empty collection of same type." <<endl; |
552 |
|
} |
553 |
< |
//splitTrackCollection1 |
553 |
> |
// ====== splitTrackCollection1 |
554 |
|
static const reco::TrackCollection s_empty_splitTrackColl1; |
555 |
|
const reco::TrackCollection *splitTrackColl1 = &s_empty_splitTrackColl1; |
556 |
|
edm::Handle<reco::TrackCollection> splitTrackCollection1Handle; |
560 |
|
} else { |
561 |
|
edm::LogInfo("Debug") << "[PVStudy] splitTrackCollection1 cannot be found -> using empty collection of same type." <<endl; |
562 |
|
} |
563 |
< |
//splitTrackCollection2 |
563 |
> |
// ====== splitTrackCollection2 |
564 |
|
static const reco::TrackCollection s_empty_splitTrackColl2; |
565 |
|
const reco::TrackCollection *splitTrackColl2 = &s_empty_splitTrackColl2; |
566 |
|
edm::Handle<reco::TrackCollection> splitTrackCollection2Handle; |
571 |
|
edm::LogInfo("Debug") << "[PVStudy] splitTrackCollection2 cannot be found -> using empty collection of same type." <<endl; |
572 |
|
} |
573 |
|
|
574 |
< |
//======================================================= |
591 |
< |
// PVTX accessors |
592 |
< |
//======================================================= |
593 |
< |
//vertexCollection |
574 |
> |
// ======= PrimaryVertexCollection |
575 |
|
static const reco::VertexCollection s_empty_vertexColl; |
576 |
|
const reco::VertexCollection *vertexColl = &s_empty_vertexColl; |
577 |
|
edm::Handle<reco::VertexCollection> vertexCollectionHandle; |
581 |
|
} else { |
582 |
|
edm::LogInfo("Debug") << "[PVStudy] vertexCollection cannot be found -> using empty collection of same type." <<endl; |
583 |
|
} |
584 |
< |
//splitVertexCollection1 |
584 |
> |
// ====== splitVertexCollection1 |
585 |
|
static const reco::VertexCollection s_empty_splitVertexColl1; |
586 |
|
const reco::VertexCollection *splitVertexColl1 = &s_empty_splitVertexColl1; |
587 |
|
edm::Handle<reco::VertexCollection> splitVertexCollection1Handle; |
591 |
|
} else { |
592 |
|
edm::LogInfo("Debug") << "[PVStudy] splitVertexCollection1 cannot be found -> using empty collection of same type." <<endl; |
593 |
|
} |
594 |
< |
//splitVertexCollection2 |
594 |
> |
// ====== splitVertexCollection2 |
595 |
|
static const reco::VertexCollection s_empty_splitVertexColl2; |
596 |
|
const reco::VertexCollection *splitVertexColl2 = &s_empty_splitVertexColl2; |
597 |
|
edm::Handle<reco::VertexCollection> splitVertexCollection2Handle; |
603 |
|
} |
604 |
|
|
605 |
|
|
606 |
< |
//======================================================= |
626 |
< |
// GET pixelVertices |
627 |
< |
//======================================================= |
628 |
< |
static const reco::VertexCollection s_empty_pixelVertexColl; |
629 |
< |
const reco::VertexCollection *pixelVertexColl = &s_empty_pixelVertexColl; |
630 |
< |
edm::Handle<reco::VertexCollection> pixelVertexCollectionHandle; |
631 |
< |
iEvent.getByLabel(pixelVertexCollectionTag_, pixelVertexCollectionHandle); |
632 |
< |
if( iEvent.getByLabel(pixelVertexCollectionTag_, pixelVertexCollectionHandle)) { |
633 |
< |
pixelVertexColl = pixelVertexCollectionHandle.product(); |
634 |
< |
} else { |
635 |
< |
edm::LogInfo("Debug") << "[PVStudy] pixelVertexCollection cannot be found. -> using empty collection of same type." <<endl; |
636 |
< |
} |
637 |
< |
|
638 |
< |
//======================================================= |
639 |
< |
// BeamSpot accessors |
640 |
< |
//======================================================= |
641 |
< |
|
606 |
> |
// ======== BeamSpot accessors |
607 |
|
edm::Handle<reco::BeamSpot> recoBeamSpotHandle; |
608 |
|
iEvent.getByLabel(bsSrc,recoBeamSpotHandle); |
609 |
|
reco::BeamSpot bs = *recoBeamSpotHandle; |
610 |
|
const Point beamSpot = recoBeamSpotHandle.isValid() ? Point(recoBeamSpotHandle->x0(), recoBeamSpotHandle->y0(), recoBeamSpotHandle->z0()) : Point(0, 0, 0); |
611 |
|
|
612 |
< |
edm::LogInfo("Debug")<<"[PVStudy] End accessing the track, beamSpot, primary vertex and pixelvertices collections"<<endl; |
612 |
> |
edm::LogInfo("Debug")<<"[PVStudy] End accessing the track, beamSpot, primary vertex collections"<<endl; |
613 |
|
|
614 |
< |
//======================================================= |
650 |
< |
// MC simvtx accessor |
651 |
< |
//======================================================= |
614 |
> |
// ========== MC simvtx accessor |
615 |
|
if (!realData_) { |
616 |
|
edm::Handle<SimVertexContainer> simVtxs; |
617 |
|
iEvent.getByLabel( simG4_, simVtxs); |
620 |
|
iEvent.getByLabel( simG4_, simTrks); |
621 |
|
} |
622 |
|
|
623 |
< |
//======================================================= |
661 |
< |
// GET PDT |
662 |
< |
//======================================================= |
623 |
> |
// ========== GET PDT |
624 |
|
try{ |
625 |
|
iSetup.getData(pdt); |
626 |
|
}catch(const Exception&){ |
627 |
|
edm::LogInfo("Debug") << "[PVStudy] Some problem occurred with the particle data table. This may not work !." <<endl; |
628 |
|
} |
629 |
< |
|
630 |
< |
//======================================================= |
631 |
< |
// Apply event cleaning for firstcoll data and MC |
632 |
< |
//======================================================= |
633 |
< |
if(applyEvtClean_) { |
634 |
< |
// =====Select on the trigger bits |
635 |
< |
edm::ESHandle<L1GtTriggerMenu> menuRcd; |
636 |
< |
iSetup.get<L1GtTriggerMenuRcd>().get(menuRcd) ; |
637 |
< |
const L1GtTriggerMenu* menu = menuRcd.product(); |
638 |
< |
edm::Handle< L1GlobalTriggerReadoutRecord > gtRecord; |
639 |
< |
iEvent.getByLabel( edm::InputTag("gtDigis"), gtRecord); |
640 |
< |
|
680 |
< |
bool isTechBit0 = false; |
681 |
< |
bool isTechBit40 = false; |
682 |
< |
bool isBeamHalo = false; |
683 |
< |
|
684 |
< |
TechnicalTriggerWord tw = gtRecord->technicalTriggerWord(); |
685 |
< |
if ( ! tw.empty() ) { |
686 |
< |
// loop over dec. bit to get total rate (no overlap) |
687 |
< |
for ( int i = 0; i < 64; ++i ) { |
688 |
< |
if ( tw[i] ) { |
689 |
< |
//cout<<"technical number "<<i<<" "<<endl; |
690 |
< |
if (i == 0) isTechBit0 = true; |
691 |
< |
if (i < 40 && i > 35) isBeamHalo = true; // The beamHalo bits are 36-39 |
692 |
< |
if (i == 40 || i == 41) isTechBit40 = true; |
693 |
< |
} |
694 |
< |
} |
695 |
< |
} // =====End select on the trigger bits |
696 |
< |
// =====Loop over the trackColl to tet the fraction of HighPurity tracks |
697 |
< |
int nTracks = 0; |
698 |
< |
int nHighPurityTracks=0; |
699 |
< |
double fHighPurity=0; |
700 |
< |
|
701 |
< |
for (unsigned int i=0; i<trackCollectionHandle->size(); i++, nTracks++) { |
702 |
< |
TrackRef tkref(trackCollectionHandle,i); |
703 |
< |
if( (tkref->qualityMask() & 4 ) == 4) ++nHighPurityTracks; |
704 |
< |
} |
705 |
< |
if(nTracks>0) |
706 |
< |
fHighPurity = double(nHighPurityTracks)/double(nTracks); |
707 |
< |
if(verbose_) |
708 |
< |
cout<<"fraction of HighPurity track is "<<fHighPurity<<endl; |
709 |
< |
|
710 |
< |
// Apply the event selection for MC events |
711 |
< |
if(realData_) { |
712 |
< |
if(!isTechBit40 || isBeamHalo || (fHighPurity<0.2&&nTracks>10) || vertexColl->begin()->isFake()) { |
713 |
< |
glob_runno_ = iEvent.id().run(); |
714 |
< |
glob_evtno_ = iEvent.id().event(); |
715 |
< |
glob_ls_ = iEvent.luminosityBlock(); |
716 |
< |
glob_bx_ = iEvent.bunchCrossing(); |
717 |
< |
return; |
718 |
< |
} |
719 |
< |
else { |
720 |
< |
// TechBit 0, beamHalo are not simulted in MC |
721 |
< |
if ( !isTechBit0 || !isTechBit40 || isBeamHalo || (fHighPurity<0.2&&nTracks>10) || vertexColl->begin()->isFake()) return; |
722 |
< |
} |
629 |
> |
|
630 |
> |
// ======= Get MC information if needed |
631 |
> |
bool MC=false; |
632 |
> |
Handle<HepMCProduct> evtMC; |
633 |
> |
if (!realData_) { |
634 |
> |
iEvent.getByLabel("generator",evtMC); |
635 |
> |
if (!evtMC.isValid()) { |
636 |
> |
MC=false; |
637 |
> |
edm::LogInfo("Debug") << "[PVStudy] no HepMCProduct found"<< endl; |
638 |
> |
} else { |
639 |
> |
edm::LogInfo("Debug") << "[PVStudy] generator HepMCProduct found"<< endl; |
640 |
> |
MC=true; |
641 |
|
} |
642 |
< |
} // End of Apply event cleaning cuts for firstcoll data |
725 |
< |
|
726 |
< |
|
642 |
> |
} |
643 |
|
|
644 |
< |
//======================================================= |
645 |
< |
// Fill trackparameters of the input tracks to pvtx fitter |
646 |
< |
//======================================================= |
644 |
> |
//======================================================================== |
645 |
> |
// Step 1: Apply event cleaning for data and MC |
646 |
> |
// WARNING: event selection cut are hard coded!! |
647 |
> |
//======================================================================== |
648 |
> |
// =====Loop over the trackColl to get the fraction of HighPurity tracks |
649 |
> |
int nTracks = 0; |
650 |
> |
int nHighPurityTracks=0; |
651 |
> |
double fHighPurity=0; |
652 |
> |
|
653 |
> |
for (unsigned int i=0; i<trackCollectionHandle->size(); i++, nTracks++) { |
654 |
> |
TrackRef tkref(trackCollectionHandle,i); |
655 |
> |
if(tkref->quality(reco::TrackBase::highPurity)) ++nHighPurityTracks; |
656 |
> |
} |
657 |
> |
if(nTracks>0) |
658 |
> |
fHighPurity = double(nHighPurityTracks)/double(nTracks); |
659 |
> |
if(verbose_) |
660 |
> |
cout<<"fraction of HighPurity track is "<<fHighPurity<<endl; |
661 |
> |
|
662 |
> |
if( (fHighPurity<fHighPuritycut_ && nTracks>10) || vertexColl->begin()->isFake()) return; |
663 |
> |
|
664 |
> |
glob_runno_ = iEvent.id().run(); |
665 |
> |
glob_evtno_ = iEvent.id().event(); |
666 |
> |
glob_ls_ = iEvent.luminosityBlock(); |
667 |
> |
glob_bx_ = iEvent.bunchCrossing(); |
668 |
> |
|
669 |
> |
//======================================================================== |
670 |
> |
// Step 2: Fill histograms for the splitting consistency checks |
671 |
> |
//======================================================================== |
672 |
> |
|
673 |
> |
// === Fill trackparameters of the input tracks to pvtx fitter |
674 |
|
edm::LogInfo("Debug")<<"[PVStudy] Start filling track parameters of the input tracks to pvtx fitter."<<endl; |
675 |
|
//fillTrackHisto(const reco::TrackCollection *trackColl, int datatype, const Point & bs) |
676 |
|
// datatype: unsplittracks (0); splittracks1 (1); splittracks2 (2); |
678 |
|
fillTrackHisto(splitTrackColl1, 1, beamSpot); |
679 |
|
fillTrackHisto(splitTrackColl2, 2, beamSpot); |
680 |
|
edm::LogInfo("Debug")<<"[PVStudy] End filling track parameters of the input tracks to pvtx fitter."<<endl; |
681 |
< |
|
682 |
< |
|
683 |
< |
//======================================================= |
741 |
< |
// Fill pixelVertices related histograms |
742 |
< |
//======================================================= |
743 |
< |
nrecPV_pxlpvtx_ = int (pixelVertexColl->size()); |
744 |
< |
if(pixelVertexColl->size()>0 && pixelVertexColl->begin()->isValid() && !(pixelVertexColl->begin()->isFake())) { |
745 |
< |
//fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, bool fillHisto, bool fillNtuple, const Point & bs) { |
746 |
< |
fillTrackHistoInPV(pixelVertexColl, 4, false, true, beamSpot); |
747 |
< |
h_pixvtx->Fill1d("dzErr_pxlpvtx", pixelVertexColl->begin()->zError()); |
748 |
< |
// Get the dZ error of the tracks in the leading pixelVertexColl |
749 |
< |
for(reco::Vertex::trackRef_iterator t = (pixelVertexColl->begin())->tracks_begin(); |
750 |
< |
t!= (pixelVertexColl->begin())->tracks_end(); t++) { |
751 |
< |
|
752 |
< |
if ( (**t).charge() < -1 || (**t).charge() > 1 ) { |
753 |
< |
// h_pvtrk->Fill1d("trkPtPV", 0.); |
754 |
< |
} |
755 |
< |
else |
756 |
< |
h_pixvtx->Fill1d("trkdzErr_pxlpvtx", (**t).dzError()); |
757 |
< |
} |
758 |
< |
// Fill the track->dz() in the PV difference from first pixelpvtx |
759 |
< |
for(reco::Vertex::trackRef_iterator t = (vertexColl->begin())->tracks_begin(); |
760 |
< |
t!= (vertexColl->begin())->tracks_end(); t++) { |
761 |
< |
// illegal charge |
762 |
< |
if ( (**t).charge() < -1 || (**t).charge() > 1 ) { |
763 |
< |
// h_pvtrk->Fill1d("trkPtPV", 0.); |
764 |
< |
} |
765 |
< |
else { |
766 |
< |
if(pixelVertexColl->size()>0) { |
767 |
< |
h_pixvtx->Fill1d("trkdz_pxlpvtxdz", (**t).dz() - pixelVertexColl->begin()->z()); |
768 |
< |
h_pixvtx->Fill1d("trkdz_pxlpvtxdz_pxlpvtxdzerr", fabs((**t).dz() - pixelVertexColl->begin()->z())/pixelVertexColl->begin()->zError()); |
769 |
< |
h_pixvtx->Fill1d("trkdz_pxlpvtxdz_trkdzerr", fabs((**t).dz() - pixelVertexColl->begin()->z())/(**t).dzError()); |
770 |
< |
h_pixvtx->Fill1d("trkdzErr_pvtx", (**t).dzError()); |
771 |
< |
} |
772 |
< |
} |
773 |
< |
} |
774 |
< |
} |
775 |
< |
|
776 |
< |
//======================================================= |
777 |
< |
// Fill number of reconstructed vertices |
778 |
< |
//======================================================= |
779 |
< |
|
681 |
> |
|
682 |
> |
|
683 |
> |
// ==== Fill number of reconstructed vertices |
684 |
|
edm::LogInfo("Debug")<<"[PVStudy] Printing vertexCollection: "<<endl; |
685 |
|
edm::LogInfo("Debug")<<"[PVStudy] Printing splitVertexCollection1: "<<endl; |
686 |
|
edm::LogInfo("Debug")<<"[PVStudy] Printing splitVertexCollection2: "<<endl; |
690 |
|
printRecVtxs(splitVertexCollection1Handle); |
691 |
|
printRecVtxs(splitVertexCollection2Handle); |
692 |
|
} |
693 |
+ |
|
694 |
|
nrecPV_ = int (vertexColl->size()); |
695 |
|
nrecPV1_spl_ = int (splitVertexColl1->size()); |
696 |
|
nrecPV2_spl_ = int (splitVertexColl2->size()); |
701 |
|
h_misc->Fill1d("nrecPVDiff", double(nrecPV1_spl_)-double(nrecPV2_spl_)); |
702 |
|
|
703 |
|
|
704 |
< |
//================================================================= |
705 |
< |
// Fill track parameter ntuple/hist for tracks used in recoVertices |
801 |
< |
//================================================================= |
802 |
< |
|
803 |
< |
//fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, bool fillHisto, bool fillNtuple, const Point & bs) { |
704 |
> |
// ======= Fill track parameter ntuple/hist for tracks used in recoVertices |
705 |
> |
//fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, const Point & bs) { |
706 |
|
if(vertexColl->size() > 0 && vertexColl->begin()->isValid() && !(vertexColl->begin()->isFake())) |
707 |
< |
fillTrackHistoInPV(vertexColl, 0, true, true, beamSpot); |
806 |
< |
|
807 |
< |
if(splitVertexColl1->size() > 0 && splitVertexColl1->begin()->isValid() && !(splitVertexColl1->begin()->isFake())) |
808 |
< |
fillTrackHistoInPV(splitVertexColl1, 1, false, true, beamSpot); |
809 |
< |
|
810 |
< |
if(splitVertexColl2->size() > 0 && splitVertexColl2->begin()->isValid() && !(splitVertexColl2->begin()->isFake())) |
811 |
< |
fillTrackHistoInPV(splitVertexColl2, 2, false, true, beamSpot); |
812 |
< |
|
813 |
< |
//======================================================= |
814 |
< |
// Compare offlinePrimaryVertices with pixelVertices |
815 |
< |
//======================================================= |
816 |
< |
if( (pixelVertexColl->size()>0 && pixelVertexColl->begin()->isValid() && !(pixelVertexColl->begin()->isFake())) |
817 |
< |
&& (vertexColl->size()>0 && vertexColl->begin()->isValid() && !(vertexColl->begin()->isFake())) ) { |
818 |
< |
h_pixvtx->Fill1d("nrecPV_minus_nrecPxlPV", double (nrecPV_ - nrecPV_pxlpvtx_)); |
819 |
< |
// difference in reconstructed position of the leading pvtx |
820 |
< |
//edm::LogInfo("Debug")<<"recx_[0] = "<< recx_[0] << "recx_pxlpvtx_[0] = "<< recx_pxlpvtx_[0]<<endl; |
821 |
< |
//edm::LogInfo("Debug")<<"recy_[0] = "<< recy_[0] << "recy_pxlpvtx_[0] = "<< recy_pxlpvtx_[0]<<endl; |
822 |
< |
h_pixvtx->Fill1d("recxPV_minus_recxPxlPV", recx_[0] - recx_pxlpvtx_[0]); |
823 |
< |
h_pixvtx->Fill1d("recyPV_minus_recyPxlPV", recy_[0] - recy_pxlpvtx_[0]); |
824 |
< |
h_pixvtx->Fill1d("reczPV_minus_reczPxlPV", recz_[0] - recz_pxlpvtx_[0]); |
825 |
< |
} |
826 |
< |
|
827 |
< |
|
828 |
< |
//========================================================== |
829 |
< |
// Fill secondary/primary min separations for multi-vertices |
830 |
< |
//========================================================== |
707 |
> |
fillTrackHistoInPV(vertexColl, 0, beamSpot); |
708 |
|
|
709 |
+ |
// ======= Fill secondary/primary min separations for multi-vertices |
710 |
|
if(vertexColl->size()>1 && vertexColl->begin()->isValid() && !(vertexColl->begin()->isFake()) ) { |
833 |
– |
|
711 |
|
double min_xsep = 9999.0; |
712 |
|
double min_ysep = 9999.0; |
713 |
|
double min_zsep = 9999.0; |
764 |
|
min_zsep_ = min_zsep; |
765 |
|
min_ntrksep_ = min_ntrksep; |
766 |
|
min_sumpt2sep_ = min_sumpt2sep; |
890 |
– |
|
891 |
– |
|
767 |
|
} // End of if(vertexColl->size()>1) { |
768 |
|
|
769 |
|
edm::LogInfo("Debug")<<"[PVStudy] End filling pvtx parameters."<<endl; |
770 |
|
|
771 |
< |
|
772 |
< |
|
773 |
< |
//======================================================= |
774 |
< |
// PrimaryVertex Matching |
775 |
< |
// 1. In z |z1-z2|< zsigncut * max(sigmaz1, sigmaz2) |
776 |
< |
// 2. |(nTrkPV1 - nTrkPV2)/(nTrkPV1+nTrkPV2)|<ntrkdiffcut_ |
902 |
< |
// Assume the first match is the primary vertex, |
903 |
< |
// since vertexColl are sorted in decreasing order of sum(pT) |
904 |
< |
//======================================================= |
771 |
> |
//======================================================================== |
772 |
> |
// Step 3: PrimaryVertex Matching |
773 |
> |
// 1. In z |z1-z2|< zsigncut * max(sigmaz1, sigmaz2) |
774 |
> |
// 2. |(nTrkPV1 - nTrkPV2)/(nTrkPV1+nTrkPV2)|<ntrkdiffcut_ |
775 |
> |
// 3. The first match is the primary vertex, which has largest sum(pT^2) |
776 |
> |
//======================================================================== |
777 |
|
|
778 |
|
edm::LogInfo("Debug")<<"[PVStudy] matching pvtxs "<<endl; |
779 |
|
reco::VertexCollection s_empty_matchedVertexColl1; |
803 |
|
if(matchVertex(recvtx1, recvtx2, zsigncut_)) { |
804 |
|
edm::LogInfo("Debug")<<"[PVStudy] The two splitted vertices match in Z. "<<endl; |
805 |
|
|
806 |
< |
int nTrkPV1 = recvtx1->tracksSize(); |
807 |
< |
int nTrkPV2 = recvtx2->tracksSize(); |
806 |
> |
int nTrkPV1, nTrkPV2; |
807 |
> |
if(useHWTrk_) { |
808 |
> |
nTrkPV1 = nHWTrkRecVtx(*recvtx1); |
809 |
> |
nTrkPV2 = nHWTrkRecVtx(*recvtx2); |
810 |
> |
} |
811 |
> |
else { |
812 |
> |
nTrkPV1 = recvtx1->tracksSize(); |
813 |
> |
nTrkPV2 = recvtx2->tracksSize(); |
814 |
> |
} |
815 |
|
double ntrkreldiff = double(nTrkPV1-nTrkPV2)/double(nTrkPV1+nTrkPV2); |
816 |
|
h_misc->Fill1d("nTrkPVDiff", nTrkPV1-nTrkPV2); |
817 |
|
h_misc->Fill1d("nTrkPVRelDiff", ntrkreldiff); |
834 |
|
} |
835 |
|
} |
836 |
|
edm::LogInfo("Debug")<<"[PVStudy] End matching pvtxs"<<endl; |
837 |
< |
|
838 |
< |
//======================================================= |
839 |
< |
// Analyze matchedVertexColls |
840 |
< |
//======================================================= |
837 |
> |
|
838 |
> |
//======================================================================== |
839 |
> |
// Step 4: Analyze matchedVertexColls |
840 |
> |
//======================================================================== |
841 |
|
edm::LogInfo("Debug")<<"[PVStudy] Begin analyzing the matchedVertexColl with size = "<< matchedVertexColl1->size()<< endl; |
842 |
|
|
843 |
+ |
// ==== If it is MC, analyze the res/pull of the unsplit vertexColl first |
844 |
+ |
if(MC){ |
845 |
+ |
// make a list of primary vertices: |
846 |
+ |
std::vector<simPrimaryVertex> simpv; |
847 |
+ |
simpv=getSimPVs(evtMC,""); |
848 |
+ |
// simpv=getSimPVs(evtMC, simVtxs, simTrks); |
849 |
+ |
h_gen->Fill1d("nsimPV", simpv.size()); |
850 |
+ |
nsimPV_ = simpv.size(); |
851 |
+ |
int isimpv = 0; |
852 |
+ |
for(std::vector<simPrimaryVertex>::iterator vsim=simpv.begin(); |
853 |
+ |
vsim!=simpv.end(); vsim++, isimpv++){ |
854 |
+ |
nsimTrkPV_[isimpv] =vsim->nGenTrk; |
855 |
+ |
simx_[isimpv] = vsim->x; |
856 |
+ |
simy_[isimpv] = vsim->y; |
857 |
+ |
simz_[isimpv] = vsim->y; |
858 |
+ |
simptsq_[isimpv] = vsim->ptsq; |
859 |
+ |
//fillMCHisto((std::vector<simPrimaryVertex>::iterator vsim, int isimpv, const reco::VertexCollection *vtxColl, int datatype) { |
860 |
+ |
fillMCHisto(vsim, isimpv, vertexColl, 3); |
861 |
+ |
} |
862 |
+ |
} |
863 |
+ |
|
864 |
|
// Compare the reconstructed vertex position and calculate resolution/pulls |
865 |
|
if(matchedVertexColl1->size() != 0 && matchedVertexColl2->size() != 0) { |
866 |
< |
//fillPvtxHisto(const reco::VertexCollection *vertexColl, int datatype, bool fillHisto, bool fillNtuple) |
867 |
< |
fillTrackHistoInPV(matchedVertexColl1, 1, true, false, beamSpot); |
868 |
< |
fillTrackHistoInPV(matchedVertexColl2, 2, true, false, beamSpot); |
866 |
> |
// ==== Analyze the splitted vtx using MC method |
867 |
> |
if(MC){ |
868 |
> |
// make a list of primary vertices: |
869 |
> |
std::vector<simPrimaryVertex> simpv; |
870 |
> |
simpv=getSimPVs(evtMC,""); |
871 |
> |
// simpv=getSimPVs(evtMC, simVtxs, simTrks); |
872 |
> |
int isimpv = 0; |
873 |
> |
for(std::vector<simPrimaryVertex>::iterator vsim=simpv.begin(); |
874 |
> |
vsim!=simpv.end(); vsim++, isimpv++){ |
875 |
> |
//fillMCHisto((std::vector<simPrimaryVertex>::iterator vsim, int isimpv, const reco::VertexCollection *vtxColl, int datatype) { |
876 |
> |
fillMCHisto(vsim, isimpv, matchedVertexColl1, 1); |
877 |
> |
fillMCHisto(vsim, isimpv, matchedVertexColl2, 2); |
878 |
> |
} |
879 |
> |
} |
880 |
> |
|
881 |
> |
// ==== Analyze res/pull two-vertex method |
882 |
> |
//fillPvtxHisto(const reco::VertexCollection *vertexColl, int datatype) |
883 |
> |
fillTrackHistoInPV(matchedVertexColl1, 1, beamSpot); |
884 |
> |
fillTrackHistoInPV(matchedVertexColl2, 2, beamSpot); |
885 |
|
|
886 |
|
reco::VertexCollection::const_iterator v1; |
887 |
|
reco::VertexCollection::const_iterator v2; |
888 |
+ |
|
889 |
|
nrecPV_twovtx_ = 0; |
890 |
|
for(v1 = matchedVertexColl1->begin(), v2 = matchedVertexColl2->begin(); |
891 |
|
v1!=matchedVertexColl1->end(), v2 != matchedVertexColl2->end(); |
898 |
|
ferror_[0] = sqrt(pow(v1->xError(),2)+pow(v2->xError(),2))/sqrt(2); |
899 |
|
ferror_[1] = sqrt(pow(v1->yError(),2)+pow(v2->yError(),2))/sqrt(2); |
900 |
|
ferror_[2] = sqrt(pow(v1->zError(),2)+pow(v2->zError(),2))/sqrt(2); |
984 |
– |
fntrk_ = (v1->tracksSize()+v2->tracksSize())/2; |
901 |
|
|
902 |
+ |
int nTrkPV1, nTrkPV2; |
903 |
+ |
if(useHWTrk_) { |
904 |
+ |
nTrkPV1 = nHWTrkRecVtx(*v1); |
905 |
+ |
nTrkPV2 = nHWTrkRecVtx(*v2); |
906 |
+ |
} |
907 |
+ |
else { |
908 |
+ |
nTrkPV1 = v1->tracksSize(); |
909 |
+ |
nTrkPV2 = v2->tracksSize(); |
910 |
+ |
} |
911 |
+ |
|
912 |
+ |
fntrk_ = (nTrkPV1 + nTrkPV2)/2; |
913 |
+ |
|
914 |
|
h_summary->Fill1d("deltax", fres_[0] ); |
915 |
|
h_summary->Fill1d("deltay", fres_[1] ); |
916 |
|
h_summary->Fill1d("deltaz", fres_[2] ); |
927 |
|
fillHisto(res(fres_[0], fres_[1], fres_[2]), |
928 |
|
error(ferror_[0], ferror_[1], ferror_[2]), |
929 |
|
fntrk_,0); |
930 |
+ |
|
931 |
|
// Fill the ntuple variables |
1003 |
– |
nTrkPV1_spl_twovtx_[nrecPV_twovtx_] = v1->tracksSize(); |
1004 |
– |
nTrkPV2_spl_twovtx_[nrecPV_twovtx_] = v2->tracksSize(); |
1005 |
– |
sumptsq1_spl_twovtx_[nrecPV_twovtx_] = sumPtSquared(*v1); |
1006 |
– |
sumptsq2_spl_twovtx_[nrecPV_twovtx_] = sumPtSquared(*v2); |
932 |
|
nTrkPV_twovtx_[nrecPV_twovtx_] = fntrk_; |
933 |
|
deltax_twovtx_[nrecPV_twovtx_] = fres_[0]; |
934 |
|
deltay_twovtx_[nrecPV_twovtx_] = fres_[1]; |
939 |
|
pullx_twovtx_[nrecPV_twovtx_] = fres_[0]/ferror_[0]; |
940 |
|
pully_twovtx_[nrecPV_twovtx_] = fres_[1]/ferror_[1]; |
941 |
|
pullz_twovtx_[nrecPV_twovtx_] = fres_[2]/ferror_[2]; |
1017 |
– |
nrecPV_twovtx_++; |
1018 |
– |
} // End of analyzing res/pull |
942 |
|
|
1020 |
– |
//======================================================= |
1021 |
– |
// Fill simulated vertices |
1022 |
– |
//======================================================= |
1023 |
– |
if (!realData_) { |
1024 |
– |
bool MC=false; |
1025 |
– |
Handle<HepMCProduct> evtMC; |
1026 |
– |
iEvent.getByLabel("generator",evtMC); |
1027 |
– |
if (!evtMC.isValid()) { |
1028 |
– |
MC=false; |
1029 |
– |
edm::LogInfo("Debug") << "[PVStudy] no HepMCProduct found"<< endl; |
1030 |
– |
} else { |
1031 |
– |
edm::LogInfo("Debug") << "[PVStudy] generator HepMCProduct found"<< endl; |
1032 |
– |
MC=true; |
1033 |
– |
} |
1034 |
– |
|
1035 |
– |
if(MC){ |
1036 |
– |
// make a list of primary vertices: |
1037 |
– |
std::vector<simPrimaryVertex> simpv; |
1038 |
– |
simpv=getSimPVs(evtMC,""); |
1039 |
– |
// simpv=getSimPVs(evtMC, simVtxs, simTrks); |
1040 |
– |
h_gen->Fill1d("nsimPV", simpv.size()); |
1041 |
– |
nsimPV_ = simpv.size(); |
1042 |
– |
|
1043 |
– |
int isimpv = 0; |
1044 |
– |
for(std::vector<simPrimaryVertex>::iterator vsim=simpv.begin(); |
1045 |
– |
vsim!=simpv.end(); vsim++, isimpv++){ |
1046 |
– |
nsimTrkPV_[isimpv] =vsim->nGenTrk; |
1047 |
– |
simx_[isimpv] = vsim->x; |
1048 |
– |
simy_[isimpv] = vsim->y; |
1049 |
– |
simz_[isimpv] = vsim->y; |
1050 |
– |
simptsq_[isimpv] = vsim->ptsq; |
1051 |
– |
|
1052 |
– |
//fillMCHisto((std::vector<simPrimaryVertex>::iterator vsim, int isimpv, const reco::VertexCollection *vtxColl, int datatype) { |
1053 |
– |
fillMCHisto(vsim, isimpv, splitVertexColl1, 1); |
1054 |
– |
fillMCHisto(vsim, isimpv, splitVertexColl2, 2); |
1055 |
– |
fillMCHisto(vsim, isimpv, vertexColl, 3); |
1056 |
– |
} |
1057 |
– |
} // End of for(std::vector<simPrimaryVertex>::iterator vsim=simpv.begin() |
1058 |
– |
} // End of if (!realData_) { |
943 |
|
|
944 |
+ |
//SplittedVertex |
945 |
+ |
nTrkPV1_spl_twovtx_[nrecPV_twovtx_] = nTrkPV1; |
946 |
+ |
ndofPV1_spl_twovtx_[nrecPV_twovtx_] = v1->ndof(); |
947 |
+ |
normchi2PV1_spl_twovtx_[nrecPV_twovtx_] = v1->normalizedChi2(); |
948 |
+ |
avgPtPV1_spl_twovtx_[nrecPV_twovtx_] = avgPtRecVtx(*v1); |
949 |
+ |
errx1_spl_twovtx_[nrecPV_twovtx_] = v1->xError(); |
950 |
+ |
erry1_spl_twovtx_[nrecPV_twovtx_] = v1->yError(); |
951 |
+ |
errz1_spl_twovtx_[nrecPV_twovtx_] = v1->zError(); |
952 |
+ |
|
953 |
+ |
nTrkPV2_spl_twovtx_[nrecPV_twovtx_] = nTrkPV2; |
954 |
+ |
ndofPV2_spl_twovtx_[nrecPV_twovtx_] = v2->ndof(); |
955 |
+ |
normchi2PV2_spl_twovtx_[nrecPV_twovtx_] = v2->normalizedChi2(); |
956 |
+ |
avgPtPV2_spl_twovtx_[nrecPV_twovtx_] = avgPtRecVtx(*v2); |
957 |
+ |
errx2_spl_twovtx_[nrecPV_twovtx_] = v2->xError(); |
958 |
+ |
erry2_spl_twovtx_[nrecPV_twovtx_] = v2->yError(); |
959 |
+ |
errz2_spl_twovtx_[nrecPV_twovtx_] = v2->zError(); |
960 |
+ |
|
961 |
+ |
nrecPV_twovtx_++; |
962 |
+ |
|
963 |
+ |
// Print some information of the two tracks events |
964 |
+ |
if(verbose_ && fntrk_ < 4) { |
965 |
+ |
cout<<"Printing matchedVertexColl1 for low ntrack bins"<<endl; |
966 |
+ |
printRecVtx(*v1); |
967 |
+ |
cout<<"Printing matchedVertexColl1 for low ntrack bins"<<endl; |
968 |
+ |
printRecVtx(*v2); |
969 |
+ |
} |
970 |
+ |
} // End of analyzing res/pull |
971 |
|
} // End of if(matchedVertexColl1->size() == 1 && matchedVertexColl2->size() == 1 ) { |
972 |
|
else |
973 |
|
edm::LogInfo("Debug")<<"[PVStudy] WARNING: Cannot find matching pvtxs"<<endl; |
975 |
|
edm::LogInfo("Debug")<<"[PVStudy] End analyzing the matchedVertexColl"<<endl; |
976 |
|
|
977 |
|
|
978 |
< |
|
979 |
< |
// Finally fill the ftree_ |
978 |
> |
//======================================================================== |
979 |
> |
// Step 5: Fill ntuple if saventuple_ is on |
980 |
> |
//======================================================================== |
981 |
|
if(saventuple_) { |
982 |
|
ftree_->Fill(); |
983 |
|
pvtxtree_->Fill(); |
1044 |
|
{ |
1045 |
|
if(vrec.isFake() || !vrec.isValid()) return false; |
1046 |
|
else |
1047 |
< |
return (fabs(vsim.z-vrec.z())<0.0500); // =500um |
1047 |
> |
return true; |
1048 |
> |
//return (fabs(vsim.z-vrec.z())<0.0500); // =500um // remove this hard cut |
1049 |
|
} |
1050 |
|
|
1051 |
|
// Match two reconstructed vertices |
1112 |
|
} |
1113 |
|
} |
1114 |
|
|
1115 |
+ |
void PVStudy::printRecVtx(const reco::Vertex & v){ |
1116 |
+ |
|
1117 |
+ |
cout << "#trk " << std::setw(3) << v.tracksSize() |
1118 |
+ |
<< " chi2 " << std::setw(4) << v.chi2() |
1119 |
+ |
<< " ndof " << std::setw(3) << v.ndof() << endl |
1120 |
+ |
<< " x " << std::setw(8) <<std::fixed << std::setprecision(4) << v.x() |
1121 |
+ |
<< " dx " << std::setw(8) << v.xError()<< endl |
1122 |
+ |
<< " y " << std::setw(8) << v.y() |
1123 |
+ |
<< " dy " << std::setw(8) << v.yError()<< endl |
1124 |
+ |
<< " z " << std::setw(8) << v.z() |
1125 |
+ |
<< " dz " << std::setw(8) << v.zError() |
1126 |
+ |
<< endl; |
1127 |
+ |
} |
1128 |
+ |
|
1129 |
|
void PVStudy::printSimTrks(const Handle<SimTrackContainer> simTrks){ |
1130 |
|
cout << " simTrks type, (momentum), vertIndex, genpartIndex" << endl; |
1131 |
|
int i=1; |
1297 |
|
} |
1298 |
|
} |
1299 |
|
|
1300 |
< |
void PVStudy::fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, bool fillHisto, bool fillNtuple, const Point & bs) { |
1300 |
> |
void PVStudy::fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, const Point & bs) { |
1301 |
|
string suffix; |
1302 |
|
suffix = ""; // for unsplit tracks |
1303 |
|
if(datatype == 1) suffix = "1_spl"; // for splittracks 1 |
1305 |
|
int ivtx = 0; |
1306 |
|
for(reco::VertexCollection::const_iterator v=vertexColl->begin(); |
1307 |
|
v!=vertexColl->end(); ++v, ++ivtx) { |
1308 |
< |
if(v->isFake()) continue; |
1309 |
< |
if(fillNtuple) { |
1310 |
< |
if(datatype == 0) { |
1311 |
< |
nTrkPV_[ivtx] = v->tracksSize(); |
1312 |
< |
sumptsq_[ivtx] = sumPtSquared(*v); |
1313 |
< |
isValid_[ivtx] = v->isValid(); |
1314 |
< |
isFake_[ivtx] = v->isFake(); |
1315 |
< |
recx_[ivtx] = v->x(); |
1316 |
< |
recy_[ivtx] = v->y(); |
1317 |
< |
recz_[ivtx] = v->z(); |
1318 |
< |
recx_err_[ivtx] = v->xError(); |
1319 |
< |
recy_err_[ivtx] = v->yError(); |
1320 |
< |
recz_err_[ivtx] = v->zError(); |
1321 |
< |
} |
1322 |
< |
if(datatype == 1) { |
1323 |
< |
nTrkPV1_spl_[ivtx] = v->tracksSize(); |
1324 |
< |
sumptsq1_spl_[ivtx] = sumPtSquared(*v); |
1325 |
< |
isValid1_spl_[ivtx] = v->isValid(); |
1326 |
< |
isFake1_spl_[ivtx] = v->isFake(); |
1400 |
< |
recx1_spl_[ivtx] = v->x(); |
1401 |
< |
recy1_spl_[ivtx] = v->y(); |
1402 |
< |
recz1_spl_[ivtx] = v->z(); |
1403 |
< |
recx1_err_spl_[ivtx] = v->xError(); |
1404 |
< |
recy1_err_spl_[ivtx] = v->yError(); |
1405 |
< |
recz1_err_spl_[ivtx] = v->zError(); |
1406 |
< |
|
1407 |
< |
} |
1408 |
< |
if(datatype == 2) { |
1409 |
< |
nTrkPV2_spl_[ivtx] = v->tracksSize(); |
1410 |
< |
sumptsq2_spl_[ivtx] = sumPtSquared(*v); |
1411 |
< |
isValid2_spl_[ivtx] = v->isValid(); |
1412 |
< |
isFake2_spl_[ivtx] = v->isFake(); |
1413 |
< |
recx2_spl_[ivtx] = v->x(); |
1414 |
< |
recy2_spl_[ivtx] = v->y(); |
1415 |
< |
recz2_spl_[ivtx] = v->z(); |
1416 |
< |
recx2_err_spl_[ivtx] = v->xError(); |
1417 |
< |
recy2_err_spl_[ivtx] = v->yError(); |
1418 |
< |
recz2_err_spl_[ivtx] = v->zError(); |
1419 |
< |
} |
1420 |
< |
if(datatype == 4) { // for pixelVertices |
1421 |
< |
nTrkPV_pxlpvtx_[ivtx] = v->tracksSize(); |
1422 |
< |
sumptsq_pxlpvtx_[ivtx] = sumPtSquared(*v); |
1423 |
< |
isValid_pxlpvtx_[ivtx] = v->isValid(); |
1424 |
< |
isFake_pxlpvtx_[ivtx] = v->isFake(); |
1425 |
< |
recx_pxlpvtx_[ivtx] = v->x(); |
1426 |
< |
recy_pxlpvtx_[ivtx] = v->y(); |
1427 |
< |
recz_pxlpvtx_[ivtx] = v->z(); |
1428 |
< |
recx_err_pxlpvtx_[ivtx] = v->xError(); |
1429 |
< |
recy_err_pxlpvtx_[ivtx] = v->yError(); |
1430 |
< |
recz_err_pxlpvtx_[ivtx] = v->zError(); |
1431 |
< |
} |
1432 |
< |
} |
1433 |
< |
} |
1434 |
< |
// For histogram only fill the leading pvtx parameters |
1435 |
< |
if(fillHisto) { |
1436 |
< |
h_pvtrk->Fill1d("nTrkPV"+suffix, vertexColl->begin()->tracksSize()); |
1437 |
< |
h_pvtrk->Fill1d("ndofPV"+suffix, vertexColl->begin()->ndof()); |
1438 |
< |
int nHWTrkPV_ = 0; |
1439 |
< |
|
1440 |
< |
try { |
1441 |
< |
for(reco::Vertex::trackRef_iterator t = vertexColl->begin()->tracks_begin(); |
1442 |
< |
t!=vertexColl->begin()->tracks_end(); t++) { |
1443 |
< |
// illegal charge |
1444 |
< |
if ( (**t).charge() < -1 || (**t).charge() > 1 ) { |
1445 |
< |
// h_pvtrk->Fill1d("trkPtPV", 0.); |
1446 |
< |
} |
1447 |
< |
else { |
1448 |
< |
h_pvtrk->Fill1d("trkPtPV"+suffix, (**t).pt()); |
1449 |
< |
h_pvtrk->Fill1d("trkDxyPV"+suffix, (**t).dxy()); |
1450 |
< |
h_pvtrk->Fill1d("trkDxyCorrPV"+suffix, (**t).dxy(bs)); |
1451 |
< |
h_pvtrk->Fill1d("trkDzPV"+suffix, (**t).dz()); |
1452 |
< |
h_pvtrk->Fill1d("trkEtaPV"+suffix, (**t).eta()); |
1453 |
< |
h_pvtrk->Fill1d("trkPhiPV"+suffix, (**t).phi()); |
1454 |
< |
|
1455 |
< |
if(vertexColl->begin()->trackWeight(*t) > 0.5) |
1456 |
< |
nHWTrkPV_++; |
1308 |
> |
if(!v->isFake()) { |
1309 |
> |
h_pvtrk->Fill1d("nTrkPV"+suffix, vertexColl->begin()->tracksSize()); |
1310 |
> |
h_pvtrk->Fill1d("ndofPV"+suffix, vertexColl->begin()->ndof()); |
1311 |
> |
h_pvtrk->Fill1d("nHWTrkPV"+suffix, nHWTrkRecVtx(*v)); |
1312 |
> |
try { |
1313 |
> |
for(reco::Vertex::trackRef_iterator t = vertexColl->begin()->tracks_begin(); |
1314 |
> |
t!=vertexColl->begin()->tracks_end(); t++) { |
1315 |
> |
// illegal charge |
1316 |
> |
if ( (**t).charge() < -1 || (**t).charge() > 1 ) { |
1317 |
> |
// h_pvtrk->Fill1d("trkPtPV", 0.); |
1318 |
> |
} |
1319 |
> |
else { |
1320 |
> |
h_pvtrk->Fill1d("trkPtPV"+suffix, (**t).pt()); |
1321 |
> |
h_pvtrk->Fill1d("trkDxyPV"+suffix, (**t).dxy()); |
1322 |
> |
h_pvtrk->Fill1d("trkDxyCorrPV"+suffix, (**t).dxy(bs)); |
1323 |
> |
h_pvtrk->Fill1d("trkDzPV"+suffix, (**t).dz()); |
1324 |
> |
h_pvtrk->Fill1d("trkEtaPV"+suffix, (**t).eta()); |
1325 |
> |
h_pvtrk->Fill1d("trkPhiPV"+suffix, (**t).phi()); |
1326 |
> |
} |
1327 |
|
} |
1328 |
|
} |
1329 |
+ |
catch (...) { |
1330 |
+ |
// exception thrown when trying to use linked track |
1331 |
+ |
// h_pvtrk->Fill1d("trkPtPV", 0.); |
1332 |
+ |
} |
1333 |
|
} |
1460 |
– |
catch (...) { |
1461 |
– |
// exception thrown when trying to use linked track |
1462 |
– |
// h_pvtrk->Fill1d("trkPtPV", 0.); |
1463 |
– |
} |
1464 |
– |
h_pvtrk->Fill1d("nHWTrkPV"+suffix, nHWTrkPV_); |
1334 |
|
} |
1466 |
– |
|
1467 |
– |
|
1335 |
|
} |
1336 |
|
|
1337 |
|
void PVStudy::fillMCHisto(std::vector<simPrimaryVertex>::iterator vsim, int isimpv, const reco::VertexCollection *vtxColl, int datatype) |
1350 |
|
suffix = "_mct"; |
1351 |
|
nrecPV_mct_ = 0; |
1352 |
|
} |
1353 |
< |
|
1353 |
> |
|
1354 |
|
//======================================================== |
1355 |
< |
// look for a matching reconstructed vertex in vertexColl |
1356 |
< |
//======================================================== |
1357 |
< |
|
1355 |
> |
// For each simulated vertex, look for a match in the vertexColl |
1356 |
> |
// If more than 1 recVtx is found, use the one with closest in Z |
1357 |
> |
//======================================================== |
1358 |
> |
|
1359 |
> |
// === look for a matching reconstructed vertex in vertexColl |
1360 |
|
for(reco::VertexCollection::const_iterator vrec=vtxColl->begin(); |
1361 |
|
vrec!=vtxColl->end(); ++vrec){ |
1493 |
– |
int nrectrk = vrec->tracksSize(); |
1362 |
|
vsim->recVtx=NULL; |
1495 |
– |
|
1363 |
|
edm::LogInfo("Debug") << "[fillMCHisto] sim primary vertex x = " << vsim->x << "; y = " << vsim->y << "; z = " << vsim->z << endl; |
1364 |
|
edm::LogInfo("Debug") << "[fillMCHisto] Is matched? " << (matchVertex(*vsim,*vrec)?"Yes":"No") << endl; |
1365 |
< |
|
1365 |
> |
|
1366 |
|
if ( matchVertex(*vsim,*vrec) ) { |
1367 |
|
vsim->recVtx=&(*vrec); |
1368 |
< |
|
1502 |
< |
// if the matching critera are fulfilled, accept the rec-vertex that is closest in z |
1368 |
> |
//if the matching critera are fulfilled, accept the rec-vertex that is closest in z |
1369 |
|
//if( ((vsim->recVtx) && (fabs(vsim->recVtx->position().z()-vsim->z)>fabs(vrec->z()-vsim->z))) |
1370 |
|
//|| (!vsim->recVtx) ) |
1371 |
|
//vsim->recVtx=&(*vrec); |
1506 |
– |
//} |
1507 |
– |
edm::LogInfo("Debug") <<"[fillMCHisto] primary matched in vertexColl" << vsim->x << " " << vsim->y << " " << vsim->z << endl; |
1508 |
– |
|
1509 |
– |
double fres_mct[3]; |
1510 |
– |
double ferror_mct[3]; |
1511 |
– |
|
1512 |
– |
fres_mct[0] = vsim->recVtx->x()-vsim->x; |
1513 |
– |
fres_mct[1] = vsim->recVtx->y()-vsim->y; |
1514 |
– |
fres_mct[2] = vsim->recVtx->z()-vsim->z; |
1515 |
– |
ferror_mct[0] = vsim->recVtx->xError(); |
1516 |
– |
ferror_mct[1] = vsim->recVtx->yError(); |
1517 |
– |
ferror_mct[2] = vsim->recVtx->zError(); |
1518 |
– |
|
1519 |
– |
h_summary->Fill1d("deltax"+suffix, fres_mct[0] ); |
1520 |
– |
h_summary->Fill1d("deltay"+suffix, fres_mct[1] ); |
1521 |
– |
h_summary->Fill1d("deltaz"+suffix, fres_mct[2] ); |
1522 |
– |
h_summary->Fill1d("pullx"+suffix, fres_mct[0]/ferror_mct[0] ); |
1523 |
– |
h_summary->Fill1d("pully"+suffix, fres_mct[1]/ferror_mct[1] ); |
1524 |
– |
h_summary->Fill1d("pullz"+suffix, fres_mct[2]/ferror_mct[2] ); |
1525 |
– |
h_summary->Fill1d("errPVx"+suffix, ferror_mct[0] ); |
1526 |
– |
h_summary->Fill1d("errPVy"+suffix, ferror_mct[1] ); |
1527 |
– |
h_summary->Fill1d("errPVz"+suffix, ferror_mct[2] ); |
1528 |
– |
pvinfo.push_back(PVStudy::PVInfo(res(fres_mct[0], fres_mct[1], fres_mct[2]), |
1529 |
– |
error(ferror_mct[0], ferror_mct[1], ferror_mct[2]), |
1530 |
– |
nrectrk)); |
1531 |
– |
// Fill histo according to its track multiplicity |
1532 |
– |
fillHisto(res(fres_mct[0], fres_mct[1], fres_mct[2]), |
1533 |
– |
error(ferror_mct[0], ferror_mct[1], ferror_mct[2]), |
1534 |
– |
nrectrk, datatype); |
1535 |
– |
|
1536 |
– |
if(saventuple_) { |
1537 |
– |
//Fill the values for variables in ftree_ |
1538 |
– |
if(datatype == 1) { |
1539 |
– |
nTrkPV_spl1_mct_[nrecPV_spl1_mct_] = nrectrk; |
1540 |
– |
deltax_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[0]; |
1541 |
– |
deltay_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[0]; |
1542 |
– |
deltaz_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[0]; |
1543 |
– |
pullx_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[0]/ferror_mct[0]; |
1544 |
– |
pully_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[1]/ferror_mct[1]; |
1545 |
– |
pullz_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[2]/ferror_mct[2]; |
1546 |
– |
nrecPV_spl1_mct_++; |
1547 |
– |
} |
1548 |
– |
|
1549 |
– |
if(datatype == 2) { |
1550 |
– |
nTrkPV_spl2_mct_[nrecPV_spl2_mct_] = nrectrk; |
1551 |
– |
deltax_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[0]; |
1552 |
– |
deltay_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[0]; |
1553 |
– |
deltaz_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[0]; |
1554 |
– |
pullx_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[0]/ferror_mct[0]; |
1555 |
– |
pully_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[1]/ferror_mct[1]; |
1556 |
– |
pullz_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[2]/ferror_mct[2]; |
1557 |
– |
nrecPV_spl2_mct_++; |
1558 |
– |
} |
1559 |
– |
if(datatype == 3) { |
1560 |
– |
nTrkPV_mct_[nrecPV_mct_] = nrectrk; |
1561 |
– |
deltax_mct_[nrecPV_mct_] = fres_mct[0]; |
1562 |
– |
deltay_mct_[nrecPV_mct_] = fres_mct[0]; |
1563 |
– |
deltaz_mct_[nrecPV_mct_] = fres_mct[0]; |
1564 |
– |
pullx_mct_[nrecPV_mct_] = fres_mct[0]/ferror_mct[0]; |
1565 |
– |
pully_mct_[nrecPV_mct_] = fres_mct[1]/ferror_mct[1]; |
1566 |
– |
pullz_mct_[nrecPV_mct_] = fres_mct[2]/ferror_mct[2]; |
1567 |
– |
nrecPV_mct_++; |
1568 |
– |
} |
1569 |
– |
} // End of if(saventuple_) { |
1570 |
– |
} // if ( matchVertex(*vsim,*vrec) ) { |
1571 |
– |
else { // no rec vertex found for this simvertex |
1572 |
– |
edm::LogInfo("Debug") <<"[fillMCHisto] primary not found " << vsim->x << " " << vsim->y << " " << vsim->z << " nGenTrk=" << vsim->nGenTrk << endl; |
1372 |
|
} |
1373 |
|
} |
1374 |
+ |
|
1375 |
+ |
// === If match found fill the residual and pulls |
1376 |
+ |
if(vsim->recVtx) { |
1377 |
+ |
edm::LogInfo("Debug") <<"[fillMCHisto] primary matched in vertexColl" << vsim->x << " " << vsim->y << " " << vsim->z << endl; |
1378 |
+ |
// if study the resolution/pull for all tracks in the PVtx |
1379 |
+ |
int nrectrk; |
1380 |
+ |
|
1381 |
+ |
if(useHWTrk_) |
1382 |
+ |
nrectrk=nHWTrkRecVtx(*(vsim->recVtx)); |
1383 |
+ |
else |
1384 |
+ |
nrectrk = int(vsim->recVtx->tracksSize()); |
1385 |
+ |
|
1386 |
+ |
double fres_mct[3]; |
1387 |
+ |
double ferror_mct[3]; |
1388 |
+ |
|
1389 |
+ |
double ndofPV = vsim->recVtx->ndof(); |
1390 |
+ |
double normchi2PV = vsim->recVtx->normalizedChi2(); |
1391 |
+ |
double avgPtPV = avgPtRecVtx(*(vsim->recVtx)); |
1392 |
+ |
|
1393 |
+ |
fres_mct[0] = vsim->recVtx->x()-vsim->x; |
1394 |
+ |
fres_mct[1] = vsim->recVtx->y()-vsim->y; |
1395 |
+ |
fres_mct[2] = vsim->recVtx->z()-vsim->z; |
1396 |
+ |
ferror_mct[0] = vsim->recVtx->xError(); |
1397 |
+ |
ferror_mct[1] = vsim->recVtx->yError(); |
1398 |
+ |
ferror_mct[2] = vsim->recVtx->zError(); |
1399 |
+ |
|
1400 |
+ |
h_summary->Fill1d("deltax"+suffix, fres_mct[0] ); |
1401 |
+ |
h_summary->Fill1d("deltay"+suffix, fres_mct[1] ); |
1402 |
+ |
h_summary->Fill1d("deltaz"+suffix, fres_mct[2] ); |
1403 |
+ |
h_summary->Fill1d("pullx"+suffix, fres_mct[0]/ferror_mct[0] ); |
1404 |
+ |
h_summary->Fill1d("pully"+suffix, fres_mct[1]/ferror_mct[1] ); |
1405 |
+ |
h_summary->Fill1d("pullz"+suffix, fres_mct[2]/ferror_mct[2] ); |
1406 |
+ |
h_summary->Fill1d("errPVx"+suffix, ferror_mct[0] ); |
1407 |
+ |
h_summary->Fill1d("errPVy"+suffix, ferror_mct[1] ); |
1408 |
+ |
h_summary->Fill1d("errPVz"+suffix, ferror_mct[2] ); |
1409 |
+ |
pvinfo.push_back(PVStudy::PVInfo(res(fres_mct[0], fres_mct[1], fres_mct[2]), |
1410 |
+ |
error(ferror_mct[0], ferror_mct[1], ferror_mct[2]), |
1411 |
+ |
nrectrk)); |
1412 |
+ |
// Fill histo according to its track multiplicity |
1413 |
+ |
fillHisto(res(fres_mct[0], fres_mct[1], fres_mct[2]), |
1414 |
+ |
error(ferror_mct[0], ferror_mct[1], ferror_mct[2]), |
1415 |
+ |
nrectrk, datatype); |
1416 |
+ |
|
1417 |
+ |
if(saventuple_) { |
1418 |
+ |
//Fill the values for variables in pvtxtree_ |
1419 |
+ |
if(datatype == 1) { |
1420 |
+ |
nTrkPV_spl1_mct_[nrecPV_spl1_mct_] = nrectrk; |
1421 |
+ |
ndofPV_spl1_mct_[nrecPV_spl1_mct_] = ndofPV; |
1422 |
+ |
normchi2PV_spl1_mct_[nrecPV_spl1_mct_] = normchi2PV; |
1423 |
+ |
avgPtPV_spl1_mct_[nrecPV_spl1_mct_] = avgPtPV; |
1424 |
+ |
deltax_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[0]; |
1425 |
+ |
deltay_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[1]; |
1426 |
+ |
deltaz_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[2]; |
1427 |
+ |
pullx_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[0]/ferror_mct[0]; |
1428 |
+ |
pully_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[1]/ferror_mct[1]; |
1429 |
+ |
pullz_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[2]/ferror_mct[2]; |
1430 |
+ |
errx_spl1_mct_[nrecPV_spl1_mct_] = ferror_mct[0]; |
1431 |
+ |
erry_spl1_mct_[nrecPV_spl1_mct_] = ferror_mct[1]; |
1432 |
+ |
errz_spl1_mct_[nrecPV_spl1_mct_] = ferror_mct[2]; |
1433 |
+ |
nrecPV_spl1_mct_++; |
1434 |
+ |
} |
1435 |
+ |
if(datatype == 2) { |
1436 |
+ |
nTrkPV_spl2_mct_[nrecPV_spl2_mct_] = nrectrk; |
1437 |
+ |
ndofPV_spl2_mct_[nrecPV_spl2_mct_] = ndofPV; |
1438 |
+ |
normchi2PV_spl2_mct_[nrecPV_spl2_mct_] = normchi2PV; |
1439 |
+ |
avgPtPV_spl2_mct_[nrecPV_spl2_mct_] = avgPtPV; |
1440 |
+ |
deltax_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[0]; |
1441 |
+ |
deltay_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[1]; |
1442 |
+ |
deltaz_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[2]; |
1443 |
+ |
pullx_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[0]/ferror_mct[0]; |
1444 |
+ |
pully_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[1]/ferror_mct[1]; |
1445 |
+ |
pullz_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[2]/ferror_mct[2]; |
1446 |
+ |
errx_spl2_mct_[nrecPV_spl2_mct_] = ferror_mct[0]; |
1447 |
+ |
erry_spl2_mct_[nrecPV_spl2_mct_] = ferror_mct[1]; |
1448 |
+ |
errz_spl2_mct_[nrecPV_spl2_mct_] = ferror_mct[2]; |
1449 |
+ |
nrecPV_spl2_mct_++; |
1450 |
+ |
} |
1451 |
+ |
if(datatype == 3) { |
1452 |
+ |
nTrkPV_mct_[nrecPV_mct_] = nrectrk; |
1453 |
+ |
ndofPV_mct_[nrecPV_mct_] = ndofPV; |
1454 |
+ |
normchi2PV_mct_[nrecPV_mct_] = normchi2PV; |
1455 |
+ |
avgPtPV_mct_[nrecPV_mct_] = avgPtPV; |
1456 |
+ |
deltax_mct_[nrecPV_mct_] = fres_mct[0]; |
1457 |
+ |
deltay_mct_[nrecPV_mct_] = fres_mct[0]; |
1458 |
+ |
deltaz_mct_[nrecPV_mct_] = fres_mct[0]; |
1459 |
+ |
pullx_mct_[nrecPV_mct_] = fres_mct[0]/ferror_mct[0]; |
1460 |
+ |
pully_mct_[nrecPV_mct_] = fres_mct[1]/ferror_mct[1]; |
1461 |
+ |
pullz_mct_[nrecPV_mct_] = fres_mct[2]/ferror_mct[2]; |
1462 |
+ |
errx_mct_[nrecPV_mct_] = ferror_mct[0]; |
1463 |
+ |
erry_mct_[nrecPV_mct_] = ferror_mct[1]; |
1464 |
+ |
errz_mct_[nrecPV_mct_] = ferror_mct[2]; |
1465 |
+ |
nrecPV_mct_++; |
1466 |
+ |
} |
1467 |
+ |
} // End of if(saventuple_) { |
1468 |
+ |
} // if ( matchVertex(*vsim,*vrec) ) { |
1469 |
+ |
else { // no rec vertex found for this simvertex |
1470 |
+ |
edm::LogInfo("Debug") <<"[fillMCHisto] primary not found " << vsim->x << " " << vsim->y << " " << vsim->z << " nGenTrk=" << vsim->nGenTrk << endl; |
1471 |
+ |
} |
1472 |
|
} |
1473 |
|
|
1474 |
+ |
|
1475 |
|
void PVStudy::SetVarToZero() { |
1578 |
– |
//Greg's variables |
1476 |
|
fntrk_ = 0; |
1477 |
|
//pvtx position (x,y,z) residual and error |
1478 |
|
for(int i = 0; i<3;i++) { |
1480 |
|
ferror_[i] = 0; |
1481 |
|
} |
1482 |
|
|
1483 |
< |
//Yanyan's variables |
1484 |
< |
// Number of reconstructed vertices |
1483 |
> |
// Event level information |
1484 |
> |
glob_runno_ = 0; |
1485 |
> |
glob_evtno_ = 0; |
1486 |
> |
glob_ls_ = 0; |
1487 |
> |
glob_bx_ = 0; |
1488 |
|
nrecPV_ = 0; |
1489 |
|
nrecPV1_spl_ = 0; |
1490 |
|
nrecPV2_spl_ = 0; |
1500 |
|
min_ntrksep_ = 9999.0; |
1501 |
|
min_sumpt2sep_ = -9999.0; |
1502 |
|
|
1503 |
< |
|
1503 |
> |
// Variables filled per Vertex |
1504 |
|
for (int i = 0; i < nMaxPVs_; i++) { |
1605 |
– |
// recoVertices with all tracks |
1606 |
– |
nTrkPV_[i] = 0; // Number of tracks in the pvtx |
1607 |
– |
sumptsq_[i] = 0; |
1608 |
– |
isValid_[i] = -1; |
1609 |
– |
isFake_[i] = -1; |
1610 |
– |
recx_[i] = 0; |
1611 |
– |
recy_[i] = 0; |
1612 |
– |
recz_[i] = 0; |
1613 |
– |
recx_err_[i] = 0; |
1614 |
– |
recy_err_[i] = 0; |
1615 |
– |
recz_err_[i] = 0; |
1616 |
– |
|
1617 |
– |
// recoVertices with splitTrack1 |
1618 |
– |
nTrkPV1_spl_[i] = 0; // Number of tracks in the pvtx |
1619 |
– |
sumptsq1_spl_[i] = 0; |
1620 |
– |
isValid1_spl_[i] = -1; |
1621 |
– |
isFake1_spl_[i] = -1; |
1622 |
– |
recx1_spl_[i] = 0; |
1623 |
– |
recy1_spl_[i] = 0; |
1624 |
– |
recz1_spl_[i] = 0; |
1625 |
– |
recx1_err_spl_[i] = 0; |
1626 |
– |
recy1_err_spl_[i] = 0; |
1627 |
– |
recz1_err_spl_[i] = 0; |
1628 |
– |
|
1629 |
– |
// recoVertices with splitTrack2 |
1630 |
– |
nTrkPV2_spl_[i] = 0; // Number of tracks in the pvtx |
1631 |
– |
sumptsq2_spl_[i] = 0; |
1632 |
– |
isValid2_spl_[i] = -1; |
1633 |
– |
isFake2_spl_[i] = -1; |
1634 |
– |
recx2_spl_[i] = 0; |
1635 |
– |
recy2_spl_[i] = 0; |
1636 |
– |
recz2_spl_[i] = 0; |
1637 |
– |
recx2_err_spl_[i] = 0; |
1638 |
– |
recy2_err_spl_[i] = 0; |
1639 |
– |
recz2_err_spl_[i] = 0; |
1640 |
– |
|
1641 |
– |
//pixelVertices |
1642 |
– |
nTrkPV_pxlpvtx_[i] = 0; // Number of tracks in the pvtx |
1643 |
– |
sumptsq_pxlpvtx_[i] = 0; |
1644 |
– |
isValid_pxlpvtx_[i] = -1; |
1645 |
– |
isFake_pxlpvtx_[i] = -1; |
1646 |
– |
recx_pxlpvtx_[i] = 0; |
1647 |
– |
recy_pxlpvtx_[i] = 0; |
1648 |
– |
recz_pxlpvtx_[i] = 0; |
1649 |
– |
recx_err_pxlpvtx_[i] = 0; |
1650 |
– |
recy_err_pxlpvtx_[i] = 0; |
1651 |
– |
recz_err_pxlpvtx_[i] = 0; |
1652 |
– |
|
1505 |
|
// matched two-vertices |
1506 |
|
nTrkPV1_spl_twovtx_[i] = 0; |
1507 |
+ |
ndofPV1_spl_twovtx_[i] = 0; |
1508 |
+ |
normchi2PV1_spl_twovtx_[i] = 0; |
1509 |
+ |
avgPtPV1_spl_twovtx_[i] = 0; |
1510 |
+ |
errx1_spl_twovtx_[i] = 0; |
1511 |
+ |
erry1_spl_twovtx_[i] = 0; |
1512 |
+ |
errz1_spl_twovtx_[i] = 0; |
1513 |
+ |
|
1514 |
|
nTrkPV2_spl_twovtx_[i] = 0; |
1515 |
< |
nTrkPV_twovtx_[i] = 0; |
1516 |
< |
sumptsq1_spl_twovtx_[i] = 0; // Sum of pT squred in the pvtx |
1517 |
< |
sumptsq2_spl_twovtx_[i] = 0; // Sum of pT squred in the pvtx |
1515 |
> |
ndofPV2_spl_twovtx_[i] = 0; |
1516 |
> |
normchi2PV2_spl_twovtx_[i] = 0; |
1517 |
> |
avgPtPV2_spl_twovtx_[i] = 0; |
1518 |
> |
errx2_spl_twovtx_[i] = 0; |
1519 |
> |
erry2_spl_twovtx_[i] = 0; |
1520 |
> |
errz2_spl_twovtx_[i] = 0; |
1521 |
> |
|
1522 |
|
deltax_twovtx_[i] = 0; |
1523 |
|
deltay_twovtx_[i] = 0; |
1524 |
|
deltaz_twovtx_[i] = 0; |
1543 |
|
pullx_mct_[i] = 0; |
1544 |
|
pully_mct_[i] = 0; |
1545 |
|
pullz_mct_[i] = 0; |
1546 |
< |
|
1546 |
> |
nTrkPV_mct_[i] = 0; |
1547 |
> |
ndofPV_mct_[i] = 0; |
1548 |
> |
normchi2PV_mct_[i] = 0; |
1549 |
> |
avgPtPV_mct_[i] = 0; |
1550 |
> |
errx_mct_[i] = 0; |
1551 |
> |
erry_mct_[i] = 0; |
1552 |
> |
errz_mct_[i] = 0; |
1553 |
> |
|
1554 |
|
deltax_spl1_mct_[i] = 0; |
1555 |
|
deltay_spl1_mct_[i] = 0; |
1556 |
|
deltaz_spl1_mct_[i] = 0; |
1557 |
|
pullx_spl1_mct_[i] = 0; |
1558 |
|
pully_spl1_mct_[i] = 0; |
1559 |
|
pullz_spl1_mct_[i] = 0; |
1560 |
< |
|
1560 |
> |
nTrkPV_spl1_mct_[i] = 0; |
1561 |
> |
ndofPV_spl1_mct_[i] = 0; |
1562 |
> |
normchi2PV_spl1_mct_[i] = 0; |
1563 |
> |
avgPtPV_spl1_mct_[i] = 0; |
1564 |
> |
errx_spl1_mct_[i] = 0; |
1565 |
> |
erry_spl1_mct_[i] = 0; |
1566 |
> |
errz_spl1_mct_[i] = 0; |
1567 |
> |
|
1568 |
|
deltax_spl2_mct_[i] = 0; |
1569 |
|
deltay_spl2_mct_[i] = 0; |
1570 |
|
deltaz_spl2_mct_[i] = 0; |
1571 |
|
pullx_spl2_mct_[i] = 0; |
1572 |
|
pully_spl2_mct_[i] = 0; |
1573 |
|
pullz_spl2_mct_[i] = 0; |
1574 |
+ |
nTrkPV_spl2_mct_[i] = 0; |
1575 |
+ |
ndofPV_spl2_mct_[i] = 0; |
1576 |
+ |
normchi2PV_spl2_mct_[i] = 0; |
1577 |
+ |
avgPtPV_spl2_mct_[i] = 0; |
1578 |
+ |
errx_spl2_mct_[i] = 0; |
1579 |
+ |
erry_spl2_mct_[i] = 0; |
1580 |
+ |
errz_spl2_mct_[i] = 0; |
1581 |
+ |
|
1582 |
|
} |
1583 |
|
} |
1584 |
|
|
1585 |
|
double PVStudy::sumPtSquared(const reco::Vertex & v) { |
1586 |
< |
double sum = 0.; |
1587 |
< |
double pT; |
1588 |
< |
for (reco::Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) { |
1589 |
< |
pT = (**it).pt(); |
1590 |
< |
sum += pT*pT; |
1586 |
> |
if(v.isFake() || (!v.isValid()) || v.tracksSize() == 0) return 0.0; |
1587 |
> |
else { |
1588 |
> |
double sum = 0.; |
1589 |
> |
double pT; |
1590 |
> |
for (reco::Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) { |
1591 |
> |
pT = (**it).pt(); |
1592 |
> |
sum += pT*pT; |
1593 |
> |
} |
1594 |
> |
return sum; |
1595 |
|
} |
1707 |
– |
return sum; |
1596 |
|
} |
1597 |
|
|
1598 |
|
|
1599 |
|
|
1600 |
+ |
double PVStudy::avgPtRecVtx(const reco::Vertex & v) { |
1601 |
+ |
|
1602 |
+ |
if(v.isFake() || !v.isValid() || v.tracksSize()==0 ) return 0; |
1603 |
+ |
else { |
1604 |
+ |
int nHWTrk = 0; |
1605 |
+ |
double sumpT = 0.; |
1606 |
+ |
for (reco::Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) { |
1607 |
+ |
if(v.trackWeight(*it) > 0.5 ) { |
1608 |
+ |
sumpT += (**it).pt(); |
1609 |
+ |
nHWTrk++; |
1610 |
+ |
} |
1611 |
+ |
} |
1612 |
+ |
if(nHWTrk > 0) |
1613 |
+ |
return sumpT/double(nHWTrk); |
1614 |
+ |
else |
1615 |
+ |
return 0; |
1616 |
+ |
} |
1617 |
+ |
} |
1618 |
+ |
|
1619 |
+ |
int PVStudy::nHWTrkRecVtx(const reco::Vertex & v) { |
1620 |
+ |
int nHWTrkPV = 0; |
1621 |
+ |
for (reco::Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) { |
1622 |
+ |
if(v.trackWeight(*it) > 0.5) |
1623 |
+ |
nHWTrkPV++; |
1624 |
+ |
} |
1625 |
+ |
return nHWTrkPV; |
1626 |
+ |
} |