clean up: removed superfluous kwargs, fixed os.system() arg seperator, white space, superfluous imports, lists as default arguments, forward TimedLoop args to Loop constructor, cleaner Task status logic

This commit is contained in:
2020-05-10 11:57:45 +00:00
parent 660ef30428
commit 4f5ce82f1e
8 changed files with 11 additions and 14 deletions
+1 -1
View File
@@ -38,7 +38,7 @@ class DIAAcquisition(BaseAcquisition):
self.client = DetectorIntegrationClient(api_address)
def acquire(self, filename=None, channels=None, n_pulses=100, use_default_dir=True, is_HG0=False, **kwargs):
def acquire(self, filename=None, channels=None, n_pulses=100, use_default_dir=True, is_HG0=False):
if filename:
if use_default_dir:
filename = os.path.join(self.default_dir, filename)
+1 -1
View File
@@ -96,7 +96,7 @@ def analyze_pedestal_on_cluster(instrument, raw_file_base, res_dir, user=None):
cmd = f"sbatch jungfrau_create_pedestals --filename {ped} --directory {res_dir} --verbosity 4"
commands.append(cmd)
cmd = "\;".join(commands)
cmd = "; ".join(commands)
os.system(cmd)
+1 -1
View File
@@ -70,7 +70,7 @@ def make_arrays(pvs, n_pulses):
shape = get_shape(val)
shape = (n_pulses,) + shape
arr = np.empty(shape, dtype)
arr = np.empty(shape, dtype)
arrays.append(arr)
return arrays
+1 -1
View File
@@ -15,7 +15,7 @@ class SwissFELPaths:
# converted: "/sf/alvra/data/p18442/res/JF_pedestals/pedestal_20200202_2046.JF02T09V02.res.h5"
# send to DIA: "/sf/alvra/data/p18442/res/JF_pedestals/pedestal_20200202_2046" where ".DETECTOR.res.h5" will be appended
pede_files = pede + "pedestal_*.res.h5"
pede_files = pede + "pedestal_*.res.h5"
default_channel_list = instrument + "config/com/channel_lists/default_channel_list"
-1
View File
@@ -1,5 +1,4 @@
import os
import colorama
from slic.utils import make_missing_dir
from slic.utils.printing import printable_dict
+2 -2
View File
@@ -13,7 +13,7 @@ def make_positions(start, end, n):
class Scanner:
def __init__(self, data_base_dir="", scan_info_dir="", default_acquisitions=[], condition=None, make_scan_sub_dir=True):
def __init__(self, data_base_dir="", scan_info_dir="", default_acquisitions=(), condition=None, make_scan_sub_dir=True):
self.data_base_dir = data_base_dir
self.scan_info_dir = scan_info_dir
self.default_acquisitions = default_acquisitions
@@ -23,7 +23,7 @@ class Scanner:
self.filename_generator = RunFilenameGenerator(scan_info_dir)
def make_scan(self, adjustables, positions, n_pulses, filename, acquisitions=[], start_immediately=True, step_info=None):
def make_scan(self, adjustables, positions, n_pulses, filename, acquisitions=(), start_immediately=True, step_info=None):
filename = self.filename_generator.get_next_run_filename(filename)
if not acquisitions:
+2 -2
View File
@@ -34,9 +34,9 @@ class Loop(Task):
class TimedLoop(Loop):
def __init__(self, func, max_time, wait_time=0, hold=True):
def __init__(self, func, max_time, **kwargs):
self.timer = Timer(max_time)
super().__init__(func, wait_time, hold)
super().__init__(func, **kwargs)
def loop(self):
while self.running and self.timer:
+3 -5
View File
@@ -29,11 +29,9 @@ class Task(BaseTask):
def status(self):
if self.thread.ident is None:
return "ready"
else:
if self.thread.isAlive():
return "running"
else:
return "done"
if self.thread.isAlive():
return "running"
return "done"
def __repr__(self):
name = typename(self)