From 5215afcea90506be8d2491dd8d82d41d34d02ea4 Mon Sep 17 00:00:00 2001
From: Sven Augustin <sven.augustin@psi.ch>
Date: Wed, 18 Nov 2020 11:53:39 +0100
Subject: [PATCH] added a FakeAcquisition for testing (can test scans when
 combined with DummyAdjustable)

---
 slic/core/acquisition/fakeacquisition.py | 50 ++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 slic/core/acquisition/fakeacquisition.py

diff --git a/slic/core/acquisition/fakeacquisition.py b/slic/core/acquisition/fakeacquisition.py
new file mode 100644
index 00000000..db86c3eb
--- /dev/null
+++ b/slic/core/acquisition/fakeacquisition.py
@@ -0,0 +1,50 @@
+from time import sleep
+
+from slic.core.task import DAQTask
+from slic.utils import typename
+
+from .baseacquisition import BaseAcquisition
+
+
+class FakeAcquisition(BaseAcquisition):
+
+    def __init__(self, instrument, pgroup):
+        self.instrument = instrument
+        self.pgroup = pgroup
+        self._stop()
+
+
+    def acquire(self, filename, detectors=None, channels=None, pvs=None, scan_info=None, n_pulses=100, wait=True):
+
+        def _acquire():
+            args = (filename, n_pulses)
+            args = ", ".join(repr(i) for i in args)
+            print("acquire({})".format(args))
+            self.running = True
+            for i in range(n_pulses):
+                if not self.running:
+                    break
+                print(f"acquire to {filename}: pulse {i}")
+                sleep(0.1)
+
+        task = DAQTask(_acquire, stopper=self._stop, filename=filename, hold=False)
+        self.current_task = task
+
+        if wait:
+            try:
+                task.wait()
+            except KeyboardInterrupt:
+                print("Stopped current DAQ task:")
+
+        return task
+
+
+    def _stop(self):
+        self.running = False
+
+
+    def __repr__(self):
+        return typename(self)
+
+
+