144def get_name_associations(list):
145
146 taggers = ["_IP3D_", "_SV1_", "_JetFitter_"]
147 old_taggers = ["_IP2D_"]
148 oneToOne_associations = {
149 '_truthLabel ': '_jet_truthLabel ',
150 '_IP3Dpb ': '_tagger_IP3D_pb ',
151 '_IP3Dpc ': '_tagger_IP3D_pc ',
152 '_IP3Dpu ': '_tagger_IP3D_pu ',
153 '_SV1pb ': '_tagger_SV1_pb ',
154 '_SV1pc ': '_tagger_SV1_pc ',
155 '_SV1pu ': '_tagger_SV1_pu ',
156 '_NTracksIP3D ': '_tagger_IP3D_NTracks_incl ',
157 '_NTracksIP2D ': '_old_taggers_IP2D_NTracks_incl ',
158 '_NGTinSvx1 ': '_tagger_SV1_nGoodTracks_incl ',
159 '_eta ': '_jet_jet_eta ',
160 '_phi ': '_jet_jet_phi ',
161 '_d0 ': '_tracks_d0_b ',
162 '_z0 ': '_tracks_z0_b ',
163 '_IP3DgradeOfTracks ': '_tagger_IP3D_gradeOfTracks_incl ',
164 '_IP2DgradeOfTracks ': '_old_taggers_IP2D_gradeOfTracks_incl ',
165 }
166 oneToOne_associations_ttbar = {
167 '_TruthBpt ': '_jet_jet_pt_b_ttbar ',
168 '_TruthLpt ': '_jet_jet_pt_l_ttbar ',
169 '_TruthCpt ': '_jet_jet_pt_c_ttbar ',
170 '_e ': '_jet_jet_e ',
171 '_pt ': '_jet_jet_pt_ttbar ',
172 }
173 oneToOne_associations_Zprime = {
174 '_TruthBpt ': '_jet_jet_pt_b_Zprime ',
175 '_TruthLpt ': '_jet_jet_pt_l_Zprime ',
176 '_TruthCpt ': '_jet_jet_pt_c_Zprime ',
177 '_e ': '_jet_jet_e_Zprime ',
178 '_pt ': '_jet_jet_pt_Zprime ',
179 }
180
181 name_associations = {}
182
183 for name in list:
184 new_name = name
185
186 for tag in taggers:
187 if tag in new_name:
188 new_name = new_name.replace(tag, "_tagger" + tag)
189
190 for tag in old_taggers:
191 if tag in new_name:
192 new_name = new_name.replace(tag, "_old_taggers" + tag)
193
194 if "_matched_pt" in new_name:
195 new_name = new_name.replace("_matched_pt", "_matched_pt_" + production_process)
196
197 if "AntiKt4EMPFlowJets" in new_name:
198 new_name = new_name.replace("AntiKt4EMPFlowJets", "AntiKt4EMPFlowJets_")
199
200 for key in oneToOne_associations:
201 if key in new_name:
202 new_name = new_name.replace(key, oneToOne_associations[key])
203
204 if production_process == 'ttbar':
205 for key in oneToOne_associations_ttbar:
206 if key in new_name:
207 new_name = new_name.replace(key, oneToOne_associations_ttbar[key])
208
209 if production_process == 'Zprime':
210 for key in oneToOne_associations_Zprime:
211 if key in new_name:
212 new_name = new_name.replace(key, oneToOne_associations_Zprime[key])
213
214
215
216
217
218 name = name.replace(" ", "")
219 new_name = new_name.replace(" ", "")
220
221
222 name_associations[name] = new_name
223
224 return name_associations
225
226
227
228