image.png

    1. #!/usr/bin/python
    3. from ROOT import gROOT,gStyle,gPad,TFrame,TLine,TCutG,TFile,TCanvas,TH1F,TH2F,TGraph
    4. import numpy as np
    5. from tqdm import tqdm
    6. from ctypes import *
    7. from array import array
    9. class evt_trig_frag(Structure):
    10.     _fields_ = [
    11.         ("slot", c_int32 *1),
    12.         ("len", c_int32 *1),
    13.         ("ts_hi", c_int32 *1),
    14.         ("ts_lo", c_int32 *1),
    15.         ("is_good_sig", c_int32 *1),
    16.         ("multi", c_int32 *1),
    17.         ("cfd_t", c_float *1),
    18.         ("cfd_t1", c_float *1),
    19.         ("cfd_t2", c_float *1),
    20.         ("base_raw", c_float *1),
    21.         ("base_smoth", c_float *1),
    22.         ("v_max", c_float *1),
    23.         ("energy", c_float *1),
    24.         ("energy1", c_float *1),
    25.         ("raise_t", c_float *1),
    26.         ("tx", c_uint32 *125),
    28.     ]
    29. evt_trig_stru = evt_trig_frag()
    31. fcut = TFile('cut_p0.root')
    32. cutg = fcut.Get('CUTG_p')
    34. rootFileInput = TFile('data_in_proc.root')
    35. tree = rootFileInput.Get('trigger_proc')
    37. tree.GetBranch('trig_proc').SetAddress(evt_trig_stru)
    39. h1 = TH2F('h1','',500,0,7,500,0,4)
    41. entries = tree.GetEntries()
    42. with open('out.txt','w') as fout:
    43.     for i in tqdm(range(entries)):
    44.         tree.GetEntry(i)
    45.         h1.Fill(evt_trig_stru.energy[0],evt_trig_stru.raise_t[0])
    46.         if cutg.IsInside(evt_trig_stru.energy[0],evt_trig_stru.raise_t[0]):
    47.             fout.write(str(i))
    48.             fout.write('\n')
    50. h1.Draw()
    51. input()