ATLAS Offline Software
Functions
check_samples_cis Namespace Reference

Functions

def readModule (part, mod, runnumber, datadir, outdir)
 
def checkSamples (histo)
 
def saveHistos (histos, part, mod, gain, runnumber, outdir)
 
def getPMTTex (pmt)
 
def cleanHisto (hist)
 
def SetupDraw ()
 
def main ()
 

Function Documentation

◆ checkSamples()

def check_samples_cis.checkSamples (   histo)

Definition at line 88 of file check_samples_cis.py.

88 def checkSamples(histo):
89 
90  maxval = 0
91  maxvalsamp = 0
92 
93  color = 'null'
94 
95  for bin in range(1, histo.GetXaxis().GetNbins()):
96  val = histo.GetBinContent(bin)
97  if val > maxval:
98  maxval = val
99  maxvalsamp = bin
100 
101  if maxvalsamp==4:
102  histo.SetFillColor(ROOT.kGreen)
103  color = 'green'
104  elif maxvalsamp==3 or maxvalsamp==5:
105  histo.SetFillColor(ROOT.kYellow)
106  color = 'yellow'
107  elif maxvalsamp==1 or maxvalsamp==2 or maxvalsamp==6 or maxvalsamp==7:
108  histo.SetFillColor(ROOT.kRed)
109  color = 'red'
110 
111  return histo, color
112 
113 

◆ cleanHisto()

def check_samples_cis.cleanHisto (   hist)

Definition at line 176 of file check_samples_cis.py.

176 def cleanHisto(hist):
177 
178  hist.GetXaxis().CenterTitle()
179  hist.GetXaxis().SetTitleOffset(1.4)
180  hist.GetXaxis().SetLabelOffset(0.021)
181  hist.GetXaxis().SetLabelSize(0.06)
182  hist.GetYaxis().CenterTitle()
183  hist.GetYaxis().SetTitleOffset(1.3)
184  hist.GetYaxis().SetLabelOffset(0.015)
185  hist.GetYaxis().SetLabelSize(0.06)
186  hist.SetFillColor(ROOT.kGray)
187 
188  return hist
189 
190 

◆ getPMTTex()

def check_samples_cis.getPMTTex (   pmt)

Definition at line 164 of file check_samples_cis.py.

164 def getPMTTex(pmt):
165 
166  pave = ROOT.TPaveText(0.7, 0.6, 0.9, 0.95)
167  pave.AddText(pmt)
168  pave.SetFillColor(0)
169 
170  pave.SetBorderSize(1)
171  pave.SetTextSize(0.15)
172  pave.SetTextAlign(12)
173 
174  return pave
175 

◆ main()

def check_samples_cis.main ( )

Definition at line 225 of file check_samples_cis.py.

225 def main():
226 
227 
228  if len(sys.argv)!=2:
229  print "Please pass a run number."
230  return True
231 
232  runnumber = int(sys.argv[1])
233 
234  datadir = '/location/of/ntuples'
235  outdir = '/where/output/will/be/sent/'
236 
237  datadir = datadir + str(runnumber) + '/'
238  outdir = outdir + str(runnumber) + '/'
239 
240  if not os.path.exists(outdir):
241  os.mkdir(outdir)
242 
243  print 'Using data from', datadir
244  print 'Ouput will be sent to', outdir
245 
246  parts = ['LBA', 'LBC', 'EBA', 'EBC']
247 
248  for part in parts:
249  for mod in range(1,65):
250  readModule(part, mod, runnumber, datadir, outdir)
251 
252  namehg = 'samples_HG_%d.ps' % runnumber
253  namelg = 'samples_LG_%d.ps' % runnumber
254  packnamehg = outdir + namehg
255  packnamelg = outdir + namelg
256 
257  can = SetupDraw()
258  can.Print('%s[' % packnamehg)
259 
260  for part in parts:
261  for mod in range(1,65):
262  pickfile = '%s%s%02dhg.dat' % (outdir, part, mod)
263  if os.path.exists(pickfile):
264  pickfile = open(pickfile, 'r')
265  page = pickle.load(pickfile)
266  page.Print(packnamehg)
267  pickfile.close()
268 
269  can.Print('%s]' % packnamehg)
270 
271  can = SetupDraw()
272  can.Print('%s[' % packnamelg)
273 
274  for part in parts:
275  for mod in range(1,65):
276  pickfile = '%s%s%02dlg.dat' % (outdir, part, mod)
277  if os.path.exists(pickfile):
278  pickfile = open(pickfile, 'r')
279  page = pickle.load(pickfile)
280  page.Print(packnamelg)
281  pickfile.close()
282 
283  can.Print('%s]' % packnamelg)
284 
285 

◆ readModule()

def check_samples_cis.readModule (   part,
  mod,
  runnumber,
  datadir,
  outdir 
)

Definition at line 10 of file check_samples_cis.py.

10 def readModule(part, mod, runnumber, datadir, outdir):
11 
12  ROOT.gROOT.SetBatch()
13  print 'Checking', part, mod
14 
15  filename = '%stiletb_%d_CIS.%s%02d.0.aan.root' % (datadir, runnumber, part, mod)
16 
17  partnames = ['LBA', 'LBC', 'EBA', 'EBC']
18  partmap = {'LBA': 'A', 'LBC': 'C', 'EBA': 'D', 'EBC': 'E'}
19 
20  histoshg = [ROOT.TH1F('%s MOD %02d PMT %02d HG' % (part, mod, int(pmt + 1)), '%02d' % (int(pmt + 1)), 7, 1, 8) for pmt in range(48)]
21  histoslg = [ROOT.TH1F('%s MOD %02d PMT %02d LG' % (part, mod, int(pmt + 1)), '%02d' % (int(pmt + 1)), 7, 1, 8) for pmt in range(48)]
22 
23  entrieshg = [0 for pmt in range(48)]
24  entrieslg = [0 for pmt in range(48)]
25 
26  for list in [histoshg, histoslg]:
27  for hist in list:
28  hist.GetXaxis().CenterLabels(ROOT.kTRUE)
29  hist = cleanHisto(hist)
30 
31 
32  if os.path.exists(filename):
33  print 'Using', filename
34  file = ROOT.TFile.Open(filename)
35  if file and not file.IsZombie():
36  tree = file.Get('h1000')
37  nevents = tree.GetEntries()
38  #nevents = 100 # for testing
39 
40  for ev in range(1, nevents):
41 
42  tree.GetEntry(ev)
43 
44  pha = tree.m_cispar[5]
45  dac = tree.m_cispar[6]
46  cap = tree.m_cispar[7]
47  if pha!=0 or cap!=100:
48  continue
49  if dac==7:
50  for pmt in range(48):
51  sampleshg = getattr(tree, 'sample%s%02dhi' % (partmap[part], mod))
52  for samp in range(7):
53  histoshg[pmt].Fill(samp+1, sampleshg[pmt*9+samp])
54  entrieshg[pmt] = entrieshg[pmt] + 1
55  if dac==128:
56  for pmt in range(48):
57  sampleslg = getattr(tree, 'sample%s%02dlo' % (partmap[part], mod))
58  for samp in range(7):
59  histoslg[pmt].Fill(samp+1, sampleslg[pmt*9+samp])
60  entrieslg[pmt] = entrieslg[pmt] + 1
61 
62 
63  for pmt in range(48):
64  if entrieshg[pmt]==0:
65  continue
66  factor = 1./float(entrieshg[pmt])
67  histoshg[pmt].Scale(factor)
68 
69  for pmt in range(48):
70  if entrieslg[pmt]==0:
71  continue
72  factor = 1./float(entrieslg[pmt])
73  histoslg[pmt].Scale(factor)
74 
75 
76  for histo in histoshg:
77  histo, color = checkSamples(histo)
78 
79  for histo in histoslg:
80  histo, color = checkSamples(histo)
81 
82 
83  saveHistos(histoshg, part, mod, 'hg', runnumber, outdir)
84  saveHistos(histoslg, part, mod, 'lg', runnumber, outdir)
85  file.Close()
86 
87 

◆ saveHistos()

def check_samples_cis.saveHistos (   histos,
  part,
  mod,
  gain,
  runnumber,
  outdir 
)

Definition at line 114 of file check_samples_cis.py.

114 def saveHistos(histos, part, mod, gain, runnumber, outdir):
115 
116 
117 
118  partnames = ['LBA', 'LBC', 'EBA', 'EBC']
119 
120  can = SetupDraw()
121  can.Divide(1,1)
122 
123  if gain == 'hg':
124  gn = 'HG'
125  elif gain == 'lg':
126  gn = 'LG'
127 
128  toptitle = ROOT.TLatex(0.42, 0.9, '%s %02d %s' % (part, mod, gn))
129  toptitle.SetTextSize(0.05)
130  toptitle.SetNDC()
131  toptitle.SetTextAlign(12)
132 
133  can.Divide(8,7)
134 
135  for hist in histos:
136  if gain == 'hg':
137  hist.SetMaximum(500)
138  elif gain == 'lg':
139  hist.SetMaximum(200)
140  can.cd(histos.index(hist)+8+1)
141  pmt = int(hist.GetTitle())
142  if pmt == '':
143  continue
144  if checkPMTValid(partnames.index(part), pmt-1)=='noninst':
145  continue
146 
147  if hist!=0:
148  hist.Draw()
149  can.Modified()
150  can.Update()
151 
152  can.Modified()
153 
154  can.cd(0)
155  toptitle.Draw()
156  can.Modified()
157  can.Update()
158 
159  pickfile = open('%s%s%02d%s.dat' % (outdir, part, mod, gain), 'w')
160  pickle.dump(can, pickfile)
161  pickfile.close()
162 
163 

◆ SetupDraw()

def check_samples_cis.SetupDraw ( )

Definition at line 191 of file check_samples_cis.py.

191 def SetupDraw():
192 
193  ROOT.gROOT.SetStyle("Plain")
194  ROOT.gStyle.SetCanvasBorderMode(0)
195  ROOT.gStyle.SetPadBorderMode(0)
196  ROOT.gStyle.SetTitleX(0.75)
197  ROOT.gStyle.SetTitleY(1.0)
198  ROOT.gStyle.SetTitleW(0.25)
199  ROOT.gStyle.SetTitleH(0.25)
200  #ROOT.gStyle.SetStatX(0.64)
201  #ROOT.gStyle.SetStatY(0.64)
202  ROOT.gStyle.SetLabelFont(42,"XYZ")
203  ROOT.gStyle.SetTextFont(42)
204  #ROOT.gStyle.SetOptStat(111110)
205  ROOT.gStyle.SetOptStat(0)
206  ROOT.gStyle.SetPalette(1)
207  ROOT.gStyle.SetTitleFont(42,"XYZ")
208  ROOT.gStyle.SetTitleBorderSize(1)
209  ROOT.gStyle.SetPadColor(0)
210  ROOT.gStyle.SetCanvasColor(0)
211  ROOT.gStyle.SetOptFit(0)
212 
213  c1 = ROOT.TCanvas()
214  c1.SetFrameBorderMode(0)
215  c1.SetBorderSize(0)
216  c1.SetBorderMode(0)
217  c1.SetFillColor(0)
218  c1.SetTickx()
219  c1.SetTicky()
220 
221  return c1
222 
223 
224 
check_samples_cis.cleanHisto
def cleanHisto(hist)
Definition: check_samples_cis.py:176
check_samples_cis.checkSamples
def checkSamples(histo)
Definition: check_samples_cis.py:88
CaloCellPos2Ntuple.int
int
Definition: CaloCellPos2Ntuple.py:24
check_samples_cis.main
def main()
Definition: check_samples_cis.py:225
check_samples_cis.readModule
def readModule(part, mod, runnumber, datadir, outdir)
Definition: check_samples_cis.py:10
check_samples_cis.SetupDraw
def SetupDraw()
Definition: check_samples_cis.py:191
plotBeamSpotVxVal.range
range
Definition: plotBeamSpotVxVal.py:195
Scale
void Scale(TH1 *h, double d=1)
Definition: comparitor.cxx:76
aux_functions.checkPMTValid
def checkPMTValid(ros, pmt)
Definition: aux_functions.py:2
Trk::open
@ open
Definition: BinningType.h:40
check_samples_cis.getPMTTex
def getPMTTex(pmt)
Definition: check_samples_cis.py:164
str
Definition: BTagTrackIpAccessor.cxx:11
check_samples_cis.saveHistos
def saveHistos(histos, part, mod, gain, runnumber, outdir)
Definition: check_samples_cis.py:114
readCCLHist.float
float
Definition: readCCLHist.py:83
Generated on Thu Jul 4 2024 22:12:28 for ATLAS Offline Software by   doxygen 1.8.18