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.9 by yygao, Thu Oct 29 02:05:54 2009 UTC vs.
Revision 1.13 by jengbou, Fri Dec 4 08:37:40 2009 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 66 | Line 66 | PVStudy::PVStudy(const edm::ParameterSet
66    //=======================================================================
67    // Get configuration for TrackTupleMaker
68    //=======================================================================
69 <  simG4_           = iConfig.getParameter<edm::InputTag>( "simG4" );
70 <  trackCollectionTag_     = iConfig.getUntrackedParameter<edm::InputTag>("trackCollection");  
71 <  splitTrackCollection1Tag_     = iConfig.getUntrackedParameter<edm::InputTag>("splitTrackCollection1");
72 <  splitTrackCollection2Tag_     = iConfig.getUntrackedParameter<edm::InputTag>("splitTrackCollection2");
69 >  simG4_                     = iConfig.getParameter<edm::InputTag>( "simG4" );
70 >  trackCollectionTag_        = iConfig.getUntrackedParameter<edm::InputTag>("trackCollection");  
71 >  splitTrackCollection1Tag_  = iConfig.getUntrackedParameter<edm::InputTag>("splitTrackCollection1");
72 >  splitTrackCollection2Tag_  = iConfig.getUntrackedParameter<edm::InputTag>("splitTrackCollection2");
73    vertexCollectionTag_       = iConfig.getUntrackedParameter<edm::InputTag>("vertexCollection");
74 <  splitVertexCollection1Tag_       = iConfig.getUntrackedParameter<edm::InputTag>("splitVertexCollection1");
75 <  splitVertexCollection2Tag_       = iConfig.getUntrackedParameter<edm::InputTag>("splitVertexCollection2");
76 <  verbose_         = iConfig.getUntrackedParameter<bool>("verbose",false);
77 <  realData_        = iConfig.getUntrackedParameter<bool>("realData",false);
78 <  analyze_         = iConfig.getUntrackedParameter<bool>("analyzeOnTheFly",false);
79 <  saventuple_         = iConfig.getUntrackedParameter<bool>("saventuple",false);  
80 <  outputfilename_  = iConfig.getUntrackedParameter<string>("OutputFileName");
81 <  ntrkdiffcut_         = iConfig.getUntrackedParameter<double>("ntrkdiffcut");
82 <  zsigncut_         = iConfig.getUntrackedParameter<int>("zsigncut");
83 <  nTrkMin_ =  iConfig.getUntrackedParameter<int>("nTrkMin");
84 <  nTrkMax_ =  iConfig.getUntrackedParameter<int>("nTrkMax");
74 >  splitVertexCollection1Tag_ = iConfig.getUntrackedParameter<edm::InputTag>("splitVertexCollection1");
75 >  splitVertexCollection2Tag_ = iConfig.getUntrackedParameter<edm::InputTag>("splitVertexCollection2");  
76 >  pixelVertexCollectionTag_  = iConfig.getParameter<edm::InputTag>("pixelVertexCollectionTag");  
77 >  verbose_                   = iConfig.getUntrackedParameter<bool>("verbose",false);
78 >  realData_                  = iConfig.getUntrackedParameter<bool>("realData",false);
79 >  analyze_                   = iConfig.getUntrackedParameter<bool>("analyzeOnTheFly",false);
80 >  saventuple_                = iConfig.getUntrackedParameter<bool>("saventuple",false);  
81 >  outputfilename_            = iConfig.getUntrackedParameter<string>("OutputFileName");
82 >  ntrkdiffcut_               = iConfig.getUntrackedParameter<double>("ntrkdiffcut");
83 >  zsigncut_                  = iConfig.getUntrackedParameter<int>("zsigncut");
84 >  nTrkMin_                   = iConfig.getUntrackedParameter<int>("nTrkMin");
85 >  nTrkMax_                   = iConfig.getUntrackedParameter<int>("nTrkMax");
86 >  histoFileName_             = iConfig.getUntrackedParameter<std::string> ("histoFileName");
87  
88    //now do what ever initialization is needed
89    pvinfo.clear();
# Line 110 | Line 112 | PVStudy::PVStudy(const edm::ParameterSet
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");
124 +
125 +    pvtxtree_->Branch("min_zsep",&min_zsep_,"min_zsep/D");
126 +    pvtxtree_->Branch("min_ntrksep",&min_ntrksep_,"min_ntrksep/D");
127 +    pvtxtree_->Branch("min_sumpt2sep",&min_sumpt2sep_,"min_sumpt2sep/D");
128 +
129      //pvtx using splittracks1
130      pvtxtree_->Branch("nrecPV1_spl",&nrecPV1_spl_,"nrecPV1_spl/I");
131 <    pvtxtree_->Branch("nTrkPV1_spl",&nTrkPV1_spl_,"nTrkPV1_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");
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");
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");
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");  
167 >
168      //Fill the variables in the twovtx pair (recvtx1, recvtx2)
169      pvtxtree_->Branch("nrecPV_twovtx",&nrecPV_twovtx_,"nrecPV_twovtx/I");
170      pvtxtree_->Branch("nTrkPV1_spl_twovtx",&nTrkPV1_spl_twovtx_,"nTrkPV1_spl_twovtx[nrecPV_twovtx]/I");
# Line 187 | Line 218 | PVStudy::PVStudy(const edm::ParameterSet
218      }
219    }
220  
190  edm::Service<TFileService> fs;
191  TFileDirectory subDir = fs->mkdir( "Summary" );
192  TFileDirectory subDir1 = fs->mkdir( "Others" );
193  //   TFileDirectory subSubDir = subDir.mkdir( "mySubSubDirectory" );
194
221    setRootStyle();
222 <  
223 <  //Book histograms for track and pvtx parameters for each splitted pvtx
224 <  //i=0 : x (as in unsplit track collection)
225 <  //i=1 : x1_spl_
226 <  //i=2 : x2_spl_
227 <  
228 <  for(int i=0;i<3;i++)
229 <    {  
230 <      string suffix;
231 <      if(i == 0) suffix = "";
232 <      else {
233 <        stringstream ss;
234 <        ss << i;
235 <        suffix =ss.str()+"_spl";  
236 <      }
237 <      h["nTrk"+suffix]   = subDir.make<TH1D>(TString("nTrk"+suffix), TString("Num of rec tracks"+suffix),300,0,300);
238 <      h["trkPt"+suffix]  = subDir.make<TH1D>(TString("trkPt"+suffix), TString("Pt of rec tracks "+suffix),100,0,100);
239 <      h["trkEta"+suffix] = subDir.make<TH1D>(TString("trkEta"+suffix), TString("#eta of rec tracks "+suffix),100,-3,3);
240 <      h["trkPhi"+suffix] = subDir.make<TH1D>(TString("trkPhi"+suffix), TString("#Phi of rec tracks "+suffix),100,-3.2,3.2);
241 <      h["trkDxy"+suffix] = subDir.make<TH1D>(TString("trkDxy"+suffix), TString("Dxy of rec tracks "+suffix),100,-5,5);  
242 <      h["trkDz"+suffix] = subDir.make<TH1D>(TString("trkDz"+suffix), TString("Dz of rec tracks "+suffix),100,-50,50);  
243 <      
218 <      h["nTrkPV"+suffix]   = subDir.make<TH1D>(TString("nTrkPV"+suffix), TString("Num of rec tracks in PV"+suffix),300,0,300);
219 <      h["trkPtPV"+suffix]  = subDir.make<TH1D>(TString("trkPtPV"+suffix), TString("Pt of rec tracks in "+suffix),100,0,100);
220 <      h["trkEtaPV"+suffix] = subDir.make<TH1D>(TString("trkEtaPV"+suffix), TString("#eta of rec tracks in PV"+suffix),100,-3,3);
221 <      h["trkPhiPV"+suffix] = subDir.make<TH1D>(TString("trkPhiPV"+suffix), TString("#Phi of rec tracks in PV"+suffix),100,-3.2,3.2);
222 <      h["trkDxyPV"+suffix] = subDir.make<TH1D>(TString("trkDxyPV"+suffix), TString("Dxy of rec tracks in PV"+suffix),100,-5,5);  
223 <      h["trkDzPV"+suffix] = subDir.make<TH1D>(TString("trkDzPV"+suffix), TString("Dz of rec tracks "+suffix),100,-50,50);
224 <      h["nrecPV"+suffix]   = subDir.make<TH1D>(TString("nrecPV"+suffix), TString("Num of rec pvtx"+suffix),50,0,50);    
225 <      suffix.clear();
222 >
223 >  // Create a root file for histograms
224 >  theFile = new TFile(histoFileName_.c_str(), "RECREATE");
225 >  // make diretories
226 >  theFile->mkdir("Summary");
227 >  theFile->cd();
228 >  theFile->mkdir("Others");
229 >  theFile->cd();
230 >
231 >  // Book Histograms:
232 >  h_pvtrk   = new PVHistograms();
233 >  h_pixvtx  = new PVHistograms();
234 >  h_misc    = new PVHistograms();
235 >  h_summary = new PVHistograms();
236 >  h_others  = new PVHistograms();
237 >
238 >  for(int i=0;i<3;i++) {  
239 >    if(i == 0) h_pvtrk->Init("pvTrk");
240 >    else {
241 >      stringstream ss;
242 >      ss << i;
243 >      h_pvtrk->Init("pvTrk",ss.str(),"spl");
244      }
245 <  h["nrecPVDiff"]     = subDir.make<TH1D>("nrecPVDiff","nrecPV1-nRecPV2",21,-10.5,10.5);
246 <  h["nTrkPVDiff"]     = subDir.make<TH1D>("nTrkPVDiff","nTrkPV1-nTrkPV2",41,-20.5,20.5);
247 <  h["nTrkPVRelDiff"]  = subDir.make<TH1D>("nTrkPVRelDiff","(nTrkPV1-nTrkPV2)/(nTrkPV1+nTrkPV2)",100,-1,1);
248 <  
245 >  }
246 >  h_pixvtx->Init("pixVtx");
247 >  h_misc->Init("misc");
248 >
249 >  // Book MC only plots
250 >  if (!realData_) {
251 >    h_gen = new PVHistograms();
252 >    h_gen->Init("generator");
253 >  }
254 >
255    //Book histograms sensitive to data/mc
256    for (vector<int>::const_iterator it= datamode.begin(); it != datamode.end() ; ++it) {
257      string suffix;
# Line 242 | Line 266 | PVStudy::PVStudy(const edm::ParameterSet
266      case 3: suffix = "_mct";
267        break;
268      }
269 <    h["deltax"+suffix] = subDir.make<TH1D>(TString("deltax"+suffix), TString("x-residual pvtx"+suffix),800,-0.04,0.04);
246 <    h["deltay"+suffix] = subDir.make<TH1D>(TString("deltay"+suffix), TString("y-residual pvtx"+suffix),800,-0.04,0.04);
247 <    h["deltaz"+suffix] = subDir.make<TH1D>(TString("deltaz"+suffix), TString("z-residual pvtx"+suffix),800,-0.04,0.04);
248 <    h["pullx"+suffix]  = subDir.make<TH1D>(TString("pullx"+suffix), TString("x-pull pvtx"+suffix),800,-10,10);
249 <    h["pully"+suffix]  = subDir.make<TH1D>(TString("pully"+suffix), TString("y-pull pvtx"+suffix),800,-10,10);
250 <    h["pullz"+suffix]  = subDir.make<TH1D>(TString("pullz"+suffix), TString("z-pull pvtx"+suffix),800,-10,10);
251 <    h["errPVx"+suffix] = subDir.make<TH1D>(TString("errPVx"+suffix), TString("X"+suffix+" vertex error"),100,0.,0.02);
252 <    h["errPVy"+suffix] = subDir.make<TH1D>(TString("errPVy"+suffix), TString("Y"+suffix+" vertex error"),100,0.,0.02);
253 <    h["errPVz"+suffix] = subDir.make<TH1D>(TString("errPVz"+suffix), TString("Z"+suffix+" vertex error"),100,0.,0.02);
269 >    h_summary->Init("summary",suffix);
270      
271      // Book residual, error and pull histograms for each nTrk bin
272      for (int ntrk=nTrkMin_;ntrk<=nTrkMax_;++ntrk) {
273        stringstream ss;
274        ss << ntrk;
275 <      string suffix2 = suffix+"_"+ss.str();
276 <      h["deltax"+suffix2]  = subDir1.make<TH1D>(TString("deltax"+suffix2), TString("x-residual of pvtx"+suffix),100,-0.02,0.02);
277 <      h["deltax"+suffix2]->GetXaxis()->SetTitle("cm");
262 <      h["deltay"+suffix2]  = subDir1.make<TH1D>(TString("deltay"+suffix2), TString("y-residual of pvtx"+suffix),100,-0.02,0.02);
263 <      h["deltay"+suffix2]->GetXaxis()->SetTitle("cm");
264 <      h["deltaz"+suffix2]  = subDir1.make<TH1D>(TString("deltaz"+suffix2), TString("z-residual of pvtx"+suffix),100,-0.02,0.02);
265 <      h["deltaz"+suffix2]->GetXaxis()->SetTitle("cm");
266 <      h["pullx"+suffix2] = subDir1.make<TH1D>(TString("pullx"+suffix2), TString("x-pull of pvtx"+suffix),100,-10.,10.);
267 <      h["pully"+suffix2] = subDir1.make<TH1D>(TString("pully"+suffix2), TString("y-pull of pvtx"+suffix),100,-10.,10.);
268 <      h["pullz"+suffix2] = subDir1.make<TH1D>(TString("pullz"+suffix2), TString("z-pull of pvtx"+suffix),100,-10.,10.);
269 <      h["errPVx"+suffix2] = subDir1.make<TH1D>(TString("errPVx"+suffix2), TString("X"+suffix+" vertex error"),100,0.,0.02);
270 <      h["errPVy"+suffix2] = subDir1.make<TH1D>(TString("errPVy"+suffix2), TString("Y"+suffix+" vertex error"),100,0.,0.02);
271 <      h["errPVz"+suffix2] = subDir1.make<TH1D>(TString("errPVz"+suffix2), TString("Z"+suffix+" vertex error"),100,0.,0.02);
272 <      suffix2.clear();
273 <    } // End of   for (int ntrk=nTrkMin_;ntrk<=nTrkMax_;++ntrk) {
274 <    
275 >      h_others->Init("others",suffix,ss.str());
276 >    }
277 >
278      // Book residual and pull histograms only when analyzeOntheFly is enabled
279      if(analyze_) {
280 <      h2["resx"+suffix+"_nTrk"]= subDir.make<TH2D>(TString("resx"+suffix+"_nTrk"), TString("x-resolution vs number of tracks in pvtx"+suffix),150,0,150,400,0.,200);
281 <      h2["resx"+suffix+"_nTrk"]->SetMarkerStyle(21);
282 <      h2["resx"+suffix+"_nTrk"]->SetMarkerColor(4);
280 <      h2["resx"+suffix+"_nTrk"]->GetXaxis()->SetTitle("Num of tracks");
281 <      h2["resx"+suffix+"_nTrk"]->GetYaxis()->SetTitle("#mum");
282 <      h2["resy"+suffix+"_nTrk"]= subDir.make<TH2D>(TString("resy"+suffix+"_nTrk"), TString("y-resolution vs number of tracks in pvtx"+suffix),150,0,150,400,0.,200);
283 <      h2["resy"+suffix+"_nTrk"]->SetMarkerStyle(21);
284 <      h2["resy"+suffix+"_nTrk"]->SetMarkerColor(4);
285 <      h2["resy"+suffix+"_nTrk"]->GetXaxis()->SetTitle("Num of tracks");
286 <      h2["resy"+suffix+"_nTrk"]->GetYaxis()->SetTitle("#mum");
287 <      h2["resz"+suffix+"_nTrk"]= subDir.make<TH2D>(TString("resz"+suffix+"_nTrk"), TString("z-resolution vs number of tracks in pvtx"+suffix),150,0,150,400,0.,200);
288 <      h2["resz"+suffix+"_nTrk"]->SetMarkerStyle(21);
289 <      h2["resz"+suffix+"_nTrk"]->SetMarkerColor(4);
290 <      h2["resz"+suffix+"_nTrk"]->GetXaxis()->SetTitle("Num of tracks");
291 <      h2["resz"+suffix+"_nTrk"]->GetYaxis()->SetTitle("#mum");
292 <      h2["pullx"+suffix+"_nTrk"]= subDir.make<TH2D>(TString("pullx"+suffix+"_nTrk"), TString("x-pull vs number of tracks"+suffix),150,0,150,100,0.,2.);
293 <      h2["pullx"+suffix+"_nTrk"]->SetMarkerStyle(21);
294 <      h2["pullx"+suffix+"_nTrk"]->SetMarkerColor(4);
295 <      h2["pullx"+suffix+"_nTrk"]->SetBit(TH1::kCanRebin);
296 <      h2["pullx"+suffix+"_nTrk"]->GetXaxis()->SetTitle("Num of tracks");
297 <      h2["pully"+suffix+"_nTrk"]= subDir.make<TH2D>(TString("pully"+suffix+"_nTrk"), TString("y-pull vs number of tracks"+suffix),150,0,150,100,0.,2.);
298 <      h2["pully"+suffix+"_nTrk"]->SetMarkerStyle(21);
299 <      h2["pully"+suffix+"_nTrk"]->SetMarkerColor(4);
300 <      h2["pully"+suffix+"_nTrk"]->SetBit(TH1::kCanRebin);
301 <      h2["pully"+suffix+"_nTrk"]->GetXaxis()->SetTitle("Num of tracks");
302 <      h2["pullz"+suffix+"_nTrk"]= subDir.make<TH2D>(TString("pullz"+suffix+"_nTrk"), TString("x-pull vs number of tracks"+suffix),150,0,150,100,0.,2.);
303 <      h2["pullz"+suffix+"_nTrk"]->SetMarkerStyle(21);
304 <      h2["pullz"+suffix+"_nTrk"]->SetMarkerColor(4);
305 <      h2["pullz"+suffix+"_nTrk"]->SetBit(TH1::kCanRebin);
306 <      h2["pullz"+suffix+"_nTrk"]->GetXaxis()->SetTitle("Num of tracks");
307 <    } // End of   if(analyze_) {
280 >      h_ana = new PVHistograms();
281 >      h_ana->InitA("analysis",suffix,"nTrk",nTrkMin_,nTrkMax_);
282 >    }
283      suffix.clear();
284 <  } // End of Book histograms sensitive to data/mc  
285 <  // Book MC only plots
311 <  if (!realData_) {  
312 <    h["genPart_T"]      = subDir.make<TH1D>("genPart_T","t component of gen particles",300,-0.5,0.5);
313 <    h["genPart_T"]->GetXaxis()->SetTitle("t (nanosecond)");
314 <    h["genPart_cT"]     = subDir.make<TH1D>("genPart_cT","ct component of gen particles",300,-150.,150.);
315 <    h["genPart_cT"]->GetXaxis()->SetTitle("ct (mm)");    
316 <    h["nsimPV"]         = subDir.make<TH1D>("nsimPV","Num of sim PV",51,-0.5,50.5);  
317 <  }
284 >  } // End of Book histograms sensitive to data/mc    
285 >
286   }
287  
288   PVStudy::~PVStudy()
289   {
290 <
291 <  // do anything here that needs to be done at desctruction time
292 <  // (e.g. close files, deallocate resources etc.)
290 >  theFile->cd();
291 >  theFile->cd("Summary");
292 >  h_pvtrk->Save();
293 >  h_pixvtx->Save();
294 >  h_gen->Save();
295 >  h_misc->Save();
296 >  h_summary->Save();
297 >  if (analyze_)
298 >    h_ana->Save();
299 >  theFile->cd();
300 >  theFile->cd("Others");
301 >  h_others->Save();
302  
303 +  theFile->Close();
304   }
305  
306   void PVStudy::setRootStyle() {
# Line 383 | Line 361 | std::vector<PVStudy::simPrimaryVertex> P
361          if( ((*mother)->status() == 3) && (!mv)) {
362            //        std::cout << "npv= " << npv << std::endl;
363            if (npv == 0) {
364 <            h["genPart_cT"]->Fill(pos.t()); // mm
365 <            h["genPart_T"]->Fill(pos.t()/299.792458); // ns
364 >            h_gen->Fill1d("genPart_cT", pos.t()); // mm
365 >            h_gen->Fill1d("genPart_T", pos.t()/299.792458); // ns
366            }
367            npv++;
368          }
# Line 451 | Line 429 | std::vector<PVStudy::simPrimaryVertex> P
429              }
430            }
431          }
432 <      }
433 <    }
432 >      }//loop MC vertices daughters
433 >    }//loop MC vertices
434    }
435    return simpv;
436   }
# Line 485 | Line 463 | std::vector<PVStudy::simPrimaryVertex> P
463                ptsq+=(*daughter)->momentum().perp()*(*daughter)->momentum().perp();
464              }
465            }
466 <          primary =  ( gv->position().t()==0);
466 >          //primary =  ( gv->position().t()==0 );
467 >          primary = true;
468 >          h_gen->Fill1d("genPart_cT", gv->position().t()); // mm
469 >          h_gen->Fill1d("genPart_T", gv->position().t()/299.792458); // ns
470 >
471            //resonance= ( gp->mother() && isResonance(gp->mother()));  // in CLHEP/HepMC days
472            // no more mother pointer in the improved HepMC GenParticle
473            resonance= ( isResonance(*(gp->production_vertex()->particles_in_const_begin())));
# Line 555 | Line 537 | PVStudy::analyze(const edm::Event& iEven
537    } else {
538      cout << "trackCollection cannot be found -> using empty collection of same type." <<endl;
539    }
540 +
541    //splitTrackCollection1
542    static const reco::TrackCollection s_empty_splitTrackColl1;
543    const reco::TrackCollection *splitTrackColl1 = &s_empty_splitTrackColl1;
# Line 565 | Line 548 | PVStudy::analyze(const edm::Event& iEven
548    } else {
549      cout << "splitTrackCollection1 cannot be found -> using empty collection of same type." <<endl;
550    }
551 +
552    //splitTrackCollection2
553    static const reco::TrackCollection s_empty_splitTrackColl2;
554    const reco::TrackCollection *splitTrackColl2 = &s_empty_splitTrackColl2;
# Line 575 | Line 559 | PVStudy::analyze(const edm::Event& iEven
559    } else {
560      cout << "splitTrackCollection2 cannot be found -> using empty collection of same type." <<endl;
561    }
562 <  
562 >
563 >
564 >  //=======================================================
565 >  // Fill trackparameters of the input tracks to pvtx fitter
566 >  //=======================================================
567 >  if(verbose_)
568 >    cout<<"Start filling track parameters of the input tracks to pvtx fitter."<<endl;
569 >  //fillTrackHisto(const reco::TrackCollection *trackColl, int datatype)
570 >  // datatype: unsplittracks (0); splittracks1 (1);  splittracks2 (2);
571 >  fillTrackHisto(trackColl, 0);
572 >  fillTrackHisto(splitTrackColl1, 1);
573 >  fillTrackHisto(splitTrackColl2, 2);
574 >  if(verbose_)
575 >    cout<<"End filling track parameters of the input tracks to pvtx fitter."<<endl;
576 >
577    //=======================================================
578    // PVTX accessors
579    //=======================================================
# Line 589 | Line 587 | PVStudy::analyze(const edm::Event& iEven
587    } else {
588      cout << "vertexCollection cannot be found -> using empty collection of same type." <<endl;
589    }
590 +
591    //splitVertexCollection1
592    static const reco::VertexCollection s_empty_splitVertexColl1;
593    const reco::VertexCollection *splitVertexColl1 = &s_empty_splitVertexColl1;
# Line 598 | Line 597 | PVStudy::analyze(const edm::Event& iEven
597      splitVertexColl1 = splitVertexCollection1Handle.product();
598    } else {
599      cout << "splitVertexCollection1 cannot be found -> using empty collection of same type." <<endl;
600 <  }
600 >  }
601 >
602    //splitVertexCollection2
603    static const reco::VertexCollection s_empty_splitVertexColl2;
604    const reco::VertexCollection *splitVertexColl2 = &s_empty_splitVertexColl2;
# Line 609 | Line 609 | PVStudy::analyze(const edm::Event& iEven
609    } else {
610      cout << "splitVertexCollection2 cannot be found -> using empty collection of same type." <<endl;
611    }
612 <
612 >  
613    if(verbose_) cout<<"End accessing the track and vertex collections"<<endl;
614    
615    //=======================================================
# Line 631 | Line 631 | PVStudy::analyze(const edm::Event& iEven
631    }catch(const Exception&){
632      cout << "Some problem occurred with the particle data table. This may not work !." <<endl;
633    }
634 +  
635 +  //=======================================================
636 +  // GET pixelVertices
637 +  //=======================================================
638 +  static const reco::VertexCollection s_empty_pixelVertexColl;
639 +  const reco::VertexCollection *pixelVertexColl = &s_empty_pixelVertexColl;
640 +  edm::Handle<reco::VertexCollection>  pixelVertexCollectionHandle;
641 +  iEvent.getByLabel(pixelVertexCollectionTag_, pixelVertexCollectionHandle);
642 +  if( iEvent.getByLabel(pixelVertexCollectionTag_, pixelVertexCollectionHandle)) {
643 +    pixelVertexColl = pixelVertexCollectionHandle.product();
644 +  } else {
645 +    cout << "pixelVertexCollection cannot be found. -> using empty collection of same type." <<endl;
646 +  }
647  
648    //=======================================================
649 <  // Fill trackparameters of the input tracks to pvtx fitter
649 >  // Fill pixelVertices related histograms
650    //=======================================================
651 <  if(verbose_)
652 <    cout<<"Start filling track parameters of the input tracks to pvtx fitter."<<endl;
653 <  //fillTrackHisto(const reco::TrackCollection *trackColl, int datatype)
654 <  // datatype: unsplittracks (0); splittracks1 (1);  splittracks2 (2);
655 <  fillTrackHisto(trackColl, 0);
656 <  fillTrackHisto(splitTrackColl1, 1);
657 <  fillTrackHisto(splitTrackColl2, 2);
658 <  if(verbose_)
659 <    cout<<"End filling track parameters of the input tracks to pvtx fitter."<<endl;
651 >  nrecPV_pxlpvtx_ = int (pixelVertexColl->size());
652 >  if(pixelVertexColl->size()>0 && pixelVertexColl->begin()->isValid() && !(pixelVertexColl->begin()->isFake())) {
653 >    //fillTrackHistoInPV(const reco::VertexCollection *vertexColl, int datatype, bool fillHisto, bool fillNtuple) {
654 >    fillTrackHistoInPV(pixelVertexColl, 4, false, true);
655 >    h_pixvtx->Fill1d("dzErr_pxlpvtx", pixelVertexColl->begin()->zError());    
656 >    // Get the dZ error of the tracks in the leading pixelVertexColl
657 >    for(reco::Vertex::trackRef_iterator t = (pixelVertexColl->begin())->tracks_begin();
658 >        t!= (pixelVertexColl->begin())->tracks_end(); t++) {
659 >      
660 >      if ( (**t).charge() < -1 || (**t).charge() > 1 ) {
661 >        //        h_pvtrk->Fill1d("trkPtPV", 0.);
662 >      }
663 >      else
664 >        h_pixvtx->Fill1d("trkdzErr_pxlpvtx", (**t).dzError());
665 >    }
666 >    // Fill the track->dz() in the PV difference from first pixelpvtx
667 >    for(reco::Vertex::trackRef_iterator t = (vertexColl->begin())->tracks_begin();
668 >        t!= (vertexColl->begin())->tracks_end(); t++) {
669 >      // illegal charge
670 >      if ( (**t).charge() < -1 || (**t).charge() > 1 ) {
671 >        //        h_pvtrk->Fill1d("trkPtPV", 0.);
672 >      }
673 >      else {
674 >        if(pixelVertexColl->size()>0) {
675 >          h_pixvtx->Fill1d("trkdz_pxlpvtxdz", (**t).dz() -  pixelVertexColl->begin()->z());
676 >          h_pixvtx->Fill1d("trkdz_pxlpvtxdz_pxlpvtxdzerr", fabs((**t).dz() -  pixelVertexColl->begin()->z())/pixelVertexColl->begin()->zError());
677 >          h_pixvtx->Fill1d("trkdz_pxlpvtxdz_trkdzerr", fabs((**t).dz() -  pixelVertexColl->begin()->z())/(**t).dzError());
678 >          h_pixvtx->Fill1d("trkdzErr_pvtx", (**t).dzError());
679 >        }
680 >      }
681 >    }
682 >  }
683  
684    //=======================================================
685 <  // Fill reconstructed vertices ootb
685 >  // Fill number of reconstructed vertices
686    //=======================================================
687 +
688    if(verbose_)  {
689      cout<<"PVStudy::analyze() Printing vertexCollection: "<<endl;
690      printRecVtxs(vertexCollectionHandle);
# Line 657 | Line 694 | PVStudy::analyze(const edm::Event& iEven
694      printRecVtxs(splitVertexCollection2Handle);
695      cout<<"Start filling pvtx parameters."<<endl;
696    }
697 <    
697 >  
698    nrecPV_ = int (vertexColl->size());
699    nrecPV1_spl_ = int (splitVertexColl1->size());
700    nrecPV2_spl_ = int (splitVertexColl2->size());
701  
702 <  h["nrecPV"]->Fill(nrecPV_);
703 <  h["nrecPV1_spl"]->Fill(nrecPV1_spl_);
704 <  h["nrecPV2_spl"]->Fill(nrecPV2_spl_);
705 <  h["nrecPVDiff"]->Fill(double(nrecPV1_spl_)-double(nrecPV2_spl_));
706 <    
702 >  h_pvtrk->Fill1d("nrecPV", nrecPV_);
703 >  h_pvtrk->Fill1d("nrecPV1_spl", nrecPV1_spl_);
704 >  h_pvtrk->Fill1d("nrecPV2_spl", nrecPV2_spl_);
705 >  h_misc->Fill1d("nrecPVDiff", double(nrecPV1_spl_)-double(nrecPV2_spl_));
706 >  
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 <  fillTrackHistoInPV(vertexColl, 0, true, true);
713 <  fillTrackHistoInPV(splitVertexColl1, 1, false, true);
714 <  fillTrackHistoInPV(splitVertexColl2, 2, false, true);
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 >    //cout<<"recx_[0] = "<< recx_[0] << "recx_pxlpvtx_[0] = "<< recx_pxlpvtx_[0]<<endl;  
730 >    //cout<<"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 >  }
735 >  
736 >
737 >
738 >  //==========================================================
739 >  // Fill secondary/primary min separations for multi-vertices
740 >  //==========================================================
741 >  
742 >  if(vertexColl->size()>1 && vertexColl->begin()->isValid() && !(vertexColl->begin()->isFake()) ) {  
743 >
744 >    double min_xsep = 9999.0;
745 >    double min_ysep = 9999.0;
746 >    double min_zsep = 9999.0;    
747 >    double min_xsep_sign = 9999.0;
748 >    double min_ysep_sign = 9999.0;
749 >    double min_zsep_sign = 9999.0;
750 >    double min_ntrksep = 9999.0;
751 >    double min_sumpt2sep = 9999999.0;
752 >    
753 >    if(verbose_) cout<<"leading pvtx z = "<< vertexColl->begin()->z() <<endl;
754 >
755 >    // Looping through the secondary vertices to find the mininum separation to the primary
756 >    for(reco::VertexCollection::const_iterator v=vertexColl->begin() + 1 ;
757 >        v!=vertexColl->end(); ++v) {  
758 >      if(v->isValid() && ! v->isFake()) {
759 >        // xsep
760 >        if(fabs(v->x()- vertexColl->begin()->x())<min_xsep)
761 >          min_xsep = fabs(v->x()- vertexColl->begin()->x());
762 >        // ysep
763 >        if(fabs(v->y()- vertexColl->begin()->y())<min_ysep)
764 >          min_ysep = fabs(v->y()- vertexColl->begin()->y());
765 >        // zsep
766 >        if(fabs(v->z()- vertexColl->begin()->z())<min_zsep)
767 >          min_zsep = fabs(v->z()- vertexColl->begin()->z());
768 >        // xsep_sign
769 >        double xsep_sign = fabs(v->x()- vertexColl->begin()->x())/max(v->xError(), vertexColl->begin()->xError());
770 >        if(xsep_sign < min_xsep_sign)
771 >          min_xsep_sign = xsep_sign;
772 >        // ysep_sign
773 >        double ysep_sign = fabs(v->y()- vertexColl->begin()->y())/max(v->yError(), vertexColl->begin()->yError());
774 >        if(ysep_sign < min_ysep_sign)
775 >          min_ysep_sign = ysep_sign;
776 >        // zsep_sign
777 >        double zsep_sign = fabs(v->z()- vertexColl->begin()->z())/max(v->zError(), vertexColl->begin()->zError());
778 >        if(zsep_sign < min_zsep_sign)
779 >          min_zsep_sign = zsep_sign;
780 >        // ntrksep
781 >        if( (double(vertexColl->begin()->tracksSize()) - double(v->tracksSize())) < min_ntrksep)
782 >          min_ntrksep = double(vertexColl->begin()->tracksSize()) - double(v->tracksSize());
783 >        // sumpt2sep
784 >        if(fabs(sumPtSquared(*v) - sumPtSquared(*(vertexColl->begin()))) < min_sumpt2sep)
785 >          min_sumpt2sep = fabs(sumPtSquared(*v) - sumPtSquared(*(vertexColl->begin())));
786 >      }
787 >    }
788 >    h_misc->Fill1d("min_xsep", min_xsep);
789 >    h_misc->Fill1d("min_ysep", min_ysep);
790 >    h_misc->Fill1d("min_zsep", min_zsep);
791 >    h_misc->Fill1d("min_xsep_sign", min_xsep_sign);
792 >    h_misc->Fill1d("min_ysep_sign", min_ysep_sign);
793 >    h_misc->Fill1d("min_zsep_sign", min_zsep_sign);
794 >    h_misc->Fill1d("min_ntrksep", min_ntrksep);
795 >    h_misc->Fill1d("min_sumpt2sep", min_sumpt2sep);
796 >
797 >    min_zsep_ = min_zsep;
798 >    min_ntrksep_ = min_ntrksep;
799 >    min_sumpt2sep_ = min_sumpt2sep;
800 >
801 >
802 >  } // End of  if(vertexColl->size()>1) {
803 >  
804    if(verbose_)
805      cout<<"End filling pvtx parameters."<<endl;
806    
# Line 690 | Line 820 | PVStudy::analyze(const edm::Event& iEven
820    reco::VertexCollection *matchedVertexColl2 = &s_empty_matchedVertexColl2;
821    matchedVertexColl2->reserve(splitVertexColl2->size());
822    
823 <  for (size_t ivtx = 0; ivtx<splitVertexCollection1Handle->size(); ++ivtx) {
823 >  bool stopmatching = false;
824 >  for (size_t ivtx = 0; ivtx<splitVertexCollection1Handle->size() && !stopmatching; ++ivtx) {
825      reco::VertexRef recvtx1(splitVertexCollection1Handle, ivtx);
826 <    for (size_t jvtx = ivtx; jvtx < splitVertexCollection2Handle->size(); ++jvtx) {
826 >    if( !(recvtx1->isValid()) || recvtx1->isFake()) break;
827 >    for (size_t jvtx = ivtx; jvtx < splitVertexCollection2Handle->size(); ++jvtx) {    
828 >      //------------------------------------------------------------------------
829 >      //== If only considering matching the first vertex of the splitVertexColl
830 >      //------------------------------------------------------------------------
831 >      if (ivtx!=0 || jvtx!=0) { stopmatching = true; break; }
832        reco::VertexRef recvtx2(splitVertexCollection2Handle, jvtx);
833 +      if( !(recvtx2->isValid()) || recvtx2->isFake()) break;
834 +      if(verbose_) {
835 +        cout<<"Matching splitVertexColl1: #  "<< ivtx<< " and splitVertexColl1: # "<<jvtx<<endl;
836 +        cout<<"recvtx1->z() = " << recvtx1->z() << "+/- "<< recvtx1->zError() << "." << endl;
837 +        cout<<"recvtx2->z() = " << recvtx2->z() << "+/- "<< recvtx2->zError() << "." << endl;
838 +      }
839        if(matchVertex(recvtx1, recvtx2, zsigncut_)) {
840          if(verbose_) {
841 <          cout<<"The two splitted vertices match in Z: "<<endl;
700 <          cout<<"recvtx1->z() = " << recvtx1->z() << "+/- "<< recvtx1->zError() << "." << endl;
701 <          cout<<"recvtx2->z() = " << recvtx2->z() << "+/- "<< recvtx2->zError() << "." << endl;
841 >          cout<<"The two splitted vertices match in Z. "<<endl;
842          }
843          int nTrkPV1 = recvtx1->tracksSize();
844          int nTrkPV2 = recvtx2->tracksSize();    
845          double ntrkreldiff = double(nTrkPV1-nTrkPV2)/double(nTrkPV1+nTrkPV2);
846 <        h["nTrkPVDiff"]->Fill(nTrkPV1-nTrkPV2);
847 <        h["nTrkPVRelDiff"]->Fill(ntrkreldiff);
846 >        h_misc->Fill1d("nTrkPVDiff", nTrkPV1-nTrkPV2);
847 >        h_misc->Fill1d("nTrkPVRelDiff", ntrkreldiff);
848          if(fabs(ntrkreldiff)<ntrkdiffcut_) {    
849            if(verbose_) {
850              cout<<"(nTrkPV1-nTrkPV2)/(nTrkPV1+nTrkPV2) = " << ntrkreldiff<<endl;
851            }
852            matchedVertexColl1->push_back(*recvtx1);
853            matchedVertexColl2->push_back(*recvtx2);
854 <          continue;
854 >          stopmatching = true; // stop the matching when the first match is found!
855 >          break;
856          }
857          else
858            cout<<"WARNING: (nTrkPV1-nTrkPV2)/(nTrkPV1+nTrkPV2) = " << ntrkreldiff<<", exceeding cut "<<ntrkdiffcut_<<endl;
# Line 731 | Line 872 | PVStudy::analyze(const edm::Event& iEven
872    if(verbose_) {
873      cout<<"Begin analyzing the matchedVertexColl with size = "<< matchedVertexColl1->size()<< endl;
874    }    
734
875    
876    // Compare the reconstructed vertex position and calculate resolution/pulls
877    if(matchedVertexColl1->size() != 0 && matchedVertexColl2->size() != 0) {  
# Line 753 | Line 893 | PVStudy::analyze(const edm::Event& iEven
893        ferror_[2] = sqrt(pow(v1->zError(),2)+pow(v2->zError(),2))/sqrt(2);
894        fntrk_ =  (v1->tracksSize()+v2->tracksSize())/2;
895        
896 <      h["deltax"]->Fill( fres_[0] );
897 <      h["deltay"]->Fill( fres_[1] );
898 <      h["deltaz"]->Fill( fres_[2] );
899 <      h["pullx"]->Fill( fres_[0]/ferror_[0]);
900 <      h["pully"]->Fill( fres_[1]/ferror_[1]);
901 <      h["pullz"]->Fill( fres_[2]/ferror_[2]);
902 <      h["errPVx"]->Fill( ferror_[0] );
903 <      h["errPVy"]->Fill( ferror_[1] );
904 <      h["errPVz"]->Fill( ferror_[2] );
896 >      h_summary->Fill1d("deltax", fres_[0] );
897 >      h_summary->Fill1d("deltay", fres_[1] );
898 >      h_summary->Fill1d("deltaz", fres_[2] );
899 >      h_summary->Fill1d("pullx", fres_[0]/ferror_[0]);
900 >      h_summary->Fill1d("pully", fres_[1]/ferror_[1]);
901 >      h_summary->Fill1d("pullz", fres_[2]/ferror_[2]);
902 >      h_summary->Fill1d("errPVx", ferror_[0] );
903 >      h_summary->Fill1d("errPVy", ferror_[1] );
904 >      h_summary->Fill1d("errPVz", ferror_[2] );
905        pvinfo.push_back(PVStudy::PVInfo(res(fres_[0], fres_[1], fres_[2]),
906                                         error(ferror_[0], ferror_[1], ferror_[2]),
907                                         fntrk_));
# Line 816 | Line 956 | PVStudy::analyze(const edm::Event& iEven
956        std::vector<simPrimaryVertex> simpv;
957        simpv=getSimPVs(evtMC,"");
958        //     simpv=getSimPVs(evtMC, simVtxs, simTrks);
959 <      h["nsimPV"]->Fill(simpv.size());
959 >      h_gen->Fill1d("nsimPV", simpv.size());
960        nsimPV_ = simpv.size();
961  
962        int isimpv = 0;
# Line 869 | Line 1009 | PVStudy::endJob() {
1009      }
1010      if(analyze_) {
1011        for (int ntrk=nTrkMin_;ntrk<=nTrkMax_;++ntrk) {
1012 +        cout << "ntrk = " << ntrk << endl;
1013          PVStudy::PVAnaInfo ipv = GetPVAnaInfo(ntrk,*it);
1014 <        if ( ipv.res_.x() > 0 ) h2["resx"+suffix+"_nTrk"]->Fill(ntrk,ipv.res_.x(), 1.);
1015 <        if ( ipv.res_.y() > 0 ) h2["resy"+suffix+"_nTrk"]->Fill(ntrk,ipv.res_.y(), 1.);
1016 <        if ( ipv.res_.z() > 0 ) h2["resz"+suffix+"_nTrk"]->Fill(ntrk,ipv.res_.z(), 1.);
1017 <        if ( ipv.pull_.x() > 0 ) h2["pullx"+suffix+"_nTrk"]->Fill(ntrk,ipv.pull_.x(), 1.);
1018 <        if ( ipv.pull_.y() > 0 ) h2["pully"+suffix+"_nTrk"]->Fill(ntrk,ipv.pull_.y(), 1.);
1019 <        if ( ipv.pull_.z() > 0 ) h2["pullz"+suffix+"_nTrk"]->Fill(ntrk,ipv.pull_.z(), 1.);
1014 >        if ( ipv.res_.x() > 0 ) h_ana->Fill2d("resx"+suffix+"_nTrk", ntrk,ipv.res_.x(), 1.);
1015 >        if ( ipv.res_.y() > 0 ) h_ana->Fill2d("resy"+suffix+"_nTrk", ntrk,ipv.res_.y(), 1.);
1016 >        if ( ipv.res_.z() > 0 ) h_ana->Fill2d("resz"+suffix+"_nTrk", ntrk,ipv.res_.z(), 1.);
1017 >        if ( ipv.pull_.x() > 0 ) h_ana->Fill2d("pullx"+suffix+"_nTrk", ntrk,ipv.pull_.x(), 1.);
1018 >        if ( ipv.pull_.y() > 0 ) h_ana->Fill2d("pully"+suffix+"_nTrk", ntrk,ipv.pull_.y(), 1.);
1019 >        if ( ipv.pull_.z() > 0 ) h_ana->Fill2d("pullz"+suffix+"_nTrk", ntrk,ipv.pull_.z(), 1.);
1020        }
1021      }
1022      suffix.clear();
# Line 993 | Line 1134 | void PVStudy::fillHisto(res r, error er,
1134    
1135    if (ntk < nTrkMin_ || ntk > nTrkMax_ ) return;
1136    // Fill histograms of res and pull of ntk
1137 <  h["deltax"+suffix]->Fill(r.x());
1138 <  h["deltay"+suffix]->Fill(r.y());
1139 <  h["deltaz"+suffix]->Fill(r.z());
1140 <  h["pullx"+suffix]->Fill(r.x()/er.x());
1141 <  h["pully"+suffix]->Fill(r.y()/er.y());
1142 <  h["pullz"+suffix]->Fill(r.z()/er.z());  
1143 <  h["errPVx"+suffix]->Fill(er.x());
1144 <  h["errPVy"+suffix]->Fill(er.y());
1145 <  h["errPVz"+suffix]->Fill(er.z());
1137 >  h_others->Fill1d("deltax"+suffix, r.x());
1138 >  h_others->Fill1d("deltay"+suffix, r.y());
1139 >  h_others->Fill1d("deltaz"+suffix, r.z());
1140 >  h_others->Fill1d("pullx"+suffix, r.x()/er.x());
1141 >  h_others->Fill1d("pully"+suffix, r.y()/er.y());
1142 >  h_others->Fill1d("pullz"+suffix, r.z()/er.z());  
1143 >  h_others->Fill1d("errPVx"+suffix, er.x());
1144 >  h_others->Fill1d("errPVy"+suffix, er.y());
1145 >  h_others->Fill1d("errPVz"+suffix, er.z());
1146   }
1147  
1148   PVStudy::PVAnaInfo PVStudy::GetPVAnaInfo(int ntk, int datamode) {
# Line 1021 | Line 1162 | PVStudy::PVAnaInfo PVStudy::GetPVAnaInfo
1162      for ( int i = 0; i < 6; ++i) {
1163        switch (i) {
1164        case 0:
1165 <        fit(h["deltax"+suffix], fpar);
1165 >        if (!h_others->ReadHisto1D("deltax"+suffix)) break;
1166 >        fit(h_others->ReadHisto1D("deltax"+suffix), fpar);
1167          data[i] = fpar[0]*cm2um;
1168          break;
1169        case 1:
1170 <        fit(h["deltay"+suffix], fpar);
1170 >        if (!h_others->ReadHisto1D("deltay"+suffix)) break;
1171 >        fit(h_others->ReadHisto1D("deltay"+suffix), fpar);
1172          data[i] = fpar[0]*cm2um;
1173          break;
1174        case 2:
1175 <        fit(h["deltaz"+suffix], fpar);
1175 >        if (!h_others->ReadHisto1D("deltaz"+suffix)) break;
1176 >        fit(h_others->ReadHisto1D("deltaz"+suffix), fpar);
1177          data[i] = fpar[0]*cm2um;
1178          break;
1179        case 3:
1180 <        fit(h["pullx"+suffix], fpar);
1180 >        if (!h_others->ReadHisto1D("pullx"+suffix)) break;
1181 >        fit(h_others->ReadHisto1D("pullx"+suffix), fpar);
1182          data[i] = fpar[0];
1183          break;
1184        case 4:
1185 <        fit(h["pully"+suffix], fpar);
1185 >        if (!h_others->ReadHisto1D("pully"+suffix)) break;
1186 >        fit(h_others->ReadHisto1D("pully"+suffix), fpar);
1187          data[i] = fpar[0];
1188          break;
1189        case 5:
1190 <        fit(h["pullz"+suffix], fpar);
1190 >        if (!h_others->ReadHisto1D("pullz"+suffix)) break;
1191 >        fit(h_others->ReadHisto1D("pullz"+suffix), fpar);
1192          data[i] = fpar[0];
1193          break;
1194        }
# Line 1060 | Line 1207 | PVStudy::PVAnaInfo PVStudy::GetPVAnaInfo
1207                              ntk);
1208   }
1209  
1210 < void PVStudy::fit(TH1 *&hdist, double fitPars[]){
1210 > void PVStudy::fit(TH1 *hdist, double fitPars[]){
1211    TAxis *axis0 = hdist->GetXaxis();
1212    int nbin = axis0->GetLast();
1213    double nOF = hdist->GetBinContent(nbin+1);
# Line 1113 | Line 1260 | void PVStudy::fillTrackHisto(const reco:
1260    if(datatype == 1) suffix = "1_spl"; // for splittracks 1
1261    if(datatype == 2) suffix = "2_spl"; // for splittracks 2
1262    
1263 <  h["nTrk"+suffix]->Fill(trackColl->size());
1263 >  h_pvtrk->Fill1d("nTrk"+suffix, trackColl->size());
1264    for(reco::TrackCollection::const_iterator itTrack = trackColl->begin();
1265        itTrack != trackColl->end();                      
1266        ++itTrack) {
1267 <    h["trkPt"+suffix]->Fill(itTrack->pt());
1268 <    h["trkDxy"+suffix]->Fill(itTrack->dxy());
1269 <    h["trkDz"+suffix]->Fill(itTrack->dz());
1270 <    h["trkEta"+suffix]->Fill(itTrack->eta());
1271 <    h["trkPhi"+suffix]->Fill(itTrack->phi());
1267 >    h_pvtrk->Fill1d("trkPt"+suffix, itTrack->pt());
1268 >    h_pvtrk->Fill1d("trkDxy"+suffix, itTrack->dxy());
1269 >    h_pvtrk->Fill1d("trkDz"+suffix, itTrack->dz());
1270 >    h_pvtrk->Fill1d("trkEta"+suffix, itTrack->eta());
1271 >    h_pvtrk->Fill1d("trkPhi"+suffix, itTrack->phi());
1272    }
1273   }
1274  
# Line 1135 | Line 1282 | void PVStudy::fillTrackHistoInPV(const r
1282        v!=vertexColl->end(); ++v, ++ivtx) {  
1283      if(fillNtuple) {
1284        if(datatype == 0) {
1285 <        nTrkPV_[ivtx] = v->tracksSize();
1285 >        nTrkPV_[ivtx] = v->tracksSize();        
1286 >        sumptsq_[ivtx] = sumPtSquared(*v);
1287 >        isValid_[ivtx] = v->isValid();
1288 >        isFake_[ivtx] = v->isFake();    
1289          recx_[ivtx] = v->x();
1290          recy_[ivtx] = v->y();
1291          recz_[ivtx] = v->z();
# Line 1144 | Line 1294 | void PVStudy::fillTrackHistoInPV(const r
1294          recz_err_[ivtx] = v->zError();
1295        }
1296        if(datatype == 1) {
1297 <        nTrkPV1_spl_[ivtx] = v->tracksSize();  
1297 >        nTrkPV1_spl_[ivtx] = v->tracksSize();  
1298 >        sumptsq1_spl_[ivtx] = sumPtSquared(*v);        
1299 >        isValid1_spl_[ivtx] = v->isValid();
1300 >        isFake1_spl_[ivtx] = v->isFake();      
1301          recx1_spl_[ivtx] = v->x();
1302          recy1_spl_[ivtx] = v->y();
1303          recz1_spl_[ivtx] = v->z();
1304          recx1_err_spl_[ivtx] = v->xError();
1305          recy1_err_spl_[ivtx] = v->yError();
1306          recz1_err_spl_[ivtx] = v->zError();
1307 +
1308        }
1309        if(datatype == 2) {
1310 <        nTrkPV2_spl_[ivtx] = v->tracksSize();  
1310 >        nTrkPV2_spl_[ivtx] = v->tracksSize();  
1311 >        sumptsq2_spl_[ivtx] = sumPtSquared(*v);        
1312 >        isValid2_spl_[ivtx] = v->isValid();
1313 >        isFake2_spl_[ivtx] = v->isFake();
1314          recx2_spl_[ivtx] = v->x();
1315          recy2_spl_[ivtx] = v->y();
1316          recz2_spl_[ivtx] = v->z();
# Line 1161 | Line 1318 | void PVStudy::fillTrackHistoInPV(const r
1318          recy2_err_spl_[ivtx] = v->yError();
1319          recz2_err_spl_[ivtx] = v->zError();
1320        }
1321 +      if(datatype == 4) { // for pixelVertices
1322 +        nTrkPV_pxlpvtx_[ivtx] = v->tracksSize();        
1323 +        sumptsq_pxlpvtx_[ivtx] = sumPtSquared(*v);      
1324 +        isValid_pxlpvtx_[ivtx] = v->isValid();
1325 +        isFake_pxlpvtx_[ivtx] = v->isFake();
1326 +        recx_pxlpvtx_[ivtx] = v->x();
1327 +        recy_pxlpvtx_[ivtx] = v->y();
1328 +        recz_pxlpvtx_[ivtx] = v->z();
1329 +        recx_err_pxlpvtx_[ivtx] = v->xError();
1330 +        recy_err_pxlpvtx_[ivtx] = v->yError();
1331 +        recz_err_pxlpvtx_[ivtx] = v->zError();
1332 +      }
1333      }
1334 <
1335 <    if(fillHisto) {
1336 <      h["nTrkPV"+suffix]->Fill(v->tracksSize());
1337 <      try {
1338 <        for(reco::Vertex::trackRef_iterator t = v->tracks_begin();
1339 <            t!=v->tracks_end(); t++) {
1340 <          // illegal charge
1341 <          if ( (**t).charge() < -1 || (**t).charge() > 1 ) {
1342 <            //    h["trkPtPV"]->Fill(0.);
1343 <          }
1344 <          else {
1345 <            h["trkPtPV"+suffix]->Fill((**t).pt());
1346 <            h["trkDxyPV"+suffix]->Fill((**t).dxy());
1347 <            h["trkDzPV"+suffix]->Fill((**t).dz());
1348 <            h["trkEtaPV"+suffix]->Fill((**t).eta());
1349 <            h["trkPhiPV"+suffix]->Fill((**t).phi());
1350 <          }
1334 >  }
1335 >  // For histogram only fill the leading pvtx parameters
1336 >  if(fillHisto) {
1337 >    h_pvtrk->Fill1d("nTrkPV"+suffix, vertexColl->begin()->tracksSize());
1338 >    try {
1339 >      for(reco::Vertex::trackRef_iterator t = vertexColl->begin()->tracks_begin();
1340 >          t!=vertexColl->begin()->tracks_end(); t++) {
1341 >        // illegal charge
1342 >        if ( (**t).charge() < -1 || (**t).charge() > 1 ) {
1343 >          //      h_pvtrk->Fill1d("trkPtPV", 0.);
1344 >        }
1345 >        else {
1346 >          h_pvtrk->Fill1d("trkPtPV"+suffix, (**t).pt());
1347 >          h_pvtrk->Fill1d("trkDxyPV"+suffix, (**t).dxy());
1348 >          h_pvtrk->Fill1d("trkDzPV"+suffix, (**t).dz());
1349 >          h_pvtrk->Fill1d("trkEtaPV"+suffix, (**t).eta());
1350 >          h_pvtrk->Fill1d("trkPhiPV"+suffix, (**t).phi());
1351          }
1352        }
1353 <      catch (...) {
1354 <        // exception thrown when trying to use linked track
1355 <        //        h["trkPtPV"]->Fill(0.);
1356 <      }
1353 >    }
1354 >    catch (...) {
1355 >      // exception thrown when trying to use linked track
1356 >      //          h_pvtrk->Fill1d("trkPtPV", 0.);
1357      }
1358    }
1359 +  
1360   }
1361  
1362   void PVStudy::fillMCHisto(std::vector<simPrimaryVertex>::iterator vsim, int isimpv, const reco::VertexCollection *vtxColl, int datatype)
# Line 1219 | Line 1389 | void PVStudy::fillMCHisto(std::vector<si
1389        cout << "sim primary vertex x = " << vsim->x << "; y = " << vsim->y << "; z = " << vsim->z <<  endl;
1390        cout << "Is matched? " << (matchVertex(*vsim,*vrec)?"Yes":"No") << endl;
1391      }
1392 <    if ( matchVertex(*vsim,*vrec) ) {
1392 >    if ( matchVertex(*vsim,*vrec) && vrec->isValid() && !vrec->isFake() ) {
1393        vsim->recVtx=&(*vrec);
1394        
1395        // if the matching critera are fulfilled, accept the rec-vertex that is closest in z
# Line 1240 | Line 1410 | void PVStudy::fillMCHisto(std::vector<si
1410        ferror_mct[1] = vsim->recVtx->yError();
1411        ferror_mct[2] = vsim->recVtx->zError();
1412        
1413 <      h["deltax"+suffix]->Fill( fres_mct[0] );
1414 <      h["deltay"+suffix]->Fill( fres_mct[1] );
1415 <      h["deltaz"+suffix]->Fill( fres_mct[2] );
1416 <      h["pullx"+suffix]->Fill( fres_mct[0]/ferror_mct[0] );
1417 <      h["pully"+suffix]->Fill( fres_mct[1]/ferror_mct[1] );
1418 <      h["pullz"+suffix]->Fill( fres_mct[2]/ferror_mct[2] );
1419 <      h["errPVx"+suffix]->Fill( ferror_mct[0] );
1420 <      h["errPVy"+suffix]->Fill( ferror_mct[1] );
1421 <      h["errPVz"+suffix]->Fill( ferror_mct[2] );
1413 >      h_summary->Fill1d("deltax"+suffix, fres_mct[0] );
1414 >      h_summary->Fill1d("deltay"+suffix, fres_mct[1] );
1415 >      h_summary->Fill1d("deltaz"+suffix, fres_mct[2] );
1416 >      h_summary->Fill1d("pullx"+suffix, fres_mct[0]/ferror_mct[0] );
1417 >      h_summary->Fill1d("pully"+suffix, fres_mct[1]/ferror_mct[1] );
1418 >      h_summary->Fill1d("pullz"+suffix, fres_mct[2]/ferror_mct[2] );
1419 >      h_summary->Fill1d("errPVx"+suffix, ferror_mct[0] );
1420 >      h_summary->Fill1d("errPVy"+suffix, ferror_mct[1] );
1421 >      h_summary->Fill1d("errPVz"+suffix, ferror_mct[2] );
1422        pvinfo.push_back(PVStudy::PVInfo(res(fres_mct[0], fres_mct[1], fres_mct[2]),
1423                                         error(ferror_mct[0], ferror_mct[1], ferror_mct[2]),
1424                                         nrectrk));
# Line 1305 | Line 1475 | void PVStudy::SetVarToZero() {
1475    fntrk_ = 0;
1476    //pvtx position (x,y,z) residual and error
1477    for(int i = 0; i<3;i++)    {
1478 <      fres_[i] = 0;
1479 <      ferror_[i] = 0;
1478 >    fres_[i] = 0;
1479 >    ferror_[i] = 0;
1480    }
1481    
1482    //Yanyan's variables
1483 <  // Number of reconstructed vertices  // for GeneralTracks
1483 >  // Number of reconstructed vertices  
1484    nrecPV_ = 0;
1485    nrecPV1_spl_ = 0;
1486    nrecPV2_spl_ = 0;
1487    nrecPV_twovtx_ = 0;
1488    nsimPV_ = 0;
1489 +
1490    nrecPV_mct_ = 0;
1491    nrecPV_spl1_mct_ = 0;
1492    nrecPV_spl2_mct_ = 0;
1493  
1494 +  // Mininum separation between the secondary pvtxes and leading pvtx
1495 +  min_zsep_ = 9999.0;
1496 +  min_ntrksep_ = 9999.0;
1497 +  min_sumpt2sep_ = -9999.0;
1498 +  
1499 +
1500    for (int i = 0; i < nMaxPVs_; i++) {
1501 <    nTrkPV_[i] = 0; // Number of tracks in the pvtx
1501 >    // recoVertices with all tracks
1502 >    nTrkPV_[i] = 0; // Number of tracks in the pvtx    
1503 >    sumptsq_[i] = 0;
1504 >    isValid_[i] = -1;
1505 >    isFake_[i] = -1;
1506      recx_[i] = 0;
1507      recy_[i] = 0;
1508      recz_[i] = 0;
1509      recx_err_[i] = 0;
1510      recy_err_[i] = 0;
1511      recz_err_[i] = 0;
1512 <
1513 <    nTrkPV1_spl_[i] = 0; // Number of tracks in the pvtx
1512 >    
1513 >    // recoVertices with splitTrack1
1514 >    nTrkPV1_spl_[i] = 0; // Number of tracks in the pvtx    
1515 >    sumptsq1_spl_[i] = 0;
1516 >    isValid1_spl_[i] = -1;
1517 >    isFake1_spl_[i] = -1;
1518      recx1_spl_[i] = 0;
1519      recy1_spl_[i] = 0;
1520      recz1_spl_[i] = 0;
1521      recx1_err_spl_[i] = 0;
1522      recy1_err_spl_[i] = 0;
1523      recz1_err_spl_[i] = 0;
1339
1340    nTrkPV2_spl_[i] = 0; // Number of tracks in the pvtx
1524    
1525 +    // recoVertices with splitTrack2
1526 +    nTrkPV2_spl_[i] = 0; // Number of tracks in the pvtx  
1527 +    sumptsq2_spl_[i] = 0;
1528 +    isValid2_spl_[i] = -1;
1529 +    isFake2_spl_[i] = -1;
1530      recx2_spl_[i] = 0;
1531      recy2_spl_[i] = 0;
1532      recz2_spl_[i] = 0;
# Line 1346 | Line 1534 | void PVStudy::SetVarToZero() {
1534      recy2_err_spl_[i] = 0;
1535      recz2_err_spl_[i] = 0;
1536      
1537 +    //pixelVertices
1538 +    nTrkPV_pxlpvtx_[i] = 0; // Number of tracks in the pvtx  
1539 +    sumptsq_pxlpvtx_[i] = 0;
1540 +    isValid_pxlpvtx_[i] = -1;
1541 +    isFake_pxlpvtx_[i] = -1;
1542 +    recx_pxlpvtx_[i] = 0;
1543 +    recy_pxlpvtx_[i] = 0;
1544 +    recz_pxlpvtx_[i] = 0;
1545 +    recx_err_pxlpvtx_[i] = 0;
1546 +    recy_err_pxlpvtx_[i] = 0;
1547 +    recz_err_pxlpvtx_[i] = 0;
1548 +
1549 +    // matched two-vertices
1550      nTrkPV1_spl_twovtx_[i] = 0;
1551      nTrkPV2_spl_twovtx_[i] = 0;
1552      nTrkPV_twovtx_[i] = 0;  
# Line 1361 | Line 1562 | void PVStudy::SetVarToZero() {
1562      pully_twovtx_[i] = 0;
1563      pullz_twovtx_[i] = 0;
1564      
1565 +    //simpv
1566      nsimTrkPV_[i] = 0;
1567      simx_[i] = 0;
1568      simy_[i] = 0;
1569      simz_[i] = 0;
1570      simptsq_[i] = 0;
1571      
1572 +    // residual and pulls with mc truth required
1573      deltax_mct_[i] = 0;
1574      deltay_mct_[i] = 0;
1575      deltaz_mct_[i] = 0;
# Line 1400 | Line 1603 | double PVStudy::sumPtSquared(const reco:
1603    }
1604    return sum;
1605   }
1606 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines