15 output = abs(phi1 - phi2)
17 output = math.pi * 2. - output
21 return (
delta_phi(phi1, phi2)**2. + (eta1 - eta2)**2.)**0.5
25 self.
tfile = ROOT.TFile(filename)
29 for each_entry
in self.
tree:
31 entry_dict[
"truth_e_ET"] = np.asarray(each_entry.truth_e_ET)/1000.
32 entry_dict[
"truth_e_phi"] = np.asarray(each_entry.truth_e_phi)
33 entry_dict[
"truth_e_eta"] = np.asarray(each_entry.truth_e_eta)
34 entry_dict[
"eg_eta"] = np.asarray(each_entry.eg_eta) * 0.1
35 entry_dict[
"eg_phi"] = np.asarray(each_entry.eg_phi) * math.pi/32.
36 entry_dict[
"eg_ET"] = np.asarray(each_entry.eg_ET)/1000.
37 entry_dict[
"eg_wstotnum"] = np.asarray(each_entry.eg_wstotnum)
38 entry_dict[
"eg_wstotden"] = np.asarray(each_entry.eg_wstotden)
39 entry_dict[
"eg_rhadnum"] = np.asarray(each_entry.eg_wstotnum)
40 entry_dict[
"eg_rhadden"] = np.asarray(each_entry.eg_wstotden)
41 entry_dict[
"eg_retanum"] = np.asarray(each_entry.eg_wstotnum)
42 entry_dict[
"eg_retaden"] = np.asarray(each_entry.eg_wstotden)
43 entry_dict[
"eg_haveseed"] = np.asarray(each_entry.eg_haveseed).astype(dtype=bool)
44 entry_dict[
"eg_nTOBs"] =
int(each_entry.eg_nTOBs)
46 entry_dict[
"truth_tauvisible_ET"] = np.asarray(each_entry.truth_tauvisible_ET)/1000.
47 entry_dict[
"truth_tauvisible_phi"] = np.asarray(each_entry.truth_tauvisible_phi)
48 entry_dict[
"truth_tauvisible_eta"] = np.asarray(each_entry.truth_tauvisible_eta)
50 entry_dict[
"eg_tauOregon_Et"] = np.asarray(each_entry.eg_tauOregon_Et)/1000.
51 entry_dict[
"eg_tauOregon_Iso"] = np.asarray(each_entry.eg_tauOregon_Iso)
52 entry_dict[
"eg_tauTLV_Et"] = np.asarray(each_entry.eg_tauTLV_Et)/1000.
53 entry_dict[
"eg_tauTLV_Iso"] = np.asarray(each_entry.eg_tauTLV_Iso)
65 for key, value
in inputs.items():
66 setattr(self, key, value)
74 for i_e
in range(len(self.truth_e_ET)):
76 for i_tob
in range(self.eg_nTOBs):
78 if self.eg_ET[i_tob] < 15.:
80 if delta_R(self.truth_e_eta[i_e], self.truth_e_phi[i_e], self.eg_eta[i_tob], self.eg_phi[i_tob]) > 0.12:
82 tob_id_tem.append(i_tob)
83 output.append(tob_id_tem)
85 for i_e
in range(len(self.truth_tauvisible_ET)):
87 for i_tob
in range(self.eg_nTOBs):
91 if delta_R(self.truth_tauvisible_eta[i_e], self.truth_tauvisible_phi[i_e], self.eg_eta[i_tob], self.eg_phi[i_tob]) > 0.12:
93 tob_id_tem.append(i_tob)
94 output.append(tob_id_tem)
99 return np.argmax(self.truth_e_ET)
101 if len(self.truth_tauvisible_ET) == 0:
103 return np.argmax(self.truth_tauvisible_ET)
108 return (abs(self.truth_e_eta[i_e]) > 1.375
and abs(self.truth_e_eta[i_e]) < 1.52)
110 return (abs(self.truth_tauvisible_eta[i_e]) > 1.375
and abs(self.truth_tauvisible_eta[i_e]) < 1.52)