Coverage for src/km3modules/topology.py: 36%

1# Filename: topology.py 

2# -*- coding: utf-8 -*- 

3# pylint: disable=locally-disabled 

4""" 

5Topology related modules. 

6 

7""" 

8 

9import numpy as np 

10 

11from km3pipe import Module 

12 

13 

14class TriggeredDUs(Module): 

15 """Check for triggered DUs.""" 

16 

17 def process(self, blob): 

18 triggered_hits = blob["Hits"].triggered_rows.to_dataframe() 

19 self.calibration.apply(triggered_hits) 

20 dus = np.unique(triggered_hits["du"]) 

21 n_dus = len(dus) 

22 

23 blob["TriggeredDUs"] = dus 

24 

25 if n_dus > 1: 

26 blob["multiple_du_event"] = True 

27 else: 

28 blob["single_du_event"] = True 

29 

30 return blob