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):