CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
HSCPTreeBuilder.cc
Go to the documentation of this file.
1 // -*- C++ -*-
2 //
3 // Package: HSCPTreeBuilder
4 // Class: HSCPTreeBuilder
5 //
13 //
14 // Original Author: Loic QUERTENMONT
15 // Created: Thu Mar 11 12:19:07 CEST 2010
16 //
17 
18 
19 #include <memory>
20 #include <cmath>
21 
30 
32 
35 
41 
48 
50 
53 
61 
64 
66 
69 
70 
74 
77 
79 
84 
87 
90 
92 
93 
94 #include "TFile.h"
95 #include "TObjString.h"
96 #include "TString.h"
97 #include "TH1F.h"
98 #include "TH2F.h"
99 #include "TProfile.h"
100 #include "TF1.h"
101 #include "TTree.h"
102 #include "TROOT.h"
103 
104 #include <ext/hash_map>
105 
106 
107 using namespace edm;
108 using namespace reco;
109 using namespace std;
110 using namespace __gnu_cxx;
111 
112 #define MAX_VERTICES 1000
113 #define MAX_HSCPS 10000
114 #define MAX_GENS 10000
115 #define MAX_ECALCRYS 10
116 
118  public:
119  explicit HSCPTreeBuilder(const edm::ParameterSet&);
120  ~HSCPTreeBuilder();
121 
122 
123  private:
124  virtual void beginJob() override ;
125  virtual bool filter(edm::Event&, const edm::EventSetup&) override;
126  virtual void endJob() override ;
127  int ClosestMuonIndex(reco::TrackRef track, std::vector<reco::MuonRef>);
128 
131 
136 
137  TTree* MyTree;
138  bool Event_triggerL1Bits[192];
139  bool Event_triggerHLTBits[128];
140  bool Event_technicalBits[64];
141  unsigned int Event_EventNumber;
142  unsigned int Event_RunNumber;
143  unsigned int Event_LumiSection;
144  unsigned int Event_BXCrossing;
145  unsigned int Event_Orbit;
146  unsigned int Event_Store;
147  unsigned int Event_Time;
150 
151  unsigned int NVertices;
152  float Vertex_x [MAX_VERTICES];
153  float Vertex_y [MAX_VERTICES];
154  float Vertex_z [MAX_VERTICES];
155  float Vertex_x_err [MAX_VERTICES];
156  float Vertex_y_err [MAX_VERTICES];
157  float Vertex_z_err [MAX_VERTICES];
158  int Vertex_TrackSize [MAX_VERTICES];
159  float Vertex_chi2 [MAX_VERTICES];
160  float Vertex_ndof [MAX_VERTICES];
161  bool Vertex_isFake [MAX_VERTICES];
162 
163  unsigned int NHSCPs;
164  bool Hscp_hasTrack [MAX_HSCPS];
165  bool Hscp_hasMuon [MAX_HSCPS];
166  bool Hscp_hasRpc [MAX_HSCPS];
167  bool Hscp_hasCalo [MAX_HSCPS];
168  int Hscp_type [MAX_HSCPS];
169  unsigned int Track_NOH [MAX_HSCPS];
170  float Track_p [MAX_HSCPS];
171  float Track_pt [MAX_HSCPS];
172  float Track_pt_err [MAX_HSCPS];
173  float Track_chi2 [MAX_HSCPS];
174  unsigned int Track_ndof [MAX_HSCPS];
175  float Track_eta [MAX_HSCPS];
176  float Track_eta_err [MAX_HSCPS];
177  float Track_phi [MAX_HSCPS];
178  float Track_phi_err [MAX_HSCPS];
179  float Track_dz [MAX_HSCPS];
180  float Track_d0 [MAX_HSCPS];
181  int Track_quality [MAX_HSCPS];
182  int Track_charge [MAX_HSCPS];
183  float Track_dEdxE1 [MAX_HSCPS];
184  float Track_dEdxE1_NOS [MAX_HSCPS];
185  unsigned int Track_dEdxE1_NOM [MAX_HSCPS];
186  float Track_dEdxE2 [MAX_HSCPS];
187  float Track_dEdxE2_NOS [MAX_HSCPS];
188  unsigned int Track_dEdxE2_NOM [MAX_HSCPS];
189  float Track_dEdxE3 [MAX_HSCPS];
190  float Track_dEdxE3_NOS [MAX_HSCPS];
191  unsigned int Track_dEdxE3_NOM [MAX_HSCPS];
192  float Track_dEdxD1 [MAX_HSCPS];
193  float Track_dEdxD1_NOS [MAX_HSCPS];
194  unsigned int Track_dEdxD1_NOM [MAX_HSCPS];
195  float Track_dEdxD2 [MAX_HSCPS];
196  float Track_dEdxD2_NOS [MAX_HSCPS];
197  unsigned int Track_dEdxD2_NOM [MAX_HSCPS];
198  float Track_dEdxD3 [MAX_HSCPS];
199  float Track_dEdxD3_NOS [MAX_HSCPS];
200  unsigned int Track_dEdxD3_NOM [MAX_HSCPS];
201  float Muon_p [MAX_HSCPS];
202  float Muon_pt [MAX_HSCPS];
203  float Muon_eta [MAX_HSCPS];
204  float Muon_phi [MAX_HSCPS];
205  int Muon_type [MAX_HSCPS];
206  bool Muon_qualityValid [MAX_HSCPS];
207  int Muon_charge [MAX_HSCPS];
208  float Muon_dt_IBeta [MAX_HSCPS];
209  float Muon_dt_IBeta_err [MAX_HSCPS];
210  float Muon_dt_fIBeta [MAX_HSCPS];
211  float Muon_dt_fIBeta_err [MAX_HSCPS];
212  int Muon_dt_ndof [MAX_HSCPS];
213  float Muon_csc_IBeta [MAX_HSCPS];
214  float Muon_csc_IBeta_err [MAX_HSCPS];
215  float Muon_csc_fIBeta [MAX_HSCPS];
216  float Muon_csc_fIBeta_err[MAX_HSCPS];
217  int Muon_csc_ndof [MAX_HSCPS];
218  float Muon_cb_IBeta [MAX_HSCPS];
219  float Muon_cb_IBeta_err [MAX_HSCPS];
220  float Muon_cb_fIBeta [MAX_HSCPS];
221  float Muon_cb_fIBeta_err [MAX_HSCPS];
222  int Muon_cb_ndof [MAX_HSCPS];
223  float Rpc_beta [MAX_HSCPS];
224 
225  float Calo_ecal_crossedE [MAX_HSCPS];
226  float Calo_ecal_beta [MAX_HSCPS];
227  float Calo_ecal_beta_err [MAX_HSCPS];
228  float Calo_ecal_invBeta_err [MAX_HSCPS];
229  float Calo_ecal_dEdx [MAX_HSCPS];
230  float Calo_ecal_time [MAX_HSCPS];
231  float Calo_ecal_time_err [MAX_HSCPS];
232  int Calo_ecal_numCrysCrossed [MAX_HSCPS];
233  float Calo_ecal_swissCrossKs [MAX_HSCPS][MAX_ECALCRYS];
234  float Calo_ecal_e1OverE9s [MAX_HSCPS][MAX_ECALCRYS];
235  float Calo_ecal_trackLengths [MAX_HSCPS][MAX_ECALCRYS];
236  float Calo_ecal_trackExitEtas [MAX_HSCPS][MAX_ECALCRYS];
237  float Calo_ecal_trackExitPhis [MAX_HSCPS][MAX_ECALCRYS];
238  float Calo_ecal_energies [MAX_HSCPS][MAX_ECALCRYS];
239  float Calo_ecal_outOfTimeEnergies[MAX_HSCPS][MAX_ECALCRYS];
240  float Calo_ecal_chi2s [MAX_HSCPS][MAX_ECALCRYS];
241  float Calo_ecal_outOfTimeChi2s [MAX_HSCPS][MAX_ECALCRYS];
242  float Calo_ecal_times [MAX_HSCPS][MAX_ECALCRYS];
243  float Calo_ecal_timeErrors [MAX_HSCPS][MAX_ECALCRYS];
244  unsigned int Calo_ecal_detIds [MAX_HSCPS][MAX_ECALCRYS];
245 
246  unsigned int NGens;
247  int Gen_pdgId [MAX_GENS];
248  float Gen_charge [MAX_GENS];
249  float Gen_p [MAX_GENS];
250  float Gen_px [MAX_GENS];
251  float Gen_py [MAX_GENS];
252  float Gen_pz [MAX_GENS];
253  float Gen_pt [MAX_GENS];
254  float Gen_eta [MAX_GENS];
255  float Gen_phi [MAX_GENS];
256  float Gen_beta [MAX_GENS];
257  float Gen_mass [MAX_GENS];
258 };
259 
261 {
262  m_HSCPsTag = iConfig.getParameter<InputTag> ("HSCParticles");
263 
264  reccordVertexInfo = iConfig.getUntrackedParameter<bool> ("reccordVertexInfo" , true );
265  reccordGenInfo = iConfig.getUntrackedParameter<bool> ("reccordGenInfo" , false );
266 
267  std::cout << "######################################################" << endl;
268  std::cout << " USE OF THE HSCPTreeBuilder is deprecated! " << endl;
269  std::cout << "better to use the HSCParticle Producer and then FWLite" << endl;
270  std::cout << "######################################################" << endl;
271 
272 }
273 
274 
276 {
277 }
278 
279 void
281 {
282  TTree::SetMaxTreeSize(1000*Long64_t(2000000000)); // authorize Trees up to 2 Terabytes
283  MyTree = tfs->make<TTree> ("HscpTree","HscpTree");
284 
285  MyTree->Branch("Event_EventNumber" ,&Event_EventNumber ,"Event_EventNumber/i");
286  MyTree->Branch("Event_RunNumber" ,&Event_RunNumber ,"Event_RunNumber/i");
287  MyTree->Branch("Event_LumiSection" ,&Event_LumiSection ,"Event_LumiSection/i");
288  MyTree->Branch("Event_BXCrossing" ,&Event_BXCrossing ,"Event_BXCrossing/i");
289  MyTree->Branch("Event_Orbit" ,&Event_Orbit ,"Event_Orbit/i");
290  MyTree->Branch("Event_Store" ,&Event_Store ,"Event_Store/i");
291  MyTree->Branch("Event_Time" ,&Event_Time ,"Event_Time/i");
292  MyTree->Branch("Event_PhysicsDeclared" ,&Event_PhysicsDeclared ,"Event_PhysicsDeclared/O");
293  MyTree->Branch("Event_technicalBits" ,Event_technicalBits ,"Event_technicalBits[64]/O");
294  MyTree->Branch("Event_triggerL1Bits" ,Event_triggerL1Bits ,"Event_triggerL1Bits[192]/O");
295  MyTree->Branch("Event_triggerHLTBits" ,Event_triggerHLTBits ,"Event_triggerHLTBits[128]/O");
296  MyTree->Branch("Event_BField" ,&Event_BField ,"Event_BField/F");
297 
298  if(reccordVertexInfo){
299  MyTree->Branch("NVertices" ,&NVertices ,"NVertices/I");
300  MyTree->Branch("Vertex_x" ,Vertex_x ,"Vertex_x[NVertices]/F");
301  MyTree->Branch("Vertex_y" ,Vertex_y ,"Vertex_y[NVertices]/F");
302  MyTree->Branch("Vertex_z" ,Vertex_z ,"Vertex_z[NVertices]/F");
303  MyTree->Branch("Vertex_x_err" ,Vertex_x_err ,"Vertex_x_err[NVertices]/F");
304  MyTree->Branch("Vertex_y_err" ,Vertex_y_err ,"Vertex_y_err[NVertices]/F");
305  MyTree->Branch("Vertex_z_err" ,Vertex_z_err ,"Vertex_z_err[NVertices]/F");
306  MyTree->Branch("Vertex_TrackSize",Vertex_TrackSize,"Vertex_TrackSize[NVertices]/I");
307  MyTree->Branch("Vertex_chi2" ,Vertex_chi2 ,"Vertex_chi2[NVertices]/F");
308  MyTree->Branch("Vertex_ndof" ,Vertex_ndof ,"Vertex_ndof[NVertices]/F");
309  MyTree->Branch("Vertex_isFake" ,Vertex_isFake ,"Vertex_isFake[NVertices]/O");
310  }
311 
312  MyTree->Branch("NHSCPs" ,&NHSCPs ,"NHSCPs/I");
313  MyTree->Branch("Hscp_hasTrack" ,Hscp_hasTrack ,"Hscp_hasTrack[NHSCPs]/O");
314  MyTree->Branch("Hscp_hasMuon" ,Hscp_hasMuon ,"Hscp_hasMuon[NHSCPs]/O");
315  MyTree->Branch("Hscp_hasRpc" ,Hscp_hasRpc ,"Hscp_hasRpc[NHSCPs]/O");
316  MyTree->Branch("Hscp_hasCalo" ,Hscp_hasCalo ,"Hscp_hasCalo[NHSCPs]/O");
317  MyTree->Branch("Hscp_type" ,Hscp_type ,"Hscp_type[NHSCPs]/I");
318  MyTree->Branch("Track_NOH" ,Track_NOH ,"Track_NOH[NHSCPs]/I");
319  MyTree->Branch("Track_p" ,Track_p ,"Track_p[NHSCPs]/F");
320  MyTree->Branch("Track_pt" ,Track_pt ,"Track_pt[NHSCPs]/F");
321  MyTree->Branch("Track_pt_err" ,Track_pt_err ,"Track_pt_err[NHSCPs]/F");
322  MyTree->Branch("Track_chi2" ,Track_chi2 ,"Track_chi2[NHSCPs]/F");
323  MyTree->Branch("Track_ndof" ,Track_ndof ,"Track_ndof[NHSCPs]/F");
324  MyTree->Branch("Track_eta" ,Track_eta ,"Track_eta[NHSCPs]/F");
325  MyTree->Branch("Track_eta_err" ,Track_eta_err ,"Track_eta_err[NHSCPs]/F");
326  MyTree->Branch("Track_phi" ,Track_phi ,"Track_phi[NHSCPs]/F");
327  MyTree->Branch("Track_phi_err" ,Track_phi_err ,"Track_phi_err[NHSCPs]/F");
328  MyTree->Branch("Track_d0" ,Track_d0 ,"Track_d0[NHSCPs]/F");
329  MyTree->Branch("Track_dz" ,Track_dz ,"Track_dz[NHSCPs]/F");
330  MyTree->Branch("Track_quality" ,Track_quality ,"Track_quality[NHSCPs]/I");
331  MyTree->Branch("Track_charge" ,Track_charge ,"Track_charge[NHSCPs]/I");
332  MyTree->Branch("Track_dEdxE1" ,Track_dEdxE1 ,"Track_dEdxE1[NHSCPs]/F");
333  MyTree->Branch("Track_dEdxE1_NOS" ,Track_dEdxE1_NOS ,"Track_dEdxE1_NOS[NHSCPs]/F");
334  MyTree->Branch("Track_dEdxE1_NOM" ,Track_dEdxE1_NOM ,"Track_dEdxE1_NOM[NHSCPs]/I");
335  MyTree->Branch("Track_dEdxE2" ,Track_dEdxE2 ,"Track_dEdxE2[NHSCPs]/F");
336  MyTree->Branch("Track_dEdxE2_NOS" ,Track_dEdxE2_NOS ,"Track_dEdxE2_NOS[NHSCPs]/F");
337  MyTree->Branch("Track_dEdxE2_NOM" ,Track_dEdxE2_NOM ,"Track_dEdxE2_NOM[NHSCPs]/I");
338  MyTree->Branch("Track_dEdxE3" ,Track_dEdxE3 ,"Track_dEdxE3[NHSCPs]/F");
339  MyTree->Branch("Track_dEdxE3_NOS" ,Track_dEdxE3_NOS ,"Track_dEdxE3_NOS[NHSCPs]/F");
340  MyTree->Branch("Track_dEdxE3_NOM" ,Track_dEdxE3_NOM ,"Track_dEdxE3_NOM[NHSCPs]/I");
341  MyTree->Branch("Track_dEdxD1" ,Track_dEdxD1 ,"Track_dEdxD1[NHSCPs]/F");
342  MyTree->Branch("Track_dEdxD1_NOS" ,Track_dEdxD1_NOS ,"Track_dEdxD1_NOS[NHSCPs]/F");
343  MyTree->Branch("Track_dEdxD1_NOM" ,Track_dEdxD1_NOM ,"Track_dEdxD1_NOM[NHSCPs]/I");
344  MyTree->Branch("Track_dEdxD2" ,Track_dEdxD2 ,"Track_dEdxD2[NHSCPs]/F");
345  MyTree->Branch("Track_dEdxD2_NOS" ,Track_dEdxD2_NOS ,"Track_dEdxD2_NOS[NHSCPs]/F");
346  MyTree->Branch("Track_dEdxD2_NOM" ,Track_dEdxD2_NOM ,"Track_dEdxD2_NOM[NHSCPs]/I");
347  MyTree->Branch("Track_dEdxD3" ,Track_dEdxD3 ,"Track_dEdxD3[NHSCPs]/F");
348  MyTree->Branch("Track_dEdxD3_NOS" ,Track_dEdxD3_NOS ,"Track_dEdxD3_NOS[NHSCPs]/F");
349  MyTree->Branch("Track_dEdxD3_NOM" ,Track_dEdxD3_NOM ,"Track_dEdxD3_NOM[NHSCPs]/I");
350  MyTree->Branch("Muon_p" ,Muon_p ,"Muon_p[NHSCPs]/F");
351  MyTree->Branch("Muon_pt" ,Muon_pt ,"Muon_pt[NHSCPs]/F");
352  MyTree->Branch("Muon_eta" ,Muon_eta ,"Muon_eta[NHSCPs]/F");
353  MyTree->Branch("Muon_phi" ,Muon_phi ,"Muon_phi[NHSCPs]/F");
354  MyTree->Branch("Muon_type" ,Muon_type ,"Muon_type[NHSCPs]/i");
355  MyTree->Branch("Muon_qualityValid" ,Muon_qualityValid ,"Muon_qualityValid[NHSCPs]/O");
356  MyTree->Branch("Muon_charge" ,Muon_charge ,"Muon_charge[NHSCPs]/i");
357  MyTree->Branch("Muon_dt_IBeta" ,Muon_dt_IBeta ,"Muon_dt_IBeta[NHSCPs]/F");
358  MyTree->Branch("Muon_dt_IBeta_err" ,Muon_dt_IBeta_err ,"Muon_dt_IBeta_err[NHSCPs]/F");
359  MyTree->Branch("Muon_dt_fIBeta" ,Muon_dt_fIBeta ,"Muon_dt_fIBeta[NHSCPs]/F");
360  MyTree->Branch("Muon_dt_fIBeta_err" ,Muon_dt_fIBeta_err ,"Muon_dt_fIBeta_err[NHSCPs]/F");
361  MyTree->Branch("Muon_dt_ndof" ,Muon_dt_ndof ,"Muon_dt_ndof[NHSCPs]/I");
362  MyTree->Branch("Muon_csc_IBeta" ,Muon_csc_IBeta ,"Muon_csc_IBeta[NHSCPs]/F");
363  MyTree->Branch("Muon_csc_IBeta_err" ,Muon_csc_IBeta_err ,"Muon_csc_IBeta_err[NHSCPs]/F");
364  MyTree->Branch("Muon_csc_fIBeta" ,Muon_csc_fIBeta ,"Muon_csc_fIBeta[NHSCPs]/F");
365  MyTree->Branch("Muon_csc_fIBeta_err",Muon_csc_fIBeta_err,"Muon_csc_fIBeta_err[NHSCPs]/F");
366  MyTree->Branch("Muon_csc_ndof" ,Muon_csc_ndof ,"Muon_csc_ndof[NHSCPs]/I");
367  MyTree->Branch("Muon_cb_IBeta" ,Muon_cb_IBeta ,"Muon_cb_IBeta[NHSCPs]/F");
368  MyTree->Branch("Muon_cb_IBeta_err" ,Muon_cb_IBeta_err ,"Muon_cb_IBeta_err[NHSCPs]/F");
369  MyTree->Branch("Muon_cb_fIBeta" ,Muon_cb_fIBeta ,"Muon_cb_fIBeta[NHSCPs]/F");
370  MyTree->Branch("Muon_cb_fIBeta_err" ,Muon_cb_fIBeta_err ,"Muon_cb_fIBeta_err[NHSCPs]/F");
371  MyTree->Branch("Muon_cb_ndof" ,Muon_cb_ndof ,"Muon_cb_ndof[NHSCPs]/I");
372 
373  MyTree->Branch("Rpc_beta" ,Rpc_beta ,"Rpc_beta[NHSCPs]/F");
374 
375  MyTree->Branch("Calo_ecal_crossedE" ,Calo_ecal_crossedE ,"Calo_ecal_crossedE[NHSCPs]/F");
376  MyTree->Branch("Calo_ecal_beta" ,Calo_ecal_beta ,"Calo_ecal_beta[NHSCPs]/F");
377  MyTree->Branch("Calo_ecal_beta_err" ,Calo_ecal_beta_err ,"Calo_ecal_beta_err[NHSCPs]/F");
378  MyTree->Branch("Calo_ecal_invBeta_err" ,Calo_ecal_invBeta_err ,"Calo_ecal_invBeta_err[NHSCPs]/F");
379  MyTree->Branch("Calo_ecal_dEdx" ,Calo_ecal_dEdx ,"Calo_ecal_dEdx[NHSCPs]/F");
380  MyTree->Branch("Calo_ecal_time" ,Calo_ecal_time ,"Calo_ecal_time[NHSCPs]/F");
381  MyTree->Branch("Calo_ecal_time_err" ,Calo_ecal_time_err ,"Calo_ecal_time_err[NHSCPs]/F");
382  MyTree->Branch("Calo_ecal_numCrysCrossed" ,Calo_ecal_numCrysCrossed,"Calo_ecal_numCrysCrossed[NHSCPs]/I");
383  MyTree->Branch("Calo_ecal_swissCrossKs" ,Calo_ecal_swissCrossKs ,"Calo_ecal_swissCrossKs[NHSCPs][10]/F");
384  MyTree->Branch("Calo_ecal_e1OverE9s" ,Calo_ecal_e1OverE9s ,"Calo_ecal_e1OverE9s[NHSCPs][10]/F");
385  MyTree->Branch("Calo_ecal_trackLengths" ,Calo_ecal_trackLengths ,"Calo_ecal_trackLengths[NHSCPs][10]/F");
386  MyTree->Branch("Calo_ecal_trackExitEtas" ,Calo_ecal_trackExitEtas ,"Calo_ecal_trackExitEtas[NHSCPs][10]/F");
387  MyTree->Branch("Calo_ecal_trackExitPhis" ,Calo_ecal_trackExitPhis ,"Calo_ecal_trackExitPhis[NHSCPs][10]/F");
388  MyTree->Branch("Calo_ecal_energies" ,Calo_ecal_energies ,"Calo_ecal_energies[NHSCPs][10]/F");
389  MyTree->Branch("Calo_ecal_outOfTimeEnergies",Calo_ecal_outOfTimeEnergies ,"Calo_ecal_outOfTimeEnergies[NHSCPs][10]/F");
390  MyTree->Branch("Calo_ecal_chi2s" ,Calo_ecal_chi2s ,"Calo_ecal_chi2s[NHSCPs][10]/F");
391  MyTree->Branch("Calo_ecal_outOfTimeChi2s" ,Calo_ecal_outOfTimeChi2s ,"Calo_ecal_outOfTimeChi2s[NHSCPs][10]/F");
392  MyTree->Branch("Calo_ecal_times" ,Calo_ecal_times ,"Calo_ecal_times[NHSCPs][10]/F");
393  MyTree->Branch("Calo_ecal_timeErrors" ,Calo_ecal_timeErrors ,"Calo_ecal_timeErrors[NHSCPs][10]/F");
394  MyTree->Branch("Calo_ecal_detIds" ,Calo_ecal_detIds ,"Calo_ecal_detIds[NHSCPs][10]/I");
395 
396  if(reccordGenInfo){
397  MyTree->Branch("NGens" ,&NGens ,"NGens/I");
398  MyTree->Branch("Gen_pdgId" ,Gen_pdgId ,"Gen_pdgId[NGens]/i");
399  MyTree->Branch("Gen_charge" ,Gen_charge ,"Gen_charge[NGens]/F");
400  MyTree->Branch("Gen_p" ,Gen_p ,"Gen_p[NGens]/F");
401  MyTree->Branch("Gen_px" ,Gen_px ,"Gen_px[NGens]/F");
402  MyTree->Branch("Gen_py" ,Gen_py ,"Gen_py[NGens]/F");
403  MyTree->Branch("Gen_pz" ,Gen_pz ,"Gen_pz[NGens]/F");
404  MyTree->Branch("Gen_pt" ,Gen_pt ,"Gen_pt[NGens]/F");
405  MyTree->Branch("Gen_eta" ,Gen_eta ,"Gen_eta[NGens]/F");
406  MyTree->Branch("Gen_phi" ,Gen_phi ,"Gen_phi[NGens]/F");
407  MyTree->Branch("Gen_beta" ,Gen_beta ,"Gen_beta[NGens]/F");
408  MyTree->Branch("Gen_mass" ,Gen_mass ,"Gen_mass[NGens]/F");
409  }
410 
411 }
412 
413 void
415 {
416 }
417 
418 
419 
420 bool
422 {
423  bool debug = false;
424  if (debug) cout << "I'm in HSCPTreeBuilder::analyze!" << endl;
425 
426 
427  Event_EventNumber = iEvent.id().event();
428  Event_RunNumber = iEvent.id().run();
429  Event_LumiSection = iEvent.eventAuxiliary().luminosityBlock();
430  Event_BXCrossing = iEvent.eventAuxiliary().bunchCrossing();
431  Event_Orbit = iEvent.eventAuxiliary().orbitNumber();
432  Event_Store = iEvent.eventAuxiliary().storeNumber();
433  Event_Time = iEvent.eventAuxiliary().time().value();
434 
435  // BField part:
437  iSetup.get<IdealMagneticFieldRecord>().get(MF);
438  const MagneticField* theMagneticField = MF.product();
439  Event_BField = fabs(theMagneticField->inTesla(GlobalPoint(0,0,0)).z());
440 
441  // L1 TRIGGER part:
443  iEvent.getByLabel("gtDigis", h_gtReadoutRecord);
444  L1GtFdlWord fdlWord = h_gtReadoutRecord->gtFdlWord();
445  TechnicalTriggerWord L1technical = fdlWord.gtTechnicalTriggerWord();
446  Event_PhysicsDeclared = h_gtReadoutRecord->gtFdlWord().physicsDeclared();
447  for (unsigned int i = 0; i < 64; ++i){ Event_technicalBits [i] = L1technical[i]; }
448  DecisionWord L1decision = fdlWord.gtDecisionWord();
449  for (unsigned int i = 0; i < 128; ++i){ Event_triggerL1Bits [i] = L1decision[i]; }
450  DecisionWordExtended L1decisionE = fdlWord.gtDecisionWordExtended();
451  for (unsigned int i = 0; i < 64; ++i){ Event_triggerL1Bits [128+i] = L1decisionE[i]; }
452 
453  // HLT TRIGGER part:
455  iEvent.getByLabel("TriggerResults", trh);
456  for(unsigned int i=0;i<trh->size() && i<128;++i){Event_triggerHLTBits[i] = trh->at(i).accept();}
457 
458  edm::Handle<reco::VertexCollection> recoVertexHandle;
459  iEvent.getByLabel("offlinePrimaryVertices", recoVertexHandle);
460  reco::VertexCollection recoVertex = *recoVertexHandle;
461 
462  if(reccordVertexInfo){
463  NVertices = 0;
464  for(unsigned int i=0;i<recoVertex.size();i++){
465  Vertex_x [NVertices] = recoVertex[i].x();
466  Vertex_y [NVertices] = recoVertex[i].y();
467  Vertex_z [NVertices] = recoVertex[i].z();
468  Vertex_x_err [NVertices] = recoVertex[i].xError();
469  Vertex_y_err [NVertices] = recoVertex[i].yError();
470  Vertex_z_err [NVertices] = recoVertex[i].zError();
471  Vertex_TrackSize[NVertices] = recoVertex[i].tracksSize();
472  Vertex_chi2 [NVertices] = recoVertex[i].chi2();
473  Vertex_ndof [NVertices] = recoVertex[i].ndof();
474  Vertex_isFake [NVertices] = recoVertex[i].isFake();
475  NVertices++;
476  }
477  }
478 
479  // Source Collection
481  iEvent.getByLabel(m_HSCPsTag, HSCPCollectionHandle);
482  susybsm::HSCParticleCollection HSCPCollection = *HSCPCollectionHandle.product();
483 
484  NHSCPs=0;
485  for(unsigned int i=0; i<HSCPCollection.size();i++){
486  susybsm::HSCParticle hscp = HSCPCollection[i];
487  reco::MuonRef muon = hscp.muonRef();
488  reco::TrackRef track = hscp.trackRef();;
489 
490  Hscp_hasTrack [NHSCPs] = hscp.hasTrackRef();
491  Hscp_hasMuon [NHSCPs] = hscp.hasMuonRef();
492  Hscp_hasRpc [NHSCPs] = hscp.hasRpcInfo();
493  Hscp_hasCalo [NHSCPs] = hscp.hasCaloInfo();
494  Hscp_type [NHSCPs] = hscp.type();
495 
496  if(track.isNonnull() && Hscp_hasTrack){
497  Track_p [NHSCPs] = track->p();
498  Track_pt [NHSCPs] = track->pt();
499  Track_pt_err [NHSCPs] = track->ptError();
500  Track_eta [NHSCPs] = track->eta();
501  Track_eta_err [NHSCPs] = track->etaError();
502  Track_phi [NHSCPs] = track->phi();
503  Track_phi_err [NHSCPs] = track->phiError();
504  Track_NOH [NHSCPs] = track->found();
505  Track_chi2 [NHSCPs] = track->chi2();
506  Track_ndof [NHSCPs] = track->ndof();
507  Track_d0 [NHSCPs] = -1.0f * track->dxy(recoVertex[0].position());
508  Track_dz [NHSCPs] = -1.0f * track->dz (recoVertex[0].position());
509  Track_quality [NHSCPs] = track->qualityMask();
510  Track_charge [NHSCPs] = track->charge();
511 /* Track_dEdxE1 [NHSCPs] = hscp.dedxEstimator1().dEdx();
512  Track_dEdxE1_NOM [NHSCPs] = hscp.dedxEstimator1().numberOfMeasurements();
513  Track_dEdxE1_NOS [NHSCPs] = hscp.dedxEstimator1().numberOfSaturatedMeasurements();
514  Track_dEdxE2 [NHSCPs] = hscp.dedxEstimator2().dEdx();
515  Track_dEdxE2_NOM [NHSCPs] = hscp.dedxEstimator2().numberOfMeasurements();
516  Track_dEdxE2_NOS [NHSCPs] = hscp.dedxEstimator2().numberOfSaturatedMeasurements();
517  Track_dEdxE3 [NHSCPs] = hscp.dedxEstimator3().dEdx();
518  Track_dEdxE3_NOM [NHSCPs] = hscp.dedxEstimator3().numberOfMeasurements();
519  Track_dEdxE3_NOS [NHSCPs] = hscp.dedxEstimator3().numberOfSaturatedMeasurements();
520  Track_dEdxD1 [NHSCPs] = hscp.dedxDiscriminator1().dEdx();
521  Track_dEdxD1_NOM [NHSCPs] = hscp.dedxDiscriminator1().numberOfMeasurements();
522  Track_dEdxD1_NOS [NHSCPs] = hscp.dedxDiscriminator1().numberOfSaturatedMeasurements();
523  Track_dEdxD2 [NHSCPs] = hscp.dedxDiscriminator2().dEdx();
524  Track_dEdxD2_NOM [NHSCPs] = hscp.dedxDiscriminator2().numberOfMeasurements();
525  Track_dEdxD2_NOS [NHSCPs] = hscp.dedxDiscriminator2().numberOfSaturatedMeasurements();
526  Track_dEdxD3 [NHSCPs] = hscp.dedxDiscriminator3().dEdx();
527  Track_dEdxD3_NOM [NHSCPs] = hscp.dedxDiscriminator3().numberOfMeasurements();
528  Track_dEdxD3_NOS [NHSCPs] = hscp.dedxDiscriminator3().numberOfSaturatedMeasurements();
529 */
530  }
531 
532  if(muon.isNonnull() && Hscp_hasMuon){
533  Muon_p [NHSCPs] = muon->p();
534  Muon_pt [NHSCPs] = muon->pt();
535  Muon_eta [NHSCPs] = muon->eta();
536  Muon_phi [NHSCPs] = muon->phi();
537  Muon_type [NHSCPs] = muon->type();
538  Muon_qualityValid [NHSCPs] = muon->isQualityValid();
539  Muon_charge [NHSCPs] = muon->charge();
540 /* Muon_dt_IBeta [NHSCPs] = hscp.muonTimeDt().inverseBeta();
541  Muon_dt_IBeta_err [NHSCPs] = hscp.muonTimeDt().inverseBetaErr();
542  Muon_dt_fIBeta [NHSCPs] = hscp.muonTimeDt().freeInverseBeta();
543  Muon_dt_fIBeta_err [NHSCPs] = hscp.muonTimeDt().freeInverseBetaErr();
544  Muon_dt_ndof [NHSCPs] = hscp.muonTimeDt().nDof();
545  Muon_csc_IBeta [NHSCPs] = hscp.muonTimeCsc().inverseBeta();
546  Muon_csc_IBeta_err [NHSCPs] = hscp.muonTimeCsc().inverseBetaErr();
547  Muon_csc_fIBeta [NHSCPs] = hscp.muonTimeCsc().freeInverseBeta();
548  Muon_csc_fIBeta_err[NHSCPs] = hscp.muonTimeCsc().freeInverseBetaErr();
549  Muon_csc_ndof [NHSCPs] = hscp.muonTimeCsc().nDof();
550  Muon_cb_IBeta [NHSCPs] = hscp.muonTimeCombined().inverseBeta();
551  Muon_cb_IBeta_err [NHSCPs] = hscp.muonTimeCombined().inverseBetaErr();
552  Muon_cb_fIBeta [NHSCPs] = hscp.muonTimeCombined().freeInverseBeta();
553  Muon_cb_fIBeta_err [NHSCPs] = hscp.muonTimeCombined().freeInverseBetaErr();
554  Muon_cb_ndof [NHSCPs] = hscp.muonTimeCombined().nDof();
555 */
556  }
557 
558  if(hscp.hasCaloInfo()){
559 // Calo_ecal_crossedE [NHSCPs] = hscp.calo().ecalCrossedEnergy;
560 // Calo_ecal_beta [NHSCPs] = hscp.calo().ecalBeta;
561 // Calo_ecal_beta_err [NHSCPs] = hscp.calo().ecalBetaError;
562 // Calo_ecal_invBeta_err [NHSCPs] = hscp.calo().ecalInvBetaError;
563 // Calo_ecal_dEdx [NHSCPs] = hscp.calo().ecalDeDx;
564 // Calo_ecal_time [NHSCPs] = hscp.calo().ecalTime;
565 // Calo_ecal_time_err [NHSCPs] = hscp.calo().ecalTimeError;
566 // Calo_ecal_numCrysCrossed[NHSCPs] = hscp.calo().ecalCrysCrossed;
567 /* for(int i=0; i < Calo_ecal_numCrysCrossed[NHSCPs] && i < MAX_ECALCRYS; ++i)
568  {
569  Calo_ecal_swissCrossKs [NHSCPs][i] = hscp.calo().ecalSwissCrossKs[i];
570  Calo_ecal_e1OverE9s [NHSCPs][i] = hscp.calo().ecalE1OverE9s[i];
571  Calo_ecal_trackLengths [NHSCPs][i] = hscp.calo().ecalTrackLengths[i];
572  GlobalPoint exitPosition = hscp.calo().ecalTrackExitPositions[i];
573  Calo_ecal_trackExitEtas [NHSCPs][i] = exitPosition.eta();
574  Calo_ecal_trackExitPhis [NHSCPs][i] = exitPosition.phi();
575  Calo_ecal_energies [NHSCPs][i] = hscp.calo().ecalEnergies[i];
576  Calo_ecal_outOfTimeEnergies[NHSCPs][i] = hscp.calo().ecalOutOfTimeEnergies[i];
577  Calo_ecal_chi2s [NHSCPs][i] = hscp.calo().ecalChi2s[i];
578  Calo_ecal_outOfTimeChi2s [NHSCPs][i] = hscp.calo().ecalOutOfTimeChi2s[i];
579  Calo_ecal_times [NHSCPs][i] = hscp.calo().ecalTimes[i];
580  Calo_ecal_timeErrors [NHSCPs][i] = hscp.calo().ecalTimeErrors[i];
581  Calo_ecal_detIds [NHSCPs][i] = hscp.calo().ecalDetIds[i];
582  }
583 */
584  }
585 
586  if(Hscp_hasRpc){
587  Rpc_beta [NHSCPs] = hscp.rpc().beta;
588  }
589 
590  NHSCPs++;
591  }
592 
593 
594  if(reccordGenInfo){
596  iEvent.getByLabel("genParticles", genParticles);
597  NGens=0;
598  for(unsigned int i=0;i<genParticles->size();i++){
599  const GenParticle & part = (*genParticles)[i];
600  if(part.status()!=1)continue;
601  if(part.pt()<5)continue;
602 // if(fabs(part.pdgId())<1000000) continue;
603 
604  Gen_pdgId [NGens] = part.pdgId();
605  Gen_charge[NGens] = part.charge();
606  Gen_p [NGens] = part.p();
607  Gen_px [NGens] = part.px();
608  Gen_py [NGens] = part.py();
609  Gen_pz [NGens] = part.pz();
610  Gen_pt [NGens] = part.pt();
611  Gen_eta [NGens] = part.eta();
612  Gen_phi [NGens] = part.phi();
613  Gen_beta [NGens] = part.p()/part.energy();
614  Gen_mass [NGens] = part.mass();
615  NGens++;
616  }
617  }
618 
619 
620  MyTree->Fill();
621  return true;
622 }
623 
624 //define this as a plug-in
626 
627 
628 
629 
630 
631 
632 
633 
634 
635 
RunNumber_t run() const
Definition: EventID.h:42
unsigned int Event_LumiSection
virtual double energy() const GCC11_FINAL
energy
T getParameter(std::string const &) const
EventNumber_t event() const
Definition: EventID.h:44
T getUntrackedParameter(std::string const &, T const &) const
int i
Definition: DBlmapReader.cc:9
const TechnicalTriggerWord & gtTechnicalTriggerWord() const
get/set technical trigger bits
Definition: L1GtFdlWord.h:149
unsigned int NGens
virtual double p() const GCC11_FINAL
magnitude of momentum vector
const RPCBetaMeasurement & rpc() const
Definition: HSCParticle.h:77
virtual GlobalVector inTesla(const GlobalPoint &gp) const =0
Field value ad specified global point, in Tesla.
virtual int pdgId() const GCC11_FINAL
PDG identifier.
Timestamp const & time() const
unsigned int Event_BXCrossing
std::vector< Vertex > VertexCollection
collection of Vertex objects
Definition: VertexFwd.h:9
tuple HSCPTreeBuilder
HSCP TREE BUILDER.
unsigned int NVertices
DEFINE_FWK_MODULE(HiMixingModule)
virtual void beginJob() override
bool hasTrackRef() const
Definition: HSCParticle.h:57
virtual double pz() const GCC11_FINAL
z coordinate of momentum vector
int bunchCrossing() const
virtual double py() const GCC11_FINAL
y coordinate of momentum vector
HSCPTreeBuilder(const edm::ParameterSet &)
static int position[TOTALCHAMBERS][3]
Definition: ReadPGInfo.cc:509
LuminosityBlockNumber_t luminosityBlock() const
void beginJob()
Definition: Breakpoints.cc:15
unsigned int Event_Orbit
#define MAX_GENS
bool isNonnull() const
Checks for non-null.
Definition: Ref.h:250
virtual bool filter(edm::Event &, const edm::EventSetup &) override
std::vector< bool > DecisionWordExtended
virtual float phi() const GCC11_FINAL
momentum azimuthal angle
std::vector< HSCParticle > HSCParticleCollection
Definition: HSCParticle.h:94
int iEvent
Definition: GenABIO.cc:243
int storeNumber() const
#define MAX_ECALCRYS
virtual int status() const GCC11_FINAL
status word
unsigned int Event_EventNumber
int orbitNumber() const
unsigned int NHSCPs
std::vector< bool > DecisionWord
typedefs
T z() const
Definition: PV3DBase.h:64
virtual double px() const GCC11_FINAL
x coordinate of momentum vector
std::vector< bool > TechnicalTriggerWord
technical trigger bits (64 bits)
#define MAX_VERTICES
const edm::EventSetup * iSetup_
const edm::Event * iEvent_
#define MAX_HSCPS
bool getByLabel(InputTag const &tag, Handle< PROD > &result) const
Definition: Event.h:390
virtual float eta() const GCC11_FINAL
momentum pseudorapidity
EventAuxiliary const & eventAuxiliary() const
Definition: Event.h:72
virtual float mass() const GCC11_FINAL
mass
reco::MuonRef muonRef() const
Definition: HSCParticle.h:74
#define debug
Definition: HDRShower.cc:19
virtual int charge() const GCC11_FINAL
electric charge
part
Definition: HCALResponse.h:20
const T & get() const
Definition: EventSetup.h:55
T const * product() const
Definition: ESHandle.h:62
const DecisionWord & gtDecisionWord() const
get/set/print algorithms bits (decision word)
Definition: L1GtFdlWord.h:171
T const * product() const
Definition: Handle.h:81
int type() const
Definition: HSCParticle.cc:5
reco::TrackRef trackRef() const
Definition: HSCParticle.h:72
edm::EventID id() const
Definition: EventBase.h:56
bool hasRpcInfo() const
Definition: HSCParticle.h:59
unsigned int Event_Store
tuple cout
Definition: gather_cfg.py:121
unsigned int Event_Time
virtual void endJob() override
volatile std::atomic< bool > shutdown_flag false
bool hasMuonRef() const
Definition: HSCParticle.h:55
const DecisionWordExtended & gtDecisionWordExtended() const
get/set extended algorithms bits (extended decision word)
Definition: L1GtFdlWord.h:203
edm::Service< TFileService > tfs
virtual float pt() const GCC11_FINAL
transverse momentum
bool hasCaloInfo() const
Definition: HSCParticle.h:60
TimeValue_t value() const
Definition: Timestamp.h:56
unsigned int Event_RunNumber