From 33b8829a091e3b92a71079d7318bf2d53559f677 Mon Sep 17 00:00:00 2001
From: jochenstahn <jochen.stahn@psi.ch>
Date: Fri, 22 Nov 2024 11:24:17 +0100
Subject: [PATCH] check for empty files introduced

---
 libeos/file_reader.py | 33 +++++++++++++++++++--------------
 1 file changed, 19 insertions(+), 14 deletions(-)

diff --git a/libeos/file_reader.py b/libeos/file_reader.py
index e77b675..e2930ae 100644
--- a/libeos/file_reader.py
+++ b/libeos/file_reader.py
@@ -20,19 +20,6 @@ try:
 except Exception:
     nb_helpers = None
 
-def get_current_per_pulse(pulseTimeS, currentTimeS, currents):
-    # add currents for early pulses and current time value after last pulse (j+1)
-    currentTimeS = np.hstack([[0], currentTimeS, [pulseTimeS[-1]+1]])
-    currents = np.hstack([[0], currents])
-    pulseCurrentS = np.zeros(pulseTimeS.shape[0], dtype=float)
-    j = 0
-    for i, ti in enumerate(pulseTimeS):
-        if ti >= currentTimeS[j+1]: 
-            j += 1
-        pulseCurrentS[i] = currents[j]
-        #print(f' {i}  {pulseTimeS[i]}  {pulseCurrentS[i]}') 
-    return pulseCurrentS
-
 class AmorData:
     """read meta-data and event streams from .hdf file(s), apply filters and conversions"""
     chopperDetectorDistance: float
@@ -196,6 +183,10 @@ class AmorData:
 
         self.read_event_stream()
         totalNumber = np.shape(self.tof_e)[0]
+        # check for empty event stream
+        if totalNumber == 0:
+             logging.error('empty event stream: can not determine end time')
+             sys.exit()
 
         self.sort_pulses()
 
@@ -251,11 +242,25 @@ class AmorData:
             self.pulseTimeS = np.append(self.pulseTimeS, tt)
             nextPulseTime = self.pulseTimeS[-1] + chopperPeriod
 
+    def get_current_per_pulse(self, pulseTimeS, currentTimeS, currents):
+        # add currents for early pulses and current time value after last pulse (j+1)
+        currentTimeS = np.hstack([[0], currentTimeS, [pulseTimeS[-1]+1]])
+        currents = np.hstack([[0], currents])
+        pulseCurrentS = np.zeros(pulseTimeS.shape[0], dtype=float)
+        j = 0
+        for i, ti in enumerate(pulseTimeS):
+            if ti >= currentTimeS[j+1]: 
+                j += 1
+            pulseCurrentS[i] = currents[j]
+            #print(f' {i}  {pulseTimeS[i]}  {pulseCurrentS[i]}') 
+        return pulseCurrentS
+
+
     def associate_pulse_with_monitor(self):
         if self.config.monitorType == 'p': # protonCharge
             self.currentTime -= np.int64(self.seriesStartTime)
             self.currentTime -= np.int64(16e9) # time offset of proton current signal
-            self.monitorPerPulse = get_current_per_pulse(self.pulseTimeS, self.currentTime, self.current) * 2*self.tau * 1e-3
+            self.monitorPerPulse = self.get_current_per_pulse(self.pulseTimeS, self.currentTime, self.current) * 2*self.tau * 1e-3
             # filter low-current pulses
             self.monitorPerPulse = np.where(self.monitorPerPulse > 2*self.tau * self.config.lowCurrentThreshold * 1e-3, self.monitorPerPulse, 0)
         elif self.config.monitorType == 't': # countingTime