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