ED_p_ED.append(h_p_ED)
h_pt_ED = ROOT.TProfile2D("h_pt_ED" + str(lay), "h_pt_ED" + str(lay), 64,
-32.5, 31.5, 64, -32.5, 31.5)
ED_pt_ED.append(h_pt_ED)
layers = [64, 32, 32, 16, 16, 8]
f1 = ROOT.TFile.Open("fpred_new_final.root")
out_image = []
out_image1 = []
out_image2 = []
out_image3 = []
for ev in range(Events):
TC = -1
En = f["Smeared_Track_Energy"][ev]
ptr = math.sqrt(En * En - 139.57018 * 139.57018)
Traj_Mark = Track.MakeTruthTrajectory(C1[ev], C2[ev], C3[ev], C4[ev])
total = []
for lay in range(6):
Traj_Mark[lay][0] = Traj_Mark[lay][0] - 1
Traj_Mark[lay][1] = Traj_Mark[lay][1] - 1
# Lay_Ev=f["RealRes_ChargedEnergy_Layer"+str(lay+1)][ev][0]
Lay_Ev = f["RealRes_TotalEnergy_Layer" + str(lay + 1)][ev][0]
# Noise_Ev=f["RealRes_Noise_Layer"+str(lay+1)][ev][0]
Lay_Ev = Lay_Ev + Noise_Ev
Topo_Ev = gT["TopoClusters" +
str(lay + 1)][ev][0][:layers[lay], :layers[lay]]
for X in range(layers[lay]):
for Y in range(layers[lay]):
if Topo_Ev[X][Y] != 0:
total.append([
lay, X, Y, Lay_Ev[X][Y], Topo_Ev[X][Y],
h_Pflow1 = ROOT.TH1D("h_Pflownew", "h_Pflownew", 50, -2, 2)
h_Energy = ROOT.TH1D("h_Energy", "h_Energy", 200, -500, 10000)
h_Eneutral = ROOT.TH1D("h_Eneutral", "h_Eneutral", 200, -500, 10000)
f1 = ROOT.TFile.Open("Epred_outN.root")
#f1 = ROOT.TFile.Open("fpred_new_final.root")
#hProf_t = f1.Get("h_Epred_prof_tot")
layers = [64, 32, 32, 16, 16, 8]
out_image = []
#for ev in range(4,5):
# print ev
for ev in range(Events):
if ev % 200 == 0: print(ev)
En = f["Smeared_Track_Energy"][ev]
ptr = math.sqrt(En * En - 139.57018 * 139.57018)
Traj_Mark = Track.MakeTruthTrajectory(C1[ev], C2[ev], C3[ev], C4[ev])
Traj_Mark_neu = Track.MakeTruthTrajectory(C1_p[ev], C2_p[ev], C3[ev],
C4[ev])
total = []
TC = -1
for lay in range(6):
Etopo_t = gEpred["PflowTC" +
str(lay + 1)][ev][0][:layers[lay], :layers[lay]]
for X in range(layers[lay]):
for Y in range(layers[lay]):
if Etopo_t[X][Y] != 0:
TC = Etopo_t[X][Y]
break
for lay in range(6):
