# Filename: topology.py
# -*- coding: utf-8 -*-
# pylint: disable=locally-disabled
"""
Topology related modules.
"""
import numpy as np
from km3pipe import Module
[docs]
class TriggeredDUs(Module):
"""Check for triggered DUs."""
[docs]
def process(self, blob):
triggered_hits = blob["Hits"].triggered_rows.to_dataframe()
self.calibration.apply(triggered_hits)
dus = np.unique(triggered_hits["du"])
n_dus = len(dus)
blob["TriggeredDUs"] = dus
if n_dus > 1:
blob["multiple_du_event"] = True
else:
blob["single_du_event"] = True
return blob