ATLAS Offline Software
SingleTrackValidation.cxx
Go to the documentation of this file.
1 /*
2  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
3 */
4 
8 
9 // In order to be able to Ntuple & cet:
10 #include "GaudiKernel/NTuple.h"
11 #include "GaudiKernel/INTupleSvc.h"
12 #include "GaudiKernel/SmartDataPtr.h"
13 
14 // To get the Magnetic Field:
15 #include "CLHEP/GenericFunctions/FixedConstant.hh"
16 
17 // To get the particle properties:
18 #include "HepPDT/ParticleDataTable.hh"
19 #include "HepPDT/ParticleData.hh"
20 
21 // To extrapolate:
22 #include "AtlasBComponent.h"
23 #include "GeoXPEngine.h"
24 
25 // To interpret LAr Geometry information:
26 #include "CaloDetDescr/CaloDetDescrElement.h"
30 
31 // pi etc
32 #include "CLHEP/Units/PhysicalConstants.h"
33 #include "AthenaKernel/Units.h"
34 
35 // System libraries
36 #include <signal.h>
37 #include <unistd.h>
38 #include <sys/types.h>
39 #include <sstream>
40 #include <fstream>
41 #include <iostream>
42 #include <sys/times.h>
43 #include <string>
44 
45 namespace Units = Athena::Units;
46 using HepGeom::Point3D;
47 using CLHEP::HepLorentzVector;
48 using CLHEP::mm;
49 using CLHEP::pi;
50 using CLHEP::twopi;
51 
52 inline int iabs (int a) { return (a>=0)? a : -a;}
53 inline int getCpu() {
54  int cpuTime = 0;
55  {
56  static const int ticksPerJiffy = sysconf(_SC_CLK_TCK)/100;
57  struct tms buff{};
58  times(&buff);
59  cpuTime=(buff.tms_utime + buff.tms_stime + buff.tms_cutime + buff.tms_cstime)/ticksPerJiffy;
60  }
61  return cpuTime;
62 }
63 
64 // We will define some actual data, later.
66 
67 public:
68  Clockwork() {}
70 
71  IPartPropSvc* partPropSvc{nullptr};
72  ITHistSvc* histSvc{nullptr};
73  const CaloCell_ID* cellId{nullptr};
74  NTuple::Tuple* nt = nullptr;
75 
76  NTuple::Item<double> eta;
77  NTuple::Item<double> pt;
78  NTuple::Item<double> phi;
79 
80  // Impact point
81  NTuple::Item<double> x;
82  NTuple::Item<double> y;
83  NTuple::Item<double> z;
84 
85  // Indices run from 0 to 11: 4 in barrel, 4 in end cap, 3 in FCAL
86  // Energy within each sampling layer (directly struck cell):
87  NTuple::Item<double> s_c00[15];
88  NTuple::Item<double> s_hits[15];
89  NTuple::Item<double> s_sumE[15];
90  NTuple::Item<double> s_deltaPhi[15];
91  NTuple::Item<double> s_sigmaPhi[15];
92  NTuple::Item<double> s_deltaEta[15];
93  NTuple::Item<double> s_sigmaEta[15];
94  NTuple::Item<double> s_t00[15];
95  NTuple::Item<double> s_widthX[15];
96  NTuple::Item<double> s_widthY[15];
97 
98  NTuple::Item<double> cpuTime;
99  NTuple::Item<double> Energy;
100  NTuple::Item<double> PDG;
101  NTuple::Item<double> RunNo;
102  NTuple::Item<double> EventNo;
103  NTuple::Item<double> E_Deposit;
104 
105  bool coolField{true};
106 };
107 
108 SingleTrackValidation::SingleTrackValidation (const std::string & name, ISvcLocator * pSvcLocator) :
109  AthAlgorithm(name,pSvcLocator),m_c(new Clockwork())
110 {
111  for (unsigned int i=0;i<162;++i) m_histos[i] = nullptr;
112 }
113 
115  if (m_c!=nullptr){ delete m_c; m_c=nullptr; }
116 }
117 
119  std::string names[162] = { "eta", "pt", "phi", "pos_x", "pos_y", "pos_z",
120  "emb0_cell", "emb1_cell", "emb2_cell", "emb3_cell", "emec0_cell", "emec1_cell", "emec2_cell", "emec3_cell",
121  "hec0_cell", "hec1_cell", "hec2_cell", "hec3_cell", "fc1_cell", "fc2_cell", "fc3_cell",
122  "emb0_hits", "emb1_hits", "emb2_hits", "emb3_hits", "emec0_hits", "emec1_hits", "emec2_hits", "emec3_hits",
123  "hec0_hits", "hec1_hits", "hec2_hits", "hec3_hits", "fc1_hits", "fc2_hits", "fc3_hits",
124  "emb0_sumE", "emb1_sumE", "emb2_sumE", "emb3_sumE", "emec0_sumE", "emec1_sumE", "emec2_sumE", "emec3_sumE",
125  "hec0_sumE", "hec1_sumE", "hec2_sumE", "hec3_sumE", "fc1_sumE", "fc2_sumE", "fc3_sumE",
126  "emb0_dPhi", "emb1_dPhi", "emb2_dPhi", "emb3_dPhi", "emec0_dPhi", "emec1_dPhi", "emec2_dPhi", "emec3_dPhi",
127  "hec0_dPhi", "hec1_dPhi", "hec2_dPhi", "hec3_dPhi", "fc1_dX", "fc2_dX", "fc3_dX",
128  "emb0_sPhi", "emb1_sPhi", "emb2_sPhi", "emb3_sPhi", "emec0_sPhi", "emec1_sPhi", "emec2_sPhi", "emec3_sPhi",
129  "hec0_sPhi", "hec1_sPhi", "hec2_sPhi", "hec3_sPhi", "fc1_sX", "fc2_sX", "fc3_sX",
130  "emb0_dEta", "emb1_dEta", "emb2_dEta", "emb3_dEta", "emec0_dEta", "emec1_dEta", "emec2_dEta", "emec3_dEta",
131  "hec0_dEta", "hec1_dEta", "hec2_dEta", "hec3_dEta", "fc1_dY", "fc2_dY", "fc3_dY",
132  "emb0_sEta", "emb1_sEta", "emb2_sEta", "emb3_sEta", "emec0_sEta", "emec1_sEta", "emec2_sEta", "emec3_sEta",
133  "hec0_sEta", "hec1_sEta", "hec2_sEta", "hec3_sEta", "fc1_sY", "fc2_sY", "fc3_sY",
134  "emb0_time", "emb1_time", "emb2_time", "emb3_time", "emec0_time", "emec1_time", "emec2_time", "emec3_time",
135  "hec0_time", "hec1_time", "hec2_time", "hec3_time", "fc1_time", "fc2_time", "fc3_time",
136  "emb0_widthX", "emb1_widthX", "emb2_widthX", "emb3_widthX", "emec0_widthX", "emec1_widthX", "emec2_widthX", "emec3_widthX",
137  "hec0_widthX", "hec1_widthX", "hec2_widthX", "hec3_widthX", "fc1_widthX", "fc2_widthX", "fc3_widthX",
138  "emb0_widthY", "emb1_widthY", "emb2_widthY", "emb3_widthY", "emec0_widthY", "emec1_widthY", "emec2_widthY", "emec3_widthY",
139  "hec0_widthY", "hec1_widthY", "hec2_widthY", "hec3_widthY", "fc1_widthY", "fc2_widthY", "fc3_widthY",
140  "cpuTime", "Energy", "PDG_ID", "RunNo", "EventNo", "E_Dep" };
141 
142  double lim[162][2] = { {0,5}, {0,100}, {-4,4}, {-1600,1600}, {-1600,1600}, {-4000,4000},
143  {0,0.25}, {0,3}, {0,6}, {0,0.1}, {0,0.25}, {0,2}, {0,3}, {0,0.1}, {0,1}, {0,10}, {0,10}, {0,10}, {0,0.1}, {0,0.1}, {0,0.1}, //cell
144  {0,100}, {0,600}, {0,600}, {0,50}, {0,50}, {0,400}, {0,500}, {0,200}, {0,100}, {0,1000}, {0,1000}, {0,100}, {0,150}, {0,50}, {0,10}, //hits
145  {0,0.4}, {0,5}, {0,10}, {0,0.2}, {0,0.1}, {0,3}, {0,5}, {0,0.2}, {0,1}, {0,10}, {0,10}, {0,0.1}, {0,1}, {0,0.1}, {0,0.1}, //sumE
146  {-500,500}, {-100,100}, {-15,15}, {-200,200}, {-3000,3000}, {-60,60}, {-25,25}, {-200,200}, {-50,50}, {-50,50}, {-50,50}, {-50,50}, {-60,60}, {-500,500}, {-200,200}, //dPhi
147  {0,1000}, {0,500}, {0,200}, {0,500}, {0,2000}, {0,250}, {0,300}, {0,500}, {0,200}, {0,200}, {0,200}, {0,200}, {0,100}, {0,100}, {0,50}, //sPhi
148  {-150,150}, {-15,15}, {-20,20}, {-200,200}, {-0,2500}, {-50,20}, {-15,15}, {-150,150}, {-50,50}, {-50,50}, {-50,50}, {-50,50}, {-60,60}, {-500,500}, {-200,200}, //dEta
149  {0,500}, {0,100}, {0,100}, {0,400}, {0,1000}, {0,150}, {0,100}, {0,400}, {0,200}, {0,200}, {0,200}, {0,200}, {0,60}, {0,100}, {0,50}, //sPhi
150  {0,750}, {0,25}, {0,20}, {0,1000}, {0,10000}, {0,40}, {0,30}, {0,1000}, {0,1000}, {0,100}, {0,100}, {0,200}, {0,10}, {0,500}, {0,100}, //time
151  {-150,150}, {-15,15}, {-20,20}, {-200,200}, {-0,2500}, {-50,20}, {-15,15}, {-150,150}, {-50,50}, {-50,50}, {-50,50}, {-50,50}, {-60,60}, {-500,500}, {-200,200}, //widthX
152  {-150,150}, {-15,15}, {-20,20}, {-200,200}, {-0,2500}, {-50,20}, {-15,15}, {-150,150}, {-50,50}, {-50,50}, {-50,50}, {-50,50}, {-60,60}, {-500,500}, {-200,200}, //widthY
153  {0,50}, {0,100}, {-25,25}, {0,10}, {0,1000}, {0,10}};
154 
155  //-------------------------------------------------------------------------//
156  // //
157  // Initialize the Particle Property Service. This is necessary in order //
158  // to obtain charge & type & other properties of the primary particle and //
159  // other particles that may turn up in the debris. //
160  // //
161  ATH_CHECK(m_ppSvc.retrieve());
162  m_c->partPropSvc = m_ppSvc.get();
163  ATH_CHECK(m_histSvc.retrieve());
164  m_c->histSvc = m_histSvc.get();
165  ATH_CHECK(detStore()->retrieve(m_c->cellId, "CaloCell_ID"));
169 
170  //----------------Now initialize the ntuples ----------------------//
171  // //
172  //==~ ~ ~==//
173  std::string filename="";
174  for (int i=0;i<162;i++){
175  m_histos[i] = new TH1F( names[i].data(), names[i].data(),50,lim[i][0],lim[i][1]);
176  filename = "/file1/Electron/";
177  filename.append(names[i]);
178  if (m_c->histSvc->regHist( filename.data() , m_histos[i] ).isFailure()){
179  ATH_MSG_WARNING( "Failed to register historam " << names[i] << ". Not sure what will happen now..." );
180  }
181  }
182 
183 
184  NTupleFilePtr file(ntupleSvc(),"/NTUPLES/FILE");
185  if (!file) throw std::runtime_error ("Ntuple MGR not open");
186  NTuple::Directory *col=ntupleSvc()->createDirectory("/NTUPLES/FILE/COL");
187  NTuplePtr nt(ntupleSvc(),"/NTUPLES/FILE/COL/SingleTrackValidation");
188  if (!nt) nt=ntupleSvc()->book(col, 1, CLID_ColumnWiseTuple, "SingleTrackValidation");
189 
190  if (nt->addItem("Eta", m_c->eta ).isFailure() ||
191  nt->addItem("Pt", m_c->pt ).isFailure() ||
192  nt->addItem("BarrelX", m_c->x ).isFailure() ||
193  nt->addItem("BarrelY", m_c->y ).isFailure() ||
194  nt->addItem("BarrelZ", m_c->z ).isFailure() ||
195  nt->addItem("Phi", m_c->phi ).isFailure() ){
196  ATH_MSG_WARNING( "Registration of some of the ntuple branches failed. No idea what will happen next..." );
197  }
198 
199 
200  char title[80];
201 
202  // Indices from 0 to 11: 4 EMB, 4 EMEC, and 4 FCAL sampling layers
203  // Handling FCAL layers separately for different titles (could do
204  // something more complicated if we wanted)
205  for (int i=0;i<15;i++){
206  if (i<12) sprintf(title,"S%i_C00",i);
207  else sprintf(title,"FC%i_C00",i-11);
208  if (nt->addItem(title,m_c->s_c00[i]).isFailure()) ATH_MSG_INFO( "Registration of a branch failed in the ntupler..." );
209  if (i<12) sprintf(title,"S%i_SumE",i);
210  else sprintf(title,"FC%i_SumE",i-11);
211  if (nt->addItem(title,m_c->s_sumE[i]).isFailure()) ATH_MSG_INFO( "Registration of a branch failed in the ntupler..." );
212  if (i<12) sprintf(title,"S%i_Hits",i);
213  else sprintf(title,"FC%i_Hits",i-11);
214  if (nt->addItem(title,m_c->s_hits[i]).isFailure()) ATH_MSG_INFO( "Registration of a branch failed in the ntupler..." );
215  if (i<12) sprintf(title,"S%i_DeltaPhi",i);
216  else sprintf(title,"FC%i_DeltaX",i-11);
217  if (nt->addItem(title,m_c->s_deltaPhi[i]).isFailure()) ATH_MSG_INFO( "Registration of a branch failed in the ntupler..." );
218  if (i<12) sprintf(title,"S%i_SigmaPhi",i);
219  else sprintf(title,"FC%i_SigmaX",i-11);
220  if (nt->addItem(title,m_c->s_sigmaPhi[i]).isFailure()) ATH_MSG_INFO( "Registration of a branch failed in the ntupler..." );
221  if (i<12) sprintf(title,"S%i_DeltaEta",i);
222  else sprintf(title,"FC%i_DeltaY",i-11);
223  if (nt->addItem(title,m_c->s_deltaEta[i]).isFailure()) ATH_MSG_INFO( "Registration of a branch failed in the ntupler..." );
224  if (i<12) sprintf(title,"S%i_SigmaEta",i);
225  else sprintf(title,"FC%i_SigmaY",i-11);
226  if (nt->addItem(title,m_c->s_sigmaEta[i]).isFailure()) ATH_MSG_INFO( "Registration of a branch failed in the ntupler..." );
227  if (i<12) sprintf(title,"S%i_Time",i);
228  else sprintf(title,"FC%i_Time",i-11);
229  if (nt->addItem(title,m_c->s_t00[i]).isFailure()) ATH_MSG_INFO( "Registration of a branch failed in the ntupler..." );
230  if (i<12) sprintf(title,"S%i_WidthX",i);
231  else sprintf(title,"FC%i_WidthX",i-11);
232  if (nt->addItem(title,m_c->s_widthX[i]).isFailure()) ATH_MSG_INFO( "Registration of a branch failed in the ntupler..." );
233  if (i<12) sprintf(title,"S%i_WidthY",i);
234  else sprintf(title,"FC%i_WidthY",i-11);
235  if (nt->addItem(title,m_c->s_widthY[i]).isFailure()) ATH_MSG_INFO( "Registration of a branch failed in the ntupler..." );
236  }
237 
238  if (nt->addItem("CPU" , m_c->cpuTime ).isFailure() ||
239  nt->addItem("TrackEnergy" , m_c->Energy ).isFailure() ||
240  nt->addItem("ParticleID" , m_c->PDG ).isFailure() ||
241  nt->addItem("Run#" , m_c->RunNo ).isFailure() ||
242  nt->addItem("Event#" , m_c->EventNo ).isFailure() ||
243  nt->addItem("DepositedEnergy", m_c->E_Deposit ).isFailure() ){
244  ATH_MSG_WARNING( "Registration of some of the ntuple branches failed. No idea what will happen next..." );
245  }
246 
247  m_c->cpuTime=0.0;
248  m_c->nt = nt;
249 
250  //==~ ~ ~==//
251  // //
252  //------------------------Done with initializations------------------------//
253 
254  return StatusCode::SUCCESS;
255 }
256 
258 
259  if (m_c->cpuTime==0) {
260  m_c->cpuTime=getCpu();
261  m_c->cpuTime+=1; // Fill the histogram with -1 for the first event
262  }
264  m_histos[156]->Fill( m_c->cpuTime/100. , 1. );
265 
266  const EventContext& context = getContext();
267  int RunNum=context.eventID().run_number();
268  int EvtNum=context.eventID().event_number();
269  double RunStr=double(RunNum);
270  double EvtStr=double(EvtNum);
271  m_c->EventNo=EvtStr;
272  m_c->RunNo=RunStr;
273  m_histos[160]->Fill(EvtStr);
274  m_histos[159]->Fill(RunNum);
275 
276  MagField::AtlasFieldCache fieldCache;
277  // Get field cache object
279  const AtlasFieldCacheCondObj* fieldCondObj{*readHandle};
280  if (fieldCondObj == nullptr) {
281  ATH_MSG_ERROR("Failed to retrieve AtlasFieldCacheCondObj with key " << m_fieldCacheCondObjInputKey.key());
282  return StatusCode::FAILURE;
283  }
284  fieldCondObj->getInitializedCache (fieldCache);
285 
286  // Get the MC Truth Information
288  for (const HepMC::GenEvent* e : *mcEvent) {
289 
290  // Get just the primary, call it "theParticle"
291  auto theParticle = *HepMC::begin(*e);
292 
293  // Fetch whatever particle properties will be used in the following:
294  const HepPDT::ParticleDataTable * dataTable = m_c->partPropSvc->PDT();
295  const HepPDT::ParticleData * particleData = dataTable->particle(iabs(theParticle->pdg_id()));
296 
297  // Get the kinematic variables:
298  HepLorentzVector momentum(theParticle->momentum().px(),
299  theParticle->momentum().py(),
300  theParticle->momentum().pz(),
301  theParticle->momentum().e());
302  Point3D<double> origin(theParticle->production_vertex()->position().x(),
303  theParticle->production_vertex()->position().y(),
304  theParticle->production_vertex()->position().z());
305  double charge = theParticle->pdg_id() > 0 ? particleData->charge() : - particleData->charge();
306  // Put Eta and Phi into the Ntuple
307  m_c->phi = theParticle->momentum().phi();
308  m_c->eta = -log(tan(theParticle->momentum().theta()/2));
309  if (!finite(m_c->eta)) m_c->eta=0;
310  m_c->pt = theParticle->momentum().perp();
311  int partID = theParticle->pdg_id();
312  double pID = double(partID);
313  m_c->PDG = pID;
314  m_c->Energy = theParticle->momentum().e();
315  m_histos[2]->Fill( theParticle->momentum().phi() );
316  double myEta = -log(tan(theParticle->momentum().theta()/2));
317  if (!finite(myEta)) m_histos[0]->Fill(0);
318  else m_histos[0]->Fill( myEta );
319  m_histos[158]->Fill( pID );
320  m_histos[1]->Fill( theParticle->momentum().perp()/Units::GeV );
321  m_histos[157]->Fill( m_c->Energy = theParticle->momentum().e()/Units::GeV );
322 
323  // Make an extrapolator:
324  const Genfun::AtlasBComponent Bx(0,&fieldCache);
325  const Genfun::AtlasBComponent By(1,&fieldCache);
326  const Genfun::AtlasBComponent Bz(2,&fieldCache);
327  GeoXPEngine extrapolator(Bx, By, Bz, origin, momentum, charge);
328 
329  // Extrapolate to the first layer in the barrel:
330  m_c->x = 0;
331  m_c->y = 0;
332  m_c->z = 0;
333  double x=0,y=0,z=0;
334  bool hitsBarrel=false;
335  //bool hitsEndcap=false;
336  for (double t = 0; t< 50; t += 0.1) {
337  x = extrapolator.x()(t);
338  y = extrapolator.y()(t);
339  z = extrapolator.z()(t);
340  double magicZ=3640.0*mm;
341  double magicR=1500.0*mm;
342  if (x*x+y*y > magicR*magicR) {
343  m_c->x = x;
344  m_c->y = y;
345  m_c->z = z;
346  hitsBarrel=true;
347  break;
348  }
349  else if (z*z > magicZ*magicZ) {
350  m_c->x = x;
351  m_c->y = y;
352  m_c->z = z;
353  //hitsEndcap=true;
354  break;
355  }
356  }
357 
358  m_histos[3]->Fill( x );
359  m_histos[4]->Fill( y );
360  m_histos[5]->Fill( z );
361 
362  // You have an x,y, and z position. Now go and get the Element corresponding to
363  // that hit position. There are four, one for each sampling layer:
364  double radImpact = std::sqrt(x*x+y*y+z*z);
365  double phiImpact = std::atan2(y,x);
366  double thetaImpact = std::acos(z/radImpact);
367  double etaImpact = -std::log(std::tan(thetaImpact/2));
368 
370  ATH_CHECK(caloMgrHandle.isValid());
371  const CaloDetDescrManager* caloMgr = *caloMgrHandle;
372  const CaloDetDescrElement *element[15]={nullptr};
373 
374  for (int i=0;i<4;i++) {
375  try {
376  element[i] = caloMgr->get_element(CaloCell_ID::LAREM,i, hitsBarrel, etaImpact, phiImpact);
377  }
378  catch (const LArID_Exception & e) {
379  std::cerr << "SingleTrackValidation EXCEPTION (LAREM)" << e.message() << std::endl;
380  }
381  }
382  for (int i=0;i<4;i++) {
383  try {
384  element[i+4] = caloMgr->get_element(CaloCell_ID::LAREM,i, hitsBarrel, etaImpact, phiImpact);
385  }
386  catch (const LArID_Exception & e) {
387  std::cerr << "SingleTrackValidation EXCEPTION (LAREM)" << e.message() << std::endl;
388  }
389  }
390  for (int i=0;i<4;i++) {
391  try {
392  element[i+8] = caloMgr->get_element(CaloCell_ID::LARHEC,i, hitsBarrel, etaImpact, phiImpact);
393  }
394  catch (const LArID_Exception & e) {
395  std::cerr << "SingleTrackValidation EXCEPTION in (LARHEC)" << e.message() << std::endl;
396  }
397  }
398  for (int i=1;i<4;i++) {
399  try {
400  element[i+11] = caloMgr->get_element(CaloCell_ID::LARFCAL,i, hitsBarrel, etaImpact, phiImpact);
401  }
402  catch (const LArID_Exception & e) {
403  std::cerr << "SingleTrackValidation EXCEPTIONin LARFCAL" << e.message() << std::endl;
404  }
405  }
406 
407 
408  // Now go and find out how much energy is there:
409  for (int z=0;z<15;z++){
410  m_c->s_c00[z]=0;
411  m_c->s_t00[z]=0;
412  }
413 
414  std::string lArKey = hitsBarrel ? "LArHitEMB" : "LArHitEMEC" ;
415 
416  double eSum [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
417  double eEta [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
418  double eEtaEta [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
419  double ePhi [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
420  double ePhiPhi [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
421  int hit_count [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
422  double eX [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
423  double eXX [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
424  double eY [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
425  double eYY [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
426  double c00 [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
427  double t00 [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
428  // double width [15]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
429  double e_dep = 0;
430 
431  for (int i=0;i<4;i++) { // Loop over the four LAr collections
432  if (i==0) {
433  lArKey="LArHitEMB";
434  } else if (i==1) {
435  lArKey="LArHitEMEC";
436  } else if (i==2) {
437  lArKey="LArHitHEC";
438  } else if (i==3) {
439  lArKey="LArHitFCAL";
440  }
441 
442  SG::ReadHandle<LArHitContainer> larHitContainer(lArKey, "StoreGateSvc");
443  if (larHitContainer.isValid()) {
444  for (const LArHit* larHit : *larHitContainer) {
445  const CaloDetDescrElement *hitElement = caloMgr->get_element(larHit->cellID());
446  int samplingLayer = m_c->cellId->sampling(larHit->cellID());
447  double energy = larHit->energy();
448 
449  for (int j=0;j<15;j++) {
450  if (hitElement==element[j]) {
451  c00[j] += energy;
452  }
453  }
454 
455  if (lArKey=="LArHitEMEC") samplingLayer += 4;
456  else if (lArKey=="LArHitHEC") samplingLayer += 8;
457  else if (lArKey=="LArHitFCAL") samplingLayer += 11;
458 
459  // Calculate phi of hit w.r.t phiImpact
460  // to avoid problems due to the 2pi modulus of phi
461  // we are thus calculating deltaphi directly, so don't subtract phiImpact again later
462  double hitPhi = hitElement->phi() - phiImpact;
463  if (hitPhi < -pi) hitPhi += twopi;
464  if (hitPhi > pi) hitPhi -= twopi;
465 
466  eSum [samplingLayer]+=energy;
467  eEta [samplingLayer]+=energy*hitElement->eta();
468  eEtaEta [samplingLayer]+=energy*hitElement->eta()*hitElement->eta();
469  ePhi [samplingLayer]+=energy*hitPhi;
470  ePhiPhi [samplingLayer]+=energy*hitPhi*hitPhi;
471  eX [samplingLayer]+=energy*hitElement->x();
472  eXX [samplingLayer]+=energy*hitElement->x()*hitElement->x();
473  eY [samplingLayer]+=energy*hitElement->y();
474  eYY [samplingLayer]+=energy*hitElement->y()*hitElement->y();
475  t00 [samplingLayer]+=energy*larHit->time();
476  hit_count[samplingLayer]+=1;
477  }
478  }
479  }
480 
481  for (int i=0;i<15;i++) {
482  if (eSum[i]!=0) eEta[i] /= eSum[i];
483  if (eSum[i]!=0) eEtaEta[i] /= eSum[i];
484  if (eSum[i]!=0) ePhi[i] /= eSum[i];
485  if (eSum[i]!=0) ePhiPhi[i] /= eSum[i];
486  if (eSum[i]!=0) eY[i] /= eSum[i];
487  if (eSum[i]!=0) eYY[i] /= eSum[i];
488  if (eSum[i]!=0) eX[i] /= eSum[i];
489  if (eSum[i]!=0) eXX[i] /= eSum[i];
490  if (eSum[i]!=0) t00[i] /= eSum[i];
491  e_dep+=eSum[i];
492  }
493 
494  double dThetaDEta = -1.0/cosh(etaImpact);
495 
496  //Fill the E Sum, center cell E, hit count fields:
497  m_c->E_Deposit=e_dep;
498  for (int z=0;z<15;z++){
499  m_c->s_sumE[z]=eSum[z];
500  m_c->s_c00[z]=c00[z];
501  m_c->s_t00[z]=t00[z];
502  m_c->s_hits[z]=hit_count[z];
503  if (z<12){
504  m_c->s_deltaPhi[z]=radImpact*std::sin(thetaImpact)*(ePhi[z]);
505  m_c->s_sigmaPhi[z]=radImpact*std::sin(thetaImpact)*std::sqrt(ePhiPhi[z]- ePhi[z]*ePhi[z]);
506  m_c->s_deltaEta[z]=radImpact*dThetaDEta*(eEta[z]-etaImpact);
507  m_c->s_sigmaEta[z]=radImpact*std::fabs(dThetaDEta)*std::sqrt(eEtaEta[z]- eEta[z]*eEta[z]);
508  } else {
509  m_c->s_deltaPhi[z]=(eX[z]-x);
510  m_c->s_sigmaPhi[z]=std::sqrt(eXX[z]- eX[z]*eX[z]);
511  m_c->s_deltaEta[z]=(eY[z]-y);
512  m_c->s_sigmaEta[z]=std::sqrt(eYY[z]-eY[z]*eY[z]);
513  }
514  m_c->s_widthX[z]=std::sqrt(eXX[z]-eX[z]*eX[z]);
515  m_c->s_widthY[z]=std::sqrt(eYY[z]-eY[z]*eY[z]);
516  }
517 
518  m_histos[161]->Fill(e_dep/Units::GeV);
519  for (int i=0;i<15;i++){
520  m_histos[6+i]->Fill( c00[i]/Units::GeV );
521  m_histos[21+i]->Fill( hit_count[i] );
522  m_histos[36+i]->Fill( eSum[i]/Units::GeV );
523  m_histos[111+i]->Fill( t00[i] );
524  m_histos[126+i]->Fill( sqrt(eXX[i]-eX[i]*eX[i]) );
525  m_histos[141+i]->Fill( sqrt(eYY[i]-eY[i]*eY[i]) );
526  if (i<8){
527  m_histos[51+i]->Fill( radImpact*std::sin(thetaImpact)*ePhi[i] );
528  m_histos[66+i]->Fill( radImpact*std::sin(thetaImpact)*std::sqrt(ePhiPhi[i]-ePhi[i]*ePhi[i]) );
529  m_histos[81+i]->Fill( radImpact*dThetaDEta*(eEta[i]-etaImpact) );
530  m_histos[96+i]->Fill( radImpact*std::fabs(dThetaDEta)*std::sqrt(eEtaEta[i]-eEta[i]*eEta[i]) );
531  } else {
532  m_histos[51+i]->Fill( eX[i]-x );
533  m_histos[66+i]->Fill( std::sqrt(eXX[i]-eX[i]*eX[i]) );
534  m_histos[81+i]->Fill( eY[i]-y );
535  m_histos[96+i]->Fill( std::sqrt(eYY[i]-eY[i]*eY[i]) );
536  }
537  }
538 
539  ATH_CHECK(ntupleSvc()->writeRecord(m_c->nt));
540 
541  }
542 
543  m_c->cpuTime= getCpu();
544 
545  return StatusCode::SUCCESS;
546 
547 }
548 
550  return StatusCode::SUCCESS;
551 }
python.PyKernel.retrieve
def retrieve(aClass, aKey=None)
Definition: PyKernel.py:110
SingleTrackValidation::Clockwork::eta
NTuple::Item< double > eta
Definition: SingleTrackValidation.cxx:76
AllowedVariables::e
e
Definition: AsgElectronSelectorTool.cxx:37
SingleTrackValidation::Clockwork::z
NTuple::Item< double > z
Definition: SingleTrackValidation.cxx:83
get_hdefs.buff
buff
Definition: get_hdefs.py:61
data
char data[hepevt_bytes_allocation_ATLAS]
Definition: HepEvt.cxx:11
CaloCell_Base_ID::LARFCAL
@ LARFCAL
Definition: CaloCell_Base_ID.h:46
GeV
#define GeV
Definition: PhysicsAnalysis/TauID/TauAnalysisTools/Root/HelperFunctions.cxx:17
SingleTrackValidation::Clockwork::s_widthX
NTuple::Item< double > s_widthX[15]
Definition: SingleTrackValidation.cxx:95
SingleTrackValidation::execute
StatusCode execute() override
Definition: SingleTrackValidation.cxx:257
SingleTrackValidation::m_ppSvc
ServiceHandle< IPartPropSvc > m_ppSvc
Definition: SingleTrackValidation.h:45
SG::ReadCondHandle
Definition: ReadCondHandle.h:44
ATH_MSG_INFO
#define ATH_MSG_INFO(x)
Definition: AthMsgStreamMacros.h:31
CaloDetDescrElement::y
float y() const
cell y
Definition: Calorimeter/CaloDetDescr/CaloDetDescr/CaloDetDescrElement.h:365
AtlasFieldCacheCondObj
Definition: AtlasFieldCacheCondObj.h:19
SingleTrackValidation::Clockwork::cellId
const CaloCell_ID * cellId
Definition: SingleTrackValidation.cxx:73
python.CreateTierZeroArgdict.partID
partID
Definition: CreateTierZeroArgdict.py:135
SingleTrackValidation::Clockwork::pt
NTuple::Item< double > pt
Definition: SingleTrackValidation.cxx:77
SingleTrackValidation::Clockwork::s_sumE
NTuple::Item< double > s_sumE[15]
Definition: SingleTrackValidation.cxx:89
SG::ReadHandle< McEventCollection >
SingleTrackValidation::m_c
Clockwork * m_c
Definition: SingleTrackValidation.h:47
SingleTrackValidation::Clockwork::coolField
bool coolField
Definition: SingleTrackValidation.cxx:105
SingleTrackValidation::Clockwork::s_t00
NTuple::Item< double > s_t00[15]
Definition: SingleTrackValidation.cxx:94
CaloDetDescrElement
This class groups all DetDescr information related to a CaloCell. Provides a generic interface for al...
Definition: Calorimeter/CaloDetDescr/CaloDetDescr/CaloDetDescrElement.h:66
LArID_Exception.h
CaloDetDescrManager_Base::get_element
const CaloDetDescrElement * get_element(const Identifier &cellId) const
get element by its identifier
Definition: CaloDetDescrManager.cxx:159
CSV_InDetExporter.new
new
Definition: CSV_InDetExporter.py:145
SingleTrackValidation::m_histos
TH1F * m_histos[162]
Definition: SingleTrackValidation.h:50
CaloCell_Base_ID::LARHEC
@ LARHEC
Definition: CaloCell_Base_ID.h:46
SingleTrackValidation::Clockwork::phi
NTuple::Item< double > phi
Definition: SingleTrackValidation.cxx:78
read_hist_ntuple.t
t
Definition: read_hist_ntuple.py:5
SG::VarHandleKey::key
const std::string & key() const
Return the StoreGate ID for the referenced object.
Definition: AthToolSupport/AsgDataHandles/Root/VarHandleKey.cxx:141
GeoXPEngine::z
const Genfun::AbsFunction & z() const
Definition: GeoXPEngine.cxx:89
SingleTrackValidation::Clockwork::s_deltaPhi
NTuple::Item< double > s_deltaPhi[15]
Definition: SingleTrackValidation.cxx:90
x
#define x
SingleTrackValidation::Clockwork::PDG
NTuple::Item< double > PDG
Definition: SingleTrackValidation.cxx:100
ReadCondHandle.h
pi
#define pi
Definition: TileMuonFitter.cxx:65
CaloCell_ID.h
AthCommonDataStore< AthCommonMsg< Algorithm > >::detStore
const ServiceHandle< StoreGateSvc > & detStore() const
The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.
Definition: AthCommonDataStore.h:95
SingleTrackValidation::Clockwork::Energy
NTuple::Item< double > Energy
Definition: SingleTrackValidation.cxx:99
SingleTrackValidation::Clockwork::EventNo
NTuple::Item< double > EventNo
Definition: SingleTrackValidation.cxx:102
SingleTrackValidation::Clockwork::s_sigmaEta
NTuple::Item< double > s_sigmaEta[15]
Definition: SingleTrackValidation.cxx:93
SingleTrackValidation::Clockwork::partPropSvc
IPartPropSvc * partPropSvc
Definition: SingleTrackValidation.cxx:71
SingleTrackValidation::initialize
StatusCode initialize() override
Definition: SingleTrackValidation.cxx:118
SingleTrackValidation::Clockwork::x
NTuple::Item< double > x
Definition: SingleTrackValidation.cxx:81
ATH_MSG_ERROR
#define ATH_MSG_ERROR(x)
Definition: AthMsgStreamMacros.h:33
ParticleGun_FastCalo_ChargeFlip_Config.energy
energy
Definition: ParticleGun_FastCalo_ChargeFlip_Config.py:78
ParticleGun_EoverP_Config.momentum
momentum
Definition: ParticleGun_EoverP_Config.py:63
SingleTrackValidation::Clockwork::y
NTuple::Item< double > y
Definition: SingleTrackValidation.cxx:82
lumiFormat.i
int i
Definition: lumiFormat.py:85
z
#define z
SingleTrackValidation::m_caloMgrKey
SG::ReadCondHandleKey< CaloDetDescrManager > m_caloMgrKey
Definition: SingleTrackValidation.h:34
CaloCell_Base_ID::sampling
int sampling(const Identifier id) const
LAr field values (NOT_VALID == invalid request)
Genfun::AtlasBComponent
Definition: AtlasBComponent.h:14
EL::StatusCode
::StatusCode StatusCode
StatusCode definition for legacy code.
Definition: PhysicsAnalysis/D3PDTools/EventLoop/EventLoop/StatusCode.h:22
python.subdetectors.mmg.names
names
Definition: mmg.py:8
covarianceTool.title
title
Definition: covarianceTool.py:542
SingleTrackValidation::Clockwork::histSvc
ITHistSvc * histSvc
Definition: SingleTrackValidation.cxx:72
file
TFile * file
Definition: tile_monitor.h:29
ATH_CHECK
#define ATH_CHECK
Definition: AthCheckMacros.h:40
GeoXPEngine.h
CaloCell_ID
Helper class for offline cell identifiers.
Definition: CaloCell_ID.h:34
drawFromPickle.tan
tan
Definition: drawFromPickle.py:36
SG::VarHandleKey::initialize
StatusCode initialize(bool used=true)
If this object is used as a property, then this should be called during the initialize phase.
Definition: AthToolSupport/AsgDataHandles/Root/VarHandleKey.cxx:103
xAOD::double
double
Definition: CompositeParticle_v1.cxx:159
SingleTrackValidation::m_truthKey
SG::ReadHandleKey< McEventCollection > m_truthKey
Definition: SingleTrackValidation.h:32
AthAlgorithm
Definition: AthAlgorithm.h:47
SG::ReadHandle::isValid
virtual bool isValid() override final
Can the handle be successfully dereferenced?
SingleTrackValidation::m_histSvc
ServiceHandle< ITHistSvc > m_histSvc
Definition: SingleTrackValidation.h:46
SingleTrackValidation::m_fieldCacheCondObjInputKey
SG::ReadCondHandleKey< AtlasFieldCacheCondObj > m_fieldCacheCondObjInputKey
Definition: SingleTrackValidation.h:40
SingleTrackValidation.h
Athena::Units
Definition: Units.h:45
twopi
constexpr double twopi
Definition: VertexPointEstimator.cxx:16
name
std::string name
Definition: Control/AthContainers/Root/debug.cxx:228
GeoXPEngine::y
const Genfun::AbsFunction & y() const
Definition: GeoXPEngine.cxx:86
charge
double charge(const T &p)
Definition: AtlasPID.h:756
SingleTrackValidation::Clockwork::s_hits
NTuple::Item< double > s_hits[15]
Definition: SingleTrackValidation.cxx:88
SingleTrackValidation::Clockwork::Clockwork
Clockwork()
Definition: SingleTrackValidation.cxx:68
SG::CondHandleKey::initialize
StatusCode initialize(bool used=true)
LArHit
Class to store hit energy and time in LAr cell from G4 simulation.
Definition: LArHit.h:25
Units.h
Wrapper to avoid constant divisions when using units.
query_example.col
col
Definition: query_example.py:7
CaloDetDescrElement::x
float x() const
cell x
Definition: Calorimeter/CaloDetDescr/CaloDetDescr/CaloDetDescrElement.h:363
GeoXPEngine::x
const Genfun::AbsFunction & x() const
Definition: GeoXPEngine.cxx:83
GeoXPEngine
Definition: GeoXPEngine.h:22
python.SystemOfUnits.mm
int mm
Definition: SystemOfUnits.py:83
SingleTrackValidation::Clockwork::cpuTime
NTuple::Item< double > cpuTime
Definition: SingleTrackValidation.cxx:98
SingleTrackValidation::Clockwork::RunNo
NTuple::Item< double > RunNo
Definition: SingleTrackValidation.cxx:101
SingleTrackValidation::Clockwork::E_Deposit
NTuple::Item< double > E_Deposit
Definition: SingleTrackValidation.cxx:103
a
TList * a
Definition: liststreamerinfos.cxx:10
SingleTrackValidation::Clockwork::s_c00
NTuple::Item< double > s_c00[15]
Definition: SingleTrackValidation.cxx:87
y
#define y
CaloDetDescrManager
This class provides the client interface for accessing the detector description information common to...
Definition: CaloDetDescrManager.h:473
ATH_MSG_WARNING
#define ATH_MSG_WARNING(x)
Definition: AthMsgStreamMacros.h:32
LArHitContainer.h
SingleTrackValidation::Clockwork::s_deltaEta
NTuple::Item< double > s_deltaEta[15]
Definition: SingleTrackValidation.cxx:92
SingleTrackValidation::SingleTrackValidation
SingleTrackValidation(const std::string &name, ISvcLocator *pSvcLocator)
Definition: SingleTrackValidation.cxx:108
SingleTrackValidation::Clockwork::nt
NTuple::Tuple * nt
Definition: SingleTrackValidation.cxx:74
iabs
int iabs(int a)
Definition: SingleTrackValidation.cxx:52
MagField::AtlasFieldCache
Local cache for magnetic field (based on MagFieldServices/AtlasFieldSvcTLS.h)
Definition: AtlasFieldCache.h:43
python.CaloCondTools.log
log
Definition: CaloCondTools.py:20
CaloCellTimeCorrFiller.filename
filename
Definition: CaloCellTimeCorrFiller.py:24
AtlasBComponent.h
HepMC::begin
GenEvent::particle_iterator begin(HepMC::GenEvent &e)
Definition: GenEvent.h:620
SingleTrackValidation::finalize
StatusCode finalize() override
Definition: SingleTrackValidation.cxx:549
SingleTrackValidation::Clockwork::s_sigmaPhi
NTuple::Item< double > s_sigmaPhi[15]
Definition: SingleTrackValidation.cxx:91
CaloDetDescrElement::eta
float eta() const
cell eta
Definition: Calorimeter/CaloDetDescr/CaloDetDescr/CaloDetDescrElement.h:344
SingleTrackValidation::Clockwork::~Clockwork
~Clockwork()
Definition: SingleTrackValidation.cxx:69
CaloDetDescrElement::phi
float phi() const
cell phi
Definition: Calorimeter/CaloDetDescr/CaloDetDescr/CaloDetDescrElement.h:346
python.TrigEgammaMonitorHelper.TH1F
def TH1F(name, title, nxbins, bins_par2, bins_par3=None, path='', **kwargs)
Definition: TrigEgammaMonitorHelper.py:24
drawFromPickle.sin
sin
Definition: drawFromPickle.py:36
beamspotnt.nt
def nt
Definition: bin/beamspotnt.py:1063
SingleTrackValidation::~SingleTrackValidation
~SingleTrackValidation()
Definition: SingleTrackValidation.cxx:114
CaloCell_Base_ID::LAREM
@ LAREM
Definition: CaloCell_Base_ID.h:46
ntupleSvc
INTupleSvc * ntupleSvc()
Definition: ServiceAccessor.h:14
StoreGateSvc.h
LArID_Exception
Exception class for LAr Identifiers.
Definition: LArID_Exception.h:20
SingleTrackValidation::Clockwork
Definition: SingleTrackValidation.cxx:65
getCpu
int getCpu()
Definition: SingleTrackValidation.cxx:53
plot_times.times
def times(fn)
Definition: plot_times.py:11
SingleTrackValidation::Clockwork::s_widthY
NTuple::Item< double > s_widthY[15]
Definition: SingleTrackValidation.cxx:96