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:
@@ -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)
|
||||
|
||||
@@ -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)
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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,5 +1,4 @@
|
||||
import os
|
||||
import colorama
|
||||
|
||||
from slic.utils import make_missing_dir
|
||||
from slic.utils.printing import printable_dict
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user