ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/Jeng/PVStudy/plugins/PVStudy.cc
(Generate patch)

Comparing UserCode/Jeng/PVStudy/plugins/PVStudy.cc (file contents):
Revision 1.17 by yygao, Tue Jan 26 12:06:49 2010 UTC vs.
Revision 1.22 by yygao, Mon Mar 29 15:27:24 2010 UTC

# Line 38 | 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 51 | Line 51 | Implementation:
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>
# Line 95 | Line 80 | PVStudy::PVStudy(const edm::ParameterSet
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 {
# Line 127 | Line 112 | PVStudy::PVStudy(const edm::ParameterSet
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");
# Line 199 | Line 144 | PVStudy::PVStudy(const edm::ParameterSet
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_) {  
# Line 216 | Line 181 | PVStudy::PVStudy(const edm::ParameterSet
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");
# Line 225 | Line 198 | PVStudy::PVStudy(const edm::ParameterSet
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");
# Line 232 | Line 213 | PVStudy::PVStudy(const edm::ParameterSet
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    
# Line 256 | Line 245 | PVStudy::PVStudy(const edm::ParameterSet
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();
# Line 269 | Line 257 | PVStudy::PVStudy(const edm::ParameterSet
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
# Line 318 | Line 305 | PVStudy::~PVStudy()
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_)
# Line 356 | Line 342 | void PVStudy::setRootStyle() {
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;
# Line 546 | Line 532 | PVStudy::analyze(const edm::Event& iEven
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;
# Line 566 | Line 550 | PVStudy::analyze(const edm::Event& iEven
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;
# Line 576 | Line 560 | PVStudy::analyze(const edm::Event& iEven
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;
# Line 587 | Line 571 | PVStudy::analyze(const edm::Event& iEven
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;
# Line 600 | Line 581 | PVStudy::analyze(const edm::Event& iEven
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;
# Line 610 | Line 591 | PVStudy::analyze(const edm::Event& iEven
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;
# Line 622 | Line 603 | PVStudy::analyze(const edm::Event& iEven
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);
# Line 657 | Line 620 | PVStudy::analyze(const edm::Event& iEven
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);
# Line 735 | Line 678 | PVStudy::analyze(const edm::Event& iEven
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;
# Line 786 | Line 690 | PVStudy::analyze(const edm::Event& iEven
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());
# Line 796 | Line 701 | PVStudy::analyze(const edm::Event& iEven
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;    
# Line 887 | Line 764 | PVStudy::analyze(const edm::Event& iEven
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;
# Line 931 | Line 803 | PVStudy::analyze(const edm::Event& iEven
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);
# Line 955 | Line 834 | PVStudy::analyze(const edm::Event& iEven
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();
# Line 981 | Line 898 | PVStudy::analyze(const edm::Event& iEven
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] );
# Line 999 | Line 927 | PVStudy::analyze(const edm::Event& iEven
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];
# Line 1014 | Line 939 | PVStudy::analyze(const edm::Event& iEven
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;
# Line 1064 | Line 975 | PVStudy::analyze(const edm::Event& iEven
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();
# Line 1132 | Line 1044 | bool PVStudy::matchVertex(const PVStudy:
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
# Line 1199 | Line 1112 | void PVStudy::printRecVtxs(const Handle<
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;
# Line 1370 | Line 1297 | void PVStudy::fillTrackHisto(const reco:
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
# Line 1378 | Line 1305 | void PVStudy::fillTrackHistoInPV(const r
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)
# Line 1483 | Line 1350 | void PVStudy::fillMCHisto(std::vector<si
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++)    {
# Line 1583 | Line 1480 | void PVStudy::SetVarToZero() {
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;
# Line 1600 | Line 1500 | void PVStudy::SetVarToZero() {
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;
# Line 1680 | Line 1543 | void PVStudy::SetVarToZero() {
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 + }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines