def process(self, event):
event.simulator = self
if self.is_display:
self.display.clear()
pfsim_particles = []
gen_particles = getattr(event, self.cfg_ana.gen_particles)
try:
self.simulator.simulate(gen_particles)
except (PropagationError, SimulationError) as err:
self.mainLogger.error(str(err) + ' -> Event discarded')
return False
pfsim_particles = self.simulator.ptcs
if self.is_display:
self.display.register(GTrajectories(pfsim_particles), layer=1)
#these are the particles before simulation
simparticles = sorted(pfsim_particles,
key=lambda ptc: ptc.e(),
reverse=True)
setattr(event, self.simname, simparticles)
#extract the tracks and clusters (extraction is prior to Colins merging step)
event.tracks = dict()
event.ecal_clusters = dict()
event.hcal_clusters = dict()
if "tracker" in self.simulator.pfinput.elements:
for element in self.simulator.pfinput.elements["tracker"]:
event.tracks[element.uniqueid] = element
if "ecal_in" in self.simulator.pfinput.elements:
for element in self.simulator.pfinput.elements["ecal_in"]:
event.ecal_clusters[element.uniqueid] = element
if "hcal_in" in self.simulator.pfinput.elements:
for element in self.simulator.pfinput.elements["hcal_in"]:
event.hcal_clusters[element.uniqueid] = element
ruler = Distance()
#create history node
#note eventually history will be created by the simulator and passed in
# as an argument and this will no longer be needed
uniqueids = list(event.tracks.keys()) + list(
event.ecal_clusters.keys()) + list(event.hcal_clusters.keys())
history = dict((idt, Node(idt)) for idt in uniqueids)
#Now merge the simulated clusters and tracks as a separate pre-stage (prior to new reconstruction)
# and set the event to point to the merged cluster
pfevent = PFEvent(event, 'tracks', 'ecal_clusters', 'hcal_clusters')
merged_ecals = MergedClusterBuilder(pfevent.ecal_clusters, ruler,
history)
setattr(event, self.mergedecalsname, merged_ecals.merged)
merged_hcals = MergedClusterBuilder(pfevent.hcal_clusters, ruler,
merged_ecals.history_nodes)
setattr(event, self.mergedhcalsname, merged_hcals.merged)
setattr(event, self.historyname, merged_hcals.history_nodes)
def merge_clusters(self, papasevent): # todo move to a separate analyzer
#For Now merge the simulated clusters as a separate pre-stage (prior to new reconstruction)
ruler = Distance()
merged_ecals = dict()
merged_hcals = dict()
ecals = papasevent.get_collection('es')
if ecals:
merged_ecals = MergedClusterBuilder(papasevent.get_collection('es'), ruler, papasevent.history).merged_clusters
hcals = papasevent.get_collection('hs')
if hcals:
merged_hcals = MergedClusterBuilder(papasevent.get_collection('hs'), ruler, papasevent.history).merged_clusters
papasevent.add_collection(merged_ecals)
papasevent.add_collection(merged_hcals)