/
Tree2Python.py
46 lines (36 loc) · 1.91 KB
/
Tree2Python.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
import numpy as np
import scipy as sp
import math
from matplotlib import pylab as plt
from ROOT import TTree, TFile, TH2D, TCanvas, TH1F, gROOT
from root_numpy import array2hist, hist2array, fill_hist, tree2array,root2array
input1="../../../storage/cc14398/Co60-10s-15Mar/Co60-10s-PS1m.root"
input2="../../../storage/cc14398/Cs137-10s-18Mar/PS1m.root"
#input2="SavedData/Shielding/Isotropic/1Mar-PSonly-Co60-33MBq-100ms-3m-Bp.root"
#input3="SavedData/Shielding/Isotropic/6Mar-Co60-37MBq-100ms-2m-Phantom.root"
#input4="SavedData/Shielding/Isotropic/6Mar-Co60-37MBq-100ms-8m-Phantom.root"
#input2="SavedData/Shielding/ParticleGun/5Mar-Co60-PartGun-Bp-1m.root"
#input3="SavedData/Shielding/ParticleGun/5Mar-Co60-PartGun-Ph-1m.root"
#input4="SavedData/Shielding/Isotropic/1Mar-PSonly-Co60-33MBq-100ms-8m-Bp.root"
#input1="../../../storage/cc14398/Co60-10s-15Mar/Co60-10s-PS1m.root"
#input2="../../../storage/cc14398/Co60-10s-15Mar/Co60-10s-PS3m.root"
#input3="../../../storage/cc14398/Co60-10s-15Mar/Co60-10s-PS7m.root"
#input4="../../../storage/cc14398/Co60-10s-15Mar/Co60-10s-PS8m.root"
KE1=root2array(input1,treename="PhaseSpace", branches="Ekine")
KE2=root2array(input2,treename="PhaseSpace",branches="Ekine")
#KE3=root2array(input3,treename="PhaseSpace",branches="Ekine")
#KE4=root2array(input4,treename="PhaseSpace",branches="Ekine")
#dX=root2array(inputFile,treename="PhaseSpace", branches="dX")
plt.hist(KE1,bins=20,histtype='step',label='Co 60')
plt.hist(KE2,bins=10,histtype='step',label='Cs 137')
#plt.hist2d(E_kinetic,dX)
plt.xlabel('Kinetic Energy (MeV)')
plt.ylabel('Counts')
#fig, ax1 = plt.subplots()
#ax1.hist([KE1,KE2,KE3], label=['1m','3m','7m'],bins=15,density=True)
#plt.hist(KE1,20,alpha=0.5,label='1m',range=[0,1.1],density=True)
#plt.hist(KE2,20,alpha=0.5,label='4m',range=[0,1.1],density=True)
#plt.hist(KE3,20,alpha=0.5,label='7m',range=[0,1.1],density=True)
plt.legend(loc=2)
plt.pause(0.01)
input("press enter to exit")