added a FakeClient to manage a (fake) run number

This commit is contained in:
2022-03-08 15:54:05 +01:00
parent ae0450c0d0
commit abe2c4af31
+22 -1
View File
@@ -15,7 +15,10 @@ class FakeAcquisition(BaseAcquisition):
self.instrument = instrument
self.pgroup = pgroup
self.client = FakeClient()
self.default_data_base_dir = "static_data"
self.default_dir = ""
self.default_detectors = shuffle([f"JFJFJFJFJFJFJFJFJFJF{i}" for i in range(2)])
self.default_channels = shuffle([f"CHCHCHCHCH{i}" for i in range(100)])
@@ -24,7 +27,14 @@ class FakeAcquisition(BaseAcquisition):
self._stop()
def acquire(self, filename, data_base_dir=None, detectors=None, channels=None, pvs=None, scan_info=None, n_pulses=100, continuous=False, wait=True):
def acquire(self, filename, data_base_dir=None, detectors=None, channels=None, pvs=None, scan_info=None, n_pulses=100, continuous=False, is_scan_step=False, wait=True):
if not is_scan_step:
run_number = self.client.next_run()
print(f"Advanced run number to {run_number}.")
else:
run_number = self.client.run_number
print(f"Continuing run number {run_number}.")
if data_base_dir is None:
print("No base directory specified, using default base directory.")
data_base_dir = self.default_data_base_dir
@@ -72,3 +82,14 @@ def shuffle(seq):
class FakeClient:
run_number = 0
def next_run(self):
self.run_number += 1
print("run number is", self.run_number)
return self.run_number