From a23c35bc4e44cdf5ca02391a35ae34aaed2124b7 Mon Sep 17 00:00:00 2001
From: Sven Augustin <sven.augustin@psi.ch>
Date: Mon, 18 Jul 2022 18:42:27 +0200
Subject: [PATCH] added time check

---
 slic/devices/general/detectors/pvdatastream2.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/slic/devices/general/detectors/pvdatastream2.py b/slic/devices/general/detectors/pvdatastream2.py
index ee4ea2e7..e0a19ce8 100644
--- a/slic/devices/general/detectors/pvdatastream2.py
+++ b/slic/devices/general/detectors/pvdatastream2.py
@@ -1,6 +1,7 @@
 from slic.utils.hastyepics import get_pv as PV
 
 from .buffer import Buffer
+from .timer import Timer
 
 
 class PVDataStream:
@@ -11,18 +12,19 @@ class PVDataStream:
         self.pv = PV(name)
 
 
-    def record(self, n):
+    def record(self, n, seconds=None):
         pv = self.pv
 
         current = pv.get()
         buf = Buffer.from_example(n, current)
+        tim = Timer(seconds)
 
         running = True
 
         def on_value_change(value=None, **kwargs):
             nonlocal running
             buf.append(value)
-            if buf.is_full:
+            if buf.is_full or tim.is_done:
                 pv.clear_callbacks()
                 running = False