diff --git a/src/toolbox_scs/misc/__init__.py b/src/toolbox_scs/misc/__init__.py index 0f4edde6c50e746b723ee7072d80998414d0534f..3ff302d5efa9d33103e834353fd4f36a0eff5dcd 100644 --- a/src/toolbox_scs/misc/__init__.py +++ b/src/toolbox_scs/misc/__init__.py @@ -1,25 +1,9 @@ -from .bunch_pattern import (extractBunchPattern, pulsePatternInfo, - repRate) -from .bunch_pattern_external import (is_sase_3, is_sase_1, - is_ppl, is_pulse_at) -from .laser_utils import positionToDelay, degToRelPower - +from .bunch_pattern import * +from .bunch_pattern_external import * +from .laser_utils import * __all__ = ( - # Functions - "extractBunchPattern", - "pulsePatternInfo", - "repRate", - "sortBAMdata", - "is_sase_3", - "is_sase_1", - "is_ppl", - "is_pulse_at", - "get_index_ppl", - "get_index_sase1", - "get_index_sase3", - "positionToDelay", - "degToRelPower", - # Classes - # Variables + bunch_pattern.__all__ + + bunch_pattern_external.__all__ + + laser_utils.__all__ ) diff --git a/src/toolbox_scs/misc/bunch_pattern.py b/src/toolbox_scs/misc/bunch_pattern.py index 3f5939b2a9c11e1b237b5b0b8b9ca3e8821ce4e5..852f9ed71adce12faaf0365f5722d18922fa295f 100644 --- a/src/toolbox_scs/misc/bunch_pattern.py +++ b/src/toolbox_scs/misc/bunch_pattern.py @@ -16,6 +16,12 @@ from extra_data import RunDirectory # import and hide variable, such that it does not alter namespace. from ..constants import mnemonics as _mnemonics_bp +__all__ = [ + 'extractBunchPattern', + 'pulsePatternInfo', + 'repRate' +] + def extractBunchPattern(bp_table=None, key='sase3', runDir=None): ''' generate the bunch pattern and number of pulses of a source directly from the diff --git a/src/toolbox_scs/misc/bunch_pattern_external.py b/src/toolbox_scs/misc/bunch_pattern_external.py index 993b58414ee475f61528e1bf4af0874c624b90a0..70d6afce8025f0d2ff271091cb23bb0be1f0e417 100644 --- a/src/toolbox_scs/misc/bunch_pattern_external.py +++ b/src/toolbox_scs/misc/bunch_pattern_external.py @@ -11,6 +11,13 @@ import logging import euxfel_bunch_pattern as ebp +__all__ = [ + 'is_sase_3', + 'is_sase_1', + 'is_ppl', + 'is_pulse_at', +] + PPL_SCS = ebp.LASER_SEED6 log = logging.getLogger(__name__) diff --git a/src/toolbox_scs/misc/laser_utils.py b/src/toolbox_scs/misc/laser_utils.py index 55e44a39839904268223b15df6cb537af77727e9..b284999e94af2921cdd4313f5be08bd996a335d5 100644 --- a/src/toolbox_scs/misc/laser_utils.py +++ b/src/toolbox_scs/misc/laser_utils.py @@ -1,3 +1,9 @@ +__all__ = [ + 'degToRelPower', + 'positionToDelay', +] + + def positionToDelay(pos, origin=0, invert = False, reflections=1): ''' converts a motor position in mm into optical delay in picosecond Inputs: