-
Notifications
You must be signed in to change notification settings - Fork 0
/
lyIrradAnalysis.py
150 lines (125 loc) · 4.79 KB
/
lyIrradAnalysis.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
import ROOT as R
import math as m
import numpy as np
import time
import os
from array import array
def Map(tf):
"""
Maps objets as dict[obj_name][0] using a TFile (tf) and TObject to browse.
"""
m = {}
for k in tf.GetListOfKeys():
n = k.GetName()
m[n] = tf.Get(n)
return m
def LYAnalysis(crystal,run):
files={}
ly , dt, unixtime = array('d'), array('d'), array('d'),
for r in run['runs']:
files[r]=R.TFile.Open(args.data+"/SourceAnalysis/SourceAnalysis_"+r+".root")
objs=Map(files[r])
unixtime.append(files[r].GetCreationDate().Convert(R.kTRUE))
if not (('charge_spill0' in objs.keys()) or ('charge_spill1' in objs.keys())):
print "Bad Run (no charge histogram): "+r
continue
if not (('filteredWaveform_spill0' in objs.keys()) or ('filteredWaveform_spill1' in objs.keys())):
print "Bad Run (no waveform): "+r
continue
histoName='charge_spill0'
if not histoName in objs.keys():
histoName='charge_spill1'
if 'charge_spill10 'in objs.keys():
histoName='charge_spill10'
if (objs[histoName].GetFunction('fTot').GetParameter(10)<1000) or (objs[histoName].GetFunction('fTot').GetParameter(10)>100000):
print "Bad Run (strange fit): "+r
continue
ly.append(objs[histoName].GetFunction('fTot').GetParameter(10))
dt.append(objs['filteredWaveform_spill0'].GetFunction("f2").GetParameter(1))
if len(ly)==0:
print "No good LY data for "+crystal
lyAvg=-9999
else:
lyAvg=sum(ly)/len(ly)
if len(dt)==0:
print "No good waveform for "+crystal
dtAvg=-9999
else:
dtAvg=sum(dt)/len(dt)
if len(unixtime)==0:
print "No good waveform for "+crystal
unixtimeAvg=-9999
else:
unixtimeAvg=int(sum(unixtime)/len(unixtime))
ref=array('d')
for r in run['refRuns']:
files[r]=R.TFile.Open(args.data+"/SourceAnalysis/SourceAnalysis_"+r+".root")
objs=Map(files[r])
if not (('charge_spill0' in objs.keys()) or ('charge_spill1' in objs.keys())):
print "Bad Run (no charge histogram): "+r
continue
if not (('filteredWaveform_spill0' in objs.keys()) or ('filteredWaveform_spill1' in objs.keys())):
print "Bad Run (no waveform): "+r
continue
histoName='charge_spill0'
if not histoName in objs.keys():
histoName='charge_spill1'
if (objs[histoName].GetFunction('fTot').GetParameter(10)<1000) or (objs[histoName].GetFunction('fTot').GetParameter(10)>100000):
print "Bad Run (strange fit): "+r
continue
ref.append(objs[histoName].GetFunction('fTot').GetParameter(10))
if len(ref)==0:
print "No good ref data for this crystal!"
refAvg=-9999
else:
refAvg=sum(ref)/len(ref)
# refAvg=11000.
ledAvg=LEDAnalysis(crystal,run['ledRuns'])
return { 'ly':lyAvg, 'dt':dtAvg, 'ref':refAvg, 'pe':ledAvg, 'time':unixtimeAvg }
def LEDAnalysis(crystal,ledRuns):
files={}
pe=array('d')
for r in ledRuns:
files[r]=R.TFile.Open(args.data+"/SinglePEAnalysis/"+r+"_simul_out.root")
objs=Map(files[r])
if not 'PMT_0' in objs.keys():
print "Bad LED Run (no fit function): "+r
continue
if ( abs(objs['PMT_0'].GetParameter(4)-35)<0.01 ):
print "Bad LED Run (fit not converging): "+r
continue
if ( objs['PMT_4'].GetParameter(1) < 1. ):
print "Bad LED Run (small LED amplitude): "+r
continue
pe.append(objs['PMT_0'].GetParameter(2))
if len(pe)==0:
print "No good LED data for "+crystal
peAvg=-9999
else:
peAvg=sum(pe)/len(pe)
return peAvg
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--output',dest='output')
parser.add_argument('--db',dest='db')
parser.add_argument('--data',dest='data')
args = parser.parse_args()
exec "from %s import crystalsDB" % args.db
crystalsDB_withData = {}
for crystal,crystalInfo in crystalsDB.items():
print "Analysing crystal "+crystal
for runInfo in crystalInfo['runs']:
tag=runInfo['tag']
lyData=LYAnalysis(crystal,runInfo)
lyData.update({'tag':"%s"%tag})
data={}
data.update(crystalInfo)
data.pop('runs',None)
data.update(lyData)
crystalsDB_withData['%s_%s'%(crystal,tag)]=data
#Save CSV using Pandas DataFrame
import pandas as pd
df=pd.DataFrame.from_dict(crystalsDB_withData,orient='index')
#df=df.drop(columns=['runs'])
print(df)
df.to_csv(args.output,header=False)