Loading [MathJax]/extensions/tex2jax.js
ATLAS Offline Software
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
Functions
LayerMaterialUtils Namespace Reference

Functions

def drawCylinderLayer (g4CylTree, cylLayer, tgCylTree, saveLocation)
 
def drawDiscLayer (g4DiscTree, discLayer, tgDiscTree, saveLocation)
 

Function Documentation

◆ drawCylinderLayer()

def LayerMaterialUtils.drawCylinderLayer (   g4CylTree,
  cylLayer,
  tgCylTree,
  saveLocation 
)

Definition at line 6 of file LayerMaterialUtils.py.

6 def drawCylinderLayer(g4CylTree,cylLayer,tgCylTree,saveLocation) :
7 
8  gROOT.cd()
9  # create a canvas first
10  clname = cylLayer.GetName()
11 
12  # now fill it
13  for evt in cylLayer :
14 
15  radius = evt.cylRadius
16  position = evt.cylPosZ
17  halfz = evt.cylHalfZ
18  binsz = evt.cylMatBinsZ
19  binsphi = evt.cylMatBinsPhi
20 
21  numbinsz = 100
22  numbinsphi = 100
23  if binsz > numbinsz : numbinsz = binsz
24  if binsphi > numbinsphi : numbinsphi = binsphi
25 
26 
27  print ('----------------------------------------------------------------')
28  print (' Layer: ', clname, ' ( LayerIndex : ',evt.cylLayerIndex, ')')
29  print (' r, halflength : ', radius, halfz)
30  print (' bins (z/phi) : ', evt.cylMatBinsZ, evt.cylMatBinsPhi)
31 
32 
33  # create the x0 histogram
34  clmap_x0 = TH2F('clmap_x0_'+clname,clname,binsz,-halfz,halfz,binsphi,-math.pi,math.pi)
35  x0max = 0.
36  # loop over bins and fill it
37  for ibin in range(evt.cylMatBinsTotal) :
38  # first get the ibin phi
39  ibinz = ibin/numbinsphi
40  ibinphi = ibin - ibinz*numbinsphi
41  x0 = evt.cylMatX0[ibin]
42  if x0 > x0max : x0max = x0
43  clmap_x0.SetBinContent(ibinz+1,ibinphi+1,x0)
44  pass
45 
46  # and draw it
47  canvasX0map = TCanvas(clname+'_x0',clname,100,100,800,1000)
48  canvasX0map.Divide(2,3)
49 
50  # ---------------------------------------------------------
51  # draw profiles first to get a good estimate for the maximum
52  canvasX0map.cd(5)
53  x0profZG4 = TProfile(clname+'x0profZG4',clname+'x0profZG4',numbinsz,-halfz+position,halfz+position) # noqa: F841
54  x0profZTG = TProfile(clname+'x0profZTG',clname+'x0profZTG',numbinsz,-halfz+position,halfz+position) # noqa: F841
55  g4CylTree.Draw('PathInX0:DensedHitZ>>'+clname+'x0profZG4','','prof')
56  tgCylTree.Draw('PathInX0:DensedHitZ>>'+clname+'x0profZTG','','prof')
57  x0profG4z = gDirectory.Get(clname+'x0profZG4')
58  x0profTGz = gDirectory.Get(clname+'x0profZTG')
59  # style
60  x0profTGz.SetMarkerStyle(8)
61  x0profTGz.SetMarkerSize(0.5)
62  x0profTGz.SetMarkerColor(kRed)
63  x0profTGz.SetLineColor(kRed)
64  x0profG4z.GetYaxis().SetTitle('t [X_{0}]')
65  x0profG4z.GetXaxis().SetTitle('z [mm]')
66  x0maxProf = 1.5*x0profG4z.GetMaximum()
67  x0profG4z.GetYaxis().SetRangeUser(0.,x0maxProf)
68  x0profG4z.Draw('hist')
69  x0profTGz.SetMarkerColor(kRed)
70  x0profTGz.SetMarkerSize(0.8)
71  x0profTGz.Draw('pe,same')
72  canvasX0map.cd(6)
73  x0profPhiG4 = TProfile(clname+'x0profPhiG4',clname+'x0profPhiG4',numbinsphi,-math.pi,math.pi) # noqa: F841
74  x0profPhiTG = TProfile(clname+'x0profPhiTG',clname+'x0profPhiTG',numbinsphi,-math.pi,math.pi) # noqa: F841
75  g4CylTree.Draw('PathInX0:DensedHitPhi>>'+clname+'x0profPhiG4','','prof')
76  tgCylTree.Draw('PathInX0:DensedHitPhi>>'+clname+'x0profPhiTG','','prof')
77  x0profG4phi = gDirectory.Get(clname+'x0profPhiG4')
78  x0profTGphi = gDirectory.Get(clname+'x0profPhiTG')
79  # style
80  x0profTGphi.SetMarkerStyle(8)
81  x0profTGphi.SetMarkerSize(0.5)
82  x0profTGphi.SetMarkerColor(kRed)
83  x0profTGphi.SetLineColor(kRed)
84  x0profG4phi.GetXaxis().SetTitle('#phi')
85  x0profG4phi.GetYaxis().SetTitle('t [X_{0}]')
86  if 1.5*x0profG4phi.GetMaximum() > x0maxProf : x0maxProf = 1.5*x0profG4phi.GetMaximum()
87  x0profG4phi.GetYaxis().SetRangeUser(0.,1.5*x0profG4phi.GetMaximum())
88  x0profG4phi.Draw('hist')
89  x0profTGphi.SetMarkerColor(kRed)
90  x0profTGphi.SetMarkerSize(0.8)
91  x0profTGphi.Draw('pe,same')
92  # ---- now the maps
93  canvasX0map.cd(1)
94  try :
95  g4CylTree.Draw('PathInX0>>'+clname+'range')
96  x0range = gDirectory.Get(clname+'range')
97  x0maxProf = 1.2*x0range.GetXaxis().GetXmax()
98  except TypeError :
99  print (' -> TTree does not exist in TrackingGeometry ')
100  x0mapG4 = TH3F(clname+'x0mapG4',clname+'x0mapG4', numbinsphi,-math.pi,math.pi,numbinsz,-halfz+position,halfz+position,50,0.,x0maxProf)
101  try :
102  g4CylTree.Draw('PathInX0:DensedHitZ:DensedHitPhi>>'+clname+'x0mapG4','','')
103  x0mapG4 = gDirectory.Get(clname+'x0mapG4')
104  x0mapG4Prof = x0mapG4.Project3DProfile('xy')
105  x0mapG4Prof.GetXaxis().SetTitle('z [mm]')
106  x0mapG4Prof.GetYaxis().SetTitle('#phi')
107  x0mapG4Prof.GetZaxis().SetTitle('t [X_{0}]')
108  x0mapG4Prof.GetZaxis().SetTitleOffset(1.5*x0mapG4Prof.GetZaxis().GetTitleOffset())
109  x0mapG4Prof.GetZaxis().SetRangeUser(0.,x0maxProf)
110  x0mapG4Prof.SetBit(kCanDelete)
111  x0mapG4Prof.SetBit(kMustCleanup)
112  x0mapG4Prof.Draw('colz')
113  except TypeError :
114  print (' -> TTree does not exist in TrackingGeometry ')
115  canvasX0map.cd(2)
116  x0mapTG = TH3F(clname+'x0mapTG',clname+'x0mapTG',numbinsphi,-math.pi,math.pi,numbinsz,-halfz+position,halfz+position,50,0.,x0maxProf)
117  try :
118  tgCylTree.Draw('PathInX0:DensedHitZ:DensedHitPhi>>'+clname+'x0mapTG','','')
119  x0mapTG = gDirectory.Get(clname+'x0mapTG')
120  x0mapTGProf = x0mapTG.Project3DProfile('xy')
121  x0mapTGProf.GetXaxis().SetTitle('z [mm]')
122  x0mapTGProf.GetYaxis().SetTitle('#phi')
123  x0mapTGProf.GetZaxis().SetTitle('t [X_{0}]')
124  x0mapTGProf.GetZaxis().SetTitleOffset(1.5*x0mapTGProf.GetZaxis().GetTitleOffset())
125  x0mapTGProf.GetZaxis().SetRangeUser(0.,x0maxProf)
126  x0mapTGProf.SetBit(kCanDelete)
127  x0mapTGProf.SetBit(kMustCleanup)
128  x0mapTGProf.Draw('colz')
129  except TypeError :
130  print (' -> TTree does not exist in Geant4 ')
131  canvasX0map.cd(3)
132  clmap_x0.GetZaxis().SetRangeUser(0.,x0max)
133  clmap_x0.GetXaxis().SetTitle('z [mm]')
134  clmap_x0.GetYaxis().SetTitle('#phi')
135  clmap_x0.GetZaxis().SetTitle('t [X_{0}]')
136  clmap_x0.GetZaxis().SetTitleOffset(1.2*clmap_x0.GetZaxis().GetTitleOffset())
137  clmap_x0.Draw('colz')
138  # ---------------------------------------------------------
139  canvasX0map.cd(4)
140  try :
141  x0mapDiff = x0mapTGProf.Clone()
142  x0mapDiff.Add(x0mapG4Prof,-1)
143  x0mapDiff.GetXaxis().SetTitle('z [mm]')
144  x0mapDiff.GetYaxis().SetTitle('#phi')
145  x0mapDiff.GetZaxis().SetTitle('#Delta(t) [X_{0}]')
146  x0mapDiff.GetZaxis().SetRangeUser(-1.,1.)
147  x0mapDiff.GetZaxis().SetTitleOffset(1.5*x0mapDiff.GetZaxis().GetTitleOffset())
148  x0mapDiff.Divide(x0mapG4Prof)
149  x0mapDiff.Draw('colz')
150  except UnboundLocalError :
151  print (' -> No TrackingGeometry/G4 histograms to divide')
152 
153 
154  canvasX0map.SaveAs(saveLocation+'/LayerMaterial_CylinderLayer_'+str(evt.cylLayerIndex)+'.png')
155 
156  canvasAnalysis = TCanvas(clname+'_analyse',clname,100,100,600,1200)
157  canvasAnalysis.Divide(1,3)
158  canvasAnalysis.cd(1)
159  cylLayer.Draw('LayerMaterialL0:LayerMaterialX0','','box')
160  canvasAnalysis.cd(2)
161  cylLayer.Draw('LayerMaterialZARho:LayerMaterialX0','','box')
162  canvasAnalysis.cd(3)
163  cylLayer.Draw('LayerMaterialZARho:LayerMaterialL0','','box')
164  canvasAnalysis.SaveAs(saveLocation+'/'+clname+'_analysis.png')
165 
166  # memory cleanup
167  gDirectory.Delete(clname+'x0mapTG')
168  gDirectory.Delete(clname+'x0mapG4')
169  gDirectory.Delete('clmap_x0_'+clname)
170  gDirectory.Delete('clmap_x0_pz_'+clname)
171  gDirectory.Delete('clmap_x0_pphi_'+clname)
172 
173  #clmap_x0_profZ.Draw('hist')
174  #canvasX0map.cd(3)
175  #clmap_x0_profPhi.Draw('hist')
176 
177 

◆ drawDiscLayer()

def LayerMaterialUtils.drawDiscLayer (   g4DiscTree,
  discLayer,
  tgDiscTree,
  saveLocation 
)

Definition at line 178 of file LayerMaterialUtils.py.

178 def drawDiscLayer(g4DiscTree,discLayer,tgDiscTree,saveLocation) :
179 
180  gROOT.cd()
181  # create a canvas first
182  dlname = discLayer.GetName()
183  # now fill it
184  for evt in discLayer :
185 
186  innerR = evt.discInnerR
187  outerR = evt.discOuterR
188  position = evt.discPositionZ
189  binsR = evt.discMatBinsR
190  binsPhi = evt.discMatBinsPhi
191  scalor = math.pi/outerR
192 
193  numbinsr = 100
194  numbinsphi = 100
195  if binsR > numbinsr : numbinsr = binsR
196  if binsPhi > numbinsphi : numbinsphi = binsPhi
197 
198  print ('----------------------------------------------------------------')
199  print (' Layer: ', dlname, ' ( LayerIndex : ',evt.discLayerIndex, ')')
200  print (' position : ', position)
201  print (' innerR/outerR : ', innerR, outerR)
202  print (' bins (r/phi) : ', binsR, binsPhi)
203 
204  # POLAR plot needs to be in -pi,pi,-pi,pi, NOT yet fixed
205 
206  # create the x0 histogram
207  dlmap_x0 = TH2F('dlmap_x0_'+dlname,dlname,binsR,-math.pi,math.pi,binsPhi,-math.pi,math.pi)
208  # the steps
209  stepR = (outerR-innerR)/binsR
210  stepPhi = 2*math.pi/binsPhi
211  x0max = 0.
212  # loop over bins and fill it
213  for ibin in range(evt.discMatBinsTotal) :
214  ibinphi = ibin/binsR
215  ibinr = ibin - ibinphi*binsR
216  #ibinz = ibin/binsphi
217  #ibinphi = ibin - ibinz*binsphi
218  x0 = evt.discMatX0[ibin]
219  if x0 > x0max : x0max = x0
220  iphival = -math.pi+(ibinphi+0.5)*stepPhi
221  irvalscal = (innerR + (ibinr+0.5)*stepR)*scalor
222  dlmap_x0.Fill(irvalscal,iphival,x0)
223  # z / phi value for the profile plots
224 
225  pass
226 
227  canvasX0map = TCanvas(dlname,dlname,100,100,800,1100)
228  canvasX0map.Divide(2,3)
229  # draw profiles tog et good estimate for x0max first
230  x0maxProf = x0max
231  canvasX0map.cd(5)
232  x0profRG4 = TProfile(dlname+'x0profRG4',dlname+'x0profRG4',numbinsr,innerR,outerR) # noqa: F841
233  x0profRTG = TProfile(dlname+'x0profRTG',dlname+'x0profRTG',numbinsr,innerR,outerR) # noqa: F841
234  g4DiscTree.Draw('PathInX0:DensedHitR>>'+dlname+'x0profRG4','','prof')
235  tgDiscTree.Draw('PathInX0:DensedHitR>>'+dlname+'x0profRTG','','prof')
236  x0profG4r = gDirectory.Get(dlname+'x0profRG4')
237  x0profTGr = gDirectory.Get(dlname+'x0profRTG')
238  # style
239  x0profTGr.SetMarkerStyle(8)
240  x0profTGr.SetMarkerSize(0.5)
241  x0profTGr.SetMarkerColor(kRed)
242  x0profTGr.SetLineColor(kRed)
243  x0profG4r.GetXaxis().SetTitle('r [mm]')
244  x0profG4r.GetYaxis().SetTitle('t [X_{0}]')
245  x0maxProf = 1.5*x0profG4r.GetMaximum()
246  x0profG4r.GetYaxis().SetRangeUser(0.,x0maxProf)
247  x0profG4r.Draw('hist')
248  x0profTGr.SetMarkerColor(kRed)
249  x0profTGr.SetMarkerSize(0.8)
250  x0profTGr.Draw('pe,same')
251  canvasX0map.cd(6)
252  x0profPhiG4 = TProfile(dlname+'x0profPhiG4',dlname+'x0profPhiG4',numbinsphi,-math.pi,math.pi) # noqa: F841
253  x0profPhiTG = TProfile(dlname+'x0profPhiTG',dlname+'x0profPhiTG',numbinsphi,-math.pi,math.pi) # noqa: F841
254  g4DiscTree.Draw('PathInX0:DensedHitPhi>>'+dlname+'x0profPhiG4','','prof')
255  tgDiscTree.Draw('PathInX0:DensedHitPhi>>'+dlname+'x0profPhiTG','','prof')
256  x0profG4phi = gDirectory.Get(dlname+'x0profPhiG4')
257  x0profTGphi = gDirectory.Get(dlname+'x0profPhiTG')
258  # style
259  x0profTGphi.SetMarkerStyle(8)
260  x0profTGphi.SetMarkerSize(0.5)
261  x0profTGphi.SetMarkerColor(kRed)
262  x0profTGphi.SetLineColor(kRed)
263  x0profG4phi.GetXaxis().SetTitle('m [mm]')
264  x0profG4phi.GetYaxis().SetTitle('t [X_{0}]')
265  if 1.5*x0profG4phi.GetMaximum() > x0maxProf : x0maxProf = 1.5*x0profG4phi.GetMaximum()
266  x0profG4phi.GetYaxis().SetRangeUser(0.,1.5*x0profG4phi.GetMaximum())
267  x0profG4phi.Draw('hist')
268  x0profTGphi.SetMarkerColor(kRed)
269  x0profTGphi.SetMarkerSize(0.8)
270  x0profTGphi.Draw('pe,same')
271 
272  canvasX0map.cd(1)
273  try :
274  g4DiscTree.Draw('PathInX0>>'+dlname+'range')
275  x0range = gDirectory.Get(dlname+'range')
276  x0maxProf = 1.2*x0range.GetXaxis().GetXmax()
277  except TypeError :
278  print (' -> TTree does not exist in TrackingGeometry ')
279  x0mapG4 = TH3F(dlname+'x0mapG4',dlname+'x0mapG4',numbinsr,-1.1*outerR,1.1*outerR,numbinsr,-1.1*outerR,1.1*outerR,50,0.,x0maxProf)
280  try :
281  g4DiscTree.Draw('PathInX0:DensedHitY:DensedHitX>>'+dlname+'x0mapG4','','')
282  x0mapG4 = gDirectory.Get(dlname+'x0mapG4')
283  x0mapG4Prof = x0mapG4.Project3DProfile('xy')
284  x0mapG4Prof.GetXaxis().SetTitle('x [mm]')
285  x0mapG4Prof.GetYaxis().SetTitle('y [mm]')
286  x0mapG4Prof.GetZaxis().SetTitle('t [X_{0}]')
287  x0mapG4Prof.GetZaxis().SetRangeUser(0.,x0maxProf)
288  x0mapG4Prof.GetZaxis().SetTitleOffset(1.5*x0mapG4Prof.GetZaxis().GetTitleOffset())
289  x0mapG4Prof.SetBit(kCanDelete)
290  x0mapG4Prof.SetBit(kMustCleanup)
291  x0mapG4Prof.Draw('colz')
292  except TypeError :
293  print (' -> TTree does not exist in TrackingGeometry ')
294  canvasX0map.cd(2)
295  try :
296  x0mapTG = TH3F(dlname+'x0mapTG',dlname+'x0mapTG',numbinsr,-1.1*outerR,1.1*outerR,numbinsr,-1.1*outerR,1.1*outerR,50,0.,x0maxProf)
297  tgDiscTree.Draw('PathInX0:DensedHitY:DensedHitX>>'+dlname+'x0mapTG','','')
298  x0mapTG = gDirectory.Get(dlname+'x0mapTG')
299  x0mapTGProf = x0mapTG.Project3DProfile('xy')
300  x0mapTGProf.GetXaxis().SetTitle('x [mm]')
301  x0mapTGProf.GetYaxis().SetTitle('y [mm]')
302  x0mapTGProf.GetZaxis().SetTitle('t [X_{0}]')
303  x0mapTGProf.GetZaxis().SetRangeUser(0.,x0maxProf)
304  x0mapTGProf.GetZaxis().SetTitleOffset(1.5*x0mapTGProf.GetZaxis().GetTitleOffset())
305  x0mapTGProf.SetBit(kCanDelete)
306  x0mapTGProf.SetBit(kMustCleanup)
307  x0mapTGProf.Draw('colz')
308  except TypeError :
309  print (' -> TTree does not exist in Geant4 ')
310  canvasX0map.cd(3)
311  dlmap_x0.SetLineColor(kWhite)
312  dlmap_x0.GetZaxis().SetRangeUser(0.,1.1*x0max)
313  dlmap_x0.Draw('polcolz')
314  canvasX0map.cd(4)
315  try :
316  x0mapDiff = x0mapTGProf.Clone()
317  x0mapDiff.Add(x0mapG4Prof,-1)
318  x0mapDiff.GetXaxis().SetTitle('x [mm]')
319  x0mapDiff.GetYaxis().SetTitle('y [mm]')
320  x0mapDiff.GetZaxis().SetTitle('t [X_{0}]')
321  x0mapDiff.GetZaxis().SetRangeUser(-1.,1.)
322  x0mapDiff.GetZaxis().SetTitleOffset(1.5*x0mapDiff.GetZaxis().GetTitleOffset())
323  x0mapDiff.Divide(x0mapG4Prof)
324  x0mapDiff.Draw('colz')
325  except UnboundLocalError :
326  print (' -> No TrackingGeometry/G4 histograms to divide')
327 
328  canvasX0map.SaveAs(saveLocation+'/LayerMaterial_DiscLayer_'+str(evt.discLayerIndex)+'.png')
329 
330  # memory cleanup
331  gDirectory.Delete(dlname+'x0mapTG')
332  gDirectory.Delete(dlname+'x0mapG4')
python.TrigEgammaMonitorHelper.TH2F
def TH2F(name, title, nxbins, bins_par2, bins_par3, bins_par4, bins_par5=None, bins_par6=None, path='', **kwargs)
Definition: TrigEgammaMonitorHelper.py:45
python.TrigEgammaMonitorHelper.TProfile
def TProfile(*args, **kwargs)
Definition: TrigEgammaMonitorHelper.py:81
LayerMaterialUtils.drawCylinderLayer
def drawCylinderLayer(g4CylTree, cylLayer, tgCylTree, saveLocation)
Definition: LayerMaterialUtils.py:6
plotBeamSpotVxVal.range
range
Definition: plotBeamSpotVxVal.py:195
LayerMaterialUtils.drawDiscLayer
def drawDiscLayer(g4DiscTree, discLayer, tgDiscTree, saveLocation)
Definition: LayerMaterialUtils.py:178
str
Definition: BTagTrackIpAccessor.cxx:11
Generated on Sat Apr 5 2025 21:55:52 for ATLAS Offline Software by   doxygen 1.8.18