ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/Jeng/PVStudy/plugins/PVStudy.cc

Comparing UserCode/Jeng/PVStudy/plugins/PVStudy.cc (file contents):
Revision 1.16 by jengbou, Thu Dec 10 17:57:31 2009 UTC vs.
Revision 1.23 by yygao, Mon May 31 14:19:45 2010 UTC

#	Line 12 | Line 12 | Implementation:
12		*/
13		//
14		// Original Author:  "Geng-yuan Jeng/UC Riverside"
15	<	//                   "Yanyan Gao/Fermilab ygao@fnal.gov"
15	>	//                    "Yanyan Gao/Fermilab ygao@fnal.gov"
16		//         Created:  Thu Aug 20 11:55:40 CDT 2009
17		// $Id$
18		//
#	Line 25 | Line 25 | Implementation:
25		#include <vector>
26		#include <iostream>
27		#include <sstream>
28	–	#include <algorithm>
28
29		// user include files
30		#include "FWCore/Framework/interface/Frameworkfwd.h"
#	Line 39 | Line 38 | Implementation:
38		#include "DataFormats/TrackReco/interface/Track.h"
39		#include "DataFormats/TrackReco/interface/TrackFwd.h"
40		#include "FWCore/ServiceRegistry/interface/Service.h"
41	<	#include "PhysicsTools/UtilAlgos/interface/TFileService.h"
41	>	#include "CommonTools/UtilAlgos/interface/TFileService.h"
42		#include "UserCode/PVStudy/interface/PVStudy.h"
43		//
44		#include "DataFormats/VertexReco/interface/Vertex.h"
#	Line 50 | Line 49 | Implementation:
49		#include <SimDataFormats/Vertex/interface/SimVertexContainer.h>
50		#include <SimDataFormats/Track/interface/SimTrack.h>
51		#include <SimDataFormats/Track/interface/SimTrackContainer.h>
52	+	// BeamSpot
53	+	#include "DataFormats/BeamSpot/interface/BeamSpot.h"
54
55		//root
56		#include <TROOT.h>
#	Line 60 | Line 61 | Implementation:
61		#include <TPad.h>
62
63		using namespace std;
64	+	typedef math::XYZTLorentzVectorF LorentzVector;
65	+	typedef math::XYZPoint Point;
66
67		PVStudy::PVStudy(const edm::ParameterSet& iConfig)
68		{
#	Line 72 | Line 75 | PVStudy::PVStudy(const edm::ParameterSet
75		splitTrackCollection2Tag_  = iConfig.getUntrackedParameter<edm::InputTag>("splitTrackCollection2");
76		vertexCollectionTag_       = iConfig.getUntrackedParameter<edm::InputTag>("vertexCollection");
77		splitVertexCollection1Tag_ = iConfig.getUntrackedParameter<edm::InputTag>("splitVertexCollection1");
78	<	splitVertexCollection2Tag_ = iConfig.getUntrackedParameter<edm::InputTag>("splitVertexCollection2");
76	<	pixelVertexCollectionTag_  = iConfig.getParameter<edm::InputTag>("pixelVertexCollectionTag");
78	>	splitVertexCollection2Tag_ = iConfig.getUntrackedParameter<edm::InputTag>("splitVertexCollection2");
79		verbose_                   = iConfig.getUntrackedParameter<bool>("verbose",false);
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");
83	–	zsigncut_                  = iConfig.getUntrackedParameter<int>("zsigncut");
86		nTrkMin_                   = iConfig.getUntrackedParameter<int>("nTrkMin");
87		nTrkMax_                   = iConfig.getUntrackedParameter<int>("nTrkMax");
88	<	histoFileName_             = iConfig.getUntrackedParameter<std::string> ("histoFileName");
89	<
88	>	zsigncut_                  = iConfig.getUntrackedParameter<int>("zsigncut");
89	>	useHWTrk_                  = iConfig.getUntrackedParameter<bool>("useHWTrk",false);
90	>	avgTrkPtInPVMin_                  = iConfig.getUntrackedParameter<double>("avgTrkPtInPVMin");
91	>	avgTrkPtInPVMax_                  = iConfig.getUntrackedParameter<double>("avgTrkPtInPVMax");
92	>	bsSrc = iConfig.getParameter< edm::InputTag >("beamSpot");
93	>
94		//now do what ever initialization is needed
95		pvinfo.clear();
96
91	–
97		// Specify the data mode vector
98		if(realData_) datamode.push_back(0);
99		else {
#	Line 97 | Line 102 | PVStudy::PVStudy(const edm::ParameterSet
102		datamode.push_back(2);
103		datamode.push_back(3);
104		}
105	<
101	<	// Create ntuple files if needed
105	>	// Create ntuple files if needed
106		if(saventuple_) {
107		file_ = TFile::Open(outputfilename_.c_str(),"RECREATE");
108		ftree_ = new TTree("mytree","mytree");
#	Line 109 | Line 113 | PVStudy::PVStudy(const edm::ParameterSet
113
114		// pvtxtree_ analyzing the pvtxs ootb
115		pvtxtree_ = new TTree("pvtxtree","pvtxtree");
112	–	//pvtx using all tracks
113	–	pvtxtree_->Branch("nrecPV",&nrecPV_,"nrecPV/I");
114	–	pvtxtree_->Branch("nTrkPV",&nTrkPV_,"nTrkPV[nrecPV]/I");
115	–	pvtxtree_->Branch("sumptsq",&sumptsq_,"sumptsq[nrecPV]/D");
116	–	pvtxtree_->Branch("isValid",&isValid_,"isValid/I");
117	–	pvtxtree_->Branch("isFake",&isFake_,"isFake/I");
118	–	pvtxtree_->Branch("recx",&recx_,"recx[nrecPV]/D");
119	–	pvtxtree_->Branch("recy",&recy_,"recy[nrecPV]/D");
120	–	pvtxtree_->Branch("recz",&recz_,"recz[nrecPV]/D");
121	–	pvtxtree_->Branch("recx_err",&recx_err_,"recx_err[nrecPV]/D");
122	–	pvtxtree_->Branch("recy_err",&recy_err_,"recy_err[nrecPV]/D");
123	–	pvtxtree_->Branch("recz_err",&recz_err_,"recz_err[nrecPV]/D");
116
117	+	// Event information for the data
118	+	pvtxtree_->Branch("glob_runno",&glob_runno_,"glob_runno/I");
119	+	pvtxtree_->Branch("glob_evtno",&glob_evtno_,"glob_evtno/I");
120	+	pvtxtree_->Branch("glob_ls",&glob_ls_,"glob_ls/I");
121	+	pvtxtree_->Branch("glob_bx",&glob_bx_,"glob_bx/I");
122	+
123	+
124	+	pvtxtree_->Branch("nrecPV",&nrecPV_,"nrecPV/I");
125	+	pvtxtree_->Branch("nrecPV1_spl",&nrecPV1_spl_,"nrecPV1_spl/I");
126	+	pvtxtree_->Branch("nrecPV2_spl",&nrecPV2_spl_,"nrecPV2_spl/I");
127	+	// Event level information
128		pvtxtree_->Branch("min_zsep",&min_zsep_,"min_zsep/D");
129		pvtxtree_->Branch("min_ntrksep",&min_ntrksep_,"min_ntrksep/D");
130		pvtxtree_->Branch("min_sumpt2sep",&min_sumpt2sep_,"min_sumpt2sep/D");
131
132	<	//pvtx using splittracks1
133	<	pvtxtree_->Branch("nrecPV1_spl",&nrecPV1_spl_,"nrecPV1_spl/I");
131	<	pvtxtree_->Branch("nTrkPV1_spl",&nTrkPV1_spl_,"nTrkPV1_spl[nrecPV1_spl]/I");
132	<	pvtxtree_->Branch("sumptsq1_spl",&sumptsq1_spl_,"sumptsq1_spl[nrecPV1_spl]/D");
133	<	pvtxtree_->Branch("isValid1_spl",&isValid1_spl_,"isValid1_spl/I");
134	<	pvtxtree_->Branch("isFake1_spl",&isFake1_spl_,"isFake1_spl/I");
135	<	pvtxtree_->Branch("recx1_spl",&recx1_spl_,"recx1_spl[nrecPV1_spl]/D");
136	<	pvtxtree_->Branch("recy1_spl",&recy1_spl_,"recy1_spl[nrecPV1_spl]/D");
137	<	pvtxtree_->Branch("recz1_spl",&recz1_spl_,"recz1_spl[nrecPV1_spl]/D");
138	<	pvtxtree_->Branch("recx1_err_spl",&recx1_err_spl_,"recx1_err_spl[nrecPV1_spl]/D");
139	<	pvtxtree_->Branch("recy1_err_spl",&recy1_err_spl_,"recy1_err_spl[nrecPV1_spl]/D");
140	<	pvtxtree_->Branch("recz1_err_spl",&recz1_err_spl_,"recz1_err_spl[nrecPV1_spl]/D");
141	<
142	<	//pvtx using splittracks2
143	<	pvtxtree_->Branch("nrecPV2_spl",&nrecPV2_spl_,"nrecPV2_spl/I");
144	<	pvtxtree_->Branch("nTrkPV2_spl",&nTrkPV2_spl_,"nTrkPV2_spl[nrecPV2_spl]/I");
145	<	pvtxtree_->Branch("sumptsq2_spl",&sumptsq2_spl_,"sumptsq2_spl[nrecPV2_spl]/D");
146	<	pvtxtree_->Branch("isValid2_spl",&isValid2_spl_,"isValid2_spl/I");
147	<	pvtxtree_->Branch("isFake2_spl",&isFake2_spl_,"isFake2_spl/I");
148	<	pvtxtree_->Branch("recx2_spl",&recx2_spl_,"recx2_spl[nrecPV2_spl]/D");
149	<	pvtxtree_->Branch("recy2_spl",&recy2_spl_,"recy2_spl[nrecPV2_spl]/D");
150	<	pvtxtree_->Branch("recz2_spl",&recz2_spl_,"recz2_spl[nrecPV2_spl]/D");
151	<	pvtxtree_->Branch("recx2_err_spl",&recx2_err_spl_,"recx2_err_spl[nrecPV2_spl]/D");
152	<	pvtxtree_->Branch("recy2_err_spl",&recy2_err_spl_,"recy2_err_spl[nrecPV2_spl]/D");
153	<	pvtxtree_->Branch("recz2_err_spl",&recz2_err_spl_,"recz2_err_spl[nrecPV2_spl]/D");
154	<
155	<	//pixeVertices
156	<	pvtxtree_->Branch("nrecPV_pxlpvtx",&nrecPV_pxlpvtx_,"nrecPV_pxlpvtx/I");
157	<	pvtxtree_->Branch("nTrkPV_pxlpvtx",&nTrkPV_pxlpvtx_,"nTrkPV_pxlpvtx[nrecPV_pxlpvtx]/I");
158	<	pvtxtree_->Branch("sumptsq_pxlpvtx",&sumptsq_pxlpvtx_,"sumptsq_pxlpvtx[nrecPV_pxlpvtx]/D");
159	<	pvtxtree_->Branch("isValid_pxlpvtx",&isValid_pxlpvtx_,"isValid_pxlpvtx/I");
160	<	pvtxtree_->Branch("isFake_pxlpvtx",&isFake_pxlpvtx_,"isFake_pxlpvtx/I");
161	<	pvtxtree_->Branch("recx_pxlpvtx",&recx_pxlpvtx_,"recx_pxlpvtx[nrecPV_pxlpvtx]/D");
162	<	pvtxtree_->Branch("recy_pxlpvtx",&recy_pxlpvtx_,"recy_pxlpvtx[nrecPV_pxlpvtx]/D");
163	<	pvtxtree_->Branch("recz_pxlpvtx",&recz_pxlpvtx_,"recz_pxlpvtx[nrecPV_pxlpvtx]/D");
164	<	pvtxtree_->Branch("recx_err_pxlpvtx",&recx_err_pxlpvtx_,"recx_err_pxlpvtx[nrecPV_pxlpvtx]/D");
165	<	pvtxtree_->Branch("recy_err_pxlpvtx",&recy_err_pxlpvtx_,"recy_err_pxlpvtx[nrecPV_pxlpvtx]/D");
166	<	pvtxtree_->Branch("recz_err_pxlpvtx",&recz_err_pxlpvtx_,"recz_err_pxlpvtx[nrecPV_pxlpvtx]/D");
132	>	//Fill the variables in the twovtx pair (recvtx1, recvtx2)
133	>	// Information related to the analyzing the two-vertex method
134
168	–	//Fill the variables in the twovtx pair (recvtx1, recvtx2)
135		pvtxtree_->Branch("nrecPV_twovtx",&nrecPV_twovtx_,"nrecPV_twovtx/I");
170	–	pvtxtree_->Branch("nTrkPV1_spl_twovtx",&nTrkPV1_spl_twovtx_,"nTrkPV1_spl_twovtx[nrecPV_twovtx]/I");
171	–	pvtxtree_->Branch("nTrkPV2_spl_twovtx",&nTrkPV2_spl_twovtx_,"nTrkPV2_spl_twovtx[nrecPV_twovtx]/I");
172	–	pvtxtree_->Branch("sumptsq1_spl_twovtx",&sumptsq1_spl_twovtx_,"sumptsq1_spl_twovtx[nrecPV_twovtx]/D");
173	–	pvtxtree_->Branch("sumptsq2_spl_twovtx",&sumptsq2_spl_twovtx_,"sumptsq2_spl_twovtx[nrecPV_twovtx]/D");
136		pvtxtree_->Branch("nTrkPV_twovtx",&nTrkPV_twovtx_,"nTrkPV_twovtx[nrecPV_twovtx]/I");
137		pvtxtree_->Branch("deltax_twovtx",&deltax_twovtx_,"deltax_twovtx[nrecPV_twovtx]/D");
138		pvtxtree_->Branch("deltay_twovtx",&deltay_twovtx_,"deltay_twovtx[nrecPV_twovtx]/D");
#	Line 181 | Line 143 | PVStudy::PVStudy(const edm::ParameterSet
143		pvtxtree_->Branch("pullx_twovtx",&pullx_twovtx_,"pullx_twovtx[nrecPV_twovtx]/D");
144		pvtxtree_->Branch("pully_twovtx",&pully_twovtx_,"pully_twovtx[nrecPV_twovtx]/D");
145		pvtxtree_->Branch("pullz_twovtx",&pullz_twovtx_,"pullz_twovtx[nrecPV_twovtx]/D");
146	+
147	+	// Information for the splitVertexColl1
148	+	pvtxtree_->Branch("nTrkPV1_spl_twovtx",&nTrkPV1_spl_twovtx_,"nTrkPV1_spl_twovtx[nrecPV_twovtx]/I");
149	+	pvtxtree_->Branch("ndofPV1_spl_twovtx",&ndofPV1_spl_twovtx_,"ndofPV1_spl_twovtx[nrecPV_twovtx]/D");
150	+	pvtxtree_->Branch("normchi2PV1_spl_twovtx",&normchi2PV1_spl_twovtx_,"normchi2PV1_spl_twovtx[nrecPV_twovtx]/D");
151	+	pvtxtree_->Branch("avgPtPV1_spl_twovtx",&avgPtPV1_spl_twovtx_,"avgPtPV1_spl_twovtx[nrecPV_twovtx]/D");
152	+	pvtxtree_->Branch("errx1_spl_twovtx",&errx1_spl_twovtx_,"errx1_spl_twovtx[nrecPV_twovtx]/D");
153	+	pvtxtree_->Branch("erry1_spl_twovtx",&erry1_spl_twovtx_,"erry1_spl_twovtx[nrecPV_twovtx]/D");
154	+	pvtxtree_->Branch("errz1_spl_twovtx",&errz1_spl_twovtx_,"errz1_spl_twovtx[nrecPV_twovtx]/D");
155	+
156	+
157	+	// Information for the splitVertexColl2
158	+	pvtxtree_->Branch("nTrkPV2_spl_twovtx",&nTrkPV2_spl_twovtx_,"nTrkPV2_spl_twovtx[nrecPV_twovtx]/I");
159	+	pvtxtree_->Branch("ndofPV2_spl_twovtx",&ndofPV2_spl_twovtx_,"ndofPV2_spl_twovtx[nrecPV_twovtx]/D");
160	+	pvtxtree_->Branch("normchi2PV2_spl_twovtx",&normchi2PV2_spl_twovtx_,"normchi2PV2_spl_twovtx[nrecPV_twovtx]/D");
161	+	pvtxtree_->Branch("avgPtPV2_spl_twovtx",&avgPtPV2_spl_twovtx_,"avgPtPV2_spl_twovtx[nrecPV_twovtx]/D");
162	+	pvtxtree_->Branch("errx2_spl_twovtx",&errx2_spl_twovtx_,"errx2_spl_twovtx[nrecPV_twovtx]/D");
163	+	pvtxtree_->Branch("erry2_spl_twovtx",&erry2_spl_twovtx_,"erry2_spl_twovtx[nrecPV_twovtx]/D");
164	+	pvtxtree_->Branch("errz2_spl_twovtx",&errz2_spl_twovtx_,"errz2_spl_twovtx[nrecPV_twovtx]/D");
165	+
166
167		// MC variables
168		if(!realData_) {
#	Line 198 | Line 180 | PVStudy::PVStudy(const edm::ParameterSet
180		pvtxtree_->Branch("deltaz_mct",&deltaz_mct_,"deltaz_mct[nrecPV_mct]/D");
181		pvtxtree_->Branch("pullx_mct",&pullx_mct_,"pullx_mct[nrecPV_mct]/D");
182		pvtxtree_->Branch("pully_mct",&pully_mct_,"pully_mct[nrecPV_mct]/D");
183	<	pvtxtree_->Branch("pullz_mct",&pullz_mct_,"pullz_mct[nrecPV_mct]/D");
183	>	pvtxtree_->Branch("pullz_mct",&pullz_mct_,"pullz_mct[nrecPV_mct]/D");
184	>	pvtxtree_->Branch("errx_mct",&errx_mct_,"errx_mct[nrecPV_mct]/D");
185	>	pvtxtree_->Branch("erry_mct",&erry_mct_,"erry_mct[nrecPV_mct]/D");
186	>	pvtxtree_->Branch("errz_mct",&errz_mct_,"errz_mct[nrecPV_mct]/D");
187	>	pvtxtree_->Branch("nTrkPV_mct",&nTrkPV_mct_,"nTrkPV_mct[nrecPV_mct]/I");
188	>	pvtxtree_->Branch("ndofPV_mct",&ndofPV_mct_,"ndofPV_mct[nrecPV_mct]/D");
189	>	pvtxtree_->Branch("normchi2PV_mct",&normchi2PV_mct_,"normchi2PV_mct[nrecPV_mct]/D");
190	>	pvtxtree_->Branch("avgPtPV_mct",&avgPtPV_mct_,"avgPtPV_mct[nrecPV_mct]/D");
191	>
192		// For pvtxs with splittracks1
193		pvtxtree_->Branch("nrecPV_spl1_mct",&nrecPV_spl1_mct_,"nrecPV_spl1_mct/I");
194		pvtxtree_->Branch("deltax_spl1_mct",&deltax_spl1_mct_,"deltax_spl1_mct[nrecPV_spl1_mct]/D");
#	Line 207 | Line 197 | PVStudy::PVStudy(const edm::ParameterSet
197		pvtxtree_->Branch("pullx_spl1_mct",&pullx_spl1_mct_,"pullx_spl1_mct[nrecPV_spl1_mct]/D");
198		pvtxtree_->Branch("pully_spl1_mct",&pully_spl1_mct_,"pully_spl1_mct[nrecPV_spl1_mct]/D");
199		pvtxtree_->Branch("pullz_spl1_mct",&pullz_spl1_mct_,"pullz_spl1_mct[nrecPV_spl1_mct]/D");
200	+	pvtxtree_->Branch("errx_spl1_mct",&errx_spl1_mct_,"errx_spl1_mct[nrecPV_spl1_mct]/D");
201	+	pvtxtree_->Branch("erry_spl1_mct",&erry_spl1_mct_,"erry_spl1_mct[nrecPV_spl1_mct]/D");
202	+	pvtxtree_->Branch("errz_spl1_mct",&errz_spl1_mct_,"errz_spl1_mct[nrecPV_spl1_mct]/D");
203	+	pvtxtree_->Branch("nTrkPV_spl1_mct",&nTrkPV_spl1_mct_,"nTrkPV_spl1_mct[nrecPV_spl1_mct]/I");
204	+	pvtxtree_->Branch("ndofPV_spl1_mct",&ndofPV_spl1_mct_,"ndofPV_spl1_mct[nrecPV_spl1_mct]/D");
205	+	pvtxtree_->Branch("normchi2PV_spl1_mct",&normchi2PV_spl1_mct_,"normchi2PV_spl1_mct[nrecPV_spl1_mct]/D");
206	+	pvtxtree_->Branch("avgPtPV_spl1_mct",&avgPtPV_spl1_mct_,"avgPtPV_spl1_mct[nrecPV_spl1_mct]/D");
207	+
208		// For pvtxs with splittracks1
209		pvtxtree_->Branch("nrecPV_spl2_mct",&nrecPV_spl2_mct_,"nrecPV_spl2_mct/I");
210		pvtxtree_->Branch("deltax_spl2_mct",&deltax_spl2_mct_,"deltax_spl2_mct[nrecPV_spl2_mct]/D");
#	Line 214 | Line 212 | PVStudy::PVStudy(const edm::ParameterSet
212		pvtxtree_->Branch("deltaz_spl2_mct",&deltaz_spl2_mct_,"deltaz_spl2_mct[nrecPV_spl2_mct]/D");
213		pvtxtree_->Branch("pullx_spl2_mct",&pullx_spl2_mct_,"pullx_spl2_mct[nrecPV_spl2_mct]/D");
214		pvtxtree_->Branch("pully_spl2_mct",&pully_spl2_mct_,"pully_spl2_mct[nrecPV_spl2_mct]/D");
215	<	pvtxtree_->Branch("pullz_spl2_mct",&pullz_spl2_mct_,"pullz_spl2_mct[nrecPV_spl2_mct]/D");
215	>	pvtxtree_->Branch("pullz_spl2_mct",&pullz_spl2_mct_,"pullz_spl2_mct[nrecPV_spl2_mct]/D");
216	>	pvtxtree_->Branch("errx_spl2_mct",&errx_spl2_mct_,"errx_spl2_mct[nrecPV_spl2_mct]/D");
217	>	pvtxtree_->Branch("erry_spl2_mct",&erry_spl2_mct_,"erry_spl2_mct[nrecPV_spl2_mct]/D");
218	>	pvtxtree_->Branch("errz_spl2_mct",&errz_spl2_mct_,"errz_spl2_mct[nrecPV_spl2_mct]/D");
219	>	pvtxtree_->Branch("nTrkPV_spl2_mct",&nTrkPV_spl2_mct_,"nTrkPV_spl2_mct[nrecPV_spl2_mct]/I");
220	>	pvtxtree_->Branch("ndofPV_spl2_mct",&ndofPV_spl2_mct_,"ndofPV_spl2_mct[nrecPV_spl2_mct]/D");
221	>	pvtxtree_->Branch("normchi2PV_spl2_mct",&normchi2PV_spl2_mct_,"normchi2PV_spl2_mct[nrecPV_spl2_mct]/D");
222	>	pvtxtree_->Branch("avgPtPV_spl2_mct",&avgPtPV_spl2_mct_,"avgPtPV_spl2_mct[nrecPV_spl2_mct]/D");
223		}
224		}
225	<
225	>
226		setRootStyle();
227
228	+	//========================================
229	+	// Booking histograms
230	+	//========================================
231	+
232		// Create a root file for histograms
233		theFile = new TFile(histoFileName_.c_str(), "RECREATE");
234		// make diretories
#	Line 234 | Line 243 | PVStudy::PVStudy(const edm::ParameterSet
243
244		// Book Histograms:
245		h_pvtrk   = new PVHistograms();
237	–	h_pixvtx  = new PVHistograms();
246		h_misc    = new PVHistograms();
247		h_summary = new PVHistograms();
248		h_others  = new PVHistograms();
#	Line 244 | Line 252 | PVStudy::PVStudy(const edm::ParameterSet
252		else {
253		stringstream ss;
254		ss << i;
255	<	h_pvtrk->Init("pvTrk",ss.str(),"spl");
255	>	h_pvtrk->Init("pvTrk", ss.str(),"spl");
256		}
257		}
250	–	h_pixvtx->Init("pixVtx");
258		h_misc->Init("misc");
259
260		// Book MC only plots
#	Line 284 | Line 291 | PVStudy::PVStudy(const edm::ParameterSet
291		if(analyze_) h_ana->InitA("analysis",suffix,"nTrk",nTrkMin_,nTrkMax_);
292		suffix.clear();
293		} // End of Book histograms sensitive to data/mc
294	+
295
296		}
297
298		PVStudy::~PVStudy()
299		{
300	+
301	+	// do anything here that needs to be done at desctruction time
302	+	// (e.g. close files, deallocate resources etc.)
303		theFile->cd();
304		theFile->cd("Summary");
305		h_pvtrk->Save();
295	–	h_pixvtx->Save();
306		h_misc->Save();
307		h_summary->Save();
308		if (!realData_)
#	Line 330 | Line 340 | void PVStudy::setRootStyle() {
340		gStyle->SetTitleSize(0.055, "");    // size for pad title; default is 0.02
341		gStyle->SetLabelSize(0.03, "XYZ");  // size for axis labels; default is 0.04
342		gStyle->SetStatFontSize(0.08);      // size for stat. box
343	<	gStyle->SetTitleFont(32, "XYZ");    // times-bold-italic font (p. 153) for axes
344	<	gStyle->SetTitleFont(32, "");       // same for pad title
345	<	gStyle->SetLabelFont(32, "XYZ");    // same for axis labels
346	<	gStyle->SetStatFont(32);            // same for stat. box
343	>	gStyle->SetTitleFont(42, "XYZ");    // times-bold-italic font (p. 153) for axes
344	>	gStyle->SetTitleFont(42, "");       // same for pad title
345	>	gStyle->SetLabelFont(42, "XYZ");    // same for axis labels
346	>	gStyle->SetStatFont(42);            // same for stat. box
347		gStyle->SetLabelOffset(0.006, "Y"); // default is 0.005
348		//
349		return;
#	Line 343 | Line 353 | void PVStudy::setRootStyle() {
353		//
354		std::vector<PVStudy::simPrimaryVertex> PVStudy::getSimPVs(const Handle<HepMCProduct> evtMC, std::string suffix="")
355		{
356	<	std::vector<PVStudy::simPrimaryVertex> simpv;
356	>	std::vector<PVStudy::simPrimaryVertex> simpv;
357		const HepMC::GenEvent* evt=evtMC->GetEvent();
358		if (evt) {
359		edm::LogInfo("SimPVs") << "[getSimPVs] process id " << evt->signal_process_id()<<endl;
#	Line 520 | Line 530 | PVStudy::analyze(const edm::Event& iEven
530		using namespace edm;
531		using namespace reco;
532
533	+	//========================================================================
534	+	// Step 0: Prepare root variables and get information from the Event
535	+	//========================================================================
536	+
537		edm::LogInfo("Debug")<<"[PVStudy]"<<endl;
524	–	//=======================================================
538		// Initialize Root-tuple variables if needed
526	–	//=======================================================
527	–	//if(saventuple_)
539		SetVarToZero();
540
541	<	//=======================================================
531	<	// Track accessors
532	<	//=======================================================
533	<	//trackCollection
541	>	// ====== TrackCollection
542		static const reco::TrackCollection s_empty_trackColl;
543		const reco::TrackCollection *trackColl = &s_empty_trackColl;
544		edm::Handle<reco::TrackCollection> trackCollectionHandle;
#	Line 540 | Line 548 | PVStudy::analyze(const edm::Event& iEven
548		} else {
549		edm::LogInfo("Debug") << "[PVStudy] trackCollection cannot be found -> using empty collection of same type." <<endl;
550		}
551	<
544	<	//splitTrackCollection1
551	>	// ====== splitTrackCollection1
552		static const reco::TrackCollection s_empty_splitTrackColl1;
553		const reco::TrackCollection *splitTrackColl1 = &s_empty_splitTrackColl1;
554		edm::Handle<reco::TrackCollection> splitTrackCollection1Handle;
#	Line 551 | Line 558 | PVStudy::analyze(const edm::Event& iEven
558		} else {
559		edm::LogInfo("Debug") << "[PVStudy] splitTrackCollection1 cannot be found -> using empty collection of same type." <<endl;
560		}
561	<
555	<	//splitTrackCollection2
561	>	// ====== splitTrackCollection2
562		static const reco::TrackCollection s_empty_splitTrackColl2;
563		const reco::TrackCollection *splitTrackColl2 = &s_empty_splitTrackColl2;
564		edm::Handle<reco::TrackCollection> splitTrackCollection2Handle;
#	Line 560 | Line 566 | PVStudy::analyze(const edm::Event& iEven
566		if( iEvent.getByLabel(splitTrackCollection2Tag_, splitTrackCollection2Handle)) {
567		splitTrackColl2 = splitTrackCollection2Handle.product();
568		} else {
569	<	edm::LogInfo("Debug") << "[PVStudy] splitTrackCollection2 cannot be found -> using empty collection of same type." <<endl;
569	>	edm::LogInfo("Debug") << "[PVStudy] splitTrackCollection2 cannot be found -> using empty collection of same type." <<endl;
570		}
571	<
572	<
567	<	//=======================================================
568	<	// Fill trackparameters of the input tracks to pvtx fitter
569	<	//=======================================================
570	<	edm::LogInfo("Debug")<<"[PVStudy] Start filling track parameters of the input tracks to pvtx fitter."<<endl;
571	<	//fillTrackHisto(const reco::TrackCollection *trackColl, int datatype)
572	<	// datatype: unsplittracks (0); splittracks1 (1);  splittracks2 (2);
573	<	fillTrackHisto(trackColl, 0);
574	<	fillTrackHisto(splitTrackColl1, 1);
575	<	fillTrackHisto(splitTrackColl2, 2);
576	<	edm::LogInfo("Debug")<<"[PVStudy] End filling track parameters of the input tracks to pvtx fitter."<<endl;
577	<
578	<	//=======================================================
579	<	// PVTX accessors
580	<	//=======================================================
581	<	//vertexCollection
571	>
572	>	// ======= PrimaryVertexCollection
573		static const reco::VertexCollection s_empty_vertexColl;
574		const reco::VertexCollection *vertexColl = &s_empty_vertexColl;
575		edm::Handle<reco::VertexCollection> vertexCollectionHandle;
#	Line 586 | Line 577 | PVStudy::analyze(const edm::Event& iEven
577		if( iEvent.getByLabel(vertexCollectionTag_, vertexCollectionHandle)) {
578		vertexColl = vertexCollectionHandle.product();
579		} else {
580	<	edm::LogInfo("Debug") << "[PVStudy] vertexCollection cannot be found -> using empty collection of same type." <<endl;
580	>	edm::LogInfo("Debug") << "[PVStudy] vertexCollection cannot be found -> using empty collection of same type." <<endl;
581		}
582	<
592	<	//splitVertexCollection1
582	>	// ====== splitVertexCollection1
583		static const reco::VertexCollection s_empty_splitVertexColl1;
584		const reco::VertexCollection *splitVertexColl1 = &s_empty_splitVertexColl1;
585		edm::Handle<reco::VertexCollection> splitVertexCollection1Handle;
#	Line 598 | Line 588 | PVStudy::analyze(const edm::Event& iEven
588		splitVertexColl1 = splitVertexCollection1Handle.product();
589		} else {
590		edm::LogInfo("Debug") << "[PVStudy] splitVertexCollection1 cannot be found -> using empty collection of same type." <<endl;
591	<	}
592	<
603	<	//splitVertexCollection2
591	>	}
592	>	// ====== splitVertexCollection2
593		static const reco::VertexCollection s_empty_splitVertexColl2;
594		const reco::VertexCollection *splitVertexColl2 = &s_empty_splitVertexColl2;
595		edm::Handle<reco::VertexCollection> splitVertexCollection2Handle;
#	Line 610 | Line 599 | PVStudy::analyze(const edm::Event& iEven
599		} else {
600		edm::LogInfo("Debug") << "[PVStudy] splitVertexCollection2 cannot be found -> using empty collection of same type." <<endl;
601		}
602	+
603	+
604	+	// ======== BeamSpot accessors
605	+	edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
606	+	iEvent.getByLabel(bsSrc,recoBeamSpotHandle);
607	+	reco::BeamSpot bs = *recoBeamSpotHandle;
608	+	const Point beamSpot = recoBeamSpotHandle.isValid() ? Point(recoBeamSpotHandle->x0(), recoBeamSpotHandle->y0(), recoBeamSpotHandle->z0()) : Point(0, 0, 0);
609
610	<	edm::LogInfo("Debug")<<"[PVStudy] End accessing the track and vertex collections"<<endl;
610	>	edm::LogInfo("Debug")<<"[PVStudy] End accessing the track, beamSpot, primary vertex collections"<<endl;
611
612	<	//=======================================================
617	<	// MC simvtx accessor
618	<	//=======================================================
612	>	// ========== MC simvtx accessor
613		if (!realData_) {
614		edm::Handle<SimVertexContainer> simVtxs;
615		iEvent.getByLabel( simG4_, simVtxs);
#	Line 624 | Line 618 | PVStudy::analyze(const edm::Event& iEven
618		iEvent.getByLabel( simG4_, simTrks);
619		}
620
621	<	//=======================================================
628	<	// GET PDT
629	<	//=======================================================
621	>	// ========== GET PDT
622		try{
623		iSetup.getData(pdt);
624		}catch(const Exception&){
625		edm::LogInfo("Debug") << "[PVStudy] Some problem occurred with the particle data table. This may not work !." <<endl;
626		}
627
628	<	//=======================================================
629	<	// GET pixelVertices
630	<	//=======================================================
631	<	static const reco::VertexCollection s_empty_pixelVertexColl;
632	<	const reco::VertexCollection *pixelVertexColl = &s_empty_pixelVertexColl;
633	<	edm::Handle<reco::VertexCollection>  pixelVertexCollectionHandle;
634	<	iEvent.getByLabel(pixelVertexCollectionTag_, pixelVertexCollectionHandle);
635	<	if( iEvent.getByLabel(pixelVertexCollectionTag_, pixelVertexCollectionHandle)) {
636	<	pixelVertexColl = pixelVertexCollectionHandle.product();
637	<	} else {
638	<	edm::LogInfo("Debug") << "[PVStudy] pixelVertexCollection cannot be found. -> using empty collection of same type." <<endl;
647	<	}
648	<
649	<	//=======================================================
650	<	// Fill pixelVertices related histograms
651	<	//=======================================================
652	<	nrecPV_pxlpvtx_ = int (pixelVertexColl->size());
653	<	if(pixelVertexColl->size()>0 && pixelVertexColl->begin()->isValid() && !(pixelVertexColl->begin()->isFake())) {
654	<	//fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, bool fillHisto, bool fillNtuple) {
655	<	fillTrackHistoInPV(pixelVertexColl, 4, false, true);
656	<	h_pixvtx->Fill1d("dzErr_pxlpvtx", pixelVertexColl->begin()->zError());
657	<	// Get the dZ error of the tracks in the leading pixelVertexColl
658	<	for(reco::Vertex::trackRef_iterator t = (pixelVertexColl->begin())->tracks_begin();
659	<	t!= (pixelVertexColl->begin())->tracks_end(); t++) {
660	<
661	<	if ( (**t).charge() < -1 || (**t).charge() > 1 ) {
662	<	//        h_pvtrk->Fill1d("trkPtPV", 0.);
663	<	}
664	<	else
665	<	h_pixvtx->Fill1d("trkdzErr_pxlpvtx", (**t).dzError());
666	<	}
667	<	// Fill the track->dz() in the PV difference from first pixelpvtx
668	<	for(reco::Vertex::trackRef_iterator t = (vertexColl->begin())->tracks_begin();
669	<	t!= (vertexColl->begin())->tracks_end(); t++) {
670	<	// illegal charge
671	<	if ( (**t).charge() < -1 || (**t).charge() > 1 ) {
672	<	//        h_pvtrk->Fill1d("trkPtPV", 0.);
673	<	}
674	<	else {
675	<	if(pixelVertexColl->size()>0) {
676	<	h_pixvtx->Fill1d("trkdz_pxlpvtxdz", (**t).dz() -  pixelVertexColl->begin()->z());
677	<	h_pixvtx->Fill1d("trkdz_pxlpvtxdz_pxlpvtxdzerr", fabs((**t).dz() -  pixelVertexColl->begin()->z())/pixelVertexColl->begin()->zError());
678	<	h_pixvtx->Fill1d("trkdz_pxlpvtxdz_trkdzerr", fabs((**t).dz() -  pixelVertexColl->begin()->z())/(**t).dzError());
679	<	h_pixvtx->Fill1d("trkdzErr_pvtx", (**t).dzError());
680	<	}
681	<	}
628	>	// ======= Get MC information if needed
629	>	bool MC=false;
630	>	Handle<HepMCProduct> evtMC;
631	>	if (!realData_) {
632	>	iEvent.getByLabel("generator",evtMC);
633	>	if (!evtMC.isValid()) {
634	>	MC=false;
635	>	edm::LogInfo("Debug") << "[PVStudy] no HepMCProduct found"<< endl;
636	>	} else {
637	>	edm::LogInfo("Debug") << "[PVStudy] generator HepMCProduct found"<< endl;
638	>	MC=true;
639		}
640		}
684	–
685	–	//=======================================================
686	–	// Fill number of reconstructed vertices
687	–	//=======================================================
641
642	+	//========================================================================
643	+	// Step 1:  Apply event cleaning for data and MC
644	+	//          WARNING: event selection cut are hard coded!!
645	+	//========================================================================
646	+
647	+	glob_runno_ = iEvent.id().run();
648	+	glob_evtno_ = iEvent.id().event();
649	+	glob_ls_   = iEvent.luminosityBlock();
650	+	glob_bx_  = iEvent.bunchCrossing();
651	+
652	+	//========================================================================
653	+	// Step 2: Fill histograms for the splitting consistency checks
654	+	//========================================================================
655	+
656	+	// === Fill trackparameters of the input tracks to pvtx fitter
657	+	edm::LogInfo("Debug")<<"[PVStudy] Start filling track parameters of the input tracks to pvtx fitter."<<endl;
658	+	//fillTrackHisto(const reco::TrackCollection *trackColl, int datatype, const Point & bs)
659	+	// datatype: unsplittracks (0); splittracks1 (1);  splittracks2 (2);
660	+	fillTrackHisto(trackColl, 0, beamSpot);
661	+	fillTrackHisto(splitTrackColl1, 1, beamSpot);
662	+	fillTrackHisto(splitTrackColl2, 2, beamSpot);
663	+	edm::LogInfo("Debug")<<"[PVStudy] End filling track parameters of the input tracks to pvtx fitter."<<endl;
664	+
665	+
666	+	// ==== Fill number of reconstructed vertices
667		edm::LogInfo("Debug")<<"[PVStudy] Printing vertexCollection: "<<endl;
668		edm::LogInfo("Debug")<<"[PVStudy] Printing splitVertexCollection1: "<<endl;
669		edm::LogInfo("Debug")<<"[PVStudy] Printing splitVertexCollection2: "<<endl;
#	Line 695 | Line 673 | PVStudy::analyze(const edm::Event& iEven
673		printRecVtxs(splitVertexCollection1Handle);
674		printRecVtxs(splitVertexCollection2Handle);
675		}
676	+
677		nrecPV_ = int (vertexColl->size());
678		nrecPV1_spl_ = int (splitVertexColl1->size());
679		nrecPV2_spl_ = int (splitVertexColl2->size());
#	Line 704 | Line 683 | PVStudy::analyze(const edm::Event& iEven
683		h_pvtrk->Fill1d("nrecPV2_spl", nrecPV2_spl_);
684		h_misc->Fill1d("nrecPVDiff", double(nrecPV1_spl_)-double(nrecPV2_spl_));
685
707	–	//=================================================================
708	–	// Fill track parameter ntuple/hist for tracks used in recoVertices
709	–	//=================================================================
710	–
711	–	//fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, bool fillHisto, bool fillNtuple) {
712	–	if(vertexColl->size() > 0 && vertexColl->begin()->isValid() && !(vertexColl->begin()->isFake()))
713	–	fillTrackHistoInPV(vertexColl, 0, true, true);
714	–
715	–	if(splitVertexColl1->size() > 0 && splitVertexColl1->begin()->isValid() && !(splitVertexColl1->begin()->isFake()))
716	–	fillTrackHistoInPV(splitVertexColl1, 1, false, true);
717	–
718	–	if(splitVertexColl1->size() > 0 && splitVertexColl2->begin()->isValid() && !(splitVertexColl2->begin()->isFake()))
719	–	fillTrackHistoInPV(splitVertexColl2, 2, false, true);
720	–
721	–
722	–	//=======================================================
723	–	// Compare offlinePrimaryVertices with pixelVertices
724	–	//=======================================================
725	–	if( (pixelVertexColl->size()>0 && pixelVertexColl->begin()->isValid() && !(pixelVertexColl->begin()->isFake()))
726	–	&& (vertexColl->size()>0 && vertexColl->begin()->isValid() && !(vertexColl->begin()->isFake())) ) {
727	–	h_pixvtx->Fill1d("nrecPV_minus_nrecPxlPV", double (nrecPV_ - nrecPV_pxlpvtx_));
728	–	// difference in reconstructed position of the leading pvtx
729	–	//edm::LogInfo("Debug")<<"recx_[0] = "<< recx_[0] << "recx_pxlpvtx_[0] = "<< recx_pxlpvtx_[0]<<endl;
730	–	//edm::LogInfo("Debug")<<"recy_[0] = "<< recy_[0] << "recy_pxlpvtx_[0] = "<< recy_pxlpvtx_[0]<<endl;
731	–	h_pixvtx->Fill1d("recxPV_minus_recxPxlPV", recx_[0] - recx_pxlpvtx_[0]);
732	–	h_pixvtx->Fill1d("recyPV_minus_recyPxlPV", recy_[0] - recy_pxlpvtx_[0]);
733	–	h_pixvtx->Fill1d("reczPV_minus_reczPxlPV", recz_[0] - recz_pxlpvtx_[0]);
734	–	}
686
687	<
688	<
689	<	//==========================================================
690	<	// Fill secondary/primary min separations for multi-vertices
740	<	//==========================================================
687	>	// ======= Fill track parameter ntuple/hist for tracks used in recoVertices
688	>	//fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, const Point & bs) {
689	>	if(vertexColl->size() > 0 && vertexColl->begin()->isValid() && !(vertexColl->begin()->isFake()))
690	>	fillTrackHistoInPV(vertexColl, 0, beamSpot);
691
692	+	// ======= Fill secondary/primary min separations for multi-vertices
693		if(vertexColl->size()>1 && vertexColl->begin()->isValid() && !(vertexColl->begin()->isFake()) ) {
743	–
694		double min_xsep = 9999.0;
695		double min_ysep = 9999.0;
696		double min_zsep = 9999.0;
#	Line 756 | Line 706 | PVStudy::analyze(const edm::Event& iEven
706		for(reco::VertexCollection::const_iterator v=vertexColl->begin() + 1 ;
707		v!=vertexColl->end(); ++v) {
708		if(v->isValid() && ! v->isFake()) {
759	–	// xsep
709		if(fabs(v->x()- vertexColl->begin()->x())<min_xsep)
710		min_xsep = fabs(v->x()- vertexColl->begin()->x());
762	–	// ysep
711		if(fabs(v->y()- vertexColl->begin()->y())<min_ysep)
712		min_ysep = fabs(v->y()- vertexColl->begin()->y());
765	–	// zsep
713		if(fabs(v->z()- vertexColl->begin()->z())<min_zsep)
714		min_zsep = fabs(v->z()- vertexColl->begin()->z());
768	–	// xsep_sign
715		double xsep_sign = fabs(v->x()- vertexColl->begin()->x())/max(v->xError(), vertexColl->begin()->xError());
716		if(xsep_sign < min_xsep_sign)
717		min_xsep_sign = xsep_sign;
772	–	// ysep_sign
718		double ysep_sign = fabs(v->y()- vertexColl->begin()->y())/max(v->yError(), vertexColl->begin()->yError());
719		if(ysep_sign < min_ysep_sign)
720		min_ysep_sign = ysep_sign;
776	–	// zsep_sign
721		double zsep_sign = fabs(v->z()- vertexColl->begin()->z())/max(v->zError(), vertexColl->begin()->zError());
722		if(zsep_sign < min_zsep_sign)
723		min_zsep_sign = zsep_sign;
780	–	// ntrksep
724		if( (double(vertexColl->begin()->tracksSize()) - double(v->tracksSize())) < min_ntrksep)
725		min_ntrksep = double(vertexColl->begin()->tracksSize()) - double(v->tracksSize());
783	–	// sumpt2sep
726		if(fabs(sumPtSquared(*v) - sumPtSquared(*(vertexColl->begin()))) < min_sumpt2sep)
727		min_sumpt2sep = fabs(sumPtSquared(*v) - sumPtSquared(*(vertexColl->begin())));
728		}
#	Line 797 | Line 739 | PVStudy::analyze(const edm::Event& iEven
739		min_zsep_ = min_zsep;
740		min_ntrksep_ = min_ntrksep;
741		min_sumpt2sep_ = min_sumpt2sep;
800	–
801	–
742		} // End of  if(vertexColl->size()>1) {
743
744		edm::LogInfo("Debug")<<"[PVStudy] End filling pvtx parameters."<<endl;
805	–
806	–	//=======================================================
807	–	//           PrimaryVertex Matching
808	–	// 1. In z  |z1-z2|< zsigncut * max(sigmaz1, sigmaz2)
809	–	// 2. |(nTrkPV1 - nTrkPV2)/(nTrkPV1+nTrkPV2)|<ntrkdiffcut_
810	–	// Assume the first match is the primary vertex,
811	–	// since vertexColl are sorted in decreasing order of sum(pT)
812	–	//=======================================================
745
746	+	//========================================================================
747	+	// Step 3:  PrimaryVertex Matching
748	+	//   1. In z  |z1-z2|< zsigncut * max(sigmaz1, sigmaz2)
749	+	//   2. |(nTrkPV1 - nTrkPV2)/(nTrkPV1+nTrkPV2)|<ntrkdiffcut_
750	+	//   3. The first match is the primary vertex, which has largest sum(pT^2)
751	+	//========================================================================
752	+
753		edm::LogInfo("Debug")<<"[PVStudy] matching pvtxs "<<endl;
754		reco::VertexCollection s_empty_matchedVertexColl1;
755		reco::VertexCollection *matchedVertexColl1 = &s_empty_matchedVertexColl1;
#	Line 833 | Line 772 | PVStudy::analyze(const edm::Event& iEven
772		edm::LogInfo("Debug")<<"[PVStudy] Matching splitVertexColl1: #  "<< ivtx<< " and splitVertexColl1: # "<<jvtx<<endl;
773		edm::LogInfo("Debug")<<"[PVStudy] recvtx1->z() = " << recvtx1->z() << "+/- "<< recvtx1->zError() << "." << endl;
774		edm::LogInfo("Debug")<<"[PVStudy] recvtx2->z() = " << recvtx2->z() << "+/- "<< recvtx2->zError() << "." << endl;
775	<
775	>
776	>	double twovtxsig = (recvtx1->z()-recvtx2->z())/max(recvtx1->zError(), recvtx2->zError());
777	>	h_misc->Fill1d("twovtxzsign", twovtxsig);
778		if(matchVertex(recvtx1, recvtx2, zsigncut_)) {
779		edm::LogInfo("Debug")<<"[PVStudy] The two splitted vertices match in Z. "<<endl;
780
781	<	int nTrkPV1 = recvtx1->tracksSize();
782	<	int nTrkPV2 = recvtx2->tracksSize();
781	>	int nTrkPV1, nTrkPV2;
782	>	if(useHWTrk_) {
783	>	nTrkPV1 = nHWTrkRecVtx(*recvtx1);
784	>	nTrkPV2 = nHWTrkRecVtx(*recvtx2);
785	>	}
786	>	else {
787	>	nTrkPV1 = recvtx1->tracksSize();
788	>	nTrkPV2 = recvtx2->tracksSize();
789	>	}
790		double ntrkreldiff = double(nTrkPV1-nTrkPV2)/double(nTrkPV1+nTrkPV2);
791		h_misc->Fill1d("nTrkPVDiff", nTrkPV1-nTrkPV2);
792		h_misc->Fill1d("nTrkPVRelDiff", ntrkreldiff);
#	Line 861 | Line 809 | PVStudy::analyze(const edm::Event& iEven
809		}
810		}
811		edm::LogInfo("Debug")<<"[PVStudy] End matching pvtxs"<<endl;
812	<
813	<	//=======================================================
814	<	//  Analyze matchedVertexColls
815	<	//=======================================================
812	>
813	>	//========================================================================
814	>	// Step 4: Analyze matchedVertexColls
815	>	//========================================================================
816		edm::LogInfo("Debug")<<"[PVStudy] Begin analyzing the matchedVertexColl with size = "<< matchedVertexColl1->size()<< endl;
817
818	+	// ==== If it is MC, analyze the res/pull of the unsplit vertexColl first
819	+	if(MC){
820	+	// make a list of primary vertices:
821	+	std::vector<simPrimaryVertex> simpv;
822	+	simpv=getSimPVs(evtMC,"");
823	+	//     simpv=getSimPVs(evtMC, simVtxs, simTrks);
824	+	h_gen->Fill1d("nsimPV", simpv.size());
825	+	nsimPV_ = simpv.size();
826	+	int isimpv = 0;
827	+	for(std::vector<simPrimaryVertex>::iterator vsim=simpv.begin();
828	+	vsim!=simpv.end(); vsim++, isimpv++){
829	+	nsimTrkPV_[isimpv]  =vsim->nGenTrk;
830	+	simx_[isimpv] = vsim->x;
831	+	simy_[isimpv] = vsim->y;
832	+	simz_[isimpv] = vsim->y;
833	+	simptsq_[isimpv] = vsim->ptsq;
834	+	fillMCHisto(vsim, isimpv, vertexColl, 3, avgTrkPtInPVMin_, avgTrkPtInPVMax_);
835	+	}
836	+	}
837	+
838		// Compare the reconstructed vertex position and calculate resolution/pulls
839		if(matchedVertexColl1->size() != 0 && matchedVertexColl2->size() != 0) {
840	<	//fillPvtxHisto(const reco::VertexCollection *vertexColl, int datatype, bool fillHisto, bool fillNtuple)
841	<	fillTrackHistoInPV(matchedVertexColl1, 1, true, false);
842	<	fillTrackHistoInPV(matchedVertexColl2, 2, true, false);
840	>	// ==== Analyze the splitted vtx using MC method
841	>	if(MC){
842	>	// make a list of primary vertices:
843	>	std::vector<simPrimaryVertex> simpv;
844	>	simpv=getSimPVs(evtMC,"");
845	>	//     simpv=getSimPVs(evtMC, simVtxs, simTrks);
846	>	int isimpv = 0;
847	>	for(std::vector<simPrimaryVertex>::iterator vsim=simpv.begin();
848	>	vsim!=simpv.end(); vsim++, isimpv++){
849	>	fillMCHisto(vsim, isimpv, matchedVertexColl1, 1, avgTrkPtInPVMin_, avgTrkPtInPVMax_);
850	>	fillMCHisto(vsim, isimpv, matchedVertexColl2, 2, avgTrkPtInPVMin_, avgTrkPtInPVMax_);
851	>	}
852	>	}
853	>
854	>	// ==== Analyze res/pull two-vertex method
855	>	//fillPvtxHisto(const reco::VertexCollection *vertexColl, int datatype)
856	>	fillTrackHistoInPV(matchedVertexColl1, 1, beamSpot);
857	>	fillTrackHistoInPV(matchedVertexColl2, 2, beamSpot);
858
859		reco::VertexCollection::const_iterator v1;
860		reco::VertexCollection::const_iterator v2;
861	+
862		nrecPV_twovtx_ = 0;
863		for(v1 = matchedVertexColl1->begin(), v2 = matchedVertexColl2->begin();
864		v1!=matchedVertexColl1->end(), v2 != matchedVertexColl2->end();
865		++v1, ++v2) {
866	+
867	+	// ==================================================================
868	+	// With the option to fill the histograms at a given average pT range
869	+	// ==================================================================
870	+	if (avgPtRecVtx(*v1) < avgTrkPtInPVMin_ || avgPtRecVtx(*v1) > avgTrkPtInPVMax_ ) continue;
871	+	if (avgPtRecVtx(*v2) < avgTrkPtInPVMin_ || avgPtRecVtx(*v2) > avgTrkPtInPVMax_ ) continue;
872	+
873	+	h_misc->Fill1d("ndofPVDiff", v1->ndof() - v2->ndof());
874	+	h_misc->Fill1d("ndofPVRelDiff", (v1->ndof()-v2->ndof())/(v1->ndof()+v2->ndof()));
875		fres_[0] = (v1->x()-v2->x())/sqrt(2.0);
876		fres_[1] = (v1->y()-v2->y())/sqrt(2.0);
877		fres_[2] = (v1->z()-v2->z())/sqrt(2.0);
878		ferror_[0] = sqrt(pow(v1->xError(),2)+pow(v2->xError(),2))/sqrt(2);
879		ferror_[1] = sqrt(pow(v1->yError(),2)+pow(v2->yError(),2))/sqrt(2);
880		ferror_[2] = sqrt(pow(v1->zError(),2)+pow(v2->zError(),2))/sqrt(2);
881	<	fntrk_ =  (v1->tracksSize()+v2->tracksSize())/2;
881	>
882	>	int nTrkPV1, nTrkPV2;
883	>	if(useHWTrk_) {
884	>	nTrkPV1 = nHWTrkRecVtx(*v1);
885	>	nTrkPV2 = nHWTrkRecVtx(*v2);
886	>	}
887	>	else {
888	>	nTrkPV1 = v1->tracksSize();
889	>	nTrkPV2 = v2->tracksSize();
890	>	}
891	>
892	>	fntrk_ = (nTrkPV1 + nTrkPV2)/2;
893
894		h_summary->Fill1d("deltax", fres_[0] );
895		h_summary->Fill1d("deltay", fres_[1] );
#	Line 900 | Line 904 | PVStudy::analyze(const edm::Event& iEven
904		error(ferror_[0], ferror_[1], ferror_[2]),
905		fntrk_));
906		// Fill histo according to its track multiplicity, set datamode = 0 for pvtx using all tracks
907	+
908		fillHisto(res(fres_[0], fres_[1], fres_[2]),
909		error(ferror_[0], ferror_[1], ferror_[2]),
910		fntrk_,0);
911	+
912		// Fill the ntuple variables
907	–	nTrkPV1_spl_twovtx_[nrecPV_twovtx_] = v1->tracksSize();
908	–	nTrkPV2_spl_twovtx_[nrecPV_twovtx_] = v2->tracksSize();
909	–	sumptsq1_spl_twovtx_[nrecPV_twovtx_] = sumPtSquared(*v1);
910	–	sumptsq2_spl_twovtx_[nrecPV_twovtx_] = sumPtSquared(*v2);
913		nTrkPV_twovtx_[nrecPV_twovtx_] = fntrk_;
914		deltax_twovtx_[nrecPV_twovtx_] = fres_[0];
915		deltay_twovtx_[nrecPV_twovtx_] = fres_[1];
#	Line 918 | Line 920 | PVStudy::analyze(const edm::Event& iEven
920		pullx_twovtx_[nrecPV_twovtx_] = fres_[0]/ferror_[0];
921		pully_twovtx_[nrecPV_twovtx_] = fres_[1]/ferror_[1];
922		pullz_twovtx_[nrecPV_twovtx_] = fres_[2]/ferror_[2];
923	+
924	+
925	+	//SplittedVertex
926	+	nTrkPV1_spl_twovtx_[nrecPV_twovtx_] = nTrkPV1;
927	+	ndofPV1_spl_twovtx_[nrecPV_twovtx_] = v1->ndof();
928	+	normchi2PV1_spl_twovtx_[nrecPV_twovtx_] = v1->normalizedChi2();
929	+	avgPtPV1_spl_twovtx_[nrecPV_twovtx_] = avgPtRecVtx(*v1);
930	+	errx1_spl_twovtx_[nrecPV_twovtx_] = v1->xError();
931	+	erry1_spl_twovtx_[nrecPV_twovtx_] = v1->yError();
932	+	errz1_spl_twovtx_[nrecPV_twovtx_] = v1->zError();
933	+
934	+	nTrkPV2_spl_twovtx_[nrecPV_twovtx_] = nTrkPV2;
935	+	ndofPV2_spl_twovtx_[nrecPV_twovtx_] = v2->ndof();
936	+	normchi2PV2_spl_twovtx_[nrecPV_twovtx_] = v2->normalizedChi2();
937	+	avgPtPV2_spl_twovtx_[nrecPV_twovtx_] = avgPtRecVtx(*v2);
938	+	errx2_spl_twovtx_[nrecPV_twovtx_] = v2->xError();
939	+	erry2_spl_twovtx_[nrecPV_twovtx_] = v2->yError();
940	+	errz2_spl_twovtx_[nrecPV_twovtx_] = v2->zError();
941	+
942		nrecPV_twovtx_++;
943	+
944	+	// Print some information of the two tracks events
945	+	if(verbose_ && fntrk_ < 4) {
946	+	cout<<"Printing matchedVertexColl1 for low ntrack bins"<<endl;
947	+	printRecVtx(*v1);
948	+	cout<<"Printing matchedVertexColl1 for low ntrack bins"<<endl;
949	+	printRecVtx(*v2);
950	+	}
951		} // End of analyzing res/pull
952		} // End of  if(matchedVertexColl1->size() == 1 && matchedVertexColl2->size() == 1 ) {
953		else
#	Line 926 | Line 955 | PVStudy::analyze(const edm::Event& iEven
955
956		edm::LogInfo("Debug")<<"[PVStudy] End analyzing the matchedVertexColl"<<endl;
957
958	<	//=======================================================
959	<	// Fill simulated vertices
960	<	//=======================================================
961	<	if (!realData_) {
933	<	bool MC=false;
934	<	Handle<HepMCProduct> evtMC;
935	<	iEvent.getByLabel("generator",evtMC);
936	<	if (!evtMC.isValid()) {
937	<	MC=false;
938	<	edm::LogInfo("Debug") << "[PVStudy] no HepMCProduct found"<< endl;
939	<	} else {
940	<	edm::LogInfo("Debug") << "[PVStudy] generator HepMCProduct found"<< endl;
941	<	MC=true;
942	<	}
943	<
944	<	if(MC){
945	<	// make a list of primary vertices:
946	<	std::vector<simPrimaryVertex> simpv;
947	<	simpv=getSimPVs(evtMC,"");
948	<	//     simpv=getSimPVs(evtMC, simVtxs, simTrks);
949	<	h_gen->Fill1d("nsimPV", simpv.size());
950	<	nsimPV_ = simpv.size();
951	<
952	<	int isimpv = 0;
953	<	for(std::vector<simPrimaryVertex>::iterator vsim=simpv.begin();
954	<	vsim!=simpv.end(); vsim++, isimpv++){
955	<	nsimTrkPV_[isimpv]  =vsim->nGenTrk;
956	<	simx_[isimpv] = vsim->x;
957	<	simy_[isimpv] = vsim->y;
958	<	simz_[isimpv] = vsim->y;
959	<	simptsq_[isimpv] = vsim->ptsq;
960	<
961	<	//fillMCHisto((std::vector<simPrimaryVertex>::iterator vsim, int isimpv, const reco::VertexCollection *vtxColl, int datatype) {
962	<	fillMCHisto(vsim, isimpv, splitVertexColl1, 1);
963	<	fillMCHisto(vsim, isimpv, splitVertexColl2, 2);
964	<	fillMCHisto(vsim, isimpv, vertexColl, 3);
965	<	}
966	<	} // End of  for(std::vector<simPrimaryVertex>::iterator vsim=simpv.begin()
967	<	} // End of   if (!realData_) {
968	<
969	<	// Finally fill the ftree_
958	>
959	>	//========================================================================
960	>	// Step 5: Fill ntuple if saventuple_ is on
961	>	//========================================================================
962		if(saventuple_) {
963		ftree_->Fill();
964		pvtxtree_->Fill();
965		}
966	+
967		}
968
969	+
970	+
971		// ------------ method called once each job just before starting event loop  ------------
972		void
973		PVStudy::beginJob()
#	Line 1018 | Line 1013 | PVStudy::endJob() {
1013		pvtxtree_->Write();
1014		file_->Close();
1015		}
1016	<	}
1022	<
1016	>
1017
1018	+	}
1019
1020		// Match a recovertex with a simvertex
1021		// ? Should the cut be parameter dependent?
#	Line 1028 | Line 1023 | PVStudy::endJob() {
1023		bool PVStudy::matchVertex(const PVStudy::simPrimaryVertex & vsim,
1024		const reco::Vertex & vrec)
1025		{
1026	<	return (fabs(vsim.z-vrec.z())<0.0500); // =500um
1026	>	if(vrec.isFake() || !vrec.isValid()) return false;
1027	>	else
1028	>	return true;
1029	>	//return (fabs(vsim.z-vrec.z())<0.0500); // =500um // remove this hard cut
1030		}
1031
1032		// Match two reconstructed vertices
#	Line 1095 | Line 1093 | void PVStudy::printRecVtxs(const Handle<
1093		}
1094		}
1095
1096	+	void PVStudy::printRecVtx(const reco::Vertex & v){
1097	+
1098	+	cout << "#trk " << std::setw(3) << v.tracksSize()
1099	+	<< " chi2 " << std::setw(4) << v.chi2()
1100	+	<< " ndof " << std::setw(3) << v.ndof() << endl
1101	+	<< " x "  << std::setw(8) <<std::fixed << std::setprecision(4) << v.x()
1102	+	<< " dx " << std::setw(8) << v.xError()<< endl
1103	+	<< " y "  << std::setw(8) << v.y()
1104	+	<< " dy " << std::setw(8) << v.yError()<< endl
1105	+	<< " z "  << std::setw(8) << v.z()
1106	+	<< " dz " << std::setw(8) << v.zError()
1107	+	<< endl;
1108	+	}
1109	+
1110		void PVStudy::printSimTrks(const Handle<SimTrackContainer> simTrks){
1111		cout <<  " simTrks   type, (momentum), vertIndex, genpartIndex"  << endl;
1112		int i=1;
#	Line 1113 | Line 1125 | void PVStudy::printSimTrks(const Handle<
1125		}
1126		}
1127
1128	<	void PVStudy::fillHisto(res r, error er, int ntk, int datamode){
1128	>	void PVStudy::fillHisto(res r, error er, int ntk, int datamode) {
1129		stringstream ss;
1130		ss << ntk;
1131		string suffix;
#	Line 1135 | Line 1147 | void PVStudy::fillHisto(res r, error er,
1147		h_others->Fill1d("errPVz"+suffix, er.z());
1148		}
1149
1150	+
1151		PVStudy::PVAnaInfo PVStudy::GetPVAnaInfo(int ntk, int datamode) {
1152		map<int,double> data;
1153		data.clear();
#	Line 1197 | Line 1210 | PVStudy::PVAnaInfo PVStudy::GetPVAnaInfo
1210		ntk);
1211		}
1212
1213	+
1214		void PVStudy::fit(TH1 *hdist, double fitPars[]){
1215		TAxis *axis0 = hdist->GetXaxis();
1216		int nbin = axis0->GetLast();
1217		double nOF = hdist->GetBinContent(nbin+1);
1218		double nUF = hdist->GetBinContent(0);
1219		//   double fitRange = axis0->GetBinUpEdge(nbin);
1220	<	double fitRange = axis0->GetXmax();
1220	>	// double fitRange = axis0->GetXmax();
1221	>	double fitRange = 2*hdist->GetRMS();
1222		double sigMax[2] = {0.,0.};
1223
1224		edm::LogInfo("Analysis") << "[Fit] Last bin = " << nbin
#	Line 1227 | Line 1242 | void PVStudy::fit(TH1 *hdist, double fit
1242
1243		TF1 *fgaus = new TF1("fgaus","gaus",-fitRange, fitRange);
1244		fgaus->SetParameter(1, 0.);
1245	<	fgaus->SetParLimits(1, -fitRange/40., fitRange/40.);
1245	>	fgaus->SetParLimits(1, -fitRange/10., fitRange/10.);
1246		fgaus->SetParLimits(2, 0., sqrt(2.)*sigMax[1]);
1247		fgaus->SetLineColor(4);
1248		hdist->Fit(fgaus,"QLRM");
#	Line 1243 | Line 1258 | void PVStudy::fit(TH1 *hdist, double fit
1258		fitPars[0] = -999;
1259		}
1260
1261	<	void PVStudy::fillTrackHisto(const reco::TrackCollection *trackColl, int datatype) {
1261	>
1262	>	void PVStudy::fillTrackHisto(const reco::TrackCollection *trackColl, int datatype, const Point & bs) {
1263		string suffix;
1264		suffix = ""; // for unsplit tracks
1265		if(datatype == 1) suffix = "1_spl"; // for splittracks 1
#	Line 1255 | Line 1271 | void PVStudy::fillTrackHisto(const reco:
1271		++itTrack) {
1272		h_pvtrk->Fill1d("trkPt"+suffix, itTrack->pt());
1273		h_pvtrk->Fill1d("trkDxy"+suffix, itTrack->dxy());
1274	+	h_pvtrk->Fill1d("trkDxyCorr"+suffix, itTrack->dxy(bs));
1275		h_pvtrk->Fill1d("trkDz"+suffix, itTrack->dz());
1276		h_pvtrk->Fill1d("trkEta"+suffix, itTrack->eta());
1277		h_pvtrk->Fill1d("trkPhi"+suffix, itTrack->phi());
1278		}
1279		}
1280
1281	<	void PVStudy::fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, bool fillHisto, bool fillNtuple) {
1281	>	void PVStudy::fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, const Point & bs) {
1282		string suffix;
1283		suffix = ""; // for unsplit tracks
1284		if(datatype == 1) suffix = "1_spl"; // for splittracks 1
#	Line 1269 | Line 1286 | void PVStudy::fillTrackHistoInPV(const r
1286		int ivtx = 0;
1287		for(reco::VertexCollection::const_iterator v=vertexColl->begin();
1288		v!=vertexColl->end(); ++v, ++ivtx) {
1289	<	if(fillNtuple) {
1290	<	if(datatype == 0) {
1291	<	nTrkPV_[ivtx] = v->tracksSize();
1292	<	sumptsq_[ivtx] = sumPtSquared(*v);
1293	<	isValid_[ivtx] = v->isValid();
1294	<	isFake_[ivtx] = v->isFake();
1295	<	recx_[ivtx] = v->x();
1296	<	recy_[ivtx] = v->y();
1297	<	recz_[ivtx] = v->z();
1298	<	recx_err_[ivtx] = v->xError();
1299	<	recy_err_[ivtx] = v->yError();
1300	<	recz_err_[ivtx] = v->zError();
1301	<	}
1302	<	if(datatype == 1) {
1303	<	nTrkPV1_spl_[ivtx] = v->tracksSize();
1304	<	sumptsq1_spl_[ivtx] = sumPtSquared(*v);
1305	<	isValid1_spl_[ivtx] = v->isValid();
1306	<	isFake1_spl_[ivtx] = v->isFake();
1307	<	recx1_spl_[ivtx] = v->x();
1291	<	recy1_spl_[ivtx] = v->y();
1292	<	recz1_spl_[ivtx] = v->z();
1293	<	recx1_err_spl_[ivtx] = v->xError();
1294	<	recy1_err_spl_[ivtx] = v->yError();
1295	<	recz1_err_spl_[ivtx] = v->zError();
1296	<
1297	<	}
1298	<	if(datatype == 2) {
1299	<	nTrkPV2_spl_[ivtx] = v->tracksSize();
1300	<	sumptsq2_spl_[ivtx] = sumPtSquared(*v);
1301	<	isValid2_spl_[ivtx] = v->isValid();
1302	<	isFake2_spl_[ivtx] = v->isFake();
1303	<	recx2_spl_[ivtx] = v->x();
1304	<	recy2_spl_[ivtx] = v->y();
1305	<	recz2_spl_[ivtx] = v->z();
1306	<	recx2_err_spl_[ivtx] = v->xError();
1307	<	recy2_err_spl_[ivtx] = v->yError();
1308	<	recz2_err_spl_[ivtx] = v->zError();
1309	<	}
1310	<	if(datatype == 4) { // for pixelVertices
1311	<	nTrkPV_pxlpvtx_[ivtx] = v->tracksSize();
1312	<	sumptsq_pxlpvtx_[ivtx] = sumPtSquared(*v);
1313	<	isValid_pxlpvtx_[ivtx] = v->isValid();
1314	<	isFake_pxlpvtx_[ivtx] = v->isFake();
1315	<	recx_pxlpvtx_[ivtx] = v->x();
1316	<	recy_pxlpvtx_[ivtx] = v->y();
1317	<	recz_pxlpvtx_[ivtx] = v->z();
1318	<	recx_err_pxlpvtx_[ivtx] = v->xError();
1319	<	recy_err_pxlpvtx_[ivtx] = v->yError();
1320	<	recz_err_pxlpvtx_[ivtx] = v->zError();
1321	<	}
1322	<	}
1323	<	}
1324	<	// For histogram only fill the leading pvtx parameters
1325	<	if(fillHisto) {
1326	<	h_pvtrk->Fill1d("nTrkPV"+suffix, vertexColl->begin()->tracksSize());
1327	<	try {
1328	<	for(reco::Vertex::trackRef_iterator t = vertexColl->begin()->tracks_begin();
1329	<	t!=vertexColl->begin()->tracks_end(); t++) {
1330	<	// illegal charge
1331	<	if ( (**t).charge() < -1 || (**t).charge() > 1 ) {
1332	<	//      h_pvtrk->Fill1d("trkPtPV", 0.);
1333	<	}
1334	<	else {
1335	<	h_pvtrk->Fill1d("trkPtPV"+suffix, (**t).pt());
1336	<	h_pvtrk->Fill1d("trkDxyPV"+suffix, (**t).dxy());
1337	<	h_pvtrk->Fill1d("trkDzPV"+suffix, (**t).dz());
1338	<	h_pvtrk->Fill1d("trkEtaPV"+suffix, (**t).eta());
1339	<	h_pvtrk->Fill1d("trkPhiPV"+suffix, (**t).phi());
1289	>	if(!v->isFake()) {
1290	>	h_pvtrk->Fill1d("nTrkPV"+suffix, vertexColl->begin()->tracksSize());
1291	>	h_pvtrk->Fill1d("ndofPV"+suffix, vertexColl->begin()->ndof());
1292	>	h_pvtrk->Fill1d("nHWTrkPV"+suffix, nHWTrkRecVtx(*v));
1293	>	try {
1294	>	for(reco::Vertex::trackRef_iterator t = vertexColl->begin()->tracks_begin();
1295	>	t!=vertexColl->begin()->tracks_end(); t++) {
1296	>	// illegal charge
1297	>	if ( (**t).charge() < -1 || (**t).charge() > 1 ) {
1298	>	//    h_pvtrk->Fill1d("trkPtPV", 0.);
1299	>	}
1300	>	else {
1301	>	h_pvtrk->Fill1d("trkPtPV"+suffix, (**t).pt());
1302	>	h_pvtrk->Fill1d("trkDxyPV"+suffix, (**t).dxy());
1303	>	h_pvtrk->Fill1d("trkDxyCorrPV"+suffix, (**t).dxy(bs));
1304	>	h_pvtrk->Fill1d("trkDzPV"+suffix, (**t).dz());
1305	>	h_pvtrk->Fill1d("trkEtaPV"+suffix, (**t).eta());
1306	>	h_pvtrk->Fill1d("trkPhiPV"+suffix, (**t).phi());
1307	>	}
1308		}
1309		}
1310	<	}
1311	<	catch (...) {
1312	<	// exception thrown when trying to use linked track
1313	<	//          h_pvtrk->Fill1d("trkPtPV", 0.);
1310	>	catch (...) {
1311	>	// exception thrown when trying to use linked track
1312	>	//        h_pvtrk->Fill1d("trkPtPV", 0.);
1313	>	}
1314		}
1315		}
1348	–
1316		}
1317
1318	<	void PVStudy::fillMCHisto(std::vector<simPrimaryVertex>::iterator vsim, int isimpv, const reco::VertexCollection *vtxColl, int datatype)
1318	>	void PVStudy::fillMCHisto(std::vector<simPrimaryVertex>::iterator vsim, int isimpv, const reco::VertexCollection *vtxColl,
1319	>	int datatype, double avgTrkPtInPVMin_, double avgTrkPtInPVMax_)
1320		{
1321		std::string suffix;
1322		// Set the vertexColl and histogram suffix according to datamode
#	Line 1364 | Line 1332 | void PVStudy::fillMCHisto(std::vector<si
1332		suffix = "_mct";
1333		nrecPV_mct_ = 0;
1334		}
1335	<
1335	>
1336		//========================================================
1337	<	//  look for a matching reconstructed vertex in vertexColl
1338	<	//========================================================
1339	<
1337	>	//  For each simulated vertex, look for a match in the vertexColl
1338	>	//  If more than 1 recVtx is found, use the one with closest in Z
1339	>	//========================================================
1340	>
1341	>	// === look for a matching reconstructed vertex in vertexColl
1342		for(reco::VertexCollection::const_iterator vrec=vtxColl->begin();
1343		vrec!=vtxColl->end(); ++vrec){
1374	–	int nrectrk = vrec->tracksSize();
1344		vsim->recVtx=NULL;
1376	–
1345		edm::LogInfo("Debug") << "[fillMCHisto] sim primary vertex x = " << vsim->x << "; y = " << vsim->y << "; z = " << vsim->z <<  endl;
1346		edm::LogInfo("Debug") << "[fillMCHisto] Is matched? " << (matchVertex(*vsim,*vrec)?"Yes":"No") << endl;
1347	<
1348	<	if ( matchVertex(*vsim,*vrec) && vrec->isValid() && !vrec->isFake() ) {
1347	>
1348	>	if ( matchVertex(*vsim,*vrec)  ) {
1349		vsim->recVtx=&(*vrec);
1350	<
1383	<	// if the matching critera are fulfilled, accept the rec-vertex that is closest in z
1350	>	//if the matching critera are fulfilled, accept the rec-vertex that is closest in z
1351		//if(    ((vsim->recVtx) && (fabs(vsim->recVtx->position().z()-vsim->z)>fabs(vrec->z()-vsim->z)))
1352		//|| (!vsim->recVtx) )
1353		//vsim->recVtx=&(*vrec);
1387	–	//}
1388	–	edm::LogInfo("Debug") <<"[fillMCHisto] primary matched in vertexColl" << vsim->x << " " << vsim->y << " " << vsim->z <<  endl;
1389	–
1390	–	double fres_mct[3];
1391	–	double ferror_mct[3];
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 ftree_
1419	–	if(datatype == 1) {
1420	–	nTrkPV_spl1_mct_[nrecPV_spl1_mct_] =   nrectrk;
1421	–	deltax_spl1_mct_[nrecPV_spl1_mct_] =  fres_mct[0];
1422	–	deltay_spl1_mct_[nrecPV_spl1_mct_] =  fres_mct[0];
1423	–	deltaz_spl1_mct_[nrecPV_spl1_mct_] =  fres_mct[0];
1424	–	pullx_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[0]/ferror_mct[0];
1425	–	pully_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[1]/ferror_mct[1];
1426	–	pullz_spl1_mct_[nrecPV_spl1_mct_] =  fres_mct[2]/ferror_mct[2];
1427	–	nrecPV_spl1_mct_++;
1428	–	}
1429	–
1430	–	if(datatype == 2) {
1431	–	nTrkPV_spl2_mct_[nrecPV_spl2_mct_] = nrectrk;
1432	–	deltax_spl2_mct_[nrecPV_spl2_mct_] =  fres_mct[0];
1433	–	deltay_spl2_mct_[nrecPV_spl2_mct_] =  fres_mct[0];
1434	–	deltaz_spl2_mct_[nrecPV_spl2_mct_] =  fres_mct[0];
1435	–	pullx_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[0]/ferror_mct[0];
1436	–	pully_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[1]/ferror_mct[1];
1437	–	pullz_spl2_mct_[nrecPV_spl2_mct_] =  fres_mct[2]/ferror_mct[2];
1438	–	nrecPV_spl2_mct_++;
1439	–	}
1440	–	if(datatype == 3) {
1441	–	nTrkPV_mct_[nrecPV_mct_] = nrectrk;
1442	–	deltax_mct_[nrecPV_mct_] =  fres_mct[0];
1443	–	deltay_mct_[nrecPV_mct_] =  fres_mct[0];
1444	–	deltaz_mct_[nrecPV_mct_] =  fres_mct[0];
1445	–	pullx_mct_[nrecPV_mct_] = fres_mct[0]/ferror_mct[0];
1446	–	pully_mct_[nrecPV_mct_] = fres_mct[1]/ferror_mct[1];
1447	–	pullz_mct_[nrecPV_mct_] =  fres_mct[2]/ferror_mct[2];
1448	–	nrecPV_mct_++;
1449	–	}
1450	–	} // End of  if(saventuple_) {
1451	–	} //  if ( matchVertex(*vsim,*vrec) ) {
1452	–	else {  // no rec vertex found for this simvertex
1453	–	edm::LogInfo("Debug") <<"[fillMCHisto] primary not found " << vsim->x << " " << vsim->y << " " << vsim->z << " nGenTrk=" << vsim->nGenTrk << endl;
1354		}
1355		}
1356	<	}
1356	>
1357	>	// === If match found fill the residual and pulls
1358	>	if(vsim->recVtx) {
1359	>	edm::LogInfo("Debug") <<"[fillMCHisto] primary matched in vertexColl" << vsim->x << " " << vsim->y << " " << vsim->z <<  endl;
1360	>	double avgPtPV = avgPtRecVtx(*(vsim->recVtx));
1361	>	if(avgPtPV>avgTrkPtInPVMax_ || avgPtPV<avgTrkPtInPVMin_) return;
1362	>
1363	>	// if study the resolution/pull for all tracks in the PVtx
1364	>	int nrectrk;
1365	>	if(useHWTrk_)
1366	>	nrectrk=nHWTrkRecVtx(*(vsim->recVtx));
1367	>	else
1368	>	nrectrk =  int(vsim->recVtx->tracksSize());
1369
1370	+	double fres_mct[3];
1371	+	double ferror_mct[3];
1372	+
1373	+	double ndofPV = vsim->recVtx->ndof();
1374	+	double normchi2PV =  vsim->recVtx->normalizedChi2();
1375
1376	+	fres_mct[0] = vsim->recVtx->x()-vsim->x;
1377	+	fres_mct[1] = vsim->recVtx->y()-vsim->y;
1378	+	fres_mct[2] = vsim->recVtx->z()-vsim->z;
1379	+	ferror_mct[0] = vsim->recVtx->xError();
1380	+	ferror_mct[1] = vsim->recVtx->yError();
1381	+	ferror_mct[2] = vsim->recVtx->zError();
1382	+
1383	+	h_summary->Fill1d("deltax"+suffix, fres_mct[0] );
1384	+	h_summary->Fill1d("deltay"+suffix, fres_mct[1] );
1385	+	h_summary->Fill1d("deltaz"+suffix, fres_mct[2] );
1386	+	h_summary->Fill1d("pullx"+suffix, fres_mct[0]/ferror_mct[0] );
1387	+	h_summary->Fill1d("pully"+suffix, fres_mct[1]/ferror_mct[1] );
1388	+	h_summary->Fill1d("pullz"+suffix, fres_mct[2]/ferror_mct[2] );
1389	+	h_summary->Fill1d("errPVx"+suffix, ferror_mct[0] );
1390	+	h_summary->Fill1d("errPVy"+suffix, ferror_mct[1] );
1391	+	h_summary->Fill1d("errPVz"+suffix, ferror_mct[2] );
1392	+	pvinfo.push_back(PVStudy::PVInfo(res(fres_mct[0], fres_mct[1], fres_mct[2]),
1393	+	error(ferror_mct[0], ferror_mct[1], ferror_mct[2]),
1394	+	nrectrk));
1395	+	// Fill histo according to its track multiplicity
1396	+	fillHisto(res(fres_mct[0], fres_mct[1], fres_mct[2]),
1397	+	error(ferror_mct[0], ferror_mct[1], ferror_mct[2]),
1398	+	nrectrk, datatype);
1399	+
1400	+	if(saventuple_) {
1401	+	//Fill the values for variables in pvtxtree_
1402	+	if(datatype == 1) {
1403	+	nTrkPV_spl1_mct_[nrecPV_spl1_mct_] =   nrectrk;
1404	+	ndofPV_spl1_mct_[nrecPV_spl1_mct_] =  ndofPV;
1405	+	normchi2PV_spl1_mct_[nrecPV_spl1_mct_] =  normchi2PV;
1406	+	avgPtPV_spl1_mct_[nrecPV_spl1_mct_] = avgPtPV;
1407	+	deltax_spl1_mct_[nrecPV_spl1_mct_] =  fres_mct[0];
1408	+	deltay_spl1_mct_[nrecPV_spl1_mct_] =  fres_mct[1];
1409	+	deltaz_spl1_mct_[nrecPV_spl1_mct_] =  fres_mct[2];
1410	+	pullx_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[0]/ferror_mct[0];
1411	+	pully_spl1_mct_[nrecPV_spl1_mct_] = fres_mct[1]/ferror_mct[1];
1412	+	pullz_spl1_mct_[nrecPV_spl1_mct_] =  fres_mct[2]/ferror_mct[2];
1413	+	errx_spl1_mct_[nrecPV_spl1_mct_] =  ferror_mct[0];
1414	+	erry_spl1_mct_[nrecPV_spl1_mct_] =  ferror_mct[1];
1415	+	errz_spl1_mct_[nrecPV_spl1_mct_] =  ferror_mct[2];
1416	+	nrecPV_spl1_mct_++;
1417	+	}
1418	+	if(datatype == 2) {
1419	+	nTrkPV_spl2_mct_[nrecPV_spl2_mct_] = nrectrk;
1420	+	ndofPV_spl2_mct_[nrecPV_spl2_mct_] =  ndofPV;
1421	+	normchi2PV_spl2_mct_[nrecPV_spl2_mct_] =  normchi2PV;
1422	+	avgPtPV_spl2_mct_[nrecPV_spl2_mct_] = avgPtPV;
1423	+	deltax_spl2_mct_[nrecPV_spl2_mct_] =  fres_mct[0];
1424	+	deltay_spl2_mct_[nrecPV_spl2_mct_] =  fres_mct[1];
1425	+	deltaz_spl2_mct_[nrecPV_spl2_mct_] =  fres_mct[2];
1426	+	pullx_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[0]/ferror_mct[0];
1427	+	pully_spl2_mct_[nrecPV_spl2_mct_] = fres_mct[1]/ferror_mct[1];
1428	+	pullz_spl2_mct_[nrecPV_spl2_mct_] =  fres_mct[2]/ferror_mct[2];
1429	+	errx_spl2_mct_[nrecPV_spl2_mct_] =  ferror_mct[0];
1430	+	erry_spl2_mct_[nrecPV_spl2_mct_] =  ferror_mct[1];
1431	+	errz_spl2_mct_[nrecPV_spl2_mct_] =  ferror_mct[2];
1432	+	nrecPV_spl2_mct_++;
1433	+	}
1434	+	if(datatype == 3) {
1435	+	nTrkPV_mct_[nrecPV_mct_] = nrectrk;
1436	+	ndofPV_mct_[nrecPV_mct_] =  ndofPV;
1437	+	normchi2PV_mct_[nrecPV_mct_] =  normchi2PV;
1438	+	avgPtPV_mct_[nrecPV_mct_] = avgPtPV;
1439	+	deltax_mct_[nrecPV_mct_] =  fres_mct[0];
1440	+	deltay_mct_[nrecPV_mct_] =  fres_mct[0];
1441	+	deltaz_mct_[nrecPV_mct_] =  fres_mct[0];
1442	+	pullx_mct_[nrecPV_mct_] = fres_mct[0]/ferror_mct[0];
1443	+	pully_mct_[nrecPV_mct_] = fres_mct[1]/ferror_mct[1];
1444	+	pullz_mct_[nrecPV_mct_] =  fres_mct[2]/ferror_mct[2];
1445	+	errx_mct_[nrecPV_mct_] =  ferror_mct[0];
1446	+	erry_mct_[nrecPV_mct_] =  ferror_mct[1];
1447	+	errz_mct_[nrecPV_mct_] =  ferror_mct[2];
1448	+	nrecPV_mct_++;
1449	+	}
1450	+	} // End of  if(saventuple_) {
1451	+	} //  if ( matchVertex(*vsim,*vrec) ) {
1452	+	else {  // no rec vertex found for this simvertex
1453	+	edm::LogInfo("Debug") <<"[fillMCHisto] primary not found " << vsim->x << " " << vsim->y << " " << vsim->z << " nGenTrk=" << vsim->nGenTrk << endl;
1454	+	}
1455	+	}
1456	+
1457	+
1458		void PVStudy::SetVarToZero() {
1460	–	//Greg's variables
1459		fntrk_ = 0;
1460		//pvtx position (x,y,z) residual and error
1461		for(int i = 0; i<3;i++)    {
#	Line 1465 | Line 1463 | void PVStudy::SetVarToZero() {
1463		ferror_[i] = 0;
1464		}
1465
1466	<	//Yanyan's variables
1467	<	// Number of reconstructed vertices
1466	>	// Event level information
1467	>	glob_runno_ = 0;
1468	>	glob_evtno_ = 0;
1469	>	glob_ls_ = 0;
1470	>	glob_bx_ = 0;
1471		nrecPV_ = 0;
1472		nrecPV1_spl_ = 0;
1473		nrecPV2_spl_ = 0;
#	Line 1482 | Line 1483 | void PVStudy::SetVarToZero() {
1483		min_ntrksep_ = 9999.0;
1484		min_sumpt2sep_ = -9999.0;
1485
1486	<
1486	>	// Variables filled per Vertex
1487		for (int i = 0; i < nMaxPVs_; i++) {
1487	–	// recoVertices with all tracks
1488	–	nTrkPV_[i] = 0; // Number of tracks in the pvtx
1489	–	sumptsq_[i] = 0;
1490	–	isValid_[i] = -1;
1491	–	isFake_[i] = -1;
1492	–	recx_[i] = 0;
1493	–	recy_[i] = 0;
1494	–	recz_[i] = 0;
1495	–	recx_err_[i] = 0;
1496	–	recy_err_[i] = 0;
1497	–	recz_err_[i] = 0;
1498	–
1499	–	// recoVertices with splitTrack1
1500	–	nTrkPV1_spl_[i] = 0; // Number of tracks in the pvtx
1501	–	sumptsq1_spl_[i] = 0;
1502	–	isValid1_spl_[i] = -1;
1503	–	isFake1_spl_[i] = -1;
1504	–	recx1_spl_[i] = 0;
1505	–	recy1_spl_[i] = 0;
1506	–	recz1_spl_[i] = 0;
1507	–	recx1_err_spl_[i] = 0;
1508	–	recy1_err_spl_[i] = 0;
1509	–	recz1_err_spl_[i] = 0;
1510	–
1511	–	// recoVertices with splitTrack2
1512	–	nTrkPV2_spl_[i] = 0; // Number of tracks in the pvtx
1513	–	sumptsq2_spl_[i] = 0;
1514	–	isValid2_spl_[i] = -1;
1515	–	isFake2_spl_[i] = -1;
1516	–	recx2_spl_[i] = 0;
1517	–	recy2_spl_[i] = 0;
1518	–	recz2_spl_[i] = 0;
1519	–	recx2_err_spl_[i] = 0;
1520	–	recy2_err_spl_[i] = 0;
1521	–	recz2_err_spl_[i] = 0;
1522	–
1523	–	//pixelVertices
1524	–	nTrkPV_pxlpvtx_[i] = 0; // Number of tracks in the pvtx
1525	–	sumptsq_pxlpvtx_[i] = 0;
1526	–	isValid_pxlpvtx_[i] = -1;
1527	–	isFake_pxlpvtx_[i] = -1;
1528	–	recx_pxlpvtx_[i] = 0;
1529	–	recy_pxlpvtx_[i] = 0;
1530	–	recz_pxlpvtx_[i] = 0;
1531	–	recx_err_pxlpvtx_[i] = 0;
1532	–	recy_err_pxlpvtx_[i] = 0;
1533	–	recz_err_pxlpvtx_[i] = 0;
1534	–
1488		// matched two-vertices
1489		nTrkPV1_spl_twovtx_[i] = 0;
1490	+	ndofPV1_spl_twovtx_[i] = 0;
1491	+	normchi2PV1_spl_twovtx_[i] = 0;
1492	+	avgPtPV1_spl_twovtx_[i] = 0;
1493	+	errx1_spl_twovtx_[i] = 0;
1494	+	erry1_spl_twovtx_[i] = 0;
1495	+	errz1_spl_twovtx_[i] = 0;
1496	+
1497		nTrkPV2_spl_twovtx_[i] = 0;
1498	<	nTrkPV_twovtx_[i] = 0;
1499	<	sumptsq1_spl_twovtx_[i] = 0; // Sum of pT squred in the pvtx
1500	<	sumptsq2_spl_twovtx_[i] = 0; // Sum of pT squred in the pvtx
1498	>	ndofPV2_spl_twovtx_[i] = 0;
1499	>	normchi2PV2_spl_twovtx_[i] = 0;
1500	>	avgPtPV2_spl_twovtx_[i] = 0;
1501	>	errx2_spl_twovtx_[i] = 0;
1502	>	erry2_spl_twovtx_[i] = 0;
1503	>	errz2_spl_twovtx_[i] = 0;
1504	>
1505		deltax_twovtx_[i] = 0;
1506		deltay_twovtx_[i] = 0;
1507		deltaz_twovtx_[i] = 0;
#	Line 1562 | Line 1526 | void PVStudy::SetVarToZero() {
1526		pullx_mct_[i] = 0;
1527		pully_mct_[i] = 0;
1528		pullz_mct_[i] = 0;
1529	<
1529	>	nTrkPV_mct_[i] = 0;
1530	>	ndofPV_mct_[i] = 0;
1531	>	normchi2PV_mct_[i] = 0;
1532	>	avgPtPV_mct_[i] = 0;
1533	>	errx_mct_[i] = 0;
1534	>	erry_mct_[i] = 0;
1535	>	errz_mct_[i] = 0;
1536	>
1537		deltax_spl1_mct_[i] = 0;
1538		deltay_spl1_mct_[i] = 0;
1539		deltaz_spl1_mct_[i] = 0;
1540		pullx_spl1_mct_[i] = 0;
1541		pully_spl1_mct_[i] = 0;
1542		pullz_spl1_mct_[i] = 0;
1543	<
1543	>	nTrkPV_spl1_mct_[i] = 0;
1544	>	ndofPV_spl1_mct_[i] = 0;
1545	>	normchi2PV_spl1_mct_[i] = 0;
1546	>	avgPtPV_spl1_mct_[i] = 0;
1547	>	errx_spl1_mct_[i] = 0;
1548	>	erry_spl1_mct_[i] = 0;
1549	>	errz_spl1_mct_[i] = 0;
1550	>
1551		deltax_spl2_mct_[i] = 0;
1552		deltay_spl2_mct_[i] = 0;
1553		deltaz_spl2_mct_[i] = 0;
1554		pullx_spl2_mct_[i] = 0;
1555		pully_spl2_mct_[i] = 0;
1556		pullz_spl2_mct_[i] = 0;
1557	+	nTrkPV_spl2_mct_[i] = 0;
1558	+	ndofPV_spl2_mct_[i] = 0;
1559	+	normchi2PV_spl2_mct_[i] = 0;
1560	+	avgPtPV_spl2_mct_[i] = 0;
1561	+	errx_spl2_mct_[i] = 0;
1562	+	erry_spl2_mct_[i] = 0;
1563	+	errz_spl2_mct_[i] = 0;
1564	+
1565		}
1580	–
1566		}
1567
1568		double PVStudy::sumPtSquared(const reco::Vertex & v)  {
1569	<	double sum = 0.;
1570	<	double pT;
1571	<	for (reco::Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) {
1572	<	pT = (**it).pt();
1573	<	sum += pT*pT;
1569	>	if(v.isFake() || (!v.isValid()) || v.tracksSize() == 0) return 0.0;
1570	>	else {
1571	>	double sum = 0.;
1572	>	double pT;
1573	>	for (reco::Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) {
1574	>	pT = (**it).pt();
1575	>	sum += pT*pT;
1576	>	}
1577	>	return sum;
1578	>	}
1579	>	}
1580	>
1581	>
1582	>
1583	>	double PVStudy::avgPtRecVtx(const reco::Vertex & v)  {
1584	>
1585	>	if(v.isFake() || !v.isValid() || v.tracksSize()==0 ) return 0;
1586	>	else {
1587	>	int nHWTrk = 0;
1588	>	double sumpT = 0.;
1589	>	for (reco::Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) {
1590	>	if(v.trackWeight(*it) > 0.5 ) {
1591	>	sumpT +=  (**it).pt();
1592	>	nHWTrk++;
1593	>	}
1594	>	}
1595	>	if(nHWTrk > 0)
1596	>	return sumpT/double(nHWTrk);
1597	>	else
1598	>	return 0;
1599		}
1590	–	return sum;
1600		}
1601
1602	+	int PVStudy::nHWTrkRecVtx(const reco::Vertex & v)  {
1603	+	int nHWTrkPV = 0;
1604	+	for (reco::Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) {
1605	+	if(v.trackWeight(*it) > 0.5)
1606	+	nHWTrkPV++;
1607	+	}
1608	+	return nHWTrkPV;
1609	+	}

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines