Monopix_Sim/lxplus_sim_newdata.py at master · TianyangWang/Monopix_Sim · GitHub

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
#To get ROOT on lxplus:
#cd /afs/cern.ch/sw/lcg/app/releases/ROOT/5.34.00/x86_64-slc5-gcc43-opt/root
#source /afs/cern.ch/sw/lcg/contrib/gcc/4.3/x86_64-slc5/setup.sh
#source bin/thisroot.sh
#cd -

import ROOT
import numpy as np

f =  ROOT.TFile.Open('pixel_hits-36x36x25.root')
t = f.Get("hitstree")
size = t.GetEntries()
print size
# cross check the tree structure


#for i, e in enumerate(t):
#    print (e.bcid, e.eta_module, e.phi_module, e.eta_index, e.phi_index, e.charge, e.trigger)
#    if i > 10 : break

arr = np.recarray((size,), dtype=[('bcid', np.uint32),
                                  ('eta_module', np.uint8),
                                  ('eta_index', np.uint16),
                                  ('phi_index', np.uint16),
                                  ('charge', np.float),
                                  ('charge_track', np.float),
                                  ('charge_noise', np.float),
                                  ('charge_xtalk', np.float),
                                  ('trigger', np.bool) ])

for i, e in enumerate(t):
    arr[i] =  (e.bcid, ord(e.eta_module), e.eta_index, e.phi_index, e.charge, e.charge_track, e.charge_noise, e.charge_xtalk, e.trigger)
    if i%100000 == 1:
        print (float(i)/size)*100, '%'

print "event number: {}".format(np.amax(arr['bcid']))
np.save('pixel_hits-36x36x25', arr)