Automatic backup triggered by new deployment

Reviewed-on: #73

.copier-answers.yml

@@ -2,7 +2,7 @@
 # It is needed to track the repo template version, and editing may break things.
 # This file will be overwritten by copier on template updates.
 
-_commit: v1.2.2
+_commit: v1.4.0
 _src_path: https://github.com/bec-project/plugin_copier_template.git
 make_commit: false
 project_name: debye_bec

.gitea/workflows/ci.yml

@@ -28,7 +28,7 @@ on:
         description: "Python version to use"
         required: false
         type: string
-        default: "3.11"
+        default: "3.12"
 
 permissions:
   pull-requests: write
@@ -44,7 +44,19 @@ jobs:
       - name: Setup Python
         uses: actions/setup-python@v5
         with:
-          python-version: "${{ inputs.PYTHON_VERSION || '3.11' }}"
+          python-version: "${{ inputs.PYTHON_VERSION || '3.12' }}"
+
+      - name: Checkout BEC Plugin Repository
+        uses: actions/checkout@v4
+        with:
+          repository: bec/debye_bec
+          ref: "${{ inputs.BEC_PLUGIN_REPO_BRANCH || github.head_ref || github.sha }}"
+          path: ./debye_bec
+
+      - name: Lint for merge conflicts from template updates
+        shell: bash
+        # Find all Copier conflicts except this line
+        run: '! grep -r "<<<<<<< before updating" | grep -v "grep -r \"<<<<<<< before updating"'
 
       - name: Checkout BEC Core
         uses: actions/checkout@v4
@@ -67,13 +79,6 @@ jobs:
           ref: "${{ inputs.BEC_WIDGETS_BRANCH || 'main' }}"
           path: ./bec_widgets
 
-      - name: Checkout BEC Plugin Repository
-        uses: actions/checkout@v4
-        with:
-          repository: bec/debye_bec
-          ref: "${{ inputs.BEC_PLUGIN_REPO_BRANCH || github.head_ref || github.sha }}"
-          path: ./debye_bec
-
       - name: Install dependencies
         shell: bash
         run: |

.gitea/workflows/create_update_pr.yml

@@ -0,0 +1,70 @@
+name: Create template upgrade PR for debye_bec
+on:
+  workflow_dispatch:
+
+permissions:
+  pull-requests: write
+
+jobs:
+  create_update_branch_and_pr:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: write
+      pull-requests: write
+
+    steps:
+      - name: Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.12'
+
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Create virtualenv
+	    run: |
+	      python -m virtualenv .venv
+
+      - name: Install tools
+        run: |
+          source .venv/bin/activate
+          pip install copier PySide6 bec_lib
+
+      - name: Perform update
+        run: |
+          source .venv/bin/activate
+          git config --global user.email "bec_ci_staging@psi.ch"
+          git config --global user.name "BEC automated CI"
+
+          branch="chore/update-template-$(python -m uuid)"
+          echo "switching to branch $branch"
+          git checkout -b $branch
+
+          echo "Running copier update..."
+          copier update --trust --defaults --conflict inline 2>&1 | tee copier.log
+          status=${PIPESTATUS[0]}
+          output="$(cat copier.log)"
+          echo $output
+          msg="$(printf '%s\n' "$output" | head -n 1)"
+
+          if ! grep -q "make_commit: true" .copier-answers.yml ; then
+            echo "Autocommit not made, committing..."
+            git add -A
+            git commit -a -m "$msg"
+          fi
+
+          if diff-index --quiet HEAD ; then
+            echo "No changes detected"
+            exit 0
+          fi
+
+          git push -u origin $branch
+          curl -X POST "https://gitea.psi.ch/api/v1/repos/${{ gitea.repository }}/pulls" \
+            -H "Authorization: token ${{ secrets.CI_REPO_WRITE }}" \
+            -H "Content-Type: application/json" \
+            -d "{
+                  \"title\": \"Template: $(echo $msg)\",
+                  \"body\": \"This PR was created by Gitea Actions\",
+                  \"head\": \"$(echo $branch)\",
+                  \"base\": \"main\"
+                }"

.gitlab-ci.yml

@@ -1,7 +0,0 @@
-include:
-- file: /templates/plugin-repo-template.yml
-  inputs:
-    name: debye_bec
-    target: debye_bec
-    branch: $CHILD_PIPELINE_BRANCH
-  project: bec/awi_utils

debye_bec/bec_ipython_client/plugins/move_to_label.py

@@ -0,0 +1,82 @@
+from __future__ import annotations
+
+import builtins
+from typing import TYPE_CHECKING
+
+from bec_lib import bec_logger
+from debye_bec.devices.absorber import STATUS as ABS_STATUS
+
+logger = bec_logger.logger
+# import builtins to avoid linter errors
+dev = builtins.__dict__.get("dev")
+
+class MoveToLabelError(Exception):
+    """Exception for the MoveToLabel function"""
+
+def move_to_label():
+    """
+    Function to move several motors to a specific position defined in the label dict.
+    """
+
+    label = get_device_conditions(label="digitalTwin")
+
+    # Get absorber status and close if open
+    logger.info("Check Frontend Absorber Status")
+    abs_was_open = dev.abs.status.get() == ABS_STATUS.OPEN
+    if abs_was_open:
+        logger.info("  Close Frontend Absorber")
+        status = dev.abs.close()
+        status.wait()
+
+    # Move Frontend Slits
+    logger.info("Move Frontend Slits into position")
+    devices = ["sldi_centerx", "sldi_centery", "sldi_gapx", "sldi_gapy"]
+    matches = {key: label[key] for key in devices if key in label}
+    statuses = []
+    for device in matches.values():
+        statuses.append(device['device'].move(device['value']))
+    for status in statuses:
+        status.wait(timeout=30)
+
+    # Move Collimating mirror
+    logger.info("Move Collimating Mirror into position")
+    if "cm_rotx" in label: # pitch
+        logger.info("  Move pitch into position")
+        surveyed_movement(
+            axis=label['cm_rotx'],
+            surveyed_axes= [
+                {'device': dev.cm_rotz, 'abs_tol': 0.1},
+            ]
+        )
+
+    # Restore absorber position
+    logger.info("Restore Frontend Absorber Status")
+    if abs_was_open:
+        status = dev.abs.open()
+        status.wait()
+
+
+def surveyed_movement(axis, surveyed_axes):
+    """
+    Moves an axis while surverying a set of axes.
+
+    Args:
+        axis (DeviceCondition): Device condition
+        surveyed_axes (list): List of dicts (same format as DeviceCondition)
+
+    Raises:
+        If during movement of axis, one of the surveyed axes moves out of tolerance.
+    """
+
+    for surv_ax in surveyed_axes:
+        surv_ax['old_value'] = surv_ax['device'].read()
+    status = axis['device'].move(axis['value'])
+    while status.status == 'RUNNING':
+        for surv_ax in surveyed_axes:
+            if abs(surv_ax['device'].read() - surv_ax['old_value']) > surv_ax['abs_tol']:
+                axis['device'].stop()
+                raise MoveToLabelError(
+                    f"During movement of {axis['device'].name}, {surv_ax['device'].name} " +
+                    f"started to move unexpectedly (old pos: {surv_ax['old_value']}, " +
+                    f"current pos: {surv_ax['device'].read()})"
+                )

debye_bec/bec_widgets/widgets/client.py

@@ -0,0 +1,41 @@
+# This file was automatically generated by generate_cli.py
+# type: ignore
+
+from __future__ import annotations
+
+from bec_lib.logger import bec_logger
+
+from bec_widgets.cli.rpc.rpc_base import RPCBase, rpc_call, rpc_timeout
+
+logger = bec_logger.logger
+
+# pylint: skip-file
+
+
+_Widgets = {
+    "DigitalTwin": "DigitalTwin",
+}
+
+
+class DigitalTwin(RPCBase):
+    """Main widget of Digital Twin"""
+
+    _IMPORT_MODULE = "debye_bec.bec_widgets.widgets.digital_twin.digital_twin"
+
+    @rpc_call
+    def remove(self):
+        """
+        Cleanup the BECConnector
+        """
+
+    @rpc_call
+    def attach(self):
+        """
+        None
+        """
+
+    @rpc_call
+    def detach(self):
+        """
+        Detach the widget from its parent dock widget (if widget is in the dock), making it a floating widget.
+        """

debye_bec/bec_widgets/widgets/designer_plugins.py

@@ -0,0 +1,13 @@
+# This file was automatically generated by generate_cli.py
+# type: ignore
+from __future__ import annotations
+
+# pylint: skip-file
+
+designer_plugins = {
+    "DigitalTwin": ("debye_bec.bec_widgets.widgets.digital_twin.digital_twin", "DigitalTwin"),
+}
+
+widget_icons = {
+    "DigitalTwin": "lightbulb",
+}

debye_bec/bec_widgets/widgets/digital_twin/calculations/calc_positions.py

@@ -0,0 +1,259 @@
+"""
+Calculates the positions of axes based on a beamline config
+"""
+
+import numpy as np
+from bec_lib import bec_logger
+
+import debye_bec.bec_widgets.widgets.digital_twin.x01da_parameters as bl
+from debye_bec.bec_widgets.widgets.digital_twin.types import ConfigDict
+
+logger = bec_logger.logger
+
+
+def calc_positions(cfg: ConfigDict) -> dict[str, dict[str, float]]:
+    """
+    Calculates the positions of axes based on a beamline config.
+
+    Args:
+        cfg(ConfigDict): Dictionary with beamline config
+
+    Returns:
+        dict[str, dict[str, float]]: Dictionary mapping device names to dictionaries
+            containing a "value" key with the corresponding float value (position).
+    """
+
+    pos = {}
+
+    ## FE slits
+    trxr = -np.arctan(cfg["h_acc"]) * bl.feSlits.center1[1]
+    trxw = (
+        (np.arctan(cfg["h_acc"]) * bl.feSlits.center1[1])
+        / bl.feSlits.center1[1]
+        * bl.feSlits.center2[1]
+    )
+    tryb = -np.arctan(cfg["v_acc"]) * bl.feSlits.center1[1]
+    tryt = (
+        (np.arctan(cfg["v_acc"]) * bl.feSlits.center1[1])
+        / bl.feSlits.center1[1]
+        * bl.feSlits.center2[1]
+    )
+
+    xgap = trxw - trxr
+    ygap = tryt - tryb
+
+    pos["sldi_gapx"] = {"value": xgap}
+    pos["sldi_gapy"] = {"value": ygap}
+
+    ## Collimating Mirror
+    obj_dist = bl.cm.center[1]  # object distance
+    beam_vs = 2 * obj_dist * np.tan(cfg["v_acc"])  # vertical size of beam after CM
+
+    # TRX
+    if cfg["cm_stripe"] in bl.cm.surface:
+        index = bl.cm.surface.index(cfg["cm_stripe"])
+    else:
+        raise ValueError(f"Requested stripe {cfg['cm_stripe']} not found in parameters!")
+    cm_trx = -(bl.cm.limOptX[0][index] + bl.cm.limOptX[1][index]) / 2
+    pos["cm_trx"] = {"value": cm_trx}
+
+    # TRY
+    height = obj_dist * np.tan(cfg["v_acc"]) ** 2 * 1 / np.tan(cfg["cm_pitch"])
+    pos["cm_try"] = {"value": height}
+
+    # Pitch
+    pos["cm_rotx"] = {
+        "value": -cfg["cm_pitch"] * 1e3
+    }  # invert and convert to mrad (same as EGU of rotx axis)
+
+    # Bending Radius
+    radius = (
+        2.0 * obj_dist / np.sin(cfg["cm_pitch"])
+    )  # Elements of modern X-ray Physics, page 108 ff.
+    pos["cm_bnd_radius"] = {"value": radius * 1e-6}  # Convert to km
+
+    ## Monochromator
+    if cfg["mo1_mode"] == "Monochromatic":
+        # Add 2x CM pitch to the bragg angle
+        bragg = cfg["mo1_bragg"]
+    elif cfg["mo1_mode"] == "Pinkbeam":
+        # Align xtal surfaces parallel to beam
+        bragg = 0
+    else:
+        raise ValueError("Monochromator mode not supported")
+    pos["mo1_bragg_angle"] = {"value": bragg / np.pi * 180}  # Bragg angle in deg
+
+    # TRY, Height
+    l = bl.mo1.xtalGap[0] / np.sin(cfg["mo1_bragg"])
+    yhor = l * np.cos(2.0 * (cfg["mo1_bragg"] + cfg["cm_pitch"]))
+    yver = yhor * np.tan(2.0 * cfg["cm_pitch"])
+
+    if cfg["mo1_mode"] == "Monochromatic":
+        beam_offset_mo1 = (
+            l * np.sin(2.0 * (cfg["mo1_bragg"] + cfg["cm_pitch"])) - yver
+        )  # Resultat ist korrekt!
+    elif cfg["mo1_mode"] == "Pinkbeam":
+        beam_offset_mo1 = 0
+    else:
+        raise ValueError("Monochromator mode not supported")
+
+    def csc(a):
+        return 1 / np.sin(a)
+
+    def cot(a):
+        return 1 / np.tan(a)
+
+    # calculate height of center of first crystal surface
+    f = bl.mo1.rotOffset  # rotation offset, mm
+    d = bl.mo1.heightOffset  # xtal height offset, mm
+    c = d * csc(cfg["mo1_bragg"]) - f * cot(cfg["mo1_bragg"])
+
+    # Calculate height of center of rotation
+    b = np.sqrt(
+        d**2 * csc(cfg["mo1_bragg"]) ** 2
+        - 2 * d * f * cot(cfg["mo1_bragg"]) * csc(cfg["mo1_bragg"])
+        + f**2 * cot(cfg["mo1_bragg"]) ** 2
+        + f**2
+    )
+    h = np.cos(np.pi / 2 - np.arctan(f / c) - cfg["mo1_bragg"] - 2 * cfg["cm_pitch"]) * b
+    h2 = ((bl.mo1.center[1] - bl.cm.center[1]) - np.sqrt(b**2 - h**2)) * np.tan(2 * cfg["cm_pitch"])
+    height_mo1_real = (
+        h + h2
+    )  # per design, the height should not change if the pitch of the CM is not changed!
+    if cfg["mo1_mode"] == "Monochromatic":
+        pass
+    elif cfg["mo1_mode"] == "Pinkbeam":
+        height_mo1_real = (
+            height_mo1_real - 13
+        )  # Move down to let beam pass between both crystal without touching copper cooler
+    else:
+        raise ValueError("Monochromator mode not supported")
+    pos["mo1_try"] = {"value": height_mo1_real}
+
+    # TRX, Crystal selection
+    if cfg["mo1_mode"] == "Monochromatic":
+        xtal = cfg["mo1_xtal"].translate(
+            str.maketrans("", "", "()")
+        )  # Remove brackets from xtal name to conform with parameters
+        if xtal in bl.mo1.xtal:
+            index = bl.mo1.xtal.index(xtal)
+        else:
+            raise ValueError(f"Requested xtal {xtal} not found in parameters!")
+        pos["mo1_trx"] = {"value": bl.mo1.xtalOffsetX[index]}
+    else:
+        pos["mo1_trx"] = {"value": 0}
+
+    diag = bl.mo1.xtalGap[0] / np.sin(cfg["mo1_bragg"])  # Calculations for Mono
+    dz = diag * np.cos(2 * (cfg["cm_pitch"] + cfg["mo1_bragg"]))
+
+    ## Slits 1
+    d = bl.opSlits1.center[1] - bl.cm.center[1] - dz
+    sl1_beam_height = d * np.tan(2 * cfg["cm_pitch"]) + beam_offset_mo1
+    pos["sl1_centery"] = {"value": sl1_beam_height}
+    pos["sl1_gapy"] = {"value": beam_vs + 1}  # Add 0.5 mm space on both sides of the beam
+
+    ## Beam Monitor 1
+    d = bl.opBM1.center[1] - bl.cm.center[1] - dz
+    bm1_beam_height = d * np.tan(2 * cfg["cm_pitch"]) + beam_offset_mo1
+    pos["bm1_try"] = {"value": bm1_beam_height}
+
+    ## Focusing Mirror
+    p = bl.fm.center[1]
+    q = cfg["smpl"] - bl.fm.center[1]
+    f = (p * q) / (p + q)  # focal length
+
+    # Bender radius
+    if cfg["fm_qy"] is None:
+        radius = 2 * q / np.sin(cfg["fm_rotx"])  # ideal bending radius for focused beam
+    else:
+        radius = (
+            2 * cfg["fm_qy"] / np.sin(cfg["fm_rotx"])
+        )  # ideal bending radius for unfocused beam
+    pos["fm_bnd_radius"] = {"value": radius * 1e-6}  # Convert to km
+
+    # Pitch
+    d = bl.fm.center[1] - bl.cm.center[1] - dz
+    fm_rotx = (
+        2 * cfg["cm_pitch"] - cfg["fm_rotx"]
+    )  # calculate pitch in absolute values (according to horizontal plane)
+    pos["fm_rotx"] = {
+        "value": -fm_rotx * 1e3
+    }  # invert and convert to mrad (same as EGU of rotx axis)
+
+    if cfg["fm_stripe"] in ("Rh (toroid)", "Pt (toroid)"):
+
+        # TRY
+        if cfg["fm_stripe"] in "Rh (toroid)":
+            r = bl.fm.r[0]
+            h_cyl = bl.fm.hToroid[0]
+        else:  # PT toroid
+            r = bl.fm.r[1]
+            h_cyl = bl.fm.hToroid[1]
+        width_beam = 2 * bl.fm.center[1] * np.tan(cfg["h_acc"] * 1e-3)
+        alpha = np.arccos(1 - width_beam**2 / (2 * r**2))
+        h = r - (r * np.cos(alpha / 2))
+        fm_beam_height = (d * np.tan(2 * cfg["cm_pitch"]) + beam_offset_mo1) * cfg["fm_gain_height"]
+        fm_height = (d * np.tan(2 * cfg["cm_pitch"]) + beam_offset_mo1 - h_cyl + h / 2) * cfg[
+            "fm_gain_height"
+        ]
+        pos["fm_try"] = {"value": fm_height}
+
+        # TRX
+        if cfg["fm_stripe"] in "Rh (toroid)":
+            x_cyl = -bl.fm.xToroid[0]
+        else:
+            x_cyl = -bl.fm.xToroid[1]
+        pos["fm_trx"] = {"value": x_cyl}
+
+    elif cfg["fm_stripe"] in ("Rh (flat)", "Pt (flat)"):
+
+        # TRY
+        fm_height = (d * np.tan(2 * cfg["cm_pitch"]) + beam_offset_mo1) * cfg["fm_gain_height"]
+        fm_beam_height = fm_height
+        pos["fm_try"] = {"value": fm_height}
+
+        # TRX
+        if cfg["fm_stripe"] in "Rh (flat)":
+            x_flat = -bl.fm.xFlat[0]
+        else:
+            x_flat = -bl.fm.xFlat[1]
+        pos["fm_trx"] = {"value": x_flat}
+
+    else:
+        raise ValueError("FM Stripe selection not valid")
+
+    pos["fm_roty"] = {"value": 0}
+    pos["fm_rotz"] = {"value": 0}
+
+    ## Slits 2
+    d = bl.opSlits2.center[1] - bl.fm.center[1]
+    sl2_beam_height = fm_beam_height - d * np.tan(-(2 * cfg["cm_pitch"] - 2 * cfg["fm_rotx"]))
+    pos["sl2_centery"] = {"value": sl2_beam_height}
+    pos["sl2_gapy"] = {"value": beam_vs + 1}  # Add 0.5 mm space on both sides of the beam
+
+    ## Beam Monitor 2
+    d = bl.opBM2.center[1] - bl.fm.center[1]
+    bm2_beam_height = fm_beam_height - d * np.tan(-(2 * cfg["cm_pitch"] - 2 * cfg["fm_rotx"]))
+    pos["bm2_try"] = {"value": bm2_beam_height}
+
+    ## Optical Table
+
+    # TRY
+    d = bl.ehWindow.center[1] - bl.fm.center[1]
+    ot_height = fm_beam_height - d * np.tan(-(2 * cfg["cm_pitch"] - 2 * cfg["fm_rotx"]))
+    pos["ot_try"] = {"value": ot_height}
+
+    # Pitch
+    ot_pitch = -(2 * cfg["cm_pitch"] - 2 * cfg["fm_rotx"])
+    pos["ot_rotx"] = {"value": ot_pitch * 1e3}
+
+    # TRZ ES1
+    ot_es1_trz = cfg["smpl"]
+    pos["ot_es1_trz"] = {"value": ot_es1_trz}
+
+    # ES0 exit window
+    pos["es0wi_try"] = {
+        "value": 5
+    }  # At 5mm, the middle of the window is 500 mm from the table (neutral position)
+
+    return pos

debye_bec/bec_widgets/widgets/digital_twin/calculations/calc_sideview.py

@@ -0,0 +1,70 @@
+"""
+Calculates the sideview coordinates based on a beamline config.
+"""
+
+import numpy as np
+
+import debye_bec.bec_widgets.widgets.digital_twin.x01da_parameters as bl
+from debye_bec.bec_widgets.widgets.digital_twin.types import ConfigDict, DataDict
+
+
+def calc_sideview(cfg: ConfigDict) -> DataDict:
+    """
+    Calculates the sideview coordinates based on a beamline config.
+
+    Args:
+        cfg(ConfigDict): Dictionary with beamline config
+
+    Returns:
+        DataDict: Sideview data
+    """
+
+    beam: DataDict = {"x": [], "y": []}
+
+    beam["x"] = []
+    beam["y"] = []
+    beam["x"].append(0)  # Source
+    beam["y"].append(bl.sourceHeight)
+    beam["x"].append(bl.cm.center[1])  # CM
+    beam["y"].append(bl.sourceHeight)
+    if cfg["mo1_mode"] in "Monochromatic":
+        diag = bl.mo1.xtalGap[0] / np.sin(cfg["mo1_bragg"])  # Calculations for Mono
+        dy = diag * np.sin(2 * (cfg["cm_pitch"] + cfg["mo1_bragg"]))
+        dz = diag * np.cos(2 * (cfg["cm_pitch"] + cfg["mo1_bragg"]))
+        beam["x"].append(bl.mo1.center[1] - dz / 2)  # Mono 1.1
+        beam["y"].append(
+            bl.sourceHeight
+            + np.tan(2 * cfg["cm_pitch"]) * (bl.mo1.center[1] - dz / 2 - bl.cm.center[1])
+        )
+        beam["x"].append(bl.mo1.center[1] + dz / 2)  # Mono 1.2
+        beam["y"].append(
+            bl.sourceHeight
+            + np.tan(2 * cfg["cm_pitch"]) * (bl.mo1.center[1] - dz / 2 - bl.cm.center[1])
+            + dy
+        )
+        beam["x"].append(bl.fm.center[1])  # FM
+        beam["y"].append(
+            bl.sourceHeight
+            + np.tan(2 * cfg["cm_pitch"]) * (bl.fm.center[1] - bl.cm.center[1] - dz)
+            + dy
+        )
+        beam["x"].append(cfg["smpl"])  # Experiment
+        beam["y"].append(
+            bl.sourceHeight
+            + np.tan(2 * cfg["cm_pitch"]) * (bl.fm.center[1] - bl.cm.center[1] - dz)
+            + dy
+            + np.tan(2 * (cfg["cm_pitch"] - cfg["fm_rotx"])) * (cfg["smpl"] - bl.fm.center[1])
+        )
+    elif cfg["mo1_mode"] == "Pinkbeam":
+        beam["x"].append(bl.fm.center[1])  # FM
+        beam["y"].append(
+            bl.sourceHeight + np.tan(2 * cfg["cm_pitch"]) * (bl.fm.center[1] - bl.cm.center[1])
+        )
+        beam["x"].append(cfg["smpl"])  # Experiment
+        beam["y"].append(
+            bl.sourceHeight
+            + np.tan(2 * cfg["cm_pitch"]) * (bl.fm.center[1] - bl.cm.center[1])
+            + np.tan(2 * (cfg["cm_pitch"] - cfg["fm_rotx"])) * (cfg["smpl"] - bl.fm.center[1])
+        )
+
+    return beam

debye_bec/bec_widgets/widgets/digital_twin/calculations/calc_surfaces.py

@@ -0,0 +1,157 @@
+"""
+Calculates the surface coordinates based on a beamline config.
+"""
+
+import re
+
+import numpy as np
+from bec_lib import bec_logger
+
+import debye_bec.bec_widgets.widgets.digital_twin.x01da_parameters as bl
+from debye_bec.bec_widgets.widgets.digital_twin.types import ConfigDict, SurfaceDict
+
+logger = bec_logger.logger
+
+
+def calc_surfaces(cfg: ConfigDict) -> SurfaceDict:
+    """
+    Calculates the surface coordinates based on a beamline config.
+
+    Args:
+        cfg(ConfigDict): Dictionary with beamline config
+
+    Returns:
+        SurfaceDict: Surface data
+    """
+
+    out: SurfaceDict = {
+        "cm": {"x": [], "y": []},
+        "mo1_1": {"x": [], "y": []},
+        "mo1_2": {"x": [], "y": []},
+        "fm": {"x": [], "y": []},
+    }
+
+    # Collimating mirror
+    l = 2 * bl.cm.center[1] * np.tan(cfg["v_acc"]) / np.sin(cfg["cm_pitch"])
+
+    w1 = 2 * (bl.cm.center[1] - l / 2) * np.tan(cfg["h_acc"])
+    w2 = 2 * (bl.cm.center[1] + l / 2) * np.tan(cfg["h_acc"])
+
+    index = bl.cm.surface.index(cfg["cm_stripe"])
+
+    cen = -cfg["cm_trx"]
+
+    out["cm"]["x"] = [cen - w1 / 2, cen - w2 / 2, cen + w2 / 2, cen + w1 / 2]
+    out["cm"]["y"] = [-l / 2, l / 2, l / 2, -l / 2]
+
+    # Monochromator
+    # calculate height of center of first crystal surface
+    c = bl.mo1.heightOffset * 1 / np.sin(cfg["mo1_bragg"]) - bl.mo1.rotOffset * 1 / np.tan(
+        cfg["mo1_bragg"]
+    )
+    e = bl.mo1.xtalGap[0] / np.tan(cfg["mo1_bragg"]) - c
+
+    xtal = cfg["mo1_xtal"].translate(
+        str.maketrans("", "", "()")
+    )  # Remove brackets from xtal name to conform with parameters
+    index = bl.mo1.xtal.index(xtal)
+
+    xtal_pos = bl.mo1.xtalOffsetX[index]
+    xtal_length_1 = bl.mo1.xtalLength1[index]
+    xtal_length_2 = bl.mo1.xtalLength2[index]
+
+    width_beam = 2 * bl.mo1.center[1] * np.tan(cfg["h_acc"])
+
+    height_beam = 2 * bl.cm.center[1] * np.tan(cfg["v_acc"])
+    w = height_beam / np.sin(cfg["mo1_bragg"])
+
+    if cfg["mo1_mode"] in "Monochromatic":
+        out["mo1_1"]["x"] = [
+            xtal_pos - width_beam / 2,
+            xtal_pos + width_beam / 2,
+            xtal_pos + width_beam / 2,
+            xtal_pos - width_beam / 2,
+        ]
+        out["mo1_1"]["y"] = [
+            xtal_length_1 / 2 - c - w / 2,
+            xtal_length_1 / 2 - c - w / 2,
+            xtal_length_1 / 2 - c + w / 2,
+            xtal_length_1 / 2 - c + w / 2,
+        ]
+        out["mo1_2"]["x"] = [
+            xtal_pos - width_beam / 2,
+            xtal_pos + width_beam / 2,
+            xtal_pos + width_beam / 2,
+            xtal_pos - width_beam / 2,
+        ]
+        out["mo1_2"]["y"] = [
+            -xtal_length_2 / 2 + e - w / 2,
+            -xtal_length_2 / 2 + e - w / 2,
+            -xtal_length_2 / 2 + e + w / 2,
+            -xtal_length_2 / 2 + e + w / 2,
+        ]
+    else:  # Pinkbeam
+        out["mo1_1"]["x"] = []
+        out["mo1_1"]["y"] = []
+        out["mo1_2"]["x"] = []
+        out["mo1_2"]["y"] = []
+
+    # Focusing mirror
+    if cfg["fm_stripe"] in ("Rh (toroid)", "Pt (toroid)"):
+        surface = bl.fm.surfaceToroid
+        stripe = re.sub(r"\s*\(.*?\)", "", cfg["fm_stripe"]).strip()
+        index = surface.index(stripe)
+        r = bl.fm.r[index]
+    else:
+        surface = bl.fm.surfaceFlat
+        stripe = re.sub(r"\s*\(.*?\)", "", cfg["fm_stripe"]).strip()
+        index = surface.index(stripe)
+        r = bl.fm.r[index]
+    off = -cfg["fm_trx"]
+
+    width_beam = 2 * bl.fm.center[1] * np.tan(cfg["h_acc"])
+
+    if cfg["fm_stripe"] in ("Rh (toroid)", "Pt (toroid)"):
+
+        l = height_beam / np.sin(cfg["fm_rotx"])
+        alpha = np.arccos(1 - width_beam**2 / (2 * r**2))
+        h = r - (r * np.cos(alpha / 2))
+        z = h / np.tan(cfg["fm_rotx"])
+
+        x = [off - width_beam / 2, off - width_beam / 2]
+        y = [l / 2 - z / 2, -l / 2 - z / 2]
+
+        res = 20
+        x_elipse = np.linspace(0, np.pi, res)
+        y_elipse = np.linspace(0, np.pi, res)
+        x_elipse = [-width_beam / 2 * np.cos(i) + off for i in x_elipse]
+        y_elipse = [width_beam * np.sin(i) * z / width_beam - l / 2 - z / 2 for i in y_elipse]
+
+        x.extend(x_elipse)
+        y.extend(y_elipse)
+
+        x.extend([off + width_beam / 2, off + width_beam / 2])
+        y.extend([-l / 2 - z / 2, l / 2 - z / 2])
+
+        res = 50
+        x_elipse = np.linspace(np.pi, 0, res)
+        y_elipse = np.linspace(np.pi, 0, res)
+        x_elipse = [-width_beam / 2 * np.cos(i) + off for i in x_elipse]
+        y_elipse = [width_beam * np.sin(i) * z / width_beam + l / 2 - z / 2 for i in y_elipse]
+
+        x.extend(x_elipse)
+        y.extend(y_elipse)
+
+        out["fm"]["x"] = x
+        out["fm"]["y"] = y
+
+    else:  # flat surface, no toroid
+        l = height_beam / np.sin(cfg["fm_rotx"])
+
+        w1 = 2 * (bl.fm.center[1] - l / 2) * np.tan(cfg["h_acc"])
+        w2 = 2 * (bl.fm.center[1] + l / 2) * np.tan(cfg["h_acc"])
+
+        out["fm"]["x"] = [off - w1 / 2, off + w1 / 2, off + w2 / 2, off - w2 / 2]
+        out["fm"]["y"] = [-l / 2, -l / 2, l / 2, l / 2]
+
+    return out

debye_bec/bec_widgets/widgets/digital_twin/calculations/calc_varia.py

@@ -0,0 +1,418 @@
+"""
+Various calculations for the digital twin
+"""
+
+import re
+from typing import Literal, cast
+
+import numpy as np
+from bec_lib import bec_logger
+from scipy.interpolate import UnivariateSpline
+from xrt.backends.raycing.physconsts import AVOGADRO, CHeVcm
+
+import debye_bec.bec_widgets.widgets.digital_twin.x01da_parameters as bl
+
+logger = bec_logger.logger
+
+H = 6.62606957e-34
+E = 1.602176634e-19
+C = 299792458
+RE = 2.8179e-15
+
+
+def sldi_gap_to_acc(sldi_gapx: float, sldi_gapy: float) -> tuple[float, float]:
+    """
+    Calculate the slits acceptance based on the gap values
+
+    Args:
+        sldi_gapx(float): GAPX value of the slits in mm
+        sldi_gapy(float): GAPY value of the slits in mm
+
+    Returns:
+        tuple[float, float]: Horizontal and vertical acceptance in rad
+    """
+    d1 = bl.feSlits.center1[1]
+    d2 = bl.feSlits.center2[1]
+    h_acc = np.tan(sldi_gapx / (d2 + d1))
+    v_acc = np.tan(sldi_gapy / (d2 + d1))
+    return h_acc, v_acc
+
+
+def cm_trx_to_stripe(cm_trx: float) -> str | None:
+    """
+    Based on the trx value of the collimating mirror, return
+    the correct stripe
+
+    Args:
+        cm_trx(float): Collimating mirror trx value
+
+    Returns
+        str | None: Stripe of the mirror, None if not found
+    """
+    cm_stripe = None
+    for name, low, high in zip(bl.cm.surface, bl.cm.limOptX[0], bl.cm.limOptX[1]):
+        if low <= cm_trx <= high:
+            cm_stripe = name
+    return cm_stripe
+
+
+def cm_stripe_to_trx(cm_stripe: str) -> float | None:
+    """
+    Based on the stripe of the collimating mirror, return
+    the trx value
+
+    Args:
+        cm_stripe(str): Stripe of the collimating mirror
+
+    Returns:
+        float | None: TRX value of the stripe. None if not found
+    """
+    for name, low, high in zip(bl.cm.surface, bl.cm.limOptX[0], bl.cm.limOptX[1]):
+        if cm_stripe == name:
+            return -(low + high) / 2
+    return None
+
+
+def fm_trx_to_stripe(fm_trx: float) -> str | None:
+    """
+    Based on the trx value of the focusing mirror, return
+    the correct stripe
+
+    Args:
+        fm_trx(float): focusing mirror trx value
+
+    Returns
+        str | None: Stripe of the mirror, None if not found
+    """
+    fm_stripe = None
+    for name, low, high in zip(bl.fm.surfaceFlat, bl.fm.limOptXFlat[1], bl.fm.limOptXFlat[0]):
+        if low <= fm_trx <= high:
+            fm_stripe = name + " (flat)"
+    for name, low, high in zip(bl.fm.surfaceToroid, bl.fm.limOptXToroid[1], bl.fm.limOptXToroid[0]):
+        if low <= fm_trx <= high:
+            fm_stripe = name + " (toroid)"
+    return fm_stripe
+
+
+def fm_stripe_to_trx(fm_stripe: str) -> float | None:
+    """
+    Based on the stripe of the focusing mirror, return
+    the trx value
+
+    Args:
+        fm_stripe(str): Stripe of the focusing mirror
+
+    Returns:
+        float | None: TRX value of the stripe. None if not found
+    """
+    for name, low, high in zip(bl.fm.surfaceFlat, bl.fm.limOptXFlat[1], bl.fm.limOptXFlat[0]):
+        if fm_stripe == name + " (flat)":
+            return (low + high) / 2
+    for name, low, high in zip(bl.fm.surfaceToroid, bl.fm.limOptXToroid[1], bl.fm.limOptXToroid[0]):
+        if fm_stripe == name + " (toroid)":
+            return -(low + high) / 2
+    return None
+
+
+def mo1_energy_resolution(xtal: Literal["Si111", "Si311"], energy: float) -> float:
+    """
+    Calculate the energy resolution of the monochromator
+
+    Args:
+        xtal(str): Xtal name. "Si111" or "Si311"
+        energy(float): Energy in eV
+
+    Returns:
+        float: Energy resolution in eV
+    """
+    index = bl.mo1.xtal.index(xtal)
+    crystal = bl.mo1.material1[index]
+
+    dtheta = np.linspace(-30, 90, 601)
+    theta = crystal.get_Bragg_angle(energy) + dtheta * 1e-6
+    refl = np.abs(crystal.get_amplitude(energy, np.sin(theta))[0]) ** 2  # single crystal
+
+    refl2 = refl**2  # DCM with parallel crystals
+
+    # FWHM of the DCM curve
+    spline = UnivariateSpline(dtheta, refl2 - refl2.max() / 2, s=0)
+    roots = cast(np.ndarray, spline.roots())
+    r1, r2 = float(roots[0]), float(roots[1])
+    fwhm_rad = (r2 - r1) * 1e-6  # µrad → rad
+
+    # Energy resolution
+    theta_b = crystal.get_Bragg_angle(energy)
+    de_over_e = fwhm_rad / np.tan(theta_b)
+    de = de_over_e * energy
+
+    # logger.info(f"DCM FWHM      : {r2-r1:.2f} µrad")
+    # logger.info(f"ΔE/E          : {dE_over_E:.2e}")
+    # logger.info(f"ΔE            : {dE:.3f} eV  at {E} eV")
+
+    return de
+
+
+def cm_reflectivity(cm_stripe: str, cm_pitch: float, energy: float) -> float:
+    """
+    Calculate the reflectivity of the mirror stripe based
+    on the pitch and energy.
+
+    Args:
+        cm_stripe(str): Mirror stripe
+        cm_pitch(float): Pitch of the mirror (beam incidence angle)
+        energy(float): Energy of the beam in eV
+
+    Returns:
+        float: Reflectivity [0-1]
+    """
+    index = bl.cm.surface.index(cm_stripe)
+    rs, _ = bl.cm.material[index].get_amplitude(energy, np.sin(cm_pitch))[0:2]
+    refl = abs(rs) ** 2
+    return refl
+
+
+def fm_reflectivity(fm_stripe: str, fm_pitch: float, energy: float) -> float:
+    """
+    Calculate the reflectivity of the mirror stripe based
+    on the pitch and energy.
+
+    Args:
+        cm_stripe(str): Mirror stripe
+        cm_pitch(float): Pitch of the mirror (beam incidence angle)
+        energy(float): Energy of the beam in eV
+
+    Returns:
+        float: Reflectivity [0-1]
+    """
+    if fm_stripe in ("Rh (toroid)", "Pt (toroid)"):
+        surface = bl.fm.surfaceToroid
+        material = bl.fm.materialToroid
+        stripe = re.sub(r"\s*\(.*?\)", "", fm_stripe).strip()
+        index = surface.index(stripe)
+    else:
+        surface = bl.fm.surfaceFlat
+        material = bl.fm.materialFlat
+        stripe = re.sub(r"\s*\(.*?\)", "", fm_stripe).strip()
+        index = surface.index(stripe)
+    rs, _ = material[index].get_amplitude(energy, np.sin(fm_pitch))[0:2]
+    refl = abs(rs) ** 2
+    return refl
+
+
+def mo1_bragg_angle(
+    mo_mode: Literal["Monochromatic", "Pinkbeam"], d_spacing: float, energy: float, cm_pitch: float
+) -> tuple[float, float]:
+    """
+    Calculate the bragg angle of the monochromator.
+    Corrects for the collimating mirror pitch.
+
+    Args:
+        mo_mode(str): Monochromator mode. "Monochromatic" or "Pinkbeam"
+        d_spacing(float): D-spacing of the crystal in Angstrom
+        energy(float): Energy of the beam in eV
+        cm_pitch(float): Pitch of collimating mirror in rad
+
+    Returns:
+        tuple[float, float]: Bragg angle and corrected bragg angle
+    """
+    wl = C * H / (E * energy)
+    val = wl / (2 * d_spacing * 1e-10)
+    bragg_angle = 0
+    if val > -1 and val < 1:
+        bragg_angle = np.asin(val)
+    if mo_mode == "Monochromatic":
+        # Add 2x CM pitch to the bragg angle
+        bragg_angle_cor = (2 * cm_pitch) + bragg_angle
+    else:
+        # Align xtal surfaces parallel to beam
+        bragg_angle_cor = 2 * cm_pitch
+    return bragg_angle, bragg_angle_cor
+
+
+def fm_ideal_pitch(
+    fm_focus: Literal["Defocused", "Focused", "Manual"],
+    fm_stripe: str,
+    smpl: float,
+    sldi_hacc: float | None = None,
+    sldi_vacc: float | None = None,
+    fm_focx: float | None = None,
+    fm_focy: float | None = None,
+) -> tuple[float, float | None]:
+    """
+    Calculates the ideal pitch for the focusing mirror depending on the
+    focusing strategy.
+    If "Defocused" is chosed, sldi_hacc, sldi_vacc, fm_focx and fm_focy
+    must be provided.
+
+    Args:
+        fm_focus(str): Focus strategy. "Defocused", "Focused" or "Manual
+        fm_stripe(str): Mirror stripe
+        smpl(float): Sample position in mm from source
+        sldi_hacc(float): Horizontal acceptance of frontend slits. Defaults to None
+        sldi_vacc(float): Vertical acceptance of frontend slits. Defaults to None
+        fm_focx(float): Requested horizontal spot size in mm. Defaults to None
+        fm_focy(float): Requested vertical spot size in mm. Defaults to None
+
+    Returns:
+        tuple[float, float | None]: Pitch of mirror in rad, qy in mm
+    """
+    p = bl.fm.center[1]  # posFM
+    q = smpl - bl.fm.center[1]  # dist posFM to posEX
+    if fm_focus in "Defocused":
+        assert sldi_hacc is not None, "sldi_hacc must be provided for Defocused mode"
+        assert sldi_vacc is not None, "sldi_vacc must be provided for Defocused mode"
+        assert fm_focx is not None, "fm_focx must be provided for Defocused mode"
+        assert fm_focy is not None, "fm_focy must be provided for Defocused mode"
+        a = 2 * np.tan(sldi_hacc) * bl.fm.center[1]  # Beam width at focusing mirror
+        b = (
+            2 * np.tan(sldi_vacc) * bl.cm.center[1]
+        )  # Beam height at focusing mirror (collimated beam)
+        x = fm_focx
+        y = fm_focy
+        qx = q + x * p / a
+        qy = q + y * p / b
+        f = (p * qx) / (p + qx)  # focal length
+    else:  # Calculate for focused beam on sample in "manual" and "focused" mode
+        qy = None
+        f = (p * q) / (p + q)  # focal length
+    pitch = 0
+    if "Rh" in fm_stripe:
+        pitch = np.arcsin(bl.fm.r[0] / (2 * f))  # ideal pitch for FM
+    if "Pt" in fm_stripe:
+        pitch = np.arcsin(bl.fm.r[1] / (2 * f))  # ideal pitch for FM
+    return pitch, qy
+
+
+def cm_critical_angle(cm_stripe: Literal["Si", "Pt", "Rh"], energy) -> float:
+    """
+    Calculate the critical angle of the mirror stripe
+
+    Args:
+        cm_stripe(str): Mirror stripe. "Si", "Pt" or "Rh"
+        energy(float): Energy in eV
+
+    Returns:
+        float: Critical angle in rad
+    """
+    if cm_stripe in "Si":
+        stripe = bl.stripeSi
+    elif cm_stripe in "Pt":
+        stripe = bl.stripePt
+    else:
+        stripe = bl.stripeRh
+    w = CHeVcm / 100 / energy  # convert energy [eV] to wavelength [m]
+    f1 = stripe.elements[0].Z + np.real(stripe.elements[0].get_f1f2(energy))
+    number_density = stripe.rho * 1e3 * AVOGADRO / (stripe.elements[0].mass / 1e3)
+    critical_angle = np.sqrt(number_density * RE * w**2 * f1 / np.pi)
+    return critical_angle
+
+
+def mirror_surface_geometries(
+    mirror: Literal["cm", "fm_toroid", "fm_flat"],
+) -> dict[str, tuple[float, float, float, float]]:
+    """
+    Return the mirror stripe geometries
+
+    Args:
+        mirror(str): Mirror. "cm", "fm_toroid" or "fm_flat"
+
+    Returns:
+        dict[str, tuple[float, float, float, float]]: Dictionary mapping surface
+        names to tuples of (x, y, width, height).
+    """
+    if mirror in "cm":
+        surface = bl.cm.surface
+        lim_opt_x = bl.cm.limOptX
+        lim_opt_y = bl.cm.limOptY
+    elif mirror in "fm_toroid":
+        surface = bl.fm.surfaceToroid
+        lim_opt_x = bl.fm.limOptXToroid
+        lim_opt_y = bl.fm.limOptYToroid
+    elif mirror in "fm_flat":
+        surface = bl.fm.surfaceFlat
+        lim_opt_x = bl.fm.limOptXFlat
+        lim_opt_y = bl.fm.limOptYFlat
+    else:
+        raise ValueError(f"Requested mirror {mirror} not available!")
+    geom = {}
+    for sf, lx, hx, ly, hy in zip(surface, lim_opt_x[0], lim_opt_x[1], lim_opt_y[0], lim_opt_y[1]):
+        geom[sf] = (lx, ly, hx - lx, hy - ly)
+    return geom
+
+
+def mo_surface_geometries(
+    mo: Literal["mo1"], plane: Literal[0, 1]
+) -> dict[str, tuple[float, float, float, float]]:
+    """
+    Return the monochromator xtal geometries
+
+    Args:
+        mo(str): Monochromator. Only "mo1" implemented
+        plane(int): Surface of xtal. 0 and 1 (First and second)
+
+    Returns:
+        dict[str, tuple[float, float, float, float]]: Dictionary mapping surface
+        names to tuples of (x, y, width, height).
+    """
+    if mo in "mo1":
+        xtal = bl.mo1.xtal
+        xtal_width = bl.mo1.xtalWidth
+        xtal_offset_x = bl.mo1.xtalOffsetX
+        if plane == 0:
+            xtal_length = bl.mo1.xtalLength1
+        else:
+            xtal_length = bl.mo1.xtalLength2
+    else:
+        return {}
+    geom = {}
+    for sf, w, offx, length in zip(xtal, xtal_width, xtal_offset_x, xtal_length):
+        geom[sf] = (offx - w / 2, -length / 2, w, length)
+    return geom
+
+
+def wall_geometries() -> list[list[float]]:
+    """
+    Return the wall geometries
+
+    Returns:
+        list[list[float]]: List of [x, y, width, height] geometry values for each wall.
+    """
+    geom = []
+    for i, _ in enumerate(bl.walls.start):
+        geom.append(
+            [
+                bl.walls.start[i],
+                bl.walls.height[i][0],
+                bl.walls.end[i] - bl.walls.start[i],
+                bl.walls.height[i][1] - bl.walls.height[i][0],
+            ]
+        )
+    return geom
+
+
+def pipe_geometries() -> list[dict[str, np.ndarray]]:
+    """
+    Return the wall geometries
+
+    Returns:
+        list[dict[str, np.ndarray]]: List of dictionaries with keys "x" and "y",
+            each containing a numpy array of two float values representing
+            the start and end coordinates of the pipe top and bottom edges.
+    """
+    pipes = []
+    for i, _ in enumerate(bl.vacuum_pipes.center):
+        top = bl.vacuum_pipes.center[i] + bl.vacuum_pipes.diameter[i] / 2 + bl.sourceHeight
+        bottom = bl.vacuum_pipes.center[i] - bl.vacuum_pipes.diameter[i] / 2 + bl.sourceHeight
+        pipes.append(
+            {
+                "x": np.array([bl.vacuum_pipes.start[i], bl.vacuum_pipes.end[i]]),
+                "y": np.array([top, top]),
+            }
+        )
+        pipes.append(
+            {
+                "x": np.array([bl.vacuum_pipes.start[i], bl.vacuum_pipes.end[i]]),
+                "y": np.array([bottom, bottom]),
+            }
+        )
+    return pipes

debye_bec/bec_widgets/widgets/digital_twin/digital_twin.py

@@ -0,0 +1,813 @@
+"""
+Digital Twin: Custom BEC widget to support the beamline alignment.
+"""
+
+import sys
+from pathlib import Path
+from typing import Literal, cast
+
+import numpy as np
+import yaml
+from bec_lib import bec_logger
+from bec_lib.endpoints import MessageEndpoints
+from bec_widgets.utils.bec_dispatcher import BECDispatcher
+from bec_widgets.utils.bec_widget import BECWidget
+from bec_widgets.utils.colors import apply_theme, get_accent_colors
+from bec_widgets.utils.error_popups import SafeSlot
+
+# pylint: disable=E0611
+from qtpy.QtCore import Qt, QTimer
+from qtpy.QtGui import QFont
+from qtpy.QtWidgets import (
+    QApplication,
+    QDialog,
+    QDialogButtonBox,
+    QHBoxLayout,
+    QLabel,
+    QPlainTextEdit,
+    QPushButton,
+    QStyle,
+    QVBoxLayout,
+    QWidget,
+)
+
+from debye_bec.bec_widgets.widgets.digital_twin.calculations.calc_positions import calc_positions
+from debye_bec.bec_widgets.widgets.digital_twin.calculations.calc_sideview import calc_sideview
+from debye_bec.bec_widgets.widgets.digital_twin.calculations.calc_surfaces import calc_surfaces
+from debye_bec.bec_widgets.widgets.digital_twin.calculations.calc_varia import (
+    cm_critical_angle,
+    cm_reflectivity,
+    cm_stripe_to_trx,
+    cm_trx_to_stripe,
+    fm_ideal_pitch,
+    fm_reflectivity,
+    fm_stripe_to_trx,
+    fm_trx_to_stripe,
+    mo1_bragg_angle,
+    mo1_energy_resolution,
+    sldi_gap_to_acc,
+)
+from debye_bec.bec_widgets.widgets.digital_twin.panels.input_panel import InputPanel
+from debye_bec.bec_widgets.widgets.digital_twin.panels.mover_panel import MoverPanel
+from debye_bec.bec_widgets.widgets.digital_twin.panels.plots import SideviewPlot, SurfacePlots
+from debye_bec.bec_widgets.widgets.digital_twin.panels.settings_panel import SettingsPanel
+from debye_bec.bec_widgets.widgets.digital_twin.types import ConfigDict
+
+logger = bec_logger.logger
+
+OFFSET_FILE = "debye_bec/debye_bec/bec_widgets/widgets/digital_twin/x01da_offsets.yaml"
+
+
+class DigitalTwin(BECWidget, QWidget):
+    """
+    Main widget of Digital Twin
+    """
+
+    PLUGIN = True
+    ICON_NAME = "lightbulb"
+
+    def __init__(self, *arg, parent=None, **kwargs):
+        super().__init__(parent=parent, theme_update=True, *arg, **kwargs)
+        self.get_bec_shortcuts()
+
+        # Check if devices are all in config
+        self.check_config()
+        self.bec_dispatcher.connect_slot(self.check_config, MessageEndpoints.device_config_update())
+
+        central = QWidget()
+        self.root_layout = QHBoxLayout(central)
+
+        self.input_widget = QWidget()
+        self.input_layout = QVBoxLayout(self.input_widget)
+        self.input = InputPanel()
+        self.settings = SettingsPanel()
+        self.input_layout.addWidget(self.input)
+        self.input_layout.addWidget(self.settings)
+
+        self.plot_widget = QWidget()
+        self.plot_layout = QVBoxLayout(self.plot_widget)
+        self.sideview_plot = SideviewPlot()
+        self.surface_plots = SurfacePlots()
+        self.plot_layout.addWidget(self.sideview_plot)
+        self.plot_layout.addWidget(self.surface_plots)
+
+        self.mover = MoverPanel(self.dev)
+
+        self.root_layout.addWidget(self.input_widget, alignment=Qt.AlignmentFlag.AlignTop)
+        self.root_layout.addWidget(self.plot_widget, alignment=Qt.AlignmentFlag.AlignTop)
+        self.root_layout.addWidget(self.mover, alignment=Qt.AlignmentFlag.AlignTop)
+
+        self.setLayout(self.root_layout)
+        self.setWindowTitle("Digital Twin")
+        self.resize(1800, 800)
+
+        self.input.energy.value_changed_connect(self.calc_assistant)
+        self.input.sldi_hacc.value_changed_connect(self.calc_assistant)
+        self.input.sldi_vacc.value_changed_connect(self.calc_assistant)
+        self.input.cm_stripe.activated_connect(self.calc_assistant)
+        self.input.cm_pitch.value_changed_connect(self.calc_assistant)
+        self.input.mo1_mode.activated_connect(self.calc_assistant)
+        self.input.mo1_xtal.activated_connect(self.calc_assistant)
+        self.input.fm_stripe.activated_connect(self.calc_assistant)
+        self.input.fm_focus.activated_connect(self.calc_assistant)
+        self.input.fm_rotx.value_changed_connect(self.calc_assistant)
+        self.input.fm_focx.value_changed_connect(self.calc_assistant)
+        self.input.fm_focy.value_changed_connect(self.calc_assistant)
+        self.input.smpl.value_changed_connect(self.calc_assistant)
+
+        self.input.adapt_reality.clicked_connect(self.adapt_reality)
+        self.settings.load_offsets.clicked_connect(self.load_offsets)
+        self.settings.show_offsets.clicked_connect(self.show_offsets)
+
+        self.bragg_angle = 0.0
+        self.qy = 0.0
+        self.offsets = {}
+
+        # Initialize all values
+        self.load_offsets(recalculate=False)
+        self.calc_assistant(identifier="init")
+
+        # Timer: update plots every 1 second
+        self._timer = QTimer(self)
+        self._timer.setInterval(100)
+        self._timer.timeout.connect(self.calc_reality)
+        self._timer.start()
+
+    def apply_theme(self, theme: Literal["dark", "light"]):
+        """
+        Apply the theme
+
+        Args:
+            theme (str): Theme, either "dark" or "light"
+        """
+        self.sideview_plot.apply_theme(theme)
+        self.surface_plots.apply_theme(theme)
+        self.mover.apply_theme(theme)
+
+    @SafeSlot()
+    def check_config(self, *args):
+        """
+        Checks the BEC config and opens a window if not all necessary
+        devices are loaded in the config. If called from a slot from
+        BEC dispatcher whenever there is a config update, stop the timer
+        that updates the plot in the background.
+        """
+        reload = (args[0] if args else {}).get("action") == "reload"
+        if reload:
+            self._timer.stop()
+        devices = [
+            "abs",
+            "sldi_gapx",
+            "sldi_gapy",
+            "cm_trx",
+            "cm_try",
+            "cm_bnd_radius",
+            "cm_rotx",
+            "mo1_bragg",
+            "mo1_trx",
+            "mo1_try",
+            "sl1_centery",
+            "sl1_gapy",
+            "bm1_try",
+            "fm_trx",
+            "fm_try",
+            "fm_bnd_radius",
+            "fm_rotx",
+            "fm_roty",
+            "fm_rotz",
+            "sl2_centery",
+            "sl2_gapy",
+            "bm2_try",
+            "ot_try",
+            "ot_rotx",
+            "es0wi_try",
+            "ot_es1_trz",
+        ]
+        while True:
+            missing = [d for d in devices if d not in self.dev]
+            if not missing:
+                break
+            dialog = QDialog()
+            dialog.setWindowTitle("Digital Twin - Config Check")
+            dialog.setFixedWidth(400)
+            layout = QVBoxLayout()
+
+            top = QHBoxLayout()
+            icon = QLabel()
+            icon_pixmap = (
+                QApplication.style()
+                .standardIcon(QStyle.StandardPixmap.SP_MessageBoxWarning)
+                .pixmap(48, 48)
+            )
+            icon.setPixmap(icon_pixmap)
+            icon.setAlignment(Qt.AlignmentFlag.AlignTop)
+            top.addWidget(icon)
+
+            text = QLabel(
+                "The current config does not include all required devices to run Digital Twin."
+                + "Reload the config with the correct devices."
+            )
+            text.setWordWrap(True)
+            text.setAlignment(Qt.AlignmentFlag.AlignTop)
+            top.addWidget(text, stretch=1)
+            layout.addLayout(top)
+
+            info = QLabel("Missing devices:\n" + ", ".join(missing))
+            info.setWordWrap(True)
+            info.setAlignment(Qt.AlignmentFlag.AlignTop)
+            layout.addWidget(info)
+            layout.addStretch()
+
+            buttons = QHBoxLayout()
+            check_again = QPushButton("Check Again")
+            close_app = QPushButton("Close Application")
+            check_again.clicked.connect(dialog.accept)
+            close_app.clicked.connect(dialog.reject)
+            buttons.addWidget(check_again)
+            buttons.addWidget(close_app)
+            layout.addLayout(buttons)
+
+            dialog.setLayout(layout)
+            dialog.show()
+            info.setMinimumHeight(info.heightForWidth(info.width()))
+            if dialog.exec_() == QDialog.DialogCode.Rejected:
+                running_app = QApplication.instance()
+                if running_app is not None:
+                    running_app.exit(0)
+        if reload:
+            self._timer.start()
+
+    @SafeSlot()
+    def calc_assistant(self, *_, **kwargs):
+        """
+        Calculates various values for the assistant.
+        If called from a qt slot, the identifier represents
+        the button pressed / value changed. Based on the identifier,
+        calculate different values.
+        Note: identifier=init calculates all values
+        """
+        identifier = kwargs["identifier"]
+        match identifier:
+            case "init":
+                self.update_mo1_mode()
+                self.calc_mo1_bragg_angle()
+                self.calc_cm_crit_pitch()
+                self.calc_cm_reflectivity()
+                self.update_fm_mode()
+                self.calc_fm_reflectivity()
+                self.calc_cm_fm_harm_suppr()
+                self.calc_fm_ideal_pitch()
+                self.calc_mo1_energy_resolution()
+            case "energy":
+                self.calc_mo1_bragg_angle()
+                self.calc_cm_crit_pitch()
+                self.calc_cm_reflectivity()
+                self.calc_fm_reflectivity()
+                self.calc_cm_fm_harm_suppr()
+                self.calc_mo1_energy_resolution()
+            case "cm_stripe":
+                self.calc_cm_crit_pitch()
+                self.calc_cm_reflectivity()
+                self.calc_cm_fm_harm_suppr()
+            case "cm_pitch":
+                self.calc_cm_reflectivity()
+                self.calc_cm_fm_harm_suppr()
+            case "mo1_mode":
+                self.update_mo1_mode()
+            case "mo1_xtal":
+                self.calc_mo1_bragg_angle()
+                self.calc_mo1_energy_resolution()
+            case "fm_focus":
+                self.update_fm_mode()
+                self.calc_fm_ideal_pitch()
+            case "fm_focx":
+                self.calc_fm_ideal_pitch()
+            case "fm_focy":
+                self.calc_fm_ideal_pitch()
+            case "fm_rotx":
+                self.calc_fm_reflectivity()
+                self.calc_cm_fm_harm_suppr()
+            case "fm_stripe":
+                self.calc_fm_reflectivity()
+                self.calc_cm_fm_harm_suppr()
+                self.calc_fm_ideal_pitch()
+            case "smpl":
+                self.calc_fm_ideal_pitch()
+        self.calc_positions()
+        self.calc_assistant_sideview()
+        self.calc_assistant_surfaces()
+
+    def get_assistant_config(self, apply_offset: bool = False) -> ConfigDict:
+        """
+        Assembles the digital twin config from the assistants input.
+
+        Args:
+            apply_offset(bool): Applies the offset values to the config.
+                                Defaults to False
+
+        Returns:
+            ConfigDict: config of the assistant
+        """
+        fm_focus = self.input.fm_focus.currentText()
+        if fm_focus in "Manual":
+            fm_rotx = self.input.fm_rotx.value()
+            fm_qy = None
+        elif fm_focus in "Focused":
+            fm_rotx = self.input.fm_rotx_ideal.value()
+            fm_qy = None
+        else:  # Focused
+            fm_rotx = self.input.fm_rotx_ideal.value()
+            fm_qy = self.qy
+
+        cm_stripe = self.input.cm_stripe.currentText()
+        cm_trx = cm_stripe_to_trx(cm_stripe)
+        fm_stripe = self.input.fm_stripe.currentText()
+        fm_trx = fm_stripe_to_trx(fm_stripe)
+
+        assert cm_trx is not None, f"No cm_trx found for given stripe {cm_stripe}!"
+        assert fm_trx is not None, f"No fm_trx found for given stripe {fm_stripe}!"
+
+        config: ConfigDict = {
+            "energy": self.input.energy.value(),
+            "h_acc": self.input.sldi_hacc.value(),
+            "v_acc": self.input.sldi_vacc.value(),
+            "cm_pitch": -self.input.cm_pitch.value(),
+            "cm_stripe": cm_stripe,
+            "cm_trx": cm_trx,
+            "mo1_mode": self.input.mo1_mode.currentText(),
+            "mo1_xtal": self.input.mo1_xtal.currentText(),
+            "mo1_bragg": self.bragg_angle,
+            "fm_rotx": -fm_rotx,
+            "fm_stripe": fm_stripe,
+            "fm_trx": fm_trx,
+            "fm_qy": fm_qy,
+            "fm_gain_height": 1,
+            "smpl": self.input.smpl.value(),
+        }
+
+        # Apply offsets
+        if apply_offset:
+            for axis, _ in config.items():
+                if axis in self.offsets:
+                    axis_offsets = self.offsets[axis]
+                    if "modifier" in axis_offsets and "offset" in axis_offsets:
+                        for idx, rng in enumerate(axis_offsets["modifier"]["range"]):
+                            if rng[0] < config[axis_offsets["modifier"]["axis"]] < rng[1]:
+                                config[axis] += axis_offsets["offset"][idx]
+                                break
+                    elif "offset" in axis_offsets:
+                        config[axis] += axis_offsets["offset"]
+
+        # Convert to SI units!
+        config["h_acc"] *= 1e-3
+        config["v_acc"] *= 1e-3
+        config["cm_pitch"] *= 1e-3
+        config["fm_rotx"] *= 1e-3
+
+        # logger.info(f'Config created: {config}')
+        return config
+
+    def get_reality_config(self) -> ConfigDict:
+        """
+        Assembles the digital twin config based on the real axis positions.
+
+        Returns:
+            ConfigDict: config of the reality
+        """
+        mo1_trx = self.dev.mo1_trx.read(cached=True)["mo1_trx"]["value"]
+        if abs(mo1_trx) > 5:
+            mo1_mode = "Monochromatic"
+        else:
+            mo1_mode = "Pinkbeam"
+        mo1_bragg = self.dev.mo1_bragg.read(cached=True)
+        sldi_gapx = self.dev.sldi_gapx.read(cached=True)["sldi_gapx"]["value"]
+        sldi_gapy = self.dev.sldi_gapy.read(cached=True)["sldi_gapy"]["value"]
+        h_acc, v_acc = sldi_gap_to_acc(sldi_gapx, sldi_gapy)
+        cm_trx = self.dev.cm_trx.read(cached=True)["cm_trx"]["value"]
+        cm_stripe = cm_trx_to_stripe(-cm_trx)
+        cm_pitch = self.dev.cm_rotx.read(cached=True)["cm_rotx"]["value"]
+        fm_trx = self.dev.fm_trx.read(cached=True)["fm_trx"]["value"]
+        fm_stripe = fm_trx_to_stripe(-fm_trx)
+        fm_rotx = self.dev.fm_rotx.read(cached=True)["fm_rotx"]["value"]
+        fm_rotx_real = 2 * cm_pitch - fm_rotx
+        smpl = self.dev.ot_es1_trz.read(cached=True)["ot_es1_trz"]["value"]
+        raw = {  # Config in SI units!
+            "energy": mo1_bragg["mo1_bragg"]["value"],
+            "h_acc": h_acc,
+            "v_acc": v_acc,
+            "cm_pitch": -cm_pitch * 1e-3,
+            "cm_stripe": cm_stripe,
+            "cm_trx": cm_trx,
+            "mo1_mode": mo1_mode,
+            "mo1_xtal": mo1_bragg["mo1_bragg_crystal_current_xtal_string"]["value"],
+            "mo1_bragg": mo1_bragg["mo1_bragg_angle"]["value"] / 180 * np.pi,
+            "fm_rotx": -fm_rotx_real * 1e-3,
+            "fm_stripe": fm_stripe,
+            "fm_trx": fm_trx,
+            "fm_qy": None,
+            "fm_gain_height": 1,
+            "smpl": smpl,
+        }
+        config = cast(ConfigDict, raw)
+        # logger.info(f'Config created: {config}')
+
+        abs_open = self.dev.abs.read(cached=True)["abs_status_string"]["value"] == "OPEN"
+        if not abs_open:
+            ready = True
+            for mover in self.mover.mover_widgets:
+                if mover.status in ("moving", "error"):
+                    ready = False
+            if ready:
+                self.mover.abs.enable_open(True)  # Enable open button
+            else:
+                self.mover.abs.enable_open(False)  # Disable open button
+        else:
+            self.mover.abs.enable_open(False)  # Disable open button
+
+        self.mover.sldi_gapx.set_feedback(sldi_gapx)
+        self.mover.sldi_gapy.set_feedback(sldi_gapy)
+        self.mover.cm_trx.set_feedback(cm_trx)
+        self.mover.cm_try.set_feedback(self.dev.cm_try.read(cached=True)["cm_try"]["value"])
+        self.mover.cm_bnd.set_feedback(
+            self.dev.cm_bnd_radius.read(cached=True)["cm_bnd_radius"]["value"]
+        )
+        self.mover.cm_rotx.set_feedback(cm_pitch)
+        self.mover.mo1_bragg_angle.set_feedback(mo1_bragg["mo1_bragg_angle"]["value"])
+        self.mover.mo1_trx.set_feedback(mo1_trx)
+        self.mover.mo1_try.set_feedback(self.dev.mo1_try.read(cached=True)["mo1_try"]["value"])
+        self.mover.sl1_centery.set_feedback(
+            self.dev.sl1_centery.read(cached=True)["sl1_centery"]["value"]
+        )
+        self.mover.sl1_gapy.set_feedback(self.dev.sl1_gapy.read(cached=True)["sl1_gapy"]["value"])
+        self.mover.bm1_try.set_feedback(self.dev.bm1_try.read(cached=True)["bm1_try"]["value"])
+        self.mover.fm_trx.set_feedback(fm_trx)
+        self.mover.fm_try.set_feedback(self.dev.fm_try.read(cached=True)["fm_try"]["value"])
+        self.mover.fm_bnd.set_feedback(
+            self.dev.fm_bnd_radius.read(cached=True)["fm_bnd_radius"]["value"]
+        )
+        self.mover.fm_rotx.set_feedback(fm_rotx)
+        self.mover.fm_roty.set_feedback(self.dev.fm_roty.read(cached=True)["fm_roty"]["value"])
+        self.mover.fm_rotz.set_feedback(self.dev.fm_rotz.read(cached=True)["fm_rotz"]["value"])
+        self.mover.sl2_centery.set_feedback(
+            self.dev.sl2_centery.read(cached=True)["sl2_centery"]["value"]
+        )
+        self.mover.sl2_gapy.set_feedback(self.dev.sl2_gapy.read(cached=True)["sl2_gapy"]["value"])
+        self.mover.bm2_try.set_feedback(self.dev.bm2_try.read(cached=True)["bm2_try"]["value"])
+        self.mover.ot_try.set_feedback(self.dev.ot_try.read(cached=True)["ot_try"]["value"])
+        self.mover.ot_rotx.set_feedback(self.dev.ot_rotx.read(cached=True)["ot_rotx"]["value"])
+        self.mover.ot_es1_trz.set_feedback(smpl)
+        self.mover.es0wi_try.set_feedback(
+            self.dev.es0wi_try.read(cached=True)["es0wi_try"]["value"]
+        )
+        self.mover.abs.set_feedback(abs_open)
+        return config
+
+    @SafeSlot()
+    def adapt_reality(self, *_):
+        """
+        Based on the real axis positions, adjust the assistant to reflect
+        the reality.
+        """
+        pos = {}
+        pos["sldi_gapx"] = self.dev.sldi_gapx.read(cached=True)["sldi_gapx"]["value"]
+        pos["sldi_gapy"] = self.dev.sldi_gapy.read(cached=True)["sldi_gapy"]["value"]
+        pos["cm_trx"] = self.dev.cm_trx.read(cached=True)["cm_trx"]["value"]
+        pos["cm_rotx"] = self.dev.cm_rotx.read(cached=True)["cm_rotx"]["value"]
+        pos["mo1_trx"] = self.dev.mo1_trx.read(cached=True)["mo1_trx"]["value"]
+        pos["fm_trx"] = self.dev.fm_trx.read(cached=True)["fm_trx"]["value"]
+        pos["fm_rotx"] = self.dev.fm_rotx.read(cached=True)["fm_rotx"]["value"]
+        pos["ot_es1_trz"] = self.dev.ot_es1_trz.read(cached=True)["ot_es1_trz"]["value"]
+
+        # Removing offsets
+        for axis, _ in pos.items():
+            if axis in self.offsets:
+                axis_offsets = self.offsets[axis]
+                if "modifier" in axis_offsets and "offset" in axis_offsets:
+                    for idx, rng in enumerate(axis_offsets["modifier"]["range"]):
+                        if rng[0] < pos[axis_offsets["modifier"]["axis"]] < rng[1]:
+                            pos[axis] -= axis_offsets["offset"][idx]
+                            break
+                elif "offset" in axis_offsets:
+                    pos[axis] -= axis_offsets["offset"]
+
+        self.input.energy.set_number(self.dev.mo1_bragg.read(cached=True)["mo1_bragg"]["value"])
+        h_acc, v_acc = sldi_gap_to_acc(pos["sldi_gapx"], pos["sldi_gapy"])
+        self.input.sldi_hacc.set_number(h_acc * 1e3)
+        self.input.sldi_vacc.set_number(v_acc * 1e3)
+        self.input.cm_stripe.set_current_text(cm_trx_to_stripe(-pos["cm_trx"]))
+        self.input.cm_pitch.set_number(pos["cm_rotx"])
+        if abs(pos["mo1_trx"]) > 5:
+            mo1_mode = "Monochromatic"
+        else:
+            mo1_mode = "Pinkbeam"
+        self.input.mo1_mode.set_current_text(mo1_mode)
+        self.input.mo1_xtal.set_current_text(
+            self.dev.mo1_bragg.read(cached=True)["mo1_bragg_crystal_current_xtal_string"]["value"]
+        )
+        self.input.fm_stripe.set_current_text(fm_trx_to_stripe(-pos["fm_trx"]))
+        self.input.fm_focus.set_current_text("Manual")
+        fm_rotx_real = 2 * pos["cm_rotx"] - pos["fm_rotx"]
+        self.input.fm_rotx.set_number(fm_rotx_real)
+        self.input.smpl.set_number(pos["ot_es1_trz"])
+        self.calc_assistant(identifier="init")
+
+    @SafeSlot()
+    def load_offsets(self, *_, recalculate: bool = True):
+        """
+        Loads or unloads the offsets from the file
+
+        Args:
+            recalculate(bool): Recalculates the assistant values after loading.
+                               Defaults to True
+        """
+
+        if self.offsets == {}:
+            # Load offsets
+            file = Path(OFFSET_FILE)
+            if not file.exists():
+                raise FileNotFoundError(f"Offset file not found: {OFFSET_FILE}")
+
+            with file.open("r", encoding="utf-8") as f:
+                data = yaml.safe_load(f)
+
+            if not isinstance(data, dict):
+                raise ValueError(f"Expected a YAML mapping, got {type(data).__name__}")
+
+            self.offsets = data
+
+            if recalculate:
+                self.calc_assistant(identifier="init")
+
+            self.settings.load_offsets.setText("Unload")
+            self.settings.offsets_status.setText("Loaded and applied")
+            self.settings.offsets_status.setColor(get_accent_colors().success.name())
+            self.settings.show_offsets.enable_button(True)
+        else:
+            # Unload offsets
+            self.offsets = {}
+            self.calc_assistant(identifier="init")
+
+            self.settings.load_offsets.setText("Load")
+            self.settings.offsets_status.setText("No offsets")
+            self.settings.offsets_status.setColor(get_accent_colors().default.name())
+            self.settings.show_offsets.enable_button(False)
+
+    @SafeSlot()
+    def show_offsets(self, *_):
+        """
+        Shows the offsets in a popup window
+        """
+        dialog = QDialog()
+        dialog.setWindowTitle("Digital Twin - Offsets")
+        dialog.setFixedWidth(500)
+        layout = QVBoxLayout(dialog)
+        layout.setSpacing(12)
+        layout.setContentsMargins(20, 20, 20, 20)
+
+        intro_label = QLabel("The offsets are saved in the digital twin BEC widget folder:")
+        intro_label.setWordWrap(True)
+        layout.addWidget(intro_label)
+
+        file = QLabel(OFFSET_FILE)
+        file.setWordWrap(True)
+        font = QFont()
+        font.setItalic(True)
+        file.setFont(font)
+        layout.addWidget(file)
+
+        text_edit = QPlainTextEdit()
+        text_edit.setReadOnly(True)
+        text_edit.setFont(QFont("Consolas", 9))
+
+        class InlineListDumper(yaml.Dumper):
+            """YAML dumper that renders all sequences on a single line."""
+
+            def represent_sequence(self, tag, sequence, *_):
+                return super().represent_sequence(tag, sequence, flow_style=True)
+
+        text_edit.setPlainText(yaml.dump(self.offsets, Dumper=InlineListDumper, sort_keys=False))
+        layout.addWidget(text_edit)
+
+        buttons = QDialogButtonBox(QDialogButtonBox.StandardButton.Close)
+        buttons.rejected.connect(dialog.reject)
+        layout.addWidget(buttons)
+
+        dialog.exec()
+
+    def update_fm_mode(self):
+        """
+        Updates the focusing mirror input group based on the
+        selection of the focus strategy.
+        """
+        fm_focus = self.input.fm_focus.currentText()
+        if fm_focus in "Manual":
+            self.input.fm_rotx.setVisible(True)
+            self.input.fm_rotx_ideal.setVisible(True)
+            self.input.fm_focx.setVisible(False)
+            self.input.fm_focy.setVisible(False)
+            self.input.fm_rotx_ideal.setLabel("Incidence Angle for focused beam")
+        elif fm_focus in "Focused":
+            self.input.fm_rotx.setVisible(False)
+            self.input.fm_rotx_ideal.setVisible(True)
+            self.input.fm_focx.setVisible(False)
+            self.input.fm_focy.setVisible(False)
+            self.input.fm_rotx_ideal.setLabel("Incidence Angle for focused beam")
+        else:  # Defocused
+            self.input.fm_rotx.setVisible(False)
+            self.input.fm_rotx_ideal.setVisible(True)
+            self.input.fm_focx.setVisible(True)
+            self.input.fm_focy.setVisible(True)
+            self.input.fm_rotx_ideal.setLabel("Incidence Angle for defocused beam")
+
+    @SafeSlot()
+    def calc_reality(self):
+        """
+        Updates the plots for the reality scene
+        """
+        config = self.get_reality_config()
+        data = calc_sideview(config)
+        self.sideview_plot.update_curves("reality", data=data)
+        surfaces = calc_surfaces(config)
+        self.surface_plots.update_surfaces(scene="reality", data=surfaces)
+
+    def calc_mo1_energy_resolution(self):
+        """
+        Calculates the energy resolution of the monochromator
+        """
+        xtal = self.input.mo1_xtal.currentText().translate(
+            str.maketrans("", "", "()")
+        )  # Remove brackets from xtal name to conform with parameters
+        xtal = cast(Literal["Si111", "Si311"], xtal)
+        energy = self.input.energy.value()
+        self.input.mo1_eres.setValue(mo1_energy_resolution(xtal, energy))
+
+    def calc_cm_reflectivity(self):
+        """
+        Calculates the collimating mirror reflectivity
+        """
+        cm_stripe = self.input.cm_stripe.currentText()
+        cm_pitch = -self.input.cm_pitch.value() * 1e-3
+        energy = self.input.energy.value()
+        self.input.cm_refl.setValue(100 * cm_reflectivity(cm_stripe, cm_pitch, energy))
+        self.input.cm_refl.setLabel(f"Reflectivity at \n{energy:.0f} eV")
+        self.input.cm_refl_harm.setValue(100 * cm_reflectivity(cm_stripe, cm_pitch, 3 * energy))
+        self.input.cm_refl_harm.setLabel(f"Reflectivity at \n{3*energy:.0f} eV")
+
+    def calc_fm_reflectivity(self):
+        """
+        Calculates the focusing mirror reflectivity
+        """
+        fm_stripe = self.input.fm_stripe.currentText()
+        fm_focus = self.input.fm_focus.currentText()
+        if fm_focus in "Manual":
+            fm_rotx = -self.input.fm_rotx.value() * 1e-3
+        else:
+            fm_rotx = -self.input.fm_rotx_ideal.value() * 1e-3
+        energy = self.input.energy.value()
+        self.input.fm_refl.setValue(100 * fm_reflectivity(fm_stripe, fm_rotx, energy))
+        self.input.fm_refl.setLabel(f"Reflectivity at \n{energy:.0f} eV")
+        self.input.fm_refl_harm.setValue(100 * fm_reflectivity(fm_stripe, fm_rotx, 3 * energy))
+        self.input.fm_refl_harm.setLabel(f"Reflectivity at \n{3*energy:.0f} eV")
+
+    def calc_cm_fm_harm_suppr(self):
+        """
+        Calculates the combined harmonics suppression of both mirrors
+        """
+        harm_suppr = (self.input.cm_refl.value() * self.input.fm_refl.value()) / (
+            self.input.cm_refl_harm.value() * self.input.fm_refl_harm.value()
+        )
+        self.input.cm_fm_harm_suppr.setValue(harm_suppr)
+        self.input.cm_fm_harm_suppr.setLabel(
+            f"Total Suppression Factor at {3 * self.input.energy.value():.0f} eV"
+        )
+
+    def calc_assistant_sideview(self):
+        """
+        Updates the sideview plot based on the assistant values
+        """
+        config = self.get_assistant_config(apply_offset=True)
+        data = calc_sideview(config)
+        self.sideview_plot.update_curves("assistant", data)
+
+    def calc_assistant_surfaces(self):
+        """
+        Updates the surface plot based on the assistant values
+        """
+        surfaces = calc_surfaces(self.get_assistant_config())
+        self.surface_plots.update_surfaces(scene="assistant", data=surfaces)
+
+    def calc_positions(self):
+        """
+        Calculates the positions for the axes based on the assistant values
+        """
+        out = calc_positions(self.get_assistant_config())
+
+        # Apply offsets
+        for axis, axis_data in out.items():
+            if axis in self.offsets:
+                axis_offsets = self.offsets[axis]
+                if "modifier" in axis_offsets and "offset" in axis_offsets:
+                    for idx, rng in enumerate(axis_offsets["modifier"]["range"]):
+                        if rng[0] < out[axis_offsets["modifier"]["axis"]]["value"] < rng[1]:
+                            axis_data["value"] += axis_offsets["offset"][idx]
+                            break
+                elif "offset" in axis_offsets:
+                    axis_data["value"] += axis_offsets["offset"]
+
+        self.mover.sldi_gapx.set_target(out["sldi_gapx"]["value"])
+        self.mover.sldi_gapy.set_target(out["sldi_gapy"]["value"])
+        self.mover.cm_trx.set_target(out["cm_trx"]["value"])
+        self.mover.cm_try.set_target(out["cm_try"]["value"])
+        self.mover.cm_bnd.set_target(out["cm_bnd_radius"]["value"])
+        self.mover.cm_rotx.set_target(out["cm_rotx"]["value"])
+        self.mover.mo1_bragg_angle.set_target(out["mo1_bragg_angle"]["value"])
+        self.mover.mo1_trx.set_target(out["mo1_trx"]["value"])
+        self.mover.mo1_try.set_target(out["mo1_try"]["value"])
+        self.mover.sl1_centery.set_target(out["sl1_centery"]["value"])
+        self.mover.sl1_gapy.set_target(out["sl1_gapy"]["value"])
+        self.mover.bm1_try.set_target(out["bm1_try"]["value"])
+        self.mover.fm_trx.set_target(out["fm_trx"]["value"])
+        self.mover.fm_try.set_target(out["fm_try"]["value"])
+        self.mover.fm_bnd.set_target(out["fm_bnd_radius"]["value"])
+        self.mover.fm_rotx.set_target(out["fm_rotx"]["value"])
+        self.mover.fm_roty.set_target(out["fm_roty"]["value"])
+        self.mover.fm_rotz.set_target(out["fm_rotz"]["value"])
+        self.mover.sl2_centery.set_target(out["sl2_centery"]["value"])
+        self.mover.sl2_gapy.set_target(out["sl2_gapy"]["value"])
+        self.mover.bm2_try.set_target(out["bm2_try"]["value"])
+        self.mover.ot_try.set_target(out["ot_try"]["value"])
+        self.mover.ot_rotx.set_target(out["ot_rotx"]["value"])
+        self.mover.ot_es1_trz.set_target(out["ot_es1_trz"]["value"])
+        self.mover.es0wi_try.set_target(out["es0wi_try"]["value"])
+
+    def calc_mo1_bragg_angle(self):
+        """
+        Calculates bragg angle in rad
+        """
+        xtal = self.input.mo1_xtal.currentText()
+        if xtal in "Si(111)":
+            d_spacing = self.dev.mo1_bragg.crystal.d_spacing_si111.read(cached=True)[
+                "mo1_bragg_crystal_d_spacing_si111"
+            ]["value"]
+        elif xtal in "Si(311)":
+            d_spacing = self.dev.mo1_bragg.crystal.d_spacing_si311.read(cached=True)[
+                "mo1_bragg_crystal_d_spacing_si311"
+            ]["value"]
+        else:
+            raise ValueError(f"Invalid xtal selection: {xtal}")
+        cm_pitch = -self.dev.cm_rotx.read(cached=True)["cm_rotx"]["value"] * 1e-3
+        mo1_mode = cast(Literal["Monochromatic", "Pinkbeam"], self.input.mo1_mode.currentText())
+        energy = self.input.energy.value()
+        theta, _ = mo1_bragg_angle(mo1_mode, d_spacing, energy, cm_pitch)
+        self.bragg_angle = theta
+        self.input.mo1_bragg_angle.setValue(theta / np.pi * 180)
+
+    def update_mo1_mode(self):
+        """
+        Updates the monochromator input group based on the
+        selection of the mode.
+        """
+        if self.input.mo1_mode.currentText() in "Monochromatic":
+            self.input.mo1_xtal.setVisible(True)
+            self.input.mo1_bragg_angle.setVisible(True)
+            self.input.mo1_eres.setVisible(True)
+        else:
+            self.input.mo1_xtal.setVisible(False)
+            self.input.mo1_bragg_angle.setVisible(False)
+            self.input.mo1_eres.setVisible(False)
+
+    def calc_fm_ideal_pitch(self):
+        """
+        Calculate the ideal pitch for the focusing mirror.
+        """
+        fm_focus = cast(
+            Literal["Defocused", "Focused", "Manual"], self.input.fm_focus.currentText()
+        )
+        fm_stripe = self.input.fm_stripe.currentText()
+        smpl = self.input.smpl.value()
+        sldi_hacc = self.input.sldi_hacc.value() * 1e-3
+        sldi_vacc = self.input.sldi_vacc.value() * 1e-3
+        fm_focx = self.input.fm_focx.value()
+        fm_focy = self.input.fm_focy.value()
+        fm_rotx, qy = fm_ideal_pitch(
+            fm_focus, fm_stripe, smpl, sldi_hacc, sldi_vacc, fm_focx, fm_focy
+        )
+        self.qy = qy
+        self.input.fm_rotx_ideal.setValue(-fm_rotx * 1e3)
+
+    def calc_cm_crit_pitch(self):
+        """
+        Calculate the critical pitch for the collimating mirror
+        """
+        cm_stripe = cast(Literal["Si", "Pt", "Rh"], self.input.cm_stripe.currentText())
+        energy = self.input.energy.value()
+        self.input.cm_pitch_critical.setValue(-cm_critical_angle(cm_stripe, energy) * 1e3)
+
+
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+    apply_theme("light")
+    dispatcher = BECDispatcher(gui_id="digital_twin")
+    win = DigitalTwin()
+    win.show()
+    sys.exit(app.exec_())

debye_bec/bec_widgets/widgets/digital_twin/digital_twin.pyproject

@@ -0,0 +1 @@
+{'files': ['digital_twin.py']}

debye_bec/bec_widgets/widgets/digital_twin/digital_twin_plugin.py

@@ -0,0 +1,57 @@
+# Copyright (C) 2022 The Qt Company Ltd.
+# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
+
+from bec_widgets.utils.bec_designer import designer_material_icon
+from qtpy.QtDesigner import QDesignerCustomWidgetInterface
+from qtpy.QtWidgets import QWidget
+
+from debye_bec.bec_widgets.widgets.digital_twin.digital_twin import DigitalTwin
+
+DOM_XML = """
+<ui language='c++'>
+    <widget class='DigitalTwin' name='digital_twin'>
+    </widget>
+</ui>
+"""
+
+
+class DigitalTwinPlugin(QDesignerCustomWidgetInterface):  # pragma: no cover
+    def __init__(self):
+        super().__init__()
+        self._form_editor = None
+
+    def createWidget(self, parent):
+        if parent is None:
+           return QWidget()
+        t = DigitalTwin(parent)
+        return t
+
+    def domXml(self):
+        return DOM_XML
+
+    def group(self):
+        return ""
+
+    def icon(self):
+        return designer_material_icon(DigitalTwin.ICON_NAME)
+
+    def includeFile(self):
+        return "digital_twin"
+
+    def initialize(self, form_editor):
+        self._form_editor = form_editor
+
+    def isContainer(self):
+        return False
+
+    def isInitialized(self):
+        return self._form_editor is not None
+
+    def name(self):
+        return "DigitalTwin"
+
+    def toolTip(self):
+        return "DigitalTwin"
+
+    def whatsThis(self):
+        return self.toolTip()

debye_bec/bec_widgets/widgets/digital_twin/panels/input_panel.py

@@ -0,0 +1,174 @@
+"""
+Panel for user inputs of the digital twin widget
+"""
+
+# pylint: disable=E0611
+from qtpy.QtWidgets import QLayout, QVBoxLayout, QWidget
+
+from debye_bec.bec_widgets.widgets.digital_twin.widgets.qt_widgets import (
+    Button,
+    ComboBox,
+    Group,
+    InputNumberField,
+    NumberIndicator,
+)
+
+
+class InputPanel(QWidget):
+    """Panel for user inputs of the digital twin widget"""
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._layout = QVBoxLayout(self)
+        self._layout.setSizeConstraint(QLayout.SetFixedSize)  # type: ignore
+
+        # Adapt to reality
+        self.adapt_reality = Button(label_button="Adapt to reality", enabled=True)
+
+        # Energy
+        self.energy = InputNumberField(
+            "energy", "Energy", unit="eV", init=8979, decimals=0, single_step=100, ll=4000, hl=65000
+        )
+
+        # FE Slits Acceptance
+        self.sldi_hacc = InputNumberField(
+            "h_acc",
+            "Horizontal",
+            unit="mrad",
+            prefix="±",
+            init=0.25,
+            decimals=3,
+            single_step=0.01,
+            ll=-0.1,
+            hl=0.9,
+        )
+        self.sldi_vacc = InputNumberField(
+            "v_acc",
+            "Vertical",
+            unit="mrad",
+            prefix="±",
+            init=0.1,
+            decimals=3,
+            single_step=0.01,
+            ll=-0.1,
+            hl=0.5,
+        )
+        self.sldi_ass_group = Group("FE Slits Acceptance", [self.sldi_hacc, self.sldi_vacc])
+
+        # Collimating mirror
+        self.cm_stripe = ComboBox("cm_stripe", "Stripe", ["Si", "Rh", "Pt"])
+        self.cm_pitch = InputNumberField(
+            "cm_pitch",
+            "Pitch",
+            unit="mrad",
+            init=-2.391,
+            decimals=3,
+            single_step=0.01,
+            ll=-4.6,
+            hl=-1.2,
+        )
+        self.cm_pitch_critical = NumberIndicator("Critical Pitch", "mrad", decimals=3)
+        self.cm_refl = NumberIndicator("Reflectivity at x eV", "%", decimals=0)
+        self.cm_refl_harm = NumberIndicator("Reflectivity at x eV", "%", decimals=0)
+        self.cm_ass_group = Group(
+            "Collimating Mirror",
+            [
+                self.cm_stripe,
+                self.cm_pitch,
+                self.cm_pitch_critical,
+                self.cm_refl,
+                self.cm_refl_harm,
+            ],
+        )
+
+        # Monochromator
+        self.mo1_mode = ComboBox("mo1_mode", "Mode", ["Monochromatic", "Pinkbeam"])
+        self.mo1_xtal = ComboBox("mo1_xtal", "Crystal", ["Si(111)", "Si(311)"])
+        self.mo1_bragg_angle = NumberIndicator("Bragg Angle", "deg", decimals=1)
+        self.mo1_eres = NumberIndicator("Energy Resolution", "eV", decimals=2)
+        self.mo1_ass_group = Group(
+            "Monochromator", [self.mo1_mode, self.mo1_xtal, self.mo1_bragg_angle, self.mo1_eres]
+        )
+
+        # Focusing Mirror
+        self.fm_stripe = ComboBox(
+            "fm_stripe", "Stripe", ["Rh (toroid)", "Rh (flat)", "Pt (toroid)", "Pt (flat)"]
+        )
+        self.fm_focus = ComboBox("fm_focus", "Focus Type", ["Manual", "Focused", "Defocused"])
+        self.fm_rotx = InputNumberField(
+            "fm_rotx",
+            "Incidence Angle",
+            unit="mrad",
+            init=-2.391,
+            decimals=3,
+            single_step=0.01,
+            ll=-10,
+            hl=2,
+        )
+        self.fm_focx = InputNumberField(
+            "fm_focx",
+            "Beam Size Horizontal",
+            unit="mm",
+            init=1,
+            decimals=1,
+            single_step=0.1,
+            ll=0,
+            hl=30,
+        )
+        self.fm_focy = InputNumberField(
+            "fm_focy",
+            "Beam Size Vertical",
+            unit="mm",
+            init=1,
+            decimals=1,
+            single_step=0.1,
+            ll=0,
+            hl=10,
+        )
+        self.fm_rotx_ideal = NumberIndicator("Incidence Angle for focused beam", "mrad", decimals=3)
+        self.fm_refl = NumberIndicator("Reflectivity at x eV", "%", decimals=0)
+        self.fm_refl_harm = NumberIndicator("Reflectivity at x eV", "%", decimals=0)
+        self.fm_ass_group = Group(
+            "Focusing Mirror",
+            [
+                self.fm_stripe,
+                self.fm_focus,
+                self.fm_rotx,
+                self.fm_focx,
+                self.fm_focy,
+                self.fm_rotx_ideal,
+                self.fm_refl,
+                self.fm_refl_harm,
+            ],
+        )
+
+        # Sample
+        self.cm_fm_harm_suppr = NumberIndicator("Total Suppression Factor at x eV", "", decimals=0)
+        self.smpl = InputNumberField(
+            "smpl",
+            "Sample Position",
+            unit="mm",
+            init=23511,
+            decimals=0,
+            single_step=100,
+            ll=23000,
+            hl=30000,
+        )
+
+        # Assemble complete assitant group
+        self.input_group = Group(
+            "User Input",
+            [
+                self.adapt_reality,
+                self.energy,
+                self.sldi_ass_group,
+                self.cm_ass_group,
+                self.mo1_ass_group,
+                self.fm_ass_group,
+                self.cm_fm_harm_suppr,
+                self.smpl,
+            ],
+        )
+
+        self._layout.addWidget(self.input_group)
+        self._layout.addStretch()

debye_bec/bec_widgets/widgets/digital_twin/panels/mover_panel.py

@@ -0,0 +1,232 @@
+"""
+Panel to move an axis to a certain position
+"""
+
+from typing import Literal
+
+# pylint: disable=E0611
+from qtpy.QtWidgets import QLayout, QVBoxLayout, QWidget
+
+from debye_bec.bec_widgets.widgets.digital_twin.widgets.move_widget import (
+    AbsorberWidget,
+    MoveWidget,
+)
+from debye_bec.bec_widgets.widgets.digital_twin.widgets.qt_widgets import Group
+
+
+class MoverPanel(QWidget):
+    """ "Panel to move an axis to a certain position"""
+
+    def __init__(self, dev, parent=None):
+        super().__init__(parent)
+        self._layout = QVBoxLayout(self)
+        self._layout.setSizeConstraint(QLayout.SetFixedSize)  # type: ignore
+
+        self.mover_widgets = []
+
+        # FE Slits
+        self.sldi_gapx = MoveWidget(
+            dev=dev, motor="sldi_gapx", label="GAPX", unit="mm", decimals=2, deadband=0.01
+        )
+        self.mover_widgets.append(self.sldi_gapx)
+
+        self.sldi_gapy = MoveWidget(
+            dev=dev, motor="sldi_gapy", label="GAPY", unit="mm", decimals=2, deadband=0.01
+        )
+        self.mover_widgets.append(self.sldi_gapy)
+
+        self.sldi_mov_group = Group("FE Slits", [self.sldi_gapx, self.sldi_gapy])
+
+        # Absorber
+        self.abs = AbsorberWidget(absorber=dev.abs, label="")
+
+        self.abs_group = Group("Absorber", [self.abs])
+
+        # Collimating mirror
+        self.cm_trx = MoveWidget(
+            dev=dev, motor="cm_trx", label="TRX", unit="mm", decimals=2, deadband=0.01
+        )
+        self.mover_widgets.append(self.cm_trx)
+
+        self.cm_try = MoveWidget(
+            dev=dev, motor="cm_try", label="TRY", unit="mm", decimals=2, deadband=0.01
+        )
+        self.mover_widgets.append(self.cm_try)
+
+        self.cm_bnd = MoveWidget(
+            dev=dev, motor="cm_bnd", label="BENDER", unit="km", decimals=2, deadband=0.2
+        )
+        self.mover_widgets.append(self.cm_bnd)
+
+        self.cm_rotx = MoveWidget(
+            dev=dev, motor="cm_rotx", label="PITCH", unit="mrad", decimals=3, deadband=0.01
+        )
+        self.mover_widgets.append(self.cm_rotx)
+
+        self.cm_mov_group = Group(
+            "Collimating Mirror", [self.cm_trx, self.cm_try, self.cm_bnd, self.cm_rotx]
+        )
+
+        # Monochromator
+        self.mo1_bragg_angle = MoveWidget(
+            dev=dev,
+            motor="mo1_bragg_angle",
+            label="Bragg Angle",
+            unit="deg",
+            decimals=3,
+            deadband=0.01,
+        )
+        self.mover_widgets.append(self.mo1_bragg_angle)
+
+        self.mo1_trx = MoveWidget(
+            dev=dev, motor="mo1_trx", label="TRX", unit="mm", decimals=2, deadband=0.01
+        )
+        self.mover_widgets.append(self.mo1_trx)
+
+        self.mo1_try = MoveWidget(
+            dev=dev, motor="mo1_try", label="TRY", unit="mm", decimals=2, deadband=0.01
+        )
+        self.mover_widgets.append(self.mo1_try)
+
+        self.mo1_mov_group = Group(
+            "Monochromator", [self.mo1_bragg_angle, self.mo1_trx, self.mo1_try]
+        )
+
+        # OP Slits 1
+        self.sl1_centery = MoveWidget(
+            dev=dev, motor="sl1_centery", label="CENTERY", unit="mm", decimals=2, deadband=0.1
+        )
+        self.mover_widgets.append(self.sl1_centery)
+
+        self.sl1_gapy = MoveWidget(
+            dev=dev, motor="sl1_gapy", label="GAPY", unit="mm", decimals=2, deadband=0.1
+        )
+        self.mover_widgets.append(self.sl1_gapy)
+
+        self.sl1_mov_group = Group("OP Slits 1", [self.sl1_centery, self.sl1_gapy])
+
+        # OP Beam Monitor 1
+        self.bm1_try = MoveWidget(
+            dev=dev, motor="bm1_try", label="TRY", unit="mm", decimals=2, deadband=0.1
+        )
+        self.mover_widgets.append(self.bm1_try)
+
+        self.bm1_mov_group = Group("OP Beam Monitor 1", [self.bm1_try])
+
+        # Focusing Mirror
+        self.fm_trx = MoveWidget(
+            dev=dev, motor="fm_trx", label="TRX", unit="mm", decimals=2, deadband=0.01
+        )
+        self.mover_widgets.append(self.fm_trx)
+
+        self.fm_try = MoveWidget(
+            dev=dev, motor="fm_try", label="TRY", unit="mm", decimals=2, deadband=0.01
+        )
+        self.mover_widgets.append(self.fm_try)
+
+        self.fm_bnd = MoveWidget(
+            dev=dev, motor="fm_bnd", label="BENDER", unit="km", decimals=2, deadband=0.2
+        )
+        self.mover_widgets.append(self.fm_bnd)
+
+        self.fm_rotx = MoveWidget(
+            dev=dev, motor="fm_rotx", label="PITCH", unit="mrad", decimals=3, deadband=0.01
+        )
+        self.mover_widgets.append(self.fm_rotx)
+
+        self.fm_roty = MoveWidget(
+            dev=dev, motor="fm_roty", label="YAW", unit="mrad", decimals=3, deadband=0.01
+        )
+        self.mover_widgets.append(self.fm_roty)
+
+        self.fm_rotz = MoveWidget(
+            dev=dev, motor="fm_rotz", label="ROLL", unit="mrad", decimals=3, deadband=0.01
+        )
+        self.mover_widgets.append(self.fm_rotz)
+
+        self.fm_mov_group = Group(
+            "Focusing Mirror",
+            [self.fm_trx, self.fm_try, self.fm_bnd, self.fm_rotx, self.fm_roty, self.fm_rotz],
+        )
+
+        # OP Slits 2
+        self.sl2_centery = MoveWidget(
+            dev=dev, motor="sl2_centery", label="CENTERY", unit="mm", decimals=2, deadband=0.1
+        )
+        self.mover_widgets.append(self.sl2_centery)
+
+        self.sl2_gapy = MoveWidget(
+            dev=dev, motor="sl2_gapy", label="GAPY", unit="mm", decimals=2, deadband=0.1
+        )
+        self.mover_widgets.append(self.sl2_gapy)
+
+        self.sl2_mov_group = Group("OP Slits 2", [self.sl2_centery, self.sl2_gapy])
+
+        # OP Beam Monitor 2
+        self.bm2_try = MoveWidget(
+            dev=dev, motor="bm2_try", label="TRY", unit="mm", decimals=2, deadband=0.1
+        )
+        self.mover_widgets.append(self.bm2_try)
+
+        self.bm2_mov_group = Group("OP Beam Monitor 2", [self.bm2_try])
+
+        # Optical Table
+        self.ot_try = MoveWidget(
+            dev=dev, motor="ot_try", label="TRY", unit="mm", decimals=2, deadband=0.2
+        )
+        self.mover_widgets.append(self.ot_try)
+
+        self.ot_rotx = MoveWidget(
+            dev=dev, motor="ot_rotx", label="ROTX", unit="mrad", decimals=3, deadband=0.05
+        )
+        self.mover_widgets.append(self.ot_rotx)
+
+        self.ot_mov_group = Group("Optical Table", [self.ot_try, self.ot_rotx])
+
+        # Experimental Station 0
+        self.es0wi_try = MoveWidget(
+            dev=dev, motor="es0wi_try", label="ES0 WI", unit="mm", decimals=0, deadband=0.1
+        )
+        self.mover_widgets.append(self.es0wi_try)
+
+        self.es0_mov_group = Group("Expperimental Station 0", [self.es0wi_try])
+
+        # Experimental Station 1
+        self.ot_es1_trz = MoveWidget(
+            dev=dev, motor="ot_es1_trz", label="ES1 TRZ", unit="mm", decimals=0, deadband=5
+        )
+        self.mover_widgets.append(self.ot_es1_trz)
+
+        self.es1_mov_group = Group("Expperimental Station 1", [self.ot_es1_trz])
+
+        # Assemble complete mover group
+        self.mover_group = Group(
+            "Mover",
+            [
+                self.sldi_mov_group,
+                self.abs_group,
+                self.cm_mov_group,
+                self.mo1_mov_group,
+                self.sl1_mov_group,
+                self.bm1_mov_group,
+                self.fm_mov_group,
+                self.sl2_mov_group,
+                self.bm2_mov_group,
+                self.ot_mov_group,
+                self.es0_mov_group,
+                self.es1_mov_group,
+            ],
+        )
+
+        self._layout.addWidget(self.mover_group)
+        self._layout.addStretch()
+
+    def apply_theme(self, theme: Literal["dark", "light"]):
+        """
+        Apply the theme
+
+        Args:
+            theme (str): Theme, either "dark" or "light"
+        """
+        for widget in self.mover_widgets:
+            widget.apply_theme(theme)

debye_bec/bec_widgets/widgets/digital_twin/panels/plots.py

@@ -0,0 +1,330 @@
+"""
+Two plot classes to plot side-view and surface-view
+"""
+
+from typing import Literal, Optional, cast
+
+import numpy as np
+import pyqtgraph as pg
+from bec_lib import bec_logger
+
+# pylint: disable=E0611
+from qtpy.QtCore import Qt
+from qtpy.QtGui import QBrush, QColor
+
+# pylint: disable=E0611
+from qtpy.QtWidgets import QApplication, QGraphicsRectItem, QHBoxLayout, QVBoxLayout, QWidget
+
+from debye_bec.bec_widgets.widgets.digital_twin.calculations.calc_varia import (
+    mirror_surface_geometries,
+    mo_surface_geometries,
+    pipe_geometries,
+    wall_geometries,
+)
+from debye_bec.bec_widgets.widgets.digital_twin.types import DataDict, SurfaceDict
+from debye_bec.bec_widgets.widgets.digital_twin.widgets.qt_widgets import Group
+
+logger = bec_logger.logger
+
+
+class SurfacePlots(QWidget):
+    """Plot widget with two curves and legend."""
+
+    def __init__(self, parent=None):
+        super().__init__(parent=parent)
+        self._layout = QHBoxLayout(self)
+
+        self.surfaces: dict[str, SurfaceDict] = {
+            "assistant": {
+                "cm": {"x": [], "y": []},
+                "mo1_1": {"x": [], "y": []},
+                "mo1_2": {"x": [], "y": []},
+                "fm": {"x": [], "y": []},
+            },
+            "reality": {
+                "cm": {"x": [], "y": []},
+                "mo1_1": {"x": [], "y": []},
+                "mo1_2": {"x": [], "y": []},
+                "fm": {"x": [], "y": []},
+            },
+        }
+
+        self.plots = {"fm": {}, "mo1_2": {}, "mo1_1": {}, "cm": {}}
+
+        self.color_impenetrable = (0, 0, 0)
+        self.colors = [(255, 255, 0), (255, 0, 255)]
+        self.text_color = (255, 255, 255)
+
+        # Create plot widgets
+        for name, widget in self.plots.items():
+            plot_widget = pg.PlotWidget()
+            plot_widget.getAxis("bottom").enableAutoSIPrefix(False)
+
+            plot_group = Group("Surface " + name, [plot_widget])
+
+            plot_widget.setLabel("left", "Z [mm]")
+            plot_widget.setLabel("bottom", "X [mm]")
+            plot_widget.setMouseEnabled(x=False, y=False)
+            plot_widget.setMenuEnabled(False)
+            plot_widget.hideButtons()
+
+            widget["widget"] = plot_widget
+            self._layout.addWidget(plot_group)
+
+        # Create surfaces
+        for idx, scene in enumerate(self.surfaces):
+            for name, _ in self.surfaces[scene].items():
+                if scene in "assistant":
+                    brush = QBrush(QColor(*self.colors[idx], 255), Qt.BrushStyle.DiagCrossPattern)
+                    pen = pg.mkPen(
+                        QColor(*self.colors[idx], 255), width=1, style=Qt.PenStyle.DashLine
+                    )
+                    z_value = 2
+                else:
+                    brush = QBrush(QColor(*self.colors[idx], 255))
+                    pen = pg.mkPen(QColor(*self.colors[idx], 255), width=1)
+                    z_value = 1
+                widget = self.plots[name]
+                self.plots[name][scene] = widget["widget"].plot(
+                    [], [], pen=pen, name=scene, brush=brush, fillLevel=0
+                )
+                self.plots[name][scene].setZValue(z_value)
+
+        self.walls = []
+        self.texts = []
+
+        self.plot_walls()
+
+        self.apply_theme()
+
+    def apply_theme(self, theme: Optional[Literal["dark", "light"]] = None):
+        """
+        Apply the theme
+
+        Args:
+            theme (Optional[str]): Theme, either "dark", "light", or None. Defaults to None.
+        """
+        if theme is None:
+            app = QApplication.instance()
+            theme = app.theme.theme  # type: ignore
+
+        bg_color = pg.getConfigOption("background")
+        fg_color = pg.getConfigOption("foreground")
+        for _, plot in self.plots.items():
+            # Background
+            plot["widget"].setBackground(bg_color)
+            # Axes (tick marks, tick labels, axis line)
+            for axis in ["left", "bottom", "right", "top"]:
+                ax = plot["widget"].getAxis(axis)
+                ax.setPen(pg.mkPen(color=fg_color))
+                ax.setTextPen(pg.mkPen(color=fg_color))
+
+        if theme == "light":
+            self.color_impenetrable = (30, 30, 30)
+            self.colors = [(79, 163, 224), (240, 128, 60)]
+            self.text_color = (255, 255, 255)
+        else:  # dark theme
+            self.color_impenetrable = (180, 180, 180)
+            self.colors = [(26, 111, 173), (212, 83, 10)]
+            self.text_color = (0, 0, 0)
+
+        for idx, scene in enumerate(self.surfaces):
+            for name, _ in self.surfaces[scene].items():
+                if scene in "assistant":
+                    brush = QBrush(QColor(*self.colors[idx], 255), Qt.BrushStyle.DiagCrossPattern)
+                    pen = pg.mkPen(
+                        QColor(*self.colors[idx], 255), width=1, style=Qt.PenStyle.DashLine
+                    )
+                else:
+                    brush = QBrush(QColor(*self.colors[idx], 255))
+                    pen = pg.mkPen(QColor(*self.colors[idx], 255), width=0)
+                self.plots[name][scene].setPen(pen)
+                self.plots[name][scene].setBrush(brush)
+
+        for wall in self.walls:
+            wall.setPen(pg.mkPen(color=self.color_impenetrable, width=2))
+            wall.setBrush(QBrush(QColor(*self.color_impenetrable)))
+
+        for text in self.texts:
+            text.setColor(self.text_color)
+
+    def plot_walls(self):
+        """Plot walls"""
+
+        def plot_surface(widget, surfaces):
+            for name, surface in surfaces.items():
+                rect = QGraphicsRectItem(*surface)
+                rect.setBrush(QBrush(QColor(*self.color_impenetrable)))
+                rect.setPen(pg.mkPen(color=self.color_impenetrable, width=2))
+                widget.addItem(rect)
+                text = pg.TextItem(name, color=self.text_color, anchor=(0.5, 0.5))
+                widget.addItem(text)
+                text.setPos(surface[0] + surface[2] / 2, surface[1] + surface[3] / 2)
+                text.setZValue(10)
+                self.walls.append(rect)
+                self.texts.append(text)
+
+        for name, plot in self.plots.items():
+            if name in "cm":
+                plot_surface(plot["widget"], mirror_surface_geometries("cm"))
+            elif name in "mo1_1":
+                plot_surface(plot["widget"], mo_surface_geometries("mo1", 0))
+            elif name in "mo1_2":
+                plot_surface(plot["widget"], mo_surface_geometries("mo1", 1))
+            elif name in "fm":
+                plot_surface(plot["widget"], mirror_surface_geometries("fm_flat"))
+                plot_surface(plot["widget"], mirror_surface_geometries("fm_toroid"))
+            else:
+                raise ValueError(f"Plot {name} not found!")
+        for name, plot in self.plots.items():
+            plot["widget"].disableAutoRange()
+
+    def update_surfaces(self, scene: Literal["assistant", "reality"], data: SurfaceDict):
+        """Update the curves of the plot
+
+        Args:
+            scene (str): The scene to update, either "assistant" or "reality".
+            data (DataDict): The new data to plot, with keys "x" and "y",
+                             each containing a list of values.
+        """
+        self.surfaces[scene] = data
+        for name, device in self.surfaces[scene].items():
+            device = cast(DataDict, device)
+            plot = self.plots[name][scene]
+            x = np.array(device["x"] + [device["x"][0]]) if len(device["x"]) != 0 else np.array([])
+            y = np.array(device["y"] + [device["y"][0]]) if len(device["y"]) != 0 else np.array([])
+            plot.setData(x=x, y=y)
+
+
+class SideviewPlot(QWidget):
+    """Plot widget with two curves and legend."""
+
+    def __init__(self, parent=None):
+        super().__init__(parent=parent)
+        self._layout = QVBoxLayout(self)
+        # self._layout.setSizeConstraint(QLayout.SetFixedSize) # type: ignore
+
+        self.plot_widget = pg.PlotWidget()
+        self.plot_widget.getAxis("bottom").enableAutoSIPrefix(False)
+        self.plot_widget.invertX(True)
+        self.plot_widget.addLegend()
+
+        self.color_impenetrable = (0, 0, 0)
+        self.colors = [(255, 255, 0), (255, 0, 255)]
+
+        self.data: dict[str, DataDict] = {
+            "assistant": {"x": [0, 1000, 2000], "y": [0, 20, 30]},
+            "reality": {"x": [0, 1000, 2000], "y": [0, 15, 50]},
+        }
+
+        self.plots = {}
+
+        self.pipes = []
+        self.walls = []
+
+        for idx, scene in enumerate(self.data.keys()):
+            if scene in "assistant":
+                pen = pg.mkPen(color=self.colors[idx], width=2, style=Qt.PenStyle.DotLine)
+                z_value = 2
+            else:
+                pen = pg.mkPen(color=self.colors[idx], width=2)
+                z_value = 1
+            self.plots[scene] = self.plot_widget.plot([], [], pen=pen, name=scene)
+            self.plots[scene].setZValue(z_value)
+
+        self.plot_group = Group("Side View", [self.plot_widget])
+
+        self.plot_widget.setLabel("left", "Height [mm]")
+        self.plot_widget.setLabel("bottom", "Distance [mm]")
+        self.plot_widget.setMouseEnabled(x=False, y=False)
+        self.plot_widget.setXRange(0, 25000, 0.1)  # pylint: disable=E1121 # type: ignore
+        self.plot_widget.setYRange(-20, 120, 0.1)  # pylint: disable=E1121 # type: ignore
+        self.plot_widget.setMenuEnabled(False)
+        self.plot_widget.hideButtons()
+
+        self._layout.addWidget(self.plot_group)
+        self._layout.addStretch()
+
+        self.plot_vacuum_pipes()
+        self.plot_walls()
+
+        self.apply_theme()
+
+    def apply_theme(self, theme: Optional[Literal["dark", "light"]] = None):
+        """
+        Apply the theme
+
+        Args:
+            theme (Optional[str]): Theme, either "dark", "light", or None. Defaults to None.
+        """
+        if theme is None:
+            app = QApplication.instance()
+            theme = app.theme.theme  # type: ignore
+
+        bg_color = pg.getConfigOption("background")
+        fg_color = pg.getConfigOption("foreground")
+        # Background
+        self.plot_widget.setBackground(bg_color)
+        # Axes (tick marks, tick labels, axis line)
+        for axis in ["left", "bottom", "right", "top"]:
+            ax = self.plot_widget.getAxis(axis)
+            ax.setPen(pg.mkPen(color=fg_color))
+            ax.setTextPen(pg.mkPen(color=fg_color))
+
+        if theme == "light":
+            self.color_impenetrable = (30, 30, 30)
+            self.colors = [(79, 163, 224), (240, 128, 60)]
+            self.text_color = (255, 255, 255)
+        else:  # dark theme
+            self.color_impenetrable = (180, 180, 180)
+            self.colors = [(26, 111, 173), (212, 83, 10)]
+            self.text_color = (0, 0, 0)
+
+        for idx, scene in enumerate(self.data):
+            if scene in "assistant":
+                brush = QBrush(QColor(*self.colors[idx], 255), Qt.BrushStyle.DiagCrossPattern)
+                pen = pg.mkPen(QColor(*self.colors[idx], 255), width=3, style=Qt.PenStyle.DashLine)
+            else:
+                brush = QBrush(QColor(*self.colors[idx], 255))
+                pen = pg.mkPen(QColor(*self.colors[idx], 255), width=3)
+            self.plots[scene].setPen(pen)
+            self.plots[scene].setBrush(brush)
+
+        for wall in self.walls:
+            wall.setPen(pg.mkPen(color=self.color_impenetrable, width=3))
+            wall.setBrush(QBrush(QColor(*self.color_impenetrable)))
+
+        for pipe in self.pipes:
+            pipe.setPen(pg.mkPen(color=self.color_impenetrable, width=3))
+
+    def plot_vacuum_pipes(self):
+        """Plot vacuum pipes"""
+        pipes = pipe_geometries()
+        for pipe in pipes:
+            self.pipes.append(
+                self.plot_widget.plot(
+                    x=pipe["x"], y=pipe["y"], pen=pg.mkPen(color=self.color_impenetrable, width=2)
+                )
+            )
+
+    def plot_walls(self):
+        """Plot walls"""
+        walls = wall_geometries()
+        for wall in walls:
+            rect = QGraphicsRectItem(wall[0], wall[1], wall[2], wall[3])
+            rect.setBrush(QBrush(QColor(*self.color_impenetrable)))
+            rect.setPen(pg.mkPen(color=self.color_impenetrable, width=2))
+            self.plot_widget.addItem(rect)
+            self.walls.append(rect)
+
+    def update_curves(self, scene: Literal["assistant", "reality"], data: DataDict):
+        """Update the curves of the plot
+
+        Args:
+            scene (str): The scene to update, either "assistant" or "reality".
+            data (DataDict): The new data to plot, with keys "x" and "y",
+                             each containing a list of values.
+        """
+        self.data[scene] = data
+        plot = self.plots[scene]
+        plot.setData(x=self.data[scene]["x"], y=self.data[scene]["y"])

debye_bec/bec_widgets/widgets/digital_twin/panels/settings_panel.py

@@ -0,0 +1,34 @@
+"""
+Settings panel for the digital twin widget
+"""
+
+# pylint: disable=E0611
+from qtpy.QtWidgets import QLayout, QVBoxLayout, QWidget
+
+from debye_bec.bec_widgets.widgets.digital_twin.widgets.qt_widgets import (
+    Button,
+    Group,
+    TextIndicator,
+)
+
+
+class SettingsPanel(QWidget):
+    """Settings panel for the digital twin widget"""
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._layout = QVBoxLayout(self)
+        self._layout.setSizeConstraint(QLayout.SetFixedSize)  # type: ignore
+
+        # Reload offsets
+        self.load_offsets = Button(label="Load Offsets", label_button="Load", enabled=True)
+        self.offsets_status = TextIndicator(label="Offsets")
+        self.show_offsets = Button(label="Show Offsets", label_button="Show", enabled=True)
+
+        # Assemble complete offset group
+        self.offset_group = Group(
+            "Axes Offsets", [self.load_offsets, self.offsets_status, self.show_offsets]
+        )
+
+        self._layout.addWidget(self.offset_group)
+        self._layout.addStretch()

debye_bec/bec_widgets/widgets/digital_twin/register_digital_twin.py

@@ -0,0 +1,15 @@
+def main():  # pragma: no cover
+    from qtpy import PYSIDE6
+
+    if not PYSIDE6:
+        print("PYSIDE6 is not available in the environment. Cannot patch designer.")
+        return
+    from PySide6.QtDesigner import QPyDesignerCustomWidgetCollection
+
+    from debye_bec.bec_widgets.widgets.digital_twin.digital_twin_plugin import DigitalTwinPlugin
+
+    QPyDesignerCustomWidgetCollection.addCustomWidget(DigitalTwinPlugin())
+
+
+if __name__ == "__main__":  # pragma: no cover
+    main()

debye_bec/bec_widgets/widgets/digital_twin/types.py

@@ -0,0 +1,73 @@
+"""Types used for the beamline config and for plotting data"""
+
+from typing import TypedDict
+
+
+class ConfigDict(TypedDict):
+    """
+    Typed dictionary representing the beamline configuration.
+
+    Attributes:
+        energy (float): Beam energy.
+        h_acc (float): Horizontal acceptance.
+        v_acc (float): Vertical acceptance.
+        cm_pitch (float): CM pitch angle.
+        cm_stripe (str): CM stripe name.
+        cm_trx (float): CM translation x.
+        mo1_mode (str): MO1 mode.
+        mo1_xtal (str): MO1 crystal.
+        mo1_bragg (float): MO1 Bragg angle.
+        fm_rotx (float): FM rotation x.
+        fm_stripe (str): FM stripe name.
+        fm_trx (float): FM translation x.
+        fm_qy (float): FM qy value.
+        fm_gain_height (int): FM gain height.
+        smpl (float): Sample value.
+    """
+
+    energy: float
+    h_acc: float
+    v_acc: float
+    cm_pitch: float
+    cm_stripe: str
+    cm_trx: float
+    mo1_mode: str
+    mo1_xtal: str
+    mo1_bragg: float
+    fm_rotx: float
+    fm_stripe: str
+    fm_trx: float
+    fm_qy: None | float
+    fm_gain_height: int
+    smpl: float
+
+
+class DataDict(TypedDict):
+    """
+    Typed dictionary representing plot data.
+
+    Attributes:
+        x (list[float]): List of x-axis values.
+        y (list[float]): List of y-axis values.
+    """
+
+    x: list
+    y: list
+
+
+class SurfaceDict(TypedDict):
+    """
+    Typed dictionary representing the surfaces of a scene,
+    grouping plot data by surface type.
+
+    Attributes:
+        cm (DataDict): Data for the cm surface.
+        mo1_1 (DataDict): Data for the mo1_1 surface.
+        mo1_2 (DataDict): Data for the mo1_2 surface.
+        fm (DataDict): Data for the fm surface.
+    """
+
+    cm: DataDict
+    mo1_1: DataDict
+    mo1_2: DataDict
+    fm: DataDict

debye_bec/bec_widgets/widgets/digital_twin/widgets/move_widget.py

@@ -0,0 +1,594 @@
+"""Move widget to display an axis and also move it through BEC"""
+
+import threading
+import time
+from typing import Literal, Optional
+
+from bec_lib import bec_logger
+from bec_qthemes import material_icon
+from bec_widgets.utils.colors import get_accent_colors
+
+# pylint: disable=E0611
+from qtpy.QtCore import Property  # type: ignore[attr-defined]
+from qtpy.QtCore import Signal  # type: ignore[attr-defined]
+from qtpy.QtCore import QObject, QPropertyAnimation, Qt, QThread
+from qtpy.QtGui import QTransform
+from qtpy.QtWidgets import QApplication, QHBoxLayout, QLabel, QPushButton, QWidget
+
+from debye_bec.devices.absorber import STATUS as ABS_STATUS
+
+logger = bec_logger.logger
+
+
+class Status:
+    """Status class for the axis"""
+
+    IN_POSITION = "in_position"  # green  mdi.check-circle
+    NOT_IN_POSITION = "not_in_position"  # orange mdi.close-circle
+    MOVING = "moving"  # blue   mdi.loading  (spinning)
+    ERROR = "error"  # red    mdi.alert-circle
+
+
+class StatusIcon(QWidget):
+    """
+    Displays a status icon using bec_qthemes Material Design Icons.
+    Handles its own spin animation for the MOVING state via QPropertyAnimation.
+    """
+
+    ICON_SIZE = 20
+
+    _ICON_MAP = {
+        Status.IN_POSITION: ("check_circle", "#27ae60"),
+        Status.NOT_IN_POSITION: ("cancel", "#e6d922"),
+        Status.ERROR: ("warning", "#e74c3c"),
+        Status.MOVING: ("cycle", "#2980b9"),
+    }
+
+    def __init__(self, parent=None):
+        super().__init__(parent=parent)
+        self._status = None
+        self._rotation = 0.0
+
+        self._label = QLabel(self)
+        self._label.setFixedSize(self.ICON_SIZE, self.ICON_SIZE)
+        self._label.setAlignment(Qt.AlignmentFlag.AlignCenter)
+        self.setFixedSize(self.ICON_SIZE, self.ICON_SIZE)
+
+        self._spin_anim = QPropertyAnimation(self, b"rotation")  # type: ignore[call-arg]
+        self._spin_anim.setStartValue(0)
+        self._spin_anim.setEndValue(360)
+        self._spin_anim.setDuration(1000)
+        self._spin_anim.setLoopCount(-1)  # Loop indefinitely
+
+        self.set_status(Status.NOT_IN_POSITION)
+
+    def get_rotation(self) -> float:
+        """
+        Return the current rotation angle in degrees.
+
+        Returns:
+            float: Rotation angle in deg
+        """
+        return self._rotation
+
+    def set_rotation(self, angle: float):
+        """
+        Set the rotation angle and update the displayed pixmap.
+
+        Rotates the current base pixmap around its center point using a smooth
+        transformation. Has no effect on the display if no base pixmap is set.
+
+        Args:
+            angle (float): Rotation angle in degrees, clockwise.
+        """
+        self._rotation = angle
+        if self._current_pixmap_base is not None:
+            cx = self._current_pixmap_base.width() / 2
+            cy = self._current_pixmap_base.height() / 2
+            t = QTransform().translate(cx, cy).rotate(angle).translate(-cx, -cy)
+            self._label.setPixmap(
+                self._current_pixmap_base.transformed(t, Qt.TransformationMode.SmoothTransformation)
+            )
+
+    rotation = Property(float, get_rotation, set_rotation)  # type: ignore[call-arg]
+
+    def set_status(self, status: str):
+        """
+        Update the widget's status and refresh the displayed icon accordingly.
+
+        Looks up the icon name and color associated with the given status from
+        ``_ICON_MAP``, renders a new pixmap, and starts or stops the spin
+        animation depending on whether the status is ``Status.MOVING``. Returns
+        early without any updates if the status has not changed.
+
+        Args:
+            status (str): The new status value. Must be a key in ``_ICON_MAP``.
+        """
+        if status == self._status:
+            return
+        self._status = status
+
+        icon_name, color = self._ICON_MAP[status]
+        icon = material_icon(
+            icon_name, size=(self.ICON_SIZE, self.ICON_SIZE), color=color, convert_to_pixmap=True
+        )
+        self._current_pixmap_base = icon
+
+        if status == Status.MOVING:
+            self._spin_anim.start()
+        else:
+            self._spin_anim.stop()
+            self._label.setPixmap(icon)
+
+
+class MotionWorker(QObject):
+    """
+    Executes motion on the specified motor and includes some safety during
+    motion for certain motors.
+    """
+
+    position_changed = Signal(float)
+    error = Signal(bool)  # True = error
+    finished = Signal(bool)  # True = reached target, False = stopped
+
+    def __init__(self, dev, motor, target_pos: float):
+        super().__init__()
+        self.dev = dev
+        self.motor = motor
+        self._target = target_pos
+        self._stop_flag = threading.Event()
+
+    def stop(self):
+        """Sets the stop flag"""
+        self._stop_flag.set()
+
+    def run(self):
+        """Prepares the movement based on the axis (motor)"""
+        match self.motor:
+            case "sldi_gapx" | "sldi_gapy" | "sldi_centerx" | "sldi_centery":
+                self.motion()
+            case "cm_trx":
+                self.motion(
+                    abs_closed=True,
+                    surveyed_axes=[{"device": self.dev["cm_roty"], "abs_tol": 0.05}],
+                )
+            case "cm_roty":
+                self.motion(
+                    abs_closed=True, surveyed_axes=[{"device": self.dev["cm_trx"], "abs_tol": 0.05}]
+                )
+            case "cm_try":
+                self.motion(
+                    abs_closed=True,
+                    surveyed_axes=[
+                        {"device": self.dev["cm_rotx"], "abs_tol": 0.05},
+                        {"device": self.dev["cm_rotz"], "abs_tol": 0.05},
+                    ],
+                )
+            case "cm_rotx":
+                self.motion(
+                    abs_closed=True,
+                    surveyed_axes=[
+                        {"device": self.dev["cm_try"], "abs_tol": 0.05},
+                        {"device": self.dev["cm_rotz"], "abs_tol": 0.05},
+                    ],
+                )
+            case "cm_rotz":
+                self.motion(
+                    abs_closed=True,
+                    surveyed_axes=[
+                        {"device": self.dev["cm_try"], "abs_tol": 0.05},
+                        {"device": self.dev["cm_rotx"], "abs_tol": 0.05},
+                    ],
+                )
+            case "cm_bnd":
+                p1 = (
+                    1 / (self.dev.cm_bnd_radius.read()["cm_bnd_radius"]["value"] * 1e3) + 0.0284
+                ) / 2e-6
+                p2 = (1 / (self._target * 1e3) + 0.0284) / 2e-6
+                self._target = p2 - p1
+                self.motion(relative=True, rb={"device": self.dev["cm_bnd_radius"]})
+            case "mo1_try" | "mo1_trx" | "mo1_roty":
+                self.motion(abs_closed=True)
+            case "mo1_bragg_angle":
+                self.motion()
+            case "sl1_centery" | "sl1_gapy" | "bm1_try":
+                self.motion()
+            case "fm_trx":
+                self.motion(
+                    abs_closed=True,
+                    surveyed_axes=[{"device": self.dev["fm_roty"], "abs_tol": 0.05}],
+                )
+            case "fm_roty":
+                self.motion(
+                    abs_closed=True, surveyed_axes=[{"device": self.dev["fm_trx"], "abs_tol": 0.05}]
+                )
+            case "fm_try":
+                self.motion(
+                    abs_closed=True,
+                    surveyed_axes=[
+                        {"device": self.dev["fm_rotx"], "abs_tol": 0.05},
+                        {"device": self.dev["fm_rotz"], "abs_tol": 0.05},
+                    ],
+                )
+            case "fm_rotx":
+                self.motion(
+                    abs_closed=True,
+                    surveyed_axes=[
+                        {"device": self.dev["fm_try"], "abs_tol": 0.05},
+                        {"device": self.dev["fm_rotz"], "abs_tol": 0.05},
+                    ],
+                )
+            case "fm_rotz":
+                self.motion(
+                    abs_closed=True,
+                    surveyed_axes=[
+                        {"device": self.dev["fm_try"], "abs_tol": 0.05},
+                        {"device": self.dev["fm_rotx"], "abs_tol": 0.05},
+                    ],
+                )
+            case "fm_bnd":
+                p1 = (
+                    1 / (self.dev.fm_bnd_radius.read()["fm_bnd_radius"]["value"] * 1e3) + 4.28e-5
+                ) / 1.84e-9
+                p2 = (1 / (self._target * 1e3) + 4.28e-5) / 1.84e-9
+                self._target = p2 - p1
+                self.motion(relative=True, rb={"device": self.dev["fm_bnd_radius"]})
+            case "sl2_centery" | "sl2_gapy" | "bm2_try":
+                self.motion()
+            case "ot_try" | "ot_rotx" | "ot_es1_trz":
+                self.motion()
+            case _:
+                logger.warning(f"Motor {self.motor} not integrated in digital twin!")
+
+    def motion(self, abs_closed: bool = False, relative: bool = False, rb=None, surveyed_axes=None):
+        """
+        Moves an axis while surverying a set of axes (if set).
+        Example surveyed_axes:
+            [{'device': bec_device_object, 'abs_tol': 0.1},]
+
+        Args:
+            surveyed_axes (list): List of dictionaries of devices
+        """
+        if abs_closed:
+            if self.dev.abs.status.get() == ABS_STATUS.OPEN:
+                status = self.dev.abs.close()
+                # TODO Set timeout to 0.001 and check if it actually raises
+                # (it should not start motion).
+                # Check of behavior of digital twin afterwards.
+                status.wait(timeout=5)
+        if surveyed_axes is not None:
+            for surv_ax in surveyed_axes:
+                surv_ax["name"] = surv_ax["device"].dotted_name
+                surv_ax["old_value"] = surv_ax["device"].read(cached=True)[surv_ax["name"]]["value"]
+        if rb is not None:
+            rb["name"] = rb["device"].dotted_name
+        status = self.dev[self.motor].move(self._target, relative=relative)
+        last_check = time.time()
+        update_interval = 0.1
+        while status.status == "RUNNING":
+            now = time.time()
+            if time.time() - last_check < update_interval:
+                time.sleep(0.01)
+            last_check = now
+            if self._stop_flag.is_set():
+                self.dev[self.motor].stop()
+                self._stop_flag.clear()
+            if rb is not None:
+                self.position_changed.emit(rb["device"].read(cached=True)[rb["name"]]["value"])
+            else:
+                self.position_changed.emit(
+                    self.dev[self.motor].read(cached=True)[self.motor]["value"]
+                )
+            if surveyed_axes is not None:
+                for surv_ax in surveyed_axes:
+                    fb = surv_ax["device"].read(cached=True)[surv_ax["name"]]["value"]
+                    if abs(fb - surv_ax["old_value"]) > surv_ax["abs_tol"]:
+                        self.dev[self.motor].stop()
+                        self.error.emit(1)
+                        break
+        self.finished.emit()
+
+
+class MoveWidget(QWidget):
+    """
+    One motor stage control group containing:
+      - Target label  (target position)
+      - Feedback label (current position)
+      - Status icon    (bec_qthemes)
+      - Start / Stop button
+    """
+
+    def __init__(self, dev, motor, label: str = "", unit=None, decimals=3, deadband=0.0):
+        super().__init__()
+        self.fb = 0.0
+        self.target = 0
+        self.dev = dev
+        self.motor = motor
+        self.deadband = deadband
+        self.status = Status.IN_POSITION
+        self._thread: QThread | None = None
+        self._worker: MotionWorker | None = None
+
+        self.text_color = (0, 0, 0)
+
+        self.unit = unit
+        self.decimals = decimals
+
+        layout = QHBoxLayout(self)
+        layout.setContentsMargins(10, 0, 0, 0)
+        layout.setSpacing(0)
+
+        # Name
+        self.label = QLabel(label)
+        self.label.setFixedWidth(100)
+        self.label.setContentsMargins(0, 0, 10, 0)
+        self.label.setWordWrap(True)
+        layout.addWidget(self.label)
+
+        # Target
+        self.target_label = QLabel("-")
+        self.target_label.setFixedWidth(100)
+        layout.addWidget(self.target_label)
+
+        # Feedback
+        self.fb_label = QLabel("-")
+        self.fb_label.setFixedWidth(100)
+        layout.addWidget(self.fb_label)
+
+        # Status icon
+        self.status_icon = StatusIcon()
+        self.status_icon.setFixedWidth(30)
+        self.status_icon.setContentsMargins(0, 0, 10, 0)
+        layout.addWidget(self.status_icon)
+
+        # Start / Stop button
+        self.btn_action = QPushButton("Move")
+        self.btn_action.setFixedWidth(90)
+        self.btn_action.setFixedHeight(20)
+        self.btn_action.clicked.connect(self._on_button_clicked)
+        layout.addWidget(self.btn_action)
+        self.btn_mode = "start"
+
+        self._apply_button_style("start")
+
+        self.apply_theme()
+
+    def apply_theme(self, theme: Optional[Literal["dark", "light"]] = None):
+        """
+        Apply the theme
+
+        Args:
+            theme (Optional[str]): Theme, either "dark", "light", or None. Defaults to None.
+        """
+        if theme is None:
+            app = QApplication.instance()
+            theme = app.theme.theme  # type: ignore
+
+        if theme == "light":
+            self.text_color = {"target": (79, 163, 224), "fb": (240, 128, 60)}
+        else:  # dark theme
+            self.text_color = {"target": (26, 111, 173), "fb": (212, 83, 10)}
+        r, g, b = self.text_color["target"]
+        self.target_label.setStyleSheet(f"QLabel {{color: rgb({r}, {g}, {b})}}")
+        r, g, b = self.text_color["fb"]
+        self.fb_label.setStyleSheet(f"QLabel {{color: rgb({r}, {g}, {b})}}")
+
+        if self.btn_mode == "start":
+            self.btn_action.setStyleSheet(
+                "QPushButton "
+                + f"{{background-color: {get_accent_colors().success.name()}; color: white;}}"
+            )
+        else:
+            self.btn_action.setStyleSheet(
+                "QPushButton "
+                + f"{{background-color: {get_accent_colors().emergency.name()}; color: white;}}"
+            )
+
+    def set_target(self, target):
+        """Change the target value in the ui"""
+        self.target = target
+        text = f"{target:.{int(self.decimals)}f}"
+        if self.unit is not None:
+            text = text + " " + self.unit
+        self.target_label.setText(text)
+        self._on_target_or_fb_changed()
+
+    def set_feedback(self, fb):
+        """Change the feedback value in the ui"""
+        if self.status != Status.MOVING:
+            self.fb = fb
+            text = f"{fb:.{int(self.decimals)}f}"
+            if self.unit is not None:
+                text = text + " " + self.unit
+            self.fb_label.setText(text)
+            self._on_target_or_fb_changed()
+
+    def _apply_button_style(self, mode: str):
+        """Apply a button style depending on if the button shows start or stop"""
+        self.btn_mode = mode
+        if mode == "start":
+            self.btn_action.setText("Move")
+            self.btn_action.setStyleSheet(
+                "QPushButton "
+                + f"{{background-color: {get_accent_colors().success.name()}; color: white;}}"
+            )
+        else:  # stop
+            self.btn_action.setText("Stop")
+            self.btn_action.setStyleSheet(
+                "QPushButton "
+                + f"{{background-color: {get_accent_colors().emergency.name()}; color: white;}}"
+            )
+
+    def _set_status(self, status: str):
+        """Set the current status icon in the ui"""
+        self.status = status
+        self.status_icon.set_status(status)
+
+    def _on_target_or_fb_changed(self):
+        """Re-evaluate in-position status whenever the target value changes."""
+        if self.status in (Status.ERROR, Status.MOVING):
+            return
+        if abs(self.fb - self.target) <= self.deadband:
+            self._set_status(Status.IN_POSITION)
+        else:
+            self._set_status(Status.NOT_IN_POSITION)
+
+    def _on_button_clicked(self):
+        """Starts or stops motion depending on current situation"""
+        if self._thread and self._thread.isRunning():
+            self._stop_motion()
+        else:
+            self._start_motion()
+
+    def _start_motion(self):
+        """Start a motion"""
+        target = self.target
+        if abs(target - self.fb) <= self.deadband:
+            self._set_status(Status.IN_POSITION)
+            return
+
+        self._set_status(Status.MOVING)
+        self._apply_button_style("stop")
+
+        self._worker = MotionWorker(self.dev, self.motor, target)
+        self._thread = QThread()
+        self._worker.moveToThread(self._thread)
+
+        self._thread.started.connect(self._worker.run)
+        self._worker.position_changed.connect(self._on_position_changed)
+        self._worker.error.connect(self._on_error)
+        self._worker.error.connect(self._thread.quit)
+        self._worker.finished.connect(self._on_motion_finished)
+        self._worker.finished.connect(self._thread.quit)
+        self._thread.finished.connect(self._cleanup_thread)
+
+        self._thread.start()
+
+    def _on_error(self):
+        """Called when an error occurs"""
+        self._set_status(Status.ERROR)
+        self._apply_button_style("start")
+
+    def _stop_motion(self):
+        """Attempts to stop the motion"""
+        if self._worker:
+            self._worker.stop()
+
+    def _on_position_changed(self, pos: float):
+        """Change the feedback value in the ui"""
+        self.fb = pos
+        text = f"{pos:.{int(self.decimals)}f}"
+        if self.unit is not None:
+            text = text + " " + self.unit
+        self.fb_label.setText(text)
+
+    def _on_motion_finished(self):
+        """Finished a movement"""
+        target = self.target
+        if self.status not in Status.ERROR:
+            if abs(self.fb - target) <= self.deadband:
+                self._set_status(Status.IN_POSITION)
+            else:
+                self._set_status(Status.NOT_IN_POSITION)
+        self._apply_button_style("start")
+
+    def _cleanup_thread(self):
+        """Cleaning up of the mover thread"""
+        if self._thread:
+            self._thread.deleteLater()
+            self._thread = None
+        if self._worker:
+            self._worker.deleteLater()
+            self._worker = None
+
+    def shutdown(self):
+        """Cleaning up of the mover when shutting down the application"""
+        if self._worker:
+            self._worker.stop()
+        if self._thread:
+            self._thread.quit()
+            self._thread.wait(2000)  # max 2 s grace period
+
+
+class AbsorberWidget(QWidget):
+    """
+    Control of the frontend absorber (only open)
+    """
+
+    def __init__(self, absorber, label: str = "Absorber"):
+        super().__init__()
+        self.absorber = absorber
+        self.fb = False
+        self.text_color = (0, 0, 0)
+
+        layout = QHBoxLayout(self)
+        layout.setContentsMargins(10, 0, 0, 0)
+        layout.setSpacing(0)
+
+        # Name
+        self.label = QLabel(label)
+        self.label.setFixedWidth(100)
+        self.label.setContentsMargins(0, 0, 10, 0)
+        self.label.setWordWrap(True)
+        layout.addWidget(self.label)
+
+        # Blank
+        self.blank_label = QLabel("")
+        self.blank_label.setFixedWidth(100)
+        layout.addWidget(self.blank_label)
+
+        # Feedback
+        self.fb_label = QLabel("-")
+        self.fb_label.setFixedWidth(100)
+        layout.addWidget(self.fb_label)
+
+        # Blank icon
+        self.blank_icon = QLabel("")
+        self.blank_icon.setFixedWidth(30)
+        self.blank_icon.setContentsMargins(0, 0, 10, 0)
+        layout.addWidget(self.blank_icon)
+
+        # Open
+        self.btn_action = QPushButton("Open")
+        self.btn_action.setFixedWidth(90)
+        self.btn_action.setFixedHeight(20)
+        self.btn_action.clicked.connect(self._on_button_clicked)
+        layout.addWidget(self.btn_action)
+
+    def set_feedback(self, fb: bool):
+        """
+        Displays the status of the absober in the ui
+
+        Args:
+            fb (bool): True will set the button to Open, False to Closed
+        """
+        self.fb = fb
+        if fb:
+            self.fb_label.setText("Open")
+            self.fb_label.setStyleSheet(f"QLabel {{color: {get_accent_colors().success.name()}}}")
+        else:
+            self.fb_label.setText("Closed")
+            self.fb_label.setStyleSheet(f"QLabel {{color: {get_accent_colors().emergency.name()}}}")
+
+    def enable_open(self, enable: bool = False):
+        """
+        Enable or disable the open/close button
+
+        Args:
+            enable (bool): Enables and disables the button
+        """
+        if enable:
+            self.btn_action.setStyleSheet(
+                "QPushButton "
+                + f"{{background-color: {get_accent_colors().success.name()}; color: white;}}"
+            )
+            self.btn_action.setEnabled(True)
+        else:  # disabled
+            self.btn_action.setStyleSheet(
+                "QPushButton {{background-color: rgb(120, 120, 120); color: white;}}"
+            )
+            self.btn_action.setDisabled(True)
+
+    def _on_button_clicked(self):
+        """Open absorber"""
+        self.absorber.open()

debye_bec/bec_widgets/widgets/digital_twin/widgets/qt_widgets.py

@@ -0,0 +1,306 @@
+"""
+Universal Qt widgets
+"""
+
+from functools import partial
+
+from bec_widgets.utils.colors import get_accent_colors
+from qtpy.QtCore import Qt
+
+# pylint: disable=E0611
+from qtpy.QtGui import QFont
+from qtpy.QtWidgets import (
+    QApplication,
+    QComboBox,
+    QDoubleSpinBox,
+    QGroupBox,
+    QHBoxLayout,
+    QLabel,
+    QPushButton,
+    QVBoxLayout,
+    QWidget,
+)
+
+
+class Group(QGroupBox):
+    def __init__(self, label, widgets):
+        super().__init__(label)
+        self.layout = QVBoxLayout(self)  # type: ignore
+        for widget in widgets:
+            self.layout.addWidget(widget)  # type: ignore
+
+
+class NumberIndicator(QWidget):
+    def __init__(self, label="", unit=None, highlight=False, decimals=3):
+        super().__init__()
+        layout = QHBoxLayout(self)
+        layout.setContentsMargins(10, 0, 0, 0)
+        layout.setSpacing(0)
+        self.label = QLabel(label)
+        self.label.setFixedWidth(140)
+        self.label.setContentsMargins(0, 0, 10, 0)
+        self.label.setWordWrap(True)
+        layout.addWidget(self.label)
+        self.val = QLabel("-")
+        self.val.setAlignment(Qt.AlignTop)  # type: ignore
+        # self.val.setFixedWidth(140)
+        layout.addWidget(self.val)
+        self.unit = unit
+        self.highlight = highlight
+        self.decimals = decimals
+        self.number = 0
+        if highlight:
+            font = QFont()
+            font.setBold(True)
+            font.setPointSize(14)
+            self.label.setFont(font)
+            self.val.setFont(font)
+
+    def value(self) -> float:
+        return self.number
+
+    def setLabel(self, label) -> None:
+        self.label.setText(label)
+
+    def setValue(self, number):
+        self.number = number
+        text = f"{number:.{int(self.decimals)}f}"
+        if self.unit is not None:
+            text = text + " " + self.unit
+        self.val.setText(text)
+
+
+class InputNumberField(QWidget):
+    def __init__(
+        self,
+        identifier="",
+        label="",
+        unit=None,
+        prefix=None,
+        init=0.0,
+        decimals=1,
+        single_step=0.1,
+        ll=-1e6,
+        hl=1e6,
+    ):
+        super().__init__()
+        layout = QHBoxLayout(self)
+        layout.setContentsMargins(10, 0, 0, 0)
+        layout.setSpacing(0)
+        self.identifier = identifier
+        self.label = QLabel(label)
+        self.label.setFixedWidth(140)
+        self.label.setContentsMargins(0, 0, 10, 0)
+        self.label.setWordWrap(True)
+        layout.addWidget(self.label)
+        self.val = QDoubleSpinBox()
+        self.val.setRange(ll, hl)
+        self.val.setDecimals(decimals)
+        self.val.setSingleStep(single_step)
+        self.val.setValue(init)
+        if unit is not None:
+            self.val.setSuffix(" " + unit)
+        if prefix is not None:
+            self.val.setPrefix(prefix + " ")
+        # self.val.setFixedWidth(140)
+        layout.addWidget(self.val)
+
+    def set_number(self, number):
+        self.val.setValue(number)
+
+    def has_focus(self) -> bool:
+        return self.val.hasFocus()
+
+    def value(self) -> float:
+        return self.val.value()
+
+    def value_changed_connect(self, func):
+        """Connect a function to the Enter/Return key press."""
+        self.val.valueChanged.connect(
+            partial(
+                func, identifier=self.identifier, value_obj=self.val, value=lambda: self.val.value()
+            )
+        )
+
+
+class ComboBox(QWidget):
+    def __init__(self, identifier="", label="", enums=[]):
+        super().__init__()
+        layout = QHBoxLayout(self)
+        layout.setContentsMargins(10, 0, 0, 0)
+        layout.setSpacing(0)
+        self.identifier = identifier
+        self.label = QLabel(label)
+        self.label.setFixedWidth(140)
+        self.label.setContentsMargins(0, 0, 10, 0)
+        self.label.setWordWrap(True)
+        layout.addWidget(self.label)
+        self.value = QComboBox()
+        for entry in enums:
+            self.value.addItem(entry)
+        layout.addWidget(self.value)
+
+    def set_current_text(self, text):
+        self.value.setCurrentText(text)
+
+    def currentText(self) -> str:
+        return self.value.currentText()
+
+    def has_focus(self) -> bool:
+        return QApplication.focusWidget() is self.value.view()
+
+    def activated_connect(self, func):
+        """Connect a function to the Enter/Return key press."""
+        self.value.activated.connect(
+            partial(
+                func,
+                identifier=self.identifier,
+                value_obj=self.value,
+                value=lambda: self.value.currentText(),
+            )
+        )
+
+    def setDisabled(self, disable):
+        self.value.setDisabled(disable)
+
+
+class Button(QWidget):
+    def __init__(self, label=None, label_button: str = "", enabled=False):
+        super().__init__()
+        layout = QHBoxLayout(self)
+        layout.setContentsMargins(10, 0, 0, 0)
+        layout.setSpacing(0)
+        if label is not None:
+            self.label = QLabel(label)
+            self.label.setFixedWidth(140)
+            layout.addWidget(self.label)
+        self.button = QPushButton(label_button)
+        if label is not None:
+            self.button.setFixedWidth(160)
+        self.enable_button(enabled)
+        layout.addWidget(self.button)
+
+    def clicked_connect(self, func):
+        """Connect a function to the button press."""
+        self.button.clicked.connect(func)
+
+    def enable_button(self, enable: bool = False):
+        if enable:
+            self.button.setStyleSheet(
+                f"QPushButton {{background-color: {get_accent_colors().default.name()}; color: white;}}"
+            )
+            self.button.setEnabled(True)
+        else:  # disabled
+            self.button.setStyleSheet(
+                "QPushButton {{background-color: rgb(120, 120, 120); color: white;}}"
+            )
+            self.button.setDisabled(True)
+
+    def setText(self, text):
+        self.button.setText(text)
+
+
+class TextIndicator(QWidget):
+    def __init__(self, label):
+        super().__init__()
+        layout = QHBoxLayout(self)
+        layout.setContentsMargins(10, 0, 0, 0)
+        layout.setSpacing(0)
+        self.label = QLabel(label)
+        self.label.setFixedWidth(140)
+        self.label.setContentsMargins(0, 0, 10, 0)
+        self.label.setWordWrap(True)
+        layout.addWidget(self.label)
+        self.text = QLabel("-")
+        self.text.setAlignment(Qt.AlignTop)  # type: ignore
+        layout.addWidget(self.text)
+
+    def setLabel(self, label) -> None:
+        self.label.setText(label)
+
+    def setText(self, text):
+        self.text.setText(text)
+
+    def setColor(self, color: str):
+        self.text.setStyleSheet(f"QLabel {{color:{color}}}")
+
+
+# class Button(QWidget):
+#     def __init__(self, label, label_button):
+#         super().__init__()
+#         layout = QHBoxLayout(self)
+#         layout.setContentsMargins(10, 0, 0, 0)
+#         layout.setSpacing(0)
+#         self.label = QLabel(label)
+#         self.label.setFixedWidth(150)
+#         layout.addWidget(self.label)
+#         self.button = QPushButton(label_button)
+#         self.button.setStyleSheet("color: black; background-color: dodgerblue;")
+#         self.button.setFixedWidth(160)
+#         layout.addWidget(self.button)
+
+#     def set_on_press(self, func):
+#         """Connect a function to the button press."""
+#         self.button.clicked.connect(func)
+
+#     def enable_button(self):
+#         self.button.setEnabled(True)
+#         self.button.setStyleSheet("color: black; background-color: dodgerblue;")
+
+#     def disable_button(self):
+#         self.button.setEnabled(False)
+#         self.button.setStyleSheet("color: black; background-color: grey;")
+
+#     def set_button_text(self, text):
+#         self.button.setText(text)
+
+# class LED(QWidget):
+#     def __init__(self, states, colors, label):
+#         super().__init__()
+#         self.states = states
+#         self.colors = colors
+#         layout = QHBoxLayout(self)
+#         layout.setContentsMargins(10, 0, 0, 0)
+#         layout.setSpacing(0)
+#         self.label = QLabel(label)
+#         self.label.setFixedWidth(150)
+#         layout.addWidget(self.label)
+#         self.led = QLabel()
+#         self.led.setFixedWidth(160)
+#         layout.addWidget(self.led)
+
+#     def apply_color(self, val):
+#         color = self.colors[self.states.index(val)]
+#         self.led.setStyleSheet(f"background-color: {color}; border: 1px solid black;")
+
+# class InputTextField(QWidget):
+#     def __init__(self, topic, label):
+#         super().__init__()
+#         self.topic = topic
+#         layout = QHBoxLayout(self)
+#         layout.setContentsMargins(10, 0, 0, 0)
+#         layout.setSpacing(0)
+#         self.label = QLabel(label)
+#         self.label.setFixedWidth(140)
+#         self.label.setContentsMargins(0, 0, 10, 0)
+#         self.label.setWordWrap(True)
+#         layout.addWidget(self.label)
+#         self.val = QLineEdit()
+#         self.val.setPlaceholderText('0')
+#         # self.val.setFixedWidth(140)
+#         layout.addWidget(self.val)
+
+#     def set_text(self, text):
+#         self.val.setText(text)
+
+#     def has_focus(self) -> bool:
+#         return self.val.hasFocus()
+
+#     def text(self) -> str:
+#         return self.val.text()
+
+#     def set_on_return(self, func):
+#         """Connect a function to the Enter/Return key press."""
+#         self.val.returnPressed.connect(
+#             partial(func, self.val, self.topic, lambda: self.val.text())
+#         )

debye_bec/bec_widgets/widgets/digital_twin/x01da_offsets.yaml

@@ -0,0 +1,50 @@
+cm_try:
+  offset: 0.15
+
+mo1_trx:
+  modifier:
+    axis: mo1_trx
+    range: [[-30, -0.1], [0.1, 30]]
+  offset: [0, 2.21]
+
+mo1_try:
+  modifier:
+    axis: mo1_trx
+    range: [[-30, -0.1], [0.1, 30]]
+  offset: [0, -1.6]
+
+sl1_centery:
+  offset: -1.8
+
+fm_trx:
+  modifier:
+    axis: fm_trx
+    range: [[-66, -31], [-24, 7], [11, 31], [38, 66]]
+  offset: [0, 0, 0, -0.16]
+
+fm_try:
+  modifier:
+    axis: fm_trx
+    range: [[-66, -31], [-24, 7], [11, 31], [38, 66]]
+  offset: [0, 0, 0, -0.45]
+
+fm_rotx:
+  modifier:
+    axis: fm_trx
+    range: [[-66, -31], [-24, 7], [11, 31], [38, 66]]
+  offset: [0, 0, 0, 0.063]
+
+fm_roty:
+  modifier:
+    axis: fm_trx
+    range: [[-66, -31], [-24, 7], [11, 31], [38, 66]]
+  offset: [0, 0, 0, -0.04]
+
+sl2_centery:
+  offset: 1.2
+
+ot_try:
+  offset: 0
+
+ot_rotx:
+  offset: 0

debye_bec/bec_widgets/widgets/digital_twin/x01da_parameters.py

@@ -0,0 +1,321 @@
+"""
+X01DA / Debye Beamline Parameters.
+This file describes the parameter of each component of the Debye beamline
+to be used for raytracing and geometrical calculations.
+"""
+
+from collections import namedtuple
+
+import numpy as np
+import xrt.backends.raycing.materials as rm
+
+# XRT definitions
+filterBeryl = rm.Material("Be", rho=1.85, kind="plate")  # pyright: ignore[reportArgumentType]
+filterDiamond = rm.Material("C", rho=3.52, kind="plate")  # pyright: ignore[reportArgumentType]
+filterGraphite = rm.Material("C", rho=2.266, kind="plate")  # pyright: ignore[reportArgumentType]
+
+stripeSi = rm.Material("Si", rho=2.33)  # pyright: ignore[reportArgumentType]
+stripePt = rm.Material("Pt", rho=21.45)  # pyright: ignore[reportArgumentType]
+stripeRh = rm.Material("Rh", rho=12.41)  # pyright: ignore[reportArgumentType]
+stripeCr = rm.Material("Cr", rho=7.14)  # pyright: ignore[reportArgumentType]
+stripePyrex = rm.Material(
+    "Si", rho=2.20
+)  # Use Si as bare element and the density of SiO2 # pyright: ignore[reportArgumentType]
+
+si111_1 = rm.CrystalSi(hkl=(1, 1, 1), tK=77)  # first xtal surface
+si311_1 = rm.CrystalSi(hkl=(3, 1, 1), tK=77)  # first xtal surface
+si333_1 = rm.CrystalSi(hkl=(3, 3, 3), tK=77)  # first xtal surface
+si511_1 = rm.CrystalSi(hkl=(5, 1, 1), tK=77)  # first xtal surface
+si111_2 = rm.CrystalSi(hkl=(1, 1, 1), tK=77)  # second xtal surface
+si311_2 = rm.CrystalSi(hkl=(3, 1, 1), tK=77)  # second xtal surface
+si333_2 = rm.CrystalSi(hkl=(3, 3, 3), tK=77)  # second xtal surface
+si511_2 = rm.CrystalSi(hkl=(5, 1, 1), tK=77)  # second xtal surface
+
+filterDiamond = rm.Material("C", rho=3.52, kind="plate")  # pyright: ignore[reportArgumentType]
+filterBe = rm.Material("Be", rho=1.85, kind="plate")  # pyright: ignore[reportArgumentType]
+filterSi3N4 = rm.Material(
+    ["Si", "N"], quantities=[3, 4], rho=3.44, kind="plate"
+)  # pyright: ignore[reportArgumentType]
+filterAl = rm.Material("Al", rho=2.69, kind="plate")  # pyright: ignore[reportArgumentType]
+filterGraphite = rm.Material("C", rho=2.266, kind="plate")  # pyright: ignore[reportArgumentType]
+
+# General parameters
+sourceHeight = 0
+
+# Synchrotron
+synchrotron = namedtuple(
+    "synchrotron", ["eE", "eI", "eEspread", "eEpsilonX", "eEpsilonZ", "betaX", "betaZ"]
+)
+
+sls1 = synchrotron(
+    eE=2.4, eI=0.4, eEspread=0.878e-3, eEpsilonX=5.63, eEpsilonZ=0.007, betaX=0.45, betaZ=14.4
+)
+
+sls2 = synchrotron(
+    eE=2.7, eI=0.4, eEspread=1.147e-3, eEpsilonX=0.156, eEpsilonZ=0.01, betaX=0.18, betaZ=4.6
+)
+
+# Source
+bendingMagnet = namedtuple("bendingMagnet", ["name", "center", "sync", "B0"])
+
+sls1_14t = bendingMagnet(name="FE-BM-SLS1-1.4T", center=(0, 0, 0), sync=sls1, B0=1.4)
+
+sls2_21t = bendingMagnet(name="FE-BM-SLS2-2.1T", center=(0, 0, 0), sync=sls2, B0=2.1)
+
+sls2_35t = bendingMagnet(name="FE-BM-SLS2-3.5T", center=(0, 0, 0), sync=sls2, B0=3.5)
+
+sls2_50t = bendingMagnet(name="FE-BM-SLS2-5.0T", center=(0, 0, 0), sync=sls2, B0=5.0)
+
+# FE slits
+fe_slits = namedtuple("slits", ["name", "center", "center1", "center2", "maxDivH", "maxDivV"])
+
+feSlits = fe_slits(
+    name="FE-SLITS",
+    center=(0, 6117, sourceHeight),
+    center1=(0, 5045, sourceHeight),
+    center2=(0, 5289.5, sourceHeight),
+    maxDivH=1.8e-3,
+    maxDivV=0.8e-3,
+)
+
+# FE Window
+filt = namedtuple(
+    "filt", ["name", "center", "pitch", "limPhysX", "limPhysY", "surface", "material", "thickness"]
+)
+
+feWindow = filt(
+    name="FE-WINDOW",
+    center=(0.0, 7020, sourceHeight),
+    pitch=np.pi / 2,
+    limPhysX=(-6, 6),
+    limPhysY=(-3.0, 3.0),
+    surface="None",
+    material=filterDiamond,
+    thickness=0.1,
+)
+feWindow = feWindow._replace(surface=f"CVD Diamond window {feWindow.thickness*1e3:0.0f} $\\mu$m")
+
+# Collimating mirror
+collimatingMirror = namedtuple(
+    "collimatingMirror",
+    [
+        "name",
+        "center",
+        "surface",
+        "material",
+        "limPhysX",
+        "limPhysY",
+        "limOptX",
+        "limOptY",
+        "R",
+        "pitch",
+        "jack1",
+        "jack2",
+        "jack3",
+        "tx1",
+        "tx2",
+    ],
+)
+
+cm = collimatingMirror(
+    name="FE-CM",
+    center=[0, 6890, sourceHeight],
+    surface=("Si", "Pt", "Rh"),
+    material=(stripeSi, stripePt, stripeRh),
+    limPhysX=(-34, 34),
+    limPhysY=(-600, 600),
+    limOptX=((-21, -7, 14), (-11, 11, 23)),
+    limOptY=((-500, -500, -500), (500, 500, 500)),
+    R=[3e6, 15e6],
+    pitch=[-5.0e-3, -0.0e-3],
+    jack1=[0.0, 7210.0, 0.0],  # Tripod X, Y, Z (global)
+    jack2=[-210.0, 8310.0, 0.0],
+    jack3=[210.0, 8310.0, 0.0],
+    tx1=[0.0, -575.5],  # X-Stage 1 [x, y] (local)
+    tx2=[0.0, 575],
+)  # X-Stage 2
+
+apertures = namedtuple("apertures", ["name", "center", "opening"])
+
+fePS = apertures(
+    name="FE-PS", center=[0, 8815, sourceHeight], opening=[-20.0, 20.0, -20.0 + 12.5, 20.0 + 12.5]
+)  # left, right, bottom, top
+
+opWbBsBlock = apertures(
+    name="OP-WB-BS-BLOCK", center=[0.0, 13860, sourceHeight], opening=[-18.0, 18.0, 25, 85.5]
+)  # left, right, bottom, top
+# opening=[-18., 18., 42, 76], # X10DA
+
+# Monochromator
+monochromator = namedtuple(
+    "monochromator",
+    [
+        "name",
+        "center",
+        "xtal",
+        "material1",
+        "material2",
+        "xtalWidth",
+        "xtalOffsetX",
+        "xtalLength1",
+        "xtalLength2",
+        "xtalGap",
+        "rotOffset",
+        "heightOffset",
+        "braggLim",
+        "jack1",
+        "jack2",
+        "jack3",
+        "tx",
+    ],
+)
+
+mo1 = monochromator(
+    name="OP-MO1",
+    center=[0.0, 11750, sourceHeight],
+    xtal=("Si311", "Si111"),
+    material1=(si311_1, si111_1),
+    material2=(si311_2, si111_2),
+    xtalWidth=(24, 24),
+    xtalOffsetX=(-21.2, 21.2),
+    xtalLength1=(55, 55),
+    xtalLength2=(105, 105),
+    xtalGap=(8, 8),
+    rotOffset=6,
+    heightOffset=8.5,
+    braggLim=[3.6, 33],
+    jack1=[0.0, 11350.0, 0.0],  # Tripod maybe not available!
+    jack2=[-400.0, 12350.0, 0.0],
+    jack3=[400.0, 12350.0, 0.0],
+    tx=0.0,
+)  # X-Stage [x]
+
+mo2 = monochromator(
+    name="OP-CCM2",
+    center=[0.0, 13250, sourceHeight],
+    xtal=("Si311", "Si111"),
+    material1=(si311_1, si111_1),
+    material2=(si311_2, si111_2),
+    xtalWidth=(24, 24),
+    xtalOffsetX=(-21, 21),
+    xtalLength1=(55, 55),
+    xtalLength2=(105, 105),
+    xtalGap=(8, 8),
+    rotOffset=6,
+    heightOffset=8.5,
+    braggLim=[3.6, 33],
+    jack1=[0.0, 13350.0, 0.0],  # Tripod maybe not available!
+    jack2=[-400.0, 14350.0, 0.0],
+    jack3=[400.0, 14350.0, 0.0],
+    tx=0.0,
+)  # X-Stage [x]
+
+# OP Slits
+op_slits = namedtuple("op_slits", ["name", "center"])
+
+opSlits1 = op_slits(name="OP-SLITS 1", center=(0, 14349.6, sourceHeight))
+
+opSlits2 = op_slits(name="OP-SLITS 2", center=(0, 18134.8, sourceHeight))
+
+# OP Beam Monitors
+op_bm = namedtuple("op_bm", ["name", "center"])
+
+opBM1 = op_bm(name="OP Beam Monitor 1", center=(0, 14599.6, sourceHeight))
+
+opBM2 = op_bm(name="OP Beam Monitor 2", center=(0, 18384.8, sourceHeight))
+
+# Focusing mirror
+focusingMirror = namedtuple(
+    "focusingMirror",
+    [
+        "name",
+        "center",
+        "surfaceToroid",
+        "materialToroid",
+        "surfaceFlat",
+        "materialFlat",
+        "limPhysXToroid",
+        "limPhysYToroid",
+        "limPhysXFlat",
+        "limPhysYFlat",
+        "limOptXToroid",
+        "limOptYToroid",
+        "limOptXFlat",
+        "limOptYFlat",
+        "R",
+        "pitch",
+        "r",
+        "xToroid",
+        "xFlat",
+        "hToroid",
+        "jack1",
+        "jack2",
+        "jack3",
+        "tx1",
+        "tx2",
+    ],
+)
+
+fm = focusingMirror(
+    name="OP-FM",
+    center=[0.0, 15670, sourceHeight],  # nominal height 58 mm above ring, SLS1!
+    surfaceToroid=("Rh", "Pt"),
+    materialToroid=(stripeRh, stripePt),
+    surfaceFlat=("Rh", "Pt"),
+    materialFlat=(stripeRh, stripePt),
+    limPhysXToroid=(-79.0, 79.0),
+    limPhysYToroid=(-575.0, 575.0),
+    limPhysXFlat=(-79.0, 79.0),
+    limPhysYFlat=(-575.0, 575.0),
+    limOptXToroid=((-38, 66), (-66, 31)),
+    limOptYToroid=((-500.0, -500.0), (500.0, 500.0)),
+    limOptXFlat=((-11.45, 23.55), (-30.45, -6.45)),
+    limOptYFlat=((-500.0, -500.0), (500.0, 500.0)),
+    R=[3e6, 15e6],
+    pitch=[-5.0e-3, 0e-3],
+    r=[35.510, 24.986],
+    xToroid=[-52, 48.5],  # offset in local x
+    xFlat=[-20.95, 8.55],
+    hToroid=[2.88, 7.15],  # depth of the cylinder at x = xCylinder1 and x = xCylinder2.
+    jack1=[-130.0, 15535 - 538.0, 0.0],
+    jack2=[130.0, 15535 + 538.0, 0.0],
+    jack3=[0.0, 15535 + 538.0, 0.0],
+    tx1=[0.0, -575.0],  # X-Stage 1 [x, y]
+    tx2=[0.0, 575.0],
+)  # X-Stage 2 [x, y]
+
+# EH Window
+ehWindow = filt(
+    name="EH-WINDOW",
+    center=(0.0, 19998.3, sourceHeight),
+    pitch=np.pi / 2,
+    limPhysX=(-20.0, 20.0),
+    limPhysY=(-4, 4),
+    surface="None",
+    material=filterSi3N4,
+    thickness=0.002,
+)
+ehWindow = ehWindow._replace(surface=f"Beryllium window {ehWindow.thickness*1e3:0.0f} $\\mu$m")
+
+# Sample
+sample = namedtuple("sample", ["name", "center"])
+
+smpl = sample(name="EH-SMPL", center=[0, 23365, sourceHeight])
+
+smpl2 = sample(name="EH-SMPL2", center=[0, 27500, sourceHeight])
+
+# Vacuum pipes
+# DN40CF ID = 35 mm oder 37 mm
+# DN50CF ID = 47.5 mm
+# DN63CF ID = 60.2 mm oder 66 mm
+# DN100CF ID = 97.4 mm oder 104 mm
+pipe = namedtuple("pipes", ["center", "diameter", "start", "end"])
+vacuum_pipes = pipe(
+    center=[27.5, (37.5 + 27.5) / 2, 37.5, 62.5, 72.5],
+    diameter=[97.4, 97.4, 97.4, 97.4, 97.4],
+    start=[10952.88, 11750 + 250, mo2.center[1] + 250, 14000, fm.center[1]],
+    end=[11750 - 250, mo2.center[1] - 250, 14000, fm.center[1], ehWindow.center[1]],
+)
+
+Walls = namedtuple("walls", ["start", "end", "height"])
+walls = Walls(start=[13999.30], end=[13999 + 75.5 + 30], height=[[-20, 25]])

debye_bec/device_configs/x01da_experimental_hutch.yaml

@@ -5,7 +5,7 @@
 ot_tryu:
   readoutPriority: baseline
   description: Optical Table Y-Translation Upstream
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES-OT:TRYU
   onFailure: retry
@@ -15,7 +15,7 @@ ot_tryu:
 ot_tryd:
   readoutPriority: baseline
   description: Optical Table Y-Translation Downstream
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES-OT:TRYD
   onFailure: retry
@@ -25,7 +25,7 @@ ot_tryd:
 ot_es1_trz:
   readoutPriority: baseline
   description: Optical Table ES1 Z-Translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-OT:TRZ
   onFailure: retry
@@ -35,7 +35,7 @@ ot_es1_trz:
 ot_es2_trz:
   readoutPriority: baseline
   description: Optical Table ES2 Z-Translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES2-OT:TRZ
   onFailure: retry
@@ -45,17 +45,17 @@ ot_es2_trz:
 ot_try:
   readoutPriority: baseline
   description: Optical Table Y-Translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES-OT:TRY
   onFailure: retry
   enabled: true
   softwareTrigger: false
 
-ot_pitch:
+ot_rotx:
   readoutPriority: baseline
   description: Optical Table Pitch
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES-OT:ROTX
   onFailure: retry
@@ -69,7 +69,7 @@ ot_pitch:
 es0wi_try:
   readoutPriority: baseline
   description: End Station 0 Exit Window Y-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES0-WI:TRY
   onFailure: retry
@@ -97,7 +97,7 @@ es0filter:
 es0sl_trxr:
   readoutPriority: baseline
   description: End Station slits X-translation Ring-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES0-SL:TRXR
   onFailure: retry
@@ -107,7 +107,7 @@ es0sl_trxr:
 es0sl_trxw:
   readoutPriority: baseline
   description: End Station slits X-translation Wall-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES0-SL:TRXW
   onFailure: retry
@@ -117,7 +117,7 @@ es0sl_trxw:
 es0sl_tryb:
   readoutPriority: baseline
   description: End Station slits Y-translation Bottom-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES0-SL:TRYB
   onFailure: retry
@@ -127,7 +127,7 @@ es0sl_tryb:
 es0sl_tryt:
   readoutPriority: baseline
   description: End Station slits X-translation Top-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES0-SL:TRYT
   onFailure: retry
@@ -137,7 +137,7 @@ es0sl_tryt:
 es0sl_center:
   readoutPriority: baseline
   description: End Station slits X-center
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES0-SL:CENTERX
   onFailure: retry
@@ -147,7 +147,7 @@ es0sl_center:
 es0sl_gapx:
   readoutPriority: baseline
   description: End Station slits X-gap
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES0-SL:GAPX
   onFailure: retry
@@ -157,7 +157,7 @@ es0sl_gapx:
 es0sl_centery:
   readoutPriority: baseline
   description: End Station slits Y-center
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES0-SL:CENTERY
   onFailure: retry
@@ -167,7 +167,7 @@ es0sl_centery:
 es0sl_gapy:
   readoutPriority: baseline
   description: End Station slits Y-gap
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES0-SL:GAPY
   onFailure: retry
@@ -195,7 +195,7 @@ es1_alignment_laser:
 es1man_trx:
   readoutPriority: baseline
   description: End Station sample manipulator X-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-MAN1:TRX
   onFailure: retry
@@ -205,7 +205,7 @@ es1man_trx:
 es1man_try:
   readoutPriority: baseline
   description: End Station sample manipulator Y-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-MAN1:TRY
   onFailure: retry
@@ -215,7 +215,7 @@ es1man_try:
 es1man_trz:
   readoutPriority: baseline
   description: End Station sample manipulator Z-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-MAN1:TRZ
   onFailure: retry
@@ -225,7 +225,7 @@ es1man_trz:
 es1man_roty:
   readoutPriority: baseline
   description: End Station sample manipulator Y-rotation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-MAN1:ROTY
   onFailure: retry
@@ -239,7 +239,7 @@ es1man_roty:
 es1arc_roty:
   readoutPriority: baseline
   description: End Station segmented arc Y-rotation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-ARC:ROTY
   onFailure: retry
@@ -249,7 +249,7 @@ es1arc_roty:
 es1det1_trx:
   readoutPriority: baseline
   description: End Station SDD 1 X-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-DET1:TRX
   onFailure: retry
@@ -259,7 +259,7 @@ es1det1_trx:
 es1bm1_trx:
   readoutPriority: baseline
   description: End Station X-ray Eye X-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-BM1:TRX
   onFailure: retry
@@ -269,7 +269,7 @@ es1bm1_trx:
 es1det2_trx:
   readoutPriority: baseline
   description: End Station SDD 2 X-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-DET2:TRX
   onFailure: retry
@@ -283,7 +283,7 @@ es1det2_trx:
 es2ma2_try:
   readoutPriority: baseline
   description: End Station ionization chamber 1+2 Y-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES2-MA2:TRY
   onFailure: retry
@@ -293,7 +293,7 @@ es2ma2_try:
 es2ma2_trz:
   readoutPriority: baseline
   description: End Station ionization chamber 1+2 Z-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES2-MA2:TRZ
   onFailure: retry
@@ -307,7 +307,7 @@ es2ma2_trz:
 es2ma3_try:
   readoutPriority: baseline
   description: End Station XRD detector Y-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES2-MA3:TRY
   onFailure: retry
@@ -386,4 +386,64 @@ es_light_toggle:
     read_pv: "X01DA-EH-LIGHT:TOGGLE"
   onFailure: retry
   enabled: true
+  softwareTrigger: false
+
+es_gas_sensor_o2:
+  readoutPriority: baseline
+  description: ES Gas Sensor O2
+  deviceClass: ophyd.EpicsSignalRO
+  deviceConfig:
+    read_pv: "X01DA-KIMESSA2:EH-O2"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
+
+es_gas_sensor_h2s:
+  readoutPriority: baseline
+  description: ES Gas Sensor H2S
+  deviceClass: ophyd.EpicsSignalRO
+  deviceConfig:
+    read_pv: "X01DA-KIMESSA2:EH-H2S"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
+
+es_gas_sensor_no2:
+  readoutPriority: baseline
+  description: ES Gas Sensor NO2
+  deviceClass: ophyd.EpicsSignalRO
+  deviceConfig:
+    read_pv: "X01DA-KIMESSA2:EH-NO2"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
+
+es_gas_sensor_co:
+  readoutPriority: baseline
+  description: ES Gas Sensor CO
+  deviceClass: ophyd.EpicsSignalRO
+  deviceConfig:
+    read_pv: "X01DA-KIMESSA2:EH-CO"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
+
+es_gas_sensor_h2:
+  readoutPriority: baseline
+  description: ES Gas Sensor H2
+  deviceClass: ophyd.EpicsSignalRO
+  deviceConfig:
+    read_pv: "X01DA-KIMESSA2:EH-H2"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
+
+es_gas_sensor_nh3:
+  readoutPriority: baseline
+  description: ES Gas Sensor NH3
+  deviceClass: ophyd.EpicsSignalRO
+  deviceConfig:
+    read_pv: "X01DA-KIMESSA2:EH-NH3"
+  onFailure: retry
+  enabled: true
   softwareTrigger: false

debye_bec/device_configs/x01da_frontend.yaml

@@ -1,4 +1,18 @@
 
+###################################
+##       Frontend Absorber       ##
+###################################
+
+abs:
+  readoutPriority: baseline
+  description: Frontend Absorber
+  deviceClass: debye_bec.devices.absorber.Absorber
+  deviceConfig:
+    prefix: "X01DA-FE-ABS1:"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
+
 ###################################
 ##        Frontend Slits         ##
 ###################################
@@ -6,7 +20,7 @@
 sldi_trxr:
   readoutPriority: baseline
   description: Front-end slit diaphragm X-translation Ring-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-SLDI:TRXR
   onFailure: retry
@@ -16,7 +30,7 @@ sldi_trxr:
 sldi_trxw:
   readoutPriority: baseline
   description: Front-end slit diaphragm X-translation Wall-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-SLDI:TRXW
   onFailure: retry
@@ -26,7 +40,7 @@ sldi_trxw:
 sldi_tryb:
   readoutPriority: baseline
   description: Front-end slit diaphragm Y-translation Bottom-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-SLDI:TRYB
   onFailure: retry
@@ -36,7 +50,7 @@ sldi_tryb:
 sldi_tryt:
   readoutPriority: baseline
   description: Front-end slit diaphragm X-translation Top-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-SLDI:TRYT
   onFailure: retry
@@ -46,7 +60,7 @@ sldi_tryt:
 sldi_centerx:
   readoutPriority: baseline
   description: Front-end slit diaphragm X-center
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-SLDI:CENTERX
   onFailure: retry
@@ -56,7 +70,7 @@ sldi_centerx:
 sldi_gapx:
   readoutPriority: baseline
   description: Front-end slit diaphragm X-gap
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-SLDI:GAPX
   onFailure: retry
@@ -66,7 +80,7 @@ sldi_gapx:
 sldi_centery:
   readoutPriority: baseline
   description: Front-end slit diaphragm Y-center
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-SLDI:CENTERY
   onFailure: retry
@@ -76,7 +90,7 @@ sldi_centery:
 sldi_gapy:
   readoutPriority: baseline
   description: Front-end slit diaphragm Y-gap
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-SLDI:GAPY
   onFailure: retry
@@ -90,7 +104,7 @@ sldi_gapy:
 cm_trxu:
   readoutPriority: baseline
   description: Collimating Mirror X-translation upstream
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:TRXU
   onFailure: retry
@@ -100,7 +114,7 @@ cm_trxu:
 cm_trxd:
   readoutPriority: baseline
   description: Collimating Mirror X-translation downstream
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:TRXD
   onFailure: retry
@@ -110,7 +124,7 @@ cm_trxd:
 cm_tryu:
   readoutPriority: baseline
   description: Collimating Mirror Y-translation upstream
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:TRYU
   onFailure: retry
@@ -120,7 +134,7 @@ cm_tryu:
 cm_trydr:
   readoutPriority: baseline
   description: Collimating Mirror Y-translation downstream ring
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:TRYDR
   onFailure: retry
@@ -130,7 +144,7 @@ cm_trydr:
 cm_trydw:
   readoutPriority: baseline
   description: Collimating Mirror Y-translation downstream wall
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:TRYDW
   onFailure: retry
@@ -140,17 +154,28 @@ cm_trydw:
 cm_bnd:
   readoutPriority: baseline
   description: Collimating Mirror bender
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:BND
   onFailure: retry
   enabled: true
   softwareTrigger: false
 
+cm_bnd_radius:
+  readoutPriority: baseline
+  description: Collimating Mirror Bending Radius
+  deviceClass: ophyd.EpicsSignalRO
+  deviceConfig:
+    read_pv: X01DA-CPCL-CM:BNDFORCE
+  onFailure: retry
+  readOnly: true
+  enabled: true
+  softwareTrigger: false
+
 cm_rotx:
   readoutPriority: baseline
   description: Collimating Morror Pitch
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:ROTX
   onFailure: retry
@@ -160,7 +185,7 @@ cm_rotx:
 cm_roty:
   readoutPriority: baseline
   description: Collimating Morror Yaw
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:ROTY
   onFailure: retry
@@ -170,7 +195,7 @@ cm_roty:
 cm_rotz:
   readoutPriority: baseline
   description: Collimating Morror Roll
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:ROTZ
   onFailure: retry
@@ -180,7 +205,7 @@ cm_rotz:
 cm_trx:
   readoutPriority: baseline
   description: Collimating Morror Center Point X
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:XTCP
   onFailure: retry
@@ -190,7 +215,7 @@ cm_trx:
 cm_try:
   readoutPriority: baseline
   description: Collimating Morror Center Point Y
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:YTCP
   onFailure: retry
@@ -200,7 +225,7 @@ cm_try:
 cm_ztcp:
   readoutPriority: baseline
   description: Collimating Morror Center Point Z
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:ZTCP
   onFailure: retry
@@ -210,7 +235,7 @@ cm_ztcp:
 cm_xstripe:
   readoutPriority: baseline
   description: Collimating Morror X Stripe
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-FE-CM:XSTRIPE
   onFailure: retry

debye_bec/device_configs/x01da_hutch_cameras.yaml

@@ -0,0 +1,34 @@
+
+###################################
+##         Hutch Cameras         ##
+###################################
+
+hutch_cam_1:
+  readoutPriority: baseline
+  description: Hutch Camera 1
+  deviceClass: debye_bec.devices.cameras.hutch_cam.HutchCam
+  deviceConfig:
+    prefix: "pcp085420"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
+
+hutch_cam_2:
+  readoutPriority: baseline
+  description: Hutch Camera 2
+  deviceClass: debye_bec.devices.cameras.hutch_cam.HutchCam
+  deviceConfig:
+    prefix: "pcp085436"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
+
+hutch_cam_3:
+  readoutPriority: baseline
+  description: Hutch Camera 3
+  deviceClass: debye_bec.devices.cameras.hutch_cam.HutchCam
+  deviceConfig:
+    prefix: "pcp085435"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false

debye_bec/device_configs/x01da_optics.yaml

@@ -3,30 +3,30 @@
 ##         Monochromator         ##
 ###################################
 
-mo_try:
+mo1_try:
   readoutPriority: baseline
   description: Monochromator Y Translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-MO1:TRY
   onFailure: retry
   enabled: true
   softwareTrigger: false
 
-mo_trx:
+mo1_trx:
   readoutPriority: baseline
   description: Monochromator X Translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-MO1:TRX
   onFailure: retry
   enabled: true
   softwareTrigger: false
 
-mo_roty:
+mo1_roty:
   readoutPriority: baseline
   description: Monochromator Yaw
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-MO1:ROTY
   onFailure: retry
@@ -40,7 +40,7 @@ mo_roty:
 sl1_trxr:
   readoutPriority: baseline
   description: Optics slits 1 X-translation Ring-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL1:TRXR
   onFailure: retry
@@ -53,7 +53,7 @@ sl1_trxr:
 sl1_trxw:
   readoutPriority: baseline
   description: Optics slits 1 X-translation Wall-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL1:TRXW
   onFailure: retry
@@ -66,7 +66,7 @@ sl1_trxw:
 sl1_tryb:
   readoutPriority: baseline
   description: Optics slits 1 Y-translation Bottom-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL1:TRYB
   onFailure: retry
@@ -79,7 +79,7 @@ sl1_tryb:
 sl1_tryt:
   readoutPriority: baseline
   description: Optics slits 1 X-translation Top-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL1:TRYT
   onFailure: retry
@@ -92,7 +92,7 @@ sl1_tryt:
 bm1_try:
   readoutPriority: baseline
   description: Beam Monitor 1 Y-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-BM1:TRY
   onFailure: retry
@@ -105,7 +105,7 @@ bm1_try:
 sl1_centerx:
   readoutPriority: baseline
   description: Optics slits 1 X-center
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL1:CENTERX
   onFailure: retry
@@ -118,7 +118,7 @@ sl1_centerx:
 sl1_gapx:
   readoutPriority: baseline
   description: Optics slits 1 X-gap
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL1:GAPX
   onFailure: retry
@@ -131,7 +131,7 @@ sl1_gapx:
 sl1_centery:
   readoutPriority: baseline
   description: Optics slits 1 Y-center
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL1:CENTERY
   onFailure: retry
@@ -144,7 +144,7 @@ sl1_centery:
 sl1_gapy:
   readoutPriority: baseline
   description: Optics slits 1 Y-gap
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL1:GAPY
   onFailure: retry
@@ -161,62 +161,78 @@ sl1_gapy:
 fm_trxu:
   readoutPriority: baseline
   description: Focusing Mirror X-translation upstream
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:TRXU
   onFailure: retry
   enabled: true
   softwareTrigger: false
+
 fm_trxd:
   readoutPriority: baseline
   description: Focusing Mirror X-translation downstream
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:TRXD
   onFailure: retry
   enabled: true
   softwareTrigger: false
+
 fm_tryd:
   readoutPriority: baseline
   description: Focusing Mirror Y-translation downstream
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:TRYD
   onFailure: retry
   enabled: true
   softwareTrigger: false
+
 fm_tryur:
   readoutPriority: baseline
   description: Focusing Mirror Y-translation upstream ring
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:TRYUR
   onFailure: retry
   enabled: true
   softwareTrigger: false
+
 fm_tryuw:
   readoutPriority: baseline
   description: Focusing Mirror Y-translation upstream wall
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:TRYUW
   onFailure: retry
   enabled: true
   softwareTrigger: false
+
 fm_bnd:
   readoutPriority: baseline
   description: Focusing Mirror bender
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:BND
   onFailure: retry
   enabled: true
   softwareTrigger: false
 
+fm_bnd_radius:
+  readoutPriority: baseline
+  description: Focusing Mirror Bending Radius
+  deviceClass: ophyd.EpicsSignalRO
+  deviceConfig:
+    read_pv: X01DA-CPCL-FM:BNDFORCE
+  onFailure: retry
+  readOnly: true
+  enabled: true
+  softwareTrigger: false
+
 fm_rotx:
   readoutPriority: baseline
   description: Focusing Morror Pitch
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:ROTX
   onFailure: retry
@@ -226,7 +242,7 @@ fm_rotx:
 fm_roty:
   readoutPriority: baseline
   description: Focusing Morror Yaw
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:ROTY
   onFailure: retry
@@ -236,27 +252,27 @@ fm_roty:
 fm_rotz:
   readoutPriority: baseline
   description: Focusing Morror Roll
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:ROTZ
   onFailure: retry
   enabled: true
   softwareTrigger: false
 
-fm_xctp:
+fm_trx:
   readoutPriority: baseline
   description: Focusing Morror Center Point X
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:XTCP
   onFailure: retry
   enabled: true
   softwareTrigger: false
 
-fm_ytcp:
+fm_try:
   readoutPriority: baseline
   description: Focusing Morror Center Point Y
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:YTCP
   onFailure: retry
@@ -266,7 +282,7 @@ fm_ytcp:
 fm_ztcp:
   readoutPriority: baseline
   description: Focusing Morror Center Point Z
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-FM:ZTCP
   onFailure: retry
@@ -280,7 +296,7 @@ fm_ztcp:
 sl2_trxr:
   readoutPriority: baseline
   description: Optics slits 2 X-translation Ring-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL2:TRXR
   onFailure: retry
@@ -293,7 +309,7 @@ sl2_trxr:
 sl2_trxw:
   readoutPriority: baseline
   description: Optics slits 2 X-translation Wall-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL2:TRXW
   onFailure: retry
@@ -306,7 +322,7 @@ sl2_trxw:
 sl2_tryb:
   readoutPriority: baseline
   description: Optics slits 2 Y-translation Bottom-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL2:TRYB
   onFailure: retry
@@ -319,7 +335,7 @@ sl2_tryb:
 sl2_tryt:
   readoutPriority: baseline
   description: Optics slits 2 X-translation Top-edge
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL2:TRYT
   onFailure: retry
@@ -332,7 +348,7 @@ sl2_tryt:
 bm2_try:
   readoutPriority: baseline
   description: Beam Monitor 2 Y-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-BM2:TRY
   onFailure: retry
@@ -345,7 +361,7 @@ bm2_try:
 sl2_centerx:
   readoutPriority: baseline
   description: Optics slits 2 X-center
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL2:CENTERX
   onFailure: retry
@@ -358,7 +374,7 @@ sl2_centerx:
 sl2_gapx:
   readoutPriority: baseline
   description: Optics slits 2 X-gap
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL2:GAPX
   onFailure: retry
@@ -371,7 +387,7 @@ sl2_gapx:
 sl2_centery:
   readoutPriority: baseline
   description: Optics slits 2 Y-center
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL2:CENTERY
   onFailure: retry
@@ -384,7 +400,7 @@ sl2_centery:
 sl2_gapy:
   readoutPriority: baseline
   description: Optics slits 2 Y-gap
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-OP-SL2:GAPY
   onFailure: retry

debye_bec/device_configs/x01da_standard_config.yaml

@@ -25,7 +25,7 @@ frontend_config:
 
 ## Bragg Monochromator
 mo1_bragg:
-  readoutPriority: monitored
+  readoutPriority: baseline
   description: Positioner for the Monochromator
   deviceClass: debye_bec.devices.mo1_bragg.mo1_bragg.Mo1Bragg
   deviceConfig:
@@ -34,14 +34,14 @@ mo1_bragg:
   enabled: true
   softwareTrigger: false
 mo1_bragg_angle:
-  readoutPriority: baseline
-  description: Positioner for the Monochromator
-  deviceClass: debye_bec.devices.mo1_bragg.mo1_bragg_angle.Mo1BraggAngle
-  deviceConfig:
-    prefix: "X01DA-OP-MO1:BRAGG:"
-  onFailure: retry
-  enabled: true
-  softwareTrigger: false
+ readoutPriority: baseline
+ description: Positioner for the Monochromator
+ deviceClass: debye_bec.devices.mo1_bragg.mo1_bragg_angle.Mo1BraggAngle
+ deviceConfig:
+   prefix: "X01DA-OP-MO1:BRAGG:"
+ onFailure: retry
+ enabled: true
+ softwareTrigger: false
 
 ## Remaining optics hutch
 optics_config:
@@ -51,7 +51,7 @@ optics_config:
 ##      Experimental Hutch       ##
 ###################################
 
-## NIDAQ
+# ## NIDAQ
 nidaq:
   readoutPriority: monitored
   description: NIDAQ backend for data reading for debye scans
@@ -67,8 +67,13 @@ xas_config:
   - !include ./x01da_xas.yaml
 
 ## XRD (Pilatus, pinhole, beamstop)
-xrd_config:
-  - !include ./x01da_xrd.yaml
+#xrd_config:
+#  - !include ./x01da_xrd.yaml
+
+# Commented out because too slow
+## Hutch cameras
+# hutch_cams:
+#   - !include ./x01da_hutch_cameras.yaml
 
 ## Remaining experimental hutch
 es_config:

debye_bec/device_configs/x01da_xas.yaml

@@ -3,35 +3,45 @@
 ##     Ionization Chambers       ##
 ###################################
 
-# ic0:
-#   readoutPriority: baseline
-#   description: Ionization chamber 0
-#   deviceClass: debye_bec.devices.ionization_chambers.ionization_chamber.IonizationChamber0
-#   deviceConfig:
-#     prefix: "X01DA-"
-#   onFailure: retry
-#   enabled: true
-#   softwareTrigger: false
+ic0:
+  readoutPriority: baseline
+  description: Ionization chamber 0
+  deviceClass: debye_bec.devices.ionization_chambers.ionization_chamber.IonizationChamber0
+  deviceConfig:
+    prefix: "X01DA-"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
 
-# ic1:
-#   readoutPriority: baseline
-#   description: Ionization chamber 1
-#   deviceClass: debye_bec.devices.ionization_chambers.ionization_chamber.IonizationChamber1
-#   deviceConfig:
-#     prefix: "X01DA-"
-#   onFailure: retry
-#   enabled: true
-#   softwareTrigger: false
+ic1:
+  readoutPriority: baseline
+  description: Ionization chamber 1
+  deviceClass: debye_bec.devices.ionization_chambers.ionization_chamber.IonizationChamber1
+  deviceConfig:
+    prefix: "X01DA-"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
 
-# ic2:
-#   readoutPriority: baseline
-#   description: Ionization chamber 2
-#   deviceClass: debye_bec.devices.ionization_chambers.ionization_chamber.IonizationChamber2
-#   deviceConfig:
-#     prefix: "X01DA-"
-#   onFailure: retry
-#   enabled: true
-#   softwareTrigger: false
+ic2:
+  readoutPriority: baseline
+  description: Ionization chamber 2
+  deviceClass: debye_bec.devices.ionization_chambers.ionization_chamber.IonizationChamber2
+  deviceConfig:
+    prefix: "X01DA-"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
+
+pips:
+  readoutPriority: baseline
+  description: Pips diode
+  deviceClass: debye_bec.devices.ionization_chambers.ionization_chamber.Pips
+  deviceConfig:
+    prefix: "X01DA-"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false
 
 ###################################
 ##    Reference Foil Changer     ##

debye_bec/device_configs/x01da_xrd.yaml

@@ -6,7 +6,7 @@
 pin1_trx:
   readoutPriority: baseline
   description: Pinhole X-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-PIN1:TRX
   onFailure: retry
@@ -17,7 +17,7 @@ pin1_trx:
 pin1_try:
   readoutPriority: baseline
   description: Pinhole Y-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-PIN1:TRY
   onFailure: retry
@@ -28,7 +28,7 @@ pin1_try:
 pin1_rotx:
   readoutPriority: baseline
   description: Pinhole X-rotation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-PIN1:ROTX
   onFailure: retry
@@ -39,7 +39,7 @@ pin1_rotx:
 pin1_roty:
   readoutPriority: baseline
   description: Pinhole Y-rotation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES1-PIN1:ROTY
   onFailure: retry
@@ -54,7 +54,7 @@ pin1_roty:
 es2bs_trx:
   readoutPriority: baseline
   description: End Station beamstop X-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES2-BS:TRX
   onFailure: retry
@@ -64,7 +64,7 @@ es2bs_trx:
 es2bs_try:
   readoutPriority: baseline
   description: End Station beamstop Y-translation
-  deviceClass: ophyd.EpicsMotor
+  deviceClass: ophyd_devices.EpicsMotorEC
   deviceConfig:
     prefix: X01DA-ES2-BS:TRY
   onFailure: retry
@@ -86,7 +86,7 @@ pilatus_curtain:
   softwareTrigger: false
 
 pilatus:
-  readoutPriority: async
+  readoutPriority: baseline
   description: Pilatus
   deviceClass: debye_bec.devices.pilatus.pilatus.Pilatus
   deviceTags:
@@ -97,12 +97,12 @@ pilatus:
   enabled: true
   softwareTrigger: true
 
-# sampl_pil:
-#   readoutPriority: baseline
-#   description: Sample to pilatus distance
-#   deviceClass: ophyd.EpicsSignalRO
-#   deviceConfig:
-#     read_pv: "X01DA-SAMPL-PIL"
-#   onFailure: retry
-#   enabled: true
-#   softwareTrigger: false
+pilatus_smpl:
+  readoutPriority: baseline
+  description: Sample to pilatus distance
+  deviceClass: ophyd.EpicsSignalRO
+  deviceConfig:
+    read_pv: "X01DA-ES2-DET:SMPLDIST"
+  onFailure: retry
+  enabled: true
+  softwareTrigger: false

debye_bec/devices/absorber.py

@@ -0,0 +1,72 @@
+"""Frontend Absorber"""
+
+from __future__ import annotations
+
+import enum
+from typing import TYPE_CHECKING
+
+from ophyd import Component as Cpt
+from ophyd import EpicsSignal, EpicsSignalRO
+from ophyd_devices import CompareStatus, DeviceStatus
+from ophyd_devices.interfaces.base_classes.psi_device_base import PSIDeviceBase
+
+if TYPE_CHECKING:
+    from bec_lib.devicemanager import ScanInfo
+
+class AbsorberError(Exception):
+    """Absorber specific exception"""
+
+class STATUS(int, enum.Enum):
+    """Absorber States"""
+
+    MOVING_CLOSE = 0
+    OPEN = 1
+    MOVING_OPEN = 2
+    CLOSED = 3
+    NOT_ENABLED = 4
+    TIMEOUT_CLOSE = 5
+    TIMEOUT_OPEN = 6
+    CLOSE_LS_LOST = 7
+    OPEN_LS_LOST = 8
+    CLOSE_LS_NOT_FREE = 9
+    OPEN_LS_NOT_FREE = 10
+    ERROR_LS = 11
+    TO_CONNECT = 12
+    MAN_OPEN = 13
+    UNDEFINED = 14
+
+class Absorber(PSIDeviceBase):
+    """Class for the Frontend Absorber"""
+
+    USER_ACCESS = ["open", "close"]
+
+    request = Cpt(EpicsSignal, suffix="REQUEST", kind="config", doc="Open/Close Absorber")
+    status = Cpt(EpicsSignalRO, suffix="STATUS", kind="normal", doc="Absorber Status")
+    status_string = Cpt(EpicsSignalRO, suffix="STATUS", kind="normal", string=True, doc="Absorber Status")
+
+    def __init__(self, *, name: str, prefix: str = "", scan_info: ScanInfo | None = None, **kwargs):
+        super().__init__(name=name, prefix=prefix, scan_info=scan_info, **kwargs)
+
+        self.timeout_for_move = 10
+        # Wait for connection on all components, ensure IOC is connected
+        self.wait_for_connection(all_signals=True, timeout=5)
+
+    def open(self) -> DeviceStatus | None:
+        """Open the Absorber"""
+        if self.status.get() == STATUS.CLOSED:
+            self.request.put(1)
+            status_open = CompareStatus(self.status, STATUS.OPEN, timeout=self.timeout_for_move)
+            status = status_open
+            return status
+        else:
+            return None
+
+    def close(self) -> DeviceStatus | None:
+        """Close the Absorber"""
+        if self.status.get() == STATUS.OPEN:
+            self.request.put(1)
+            status_close = CompareStatus(self.status, STATUS.CLOSED, timeout=self.timeout_for_move)
+            status = status_close
+            return status
+        else:
+            return None

debye_bec/devices/cameras/basler_cam.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING
 
-from ophyd import ADBase
+from ophyd import ADBase, EpicsSignalRO
 from ophyd import ADComponent as ADCpt
 from ophyd import Component as Cpt
 from ophyd_devices import PreviewSignal
@@ -20,6 +20,16 @@ if TYPE_CHECKING:  # pragma: no cover
 class BaslerCamBase(ADBase):
     """BaslerCam Base class."""
 
+    cam_detector_state_string = Cpt(EpicsSignalRO, suffix="cam1:DetectorState_RBV", string=True)
+
+    _default_configuration_attrs = [
+        'cam1.acquire_time',
+        'cam1.detector_state',
+        'cam_detector_state_string',
+        'cam1.gain',
+        'cam1.model',
+    ]
+
     cam1 = ADCpt(AravisDetectorCam, "cam1:")
     image1 = ADCpt(ImagePlugin_V35, "image1:")
 

debye_bec/devices/cameras/hutch_cam.py

@@ -0,0 +1,86 @@
+"""EH Hutch Cameras"""
+
+from __future__ import annotations
+
+import threading
+from typing import TYPE_CHECKING
+
+import cv2
+from bec_lib.file_utils import get_full_path
+from bec_lib.logger import bec_logger
+from bec_server.scan_server.scans.scan_base import ScanInfo as ScanServerScanInfo
+from ophyd_devices import DeviceStatus
+from ophyd_devices.interfaces.base_classes.psi_device_base import PSIDeviceBase
+
+from debye_bec.devices.utils.utils import fetch_scan_info
+
+if TYPE_CHECKING:  # pragma: no cover
+    from bec_lib.devicemanager import ScanInfo
+    from bec_lib.messages import ScanStatusMessage
+
+logger = bec_logger.logger
+
+CAM_USERNAME = "camera_user"
+CAM_PASSWORD = "camera_user1"
+CAM_PORT = 554
+
+
+class HutchCam(PSIDeviceBase):
+    """Class for the Hutch Cameras"""
+
+    # image = Cpt(Signal, name='image', kind='config')
+
+    def __init__(self, *, name: str, prefix: str = "", scan_info: ScanInfo | None = None, **kwargs):
+        super().__init__(name=name, scan_info=scan_info, **kwargs)
+        self.scan_parameters: ScanServerScanInfo = None
+        self.hostname = prefix
+        self.status = None
+
+    # pylint: disable=E1101
+    def on_connected(self) -> None:
+        """
+        Called after the device is connected and its signals are connected.
+        Default values for signals should be set here.
+        """
+        rtsp_url = f"rtsp://{CAM_USERNAME}:{CAM_PASSWORD}@{self.hostname}.psi.ch:{CAM_PORT}/rtpstream/config1"
+        cap = cv2.VideoCapture(f"{rtsp_url}?tcp")
+        if not cap.isOpened():
+            logger.error(self, "Connection Failed", "Could not connect to the camera stream.")
+            return
+        cap.release()
+
+    def on_stage(self) -> DeviceStatus:
+        """Called while staging the device."""
+        self.scan_parameters = fetch_scan_info(self.scan_info)
+        file_path = get_full_path(self.scan_info, name="hutch_cam_" + self.hostname).removesuffix(
+            "h5"
+        )
+
+        self.status = DeviceStatus(self)
+
+        thread = threading.Thread(
+            target=self._save_picture, args=(file_path, self.status), daemon=True
+        )
+        thread.start()
+
+        return self.status
+
+    def _save_picture(self, file_path, status):
+        try:
+            logger.info(f"Capture from camera {self.hostname}")
+            rtsp_url = f"rtsp://{CAM_USERNAME}:{CAM_PASSWORD}@{self.hostname}.psi.ch:{CAM_PORT}/rtpstream/config1"
+            cap = cv2.VideoCapture(f"{rtsp_url}?tcp")
+            if not cap.isOpened():
+                logger.error("Connection Failed", "Could not connect to the camera stream.")
+                return
+            logger.info(f"Connection to camera {self.hostname} established")
+            ret, frame = cap.readAsync()
+            cap.release()
+            if not ret:
+                logger.error("Capture Failed", "Failed to capture image from camera.")
+                return
+            cv2.imwrite(file_path + "png", frame)
+            status.set_finished()
+            logger.info(f"Capture from camera {self.hostname} done")
+        except Exception as e:
+            status.set_exception(e)

debye_bec/devices/cameras/prosilica_cam.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING
 
-from ophyd import ADBase
+from ophyd import ADBase, EpicsSignalRO
 from ophyd import ADComponent as ADCpt
 from ophyd import Component as Cpt
 from ophyd_devices import PreviewSignal
@@ -20,6 +20,16 @@ if TYPE_CHECKING:  # pragma: no cover
 class ProsilicaCamBase(ADBase):
     """Base class for Prosilica cameras."""
 
+    cam_detector_state_string = Cpt(EpicsSignalRO, suffix="cam1:DetectorState_RBV", string=True)
+
+    _default_configuration_attrs = [
+        'cam1.acquire_time',
+        'cam1.detector_state',
+        'cam_detector_state_string',
+        'cam1.gain',
+        'cam1.model',
+    ]
+
     cam1 = ADCpt(ProsilicaDetectorCam, "cam1:")
     image1 = ADCpt(ImagePlugin_V35, "image1:")
 

debye_bec/devices/ionization_chambers/ionization_chamber.py

@@ -9,8 +9,7 @@ from ophyd import Component as Cpt
 from ophyd import Device
 from ophyd import DynamicDeviceComponent as Dcpt
 from ophyd import EpicsSignal, EpicsSignalRO, EpicsSignalWithRBV
-from ophyd.status import DeviceStatus, SubscriptionStatus
-from ophyd_devices import CompareStatus, TransitionStatus
+from ophyd_devices import CompareStatus, DeviceStatus, SubscriptionStatus, TransitionStatus
 from ophyd_devices.interfaces.base_classes.psi_device_base import PSIDeviceBase
 from typeguard import typechecked
 
@@ -34,22 +33,24 @@ class EpicsSignalSplit(EpicsSignal):
 class GasMixSetupControl(Device):
     """GasMixSetup Control for Inonization Chamber 0"""
 
-    gas1_req = Cpt(EpicsSignalWithRBV, suffix="Gas1Req", kind="config", doc="Gas 1 requirement")
+    gas1_req = Cpt(EpicsSignalWithRBV, suffix="Gas1Req", kind="omitted", doc="Gas 1 requirement")
     conc1_req = Cpt(
-        EpicsSignalWithRBV, suffix="Conc1Req", kind="config", doc="Concentration 1 requirement"
+        EpicsSignalWithRBV, suffix="Conc1Req", kind="omitted", doc="Concentration 1 requirement"
     )
-    gas2_req = Cpt(EpicsSignalWithRBV, suffix="Gas2Req", kind="config", doc="Gas 2 requirement")
+    gas2_req = Cpt(EpicsSignalWithRBV, suffix="Gas2Req", kind="omitted", doc="Gas 2 requirement")
     conc2_req = Cpt(
-        EpicsSignalWithRBV, suffix="Conc2Req", kind="config", doc="Concentration 2 requirement"
+        EpicsSignalWithRBV, suffix="Conc2Req", kind="omitted", doc="Concentration 2 requirement"
     )
     press_req = Cpt(
-        EpicsSignalWithRBV, suffix="PressReq", kind="config", doc="Pressure requirement"
+        EpicsSignalWithRBV, suffix="PressReq", kind="omitted", doc="Pressure requirement"
     )
     fill = Cpt(EpicsSignal, suffix="Fill", kind="config", doc="Fill the chamber")
     status = Cpt(EpicsSignalRO, suffix="Status", kind="config", doc="Status")
     gas1 = Cpt(EpicsSignalRO, suffix="Gas1", kind="config", doc="Gas 1")
+    gas1_string = Cpt(EpicsSignalRO, suffix="Gas1", kind="config", doc="Gas 1", string=True)
     conc1 = Cpt(EpicsSignalRO, suffix="Conc1", kind="config", doc="Concentration 1")
     gas2 = Cpt(EpicsSignalRO, suffix="Gas2", kind="config", doc="Gas 2")
+    gas2_string = Cpt(EpicsSignalRO, suffix="Gas2", kind="config", doc="Gas 2", string=True)
     conc2 = Cpt(EpicsSignalRO, suffix="Conc2", kind="config", doc="Concentration 2")
     press = Cpt(EpicsSignalRO, suffix="PressTransm", kind="config", doc="Current Pressure")
 
@@ -85,10 +86,25 @@ class IonizationChamber0(PSIDeviceBase):
             (f"ES:AMP5004:cFilter{num}_ENUM"),
             {"kind": "config", "doc": f"Filter of ch{num} -> IC{num-1}"},
         ),
+        "cOnOff_string": (
+            EpicsSignal,
+            (f"ES:AMP5004.cOnOff{num}"),
+            {"kind": "config", "doc": f"Enable ch{num} -> IC{num-1}", "string": True},
+        ),
+        "cGain_ENUM_string": (
+            EpicsSignalWithRBV,
+            (f"ES:AMP5004:cGain{num}_ENUM"),
+            {"kind": "config", "doc": f"Gain of ch{num} -> IC{num-1}", "string": True},
+        ),
+        "cFilter_ENUM_string": (
+            EpicsSignalWithRBV,
+            (f"ES:AMP5004:cFilter{num}_ENUM"),
+            {"kind": "config", "doc": f"Filter of ch{num} -> IC{num-1}", "string": True},
+        ),
     }
     amp = Dcpt(amp_signals)
     gmes = Cpt(GasMixSetupControl, suffix=f"ES-GMES:IC{num-1}")
-    gmes_status = Cpt(EpicsSignalRO, suffix="ES-GMES:StatusMsg0", kind="config", doc="Status")
+    gmes_status_msg = Cpt(EpicsSignalRO, suffix="ES-GMES:StatusMsg0", kind="config", doc="Status")
     hv = Cpt(HighVoltageSuppliesControl, suffix=f"ES1-IC{num-1}:")
     hv_en_signals = {
         "ext_ena": (
@@ -105,7 +121,7 @@ class IonizationChamber0(PSIDeviceBase):
         super().__init__(name=name, prefix=prefix, scan_info=scan_info, **kwargs)
 
     @typechecked
-    def set_gain(self, gain: Literal["1e6", "1e7", "5e7", "1e8", "1e9"] | AmplifierGain) -> None:
+    def set_gain(self, gain: Literal["1e6", "1e7", "5e7", "1e8", "1e9"]) -> None:
         """Configure the gain setting of the specified channel
 
         Args:
@@ -131,10 +147,7 @@ class IonizationChamber0(PSIDeviceBase):
                 self.amp.cGain_ENUM.put(AmplifierGain.G1E9)
 
     def set_filter(
-        self,
-        value: (
-            Literal["1us", "3us", "10us", "30us", "100us", "300us", "1ms", "3ms"] | AmplifierFilter
-        ),
+        self, value: Literal["1us", "3us", "10us", "30us", "100us", "300us", "1ms", "3ms"]
     ) -> None:
         """Configure the filter setting of the specified channel
 
@@ -279,10 +292,25 @@ class IonizationChamber1(IonizationChamber0):
             (f"ES:AMP5004:cFilter{num}_ENUM"),
             {"kind": "config", "doc": f"Filter of ch{num} -> IC{num-1}"},
         ),
+        "cOnOff_string": (
+            EpicsSignal,
+            (f"ES:AMP5004.cOnOff{num}"),
+            {"kind": "config", "doc": f"Enable ch{num} -> IC{num-1}", "string": True},
+        ),
+        "cGain_ENUM_string": (
+            EpicsSignalWithRBV,
+            (f"ES:AMP5004:cGain{num}_ENUM"),
+            {"kind": "config", "doc": f"Gain of ch{num} -> IC{num-1}", "string": True},
+        ),
+        "cFilter_ENUM_string": (
+            EpicsSignalWithRBV,
+            (f"ES:AMP5004:cFilter{num}_ENUM"),
+            {"kind": "config", "doc": f"Filter of ch{num} -> IC{num-1}", "string": True},
+        ),
     }
     amp = Dcpt(amp_signals)
     gmes = Cpt(GasMixSetupControl, suffix=f"ES-GMES:IC{num-1}")
-    gmes_status = Cpt(EpicsSignalRO, suffix="ES-GMES:StatusMsg0", kind="config", doc="Status")
+    gmes_status_msg = Cpt(EpicsSignalRO, suffix="ES-GMES:StatusMsg0", kind="config", doc="Status")
     hv = Cpt(HighVoltageSuppliesControl, suffix=f"ES2-IC{num-1}:")
     hv_en_signals = {
         "ext_ena": (
@@ -315,10 +343,25 @@ class IonizationChamber2(IonizationChamber0):
             (f"ES:AMP5004:cFilter{num}_ENUM"),
             {"kind": "config", "doc": f"Filter of ch{num} -> IC{num-1}"},
         ),
+        "cOnOff_string": (
+            EpicsSignal,
+            (f"ES:AMP5004.cOnOff{num}"),
+            {"kind": "config", "doc": f"Enable ch{num} -> IC{num-1}", "string": True},
+        ),
+        "cGain_ENUM_string": (
+            EpicsSignalWithRBV,
+            (f"ES:AMP5004:cGain{num}_ENUM"),
+            {"kind": "config", "doc": f"Gain of ch{num} -> IC{num-1}", "string": True},
+        ),
+        "cFilter_ENUM_string": (
+            EpicsSignalWithRBV,
+            (f"ES:AMP5004:cFilter{num}_ENUM"),
+            {"kind": "config", "doc": f"Filter of ch{num} -> IC{num-1}", "string": True},
+        ),
     }
     amp = Dcpt(amp_signals)
     gmes = Cpt(GasMixSetupControl, suffix=f"ES-GMES:IC{num-1}")
-    gmes_status = Cpt(EpicsSignalRO, suffix="ES-GMES:StatusMsg0", kind="config", doc="Status")
+    gmes_status_msg = Cpt(EpicsSignalRO, suffix="ES-GMES:StatusMsg0", kind="config", doc="Status")
     hv = Cpt(HighVoltageSuppliesControl, suffix=f"ES2-IC{num-1}:")
     hv_en_signals = {
         "ext_ena": (
@@ -329,3 +372,63 @@ class IonizationChamber2(IonizationChamber0):
         "ena": (EpicsSignal, "ES2-IC12:HV-Ena", {"kind": "config", "doc": "Enable signal of HV"}),
     }
     hv_en = Dcpt(hv_en_signals)
+
+class Pips(IonizationChamber0):
+    """Pips, prefix should be 'X01DA-'."""
+
+    USER_ACCESS = ["set_gain", "set_filter"]
+
+    num = 4
+    amp_signals = {
+        "cOnOff": (
+            EpicsSignal,
+            (f"ES:AMP5004.cOnOff{num}"),
+            {"kind": "config", "doc": f"Enable ch{num} -> IC{num-1}"},
+        ),
+        "cGain_ENUM": (
+            EpicsSignalWithRBV,
+            (f"ES:AMP5004:cGain{num}_ENUM"),
+            {"kind": "config", "doc": f"Gain of ch{num} -> IC{num-1}"},
+        ),
+        "cFilter_ENUM": (
+            EpicsSignalWithRBV,
+            (f"ES:AMP5004:cFilter{num}_ENUM"),
+            {"kind": "config", "doc": f"Filter of ch{num} -> IC{num-1}"},
+        ),
+        "cOnOff_string": (
+            EpicsSignal,
+            (f"ES:AMP5004.cOnOff{num}"),
+            {"kind": "config", "doc": f"Enable ch{num} -> IC{num-1}", "string": True},
+        ),
+        "cGain_ENUM_string": (
+            EpicsSignalWithRBV,
+            (f"ES:AMP5004:cGain{num}_ENUM"),
+            {"kind": "config", "doc": f"Gain of ch{num} -> IC{num-1}", "string": True},
+        ),
+        "cFilter_ENUM_string": (
+            EpicsSignalWithRBV,
+            (f"ES:AMP5004:cFilter{num}_ENUM"),
+            {"kind": "config", "doc": f"Filter of ch{num} -> IC{num-1}", "string": True},
+        ),
+    }
+    amp = Dcpt(amp_signals)
+    gmes = None
+    gmes_status_msg = None
+    hv = None
+    hv_en_signals = None
+    hv_en = None
+
+    @typechecked
+    def set_hv(self, *_) -> None:
+        """Not available for the PIPS"""
+        return None
+
+    @typechecked
+    def set_grid(self, *_) -> None:
+        """Not available for the PIPS"""
+        return None
+
+    @typechecked
+    def fill(self, *_) -> None:
+        """Not available for the PIPS"""
+        return None

debye_bec/devices/mo1_bragg/mo1_bragg.py

@@ -9,16 +9,16 @@ used to ensure that the action is executed completely. This is believed
 to allow for a more stable execution of the action."""
 
 import time
-from typing import Any, Literal
+from typing import Literal
 
 from bec_lib.devicemanager import ScanInfo
 from bec_lib.logger import bec_logger
+from bec_server.scan_server.scans.scan_base import ScanInfo as ScanServerScanInfo
 from ophyd import Component as Cpt
-from ophyd import DeviceStatus, Signal, StatusBase
-from ophyd.status import SubscriptionStatus, WaitTimeoutError
+from ophyd import DeviceStatus, StatusBase
+from ophyd.status import WaitTimeoutError
 from ophyd_devices import CompareStatus, ProgressSignal, TransitionStatus
 from ophyd_devices.interfaces.base_classes.psi_device_base import PSIDeviceBase
-from ophyd_devices.utils.errors import DeviceStopError
 from pydantic import BaseModel, Field
 from typeguard import typechecked
 
@@ -34,6 +34,7 @@ from debye_bec.devices.mo1_bragg.mo1_bragg_enums import (
     TriggerControlSource,
 )
 from debye_bec.devices.mo1_bragg.mo1_bragg_utils import compute_spline
+from debye_bec.devices.utils.utils import fetch_scan_info
 
 # Initialise logger
 logger = bec_logger.logger
@@ -45,36 +46,6 @@ class Mo1BraggError(Exception):
     """Exception for the Mo1 Bragg positioner"""
 
 
-########## Scan Parameter Model  ##########
-
-
-class ScanParameter(BaseModel):
-    """Dataclass to store the scan parameters for the Mo1 Bragg positioner.
-    This needs to be in sync with the kwargs of the MO1 Bragg scans from Debye, to
-    ensure that the scan parameters are correctly set. Any changes in the scan kwargs,
-    i.e. renaming or adding new parameters, need to be represented here as well."""
-
-    scan_time: float | None = Field(None, description="Scan time for a half oscillation")
-    scan_duration: float | None = Field(None, description="Duration of the scan")
-    break_enable_low: bool | None = Field(
-        None, description="Break enabled for low, should be PV trig_ena_lo_enum"
-    )  # trig_enable_low: bool = None
-    break_enable_high: bool | None = Field(
-        None, description="Break enabled for high, should be PV trig_ena_hi_enum"
-    )  # trig_enable_high: bool = None
-    break_time_low: float | None = Field(None, description="Break time low energy/angle")
-    break_time_high: float | None = Field(None, description="Break time high energy/angle")
-    cycle_low: int | None = Field(None, description="Cycle for low energy/angle")
-    cycle_high: int | None = Field(None, description="Cycle for high energy/angle")
-    exp_time: float | None = Field(None, description="XRD trigger period")
-    n_of_trigger: int | None = Field(None, description="Amount of XRD triggers")
-    start: float | None = Field(None, description="Start value for energy/angle")
-    stop: float | None = Field(None, description="Stop value for energy/angle")
-    p_kink: float | None = Field(None, description="P Kink")
-    e_kink: float | None = Field(None, description="Energy Kink")
-    model_config: dict = {"validate_assignment": True}
-
-
 ########### Mo1 Bragg Motor Class ###########
 
 
@@ -86,7 +57,7 @@ class Mo1Bragg(PSIDeviceBase, Mo1BraggPositioner):
 
     progress_signal = Cpt(ProgressSignal, name="progress_signal")
 
-    USER_ACCESS = ["set_advanced_xas_settings", "set_xtal"]
+    USER_ACCESS = ["set_advanced_xas_settings", "set_xtal", "convert_angle_energy"]
 
     def __init__(self, name: str, prefix: str = "", scan_info: ScanInfo | None = None, **kwargs):  # type: ignore
         """
@@ -97,8 +68,15 @@ class Mo1Bragg(PSIDeviceBase, Mo1BraggPositioner):
             scan_info (ScanInfo): The scan info to use.
         """
         super().__init__(name=name, scan_info=scan_info, prefix=prefix, **kwargs)
-        self.scan_parameter = ScanParameter()
+        self.scan_parameters: ScanServerScanInfo = None
         self.timeout_for_pvwait = 7.5
+        self.valid_scan_names = [
+            "xas_simple_scan",
+            "xas_simple_scan_with_xrd",
+            "xas_advanced_scan",
+            "xas_advanced_scan_with_xrd",
+            "nidaq_continuous_scan",
+        ]
 
     ########################################
     #  Beamline Specific Implementations   #
@@ -125,94 +103,187 @@ class Mo1Bragg(PSIDeviceBase, Mo1BraggPositioner):
 
         Information about the upcoming scan can be accessed from the scan_info (self.scan_info.msg) object.
         """
+        self.scan_parameters = fetch_scan_info(self.scan_info)
         if self.scan_control.scan_msg.get() != ScanControlLoadMessage.PENDING:
             status = CompareStatus(self.scan_control.scan_msg, ScanControlLoadMessage.PENDING)
             self.cancel_on_stop(status)
             self.scan_control.scan_val_reset.put(1)
             status.wait(timeout=self.timeout_for_pvwait)
 
-        scan_name = self.scan_info.msg.scan_name
-        self._update_scan_parameter()
-        if scan_name == "xas_simple_scan":
-            self.set_xas_settings(
-                low=self.scan_parameter.start,
-                high=self.scan_parameter.stop,
-                scan_time=self.scan_parameter.scan_time,
-            )
-            self.set_trig_settings(
-                enable_low=False,
-                enable_high=False,
-                break_time_low=0,
-                break_time_high=0,
-                cycle_low=0,
-                cycle_high=0,
-                exp_time=0,
-                n_of_trigger=0,
-            )
-            self.set_scan_control_settings(
-                mode=ScanControlMode.SIMPLE, scan_duration=self.scan_parameter.scan_duration
-            )
-        elif scan_name == "xas_simple_scan_with_xrd":
-            self.set_xas_settings(
-                low=self.scan_parameter.start,
-                high=self.scan_parameter.stop,
-                scan_time=self.scan_parameter.scan_time,
-            )
-            self.set_trig_settings(
-                enable_low=self.scan_parameter.break_enable_low,
-                enable_high=self.scan_parameter.break_enable_high,
-                break_time_low=self.scan_parameter.break_time_low,
-                break_time_high=self.scan_parameter.break_time_high,
-                cycle_low=self.scan_parameter.cycle_low,
-                cycle_high=self.scan_parameter.cycle_high,
-                exp_time=self.scan_parameter.exp_time,
-                n_of_trigger=self.scan_parameter.n_of_trigger,
-            )
-            self.set_scan_control_settings(
-                mode=ScanControlMode.SIMPLE, scan_duration=self.scan_parameter.scan_duration
-            )
-        elif scan_name == "xas_advanced_scan":
-            self.set_advanced_xas_settings(
-                low=self.scan_parameter.start,
-                high=self.scan_parameter.stop,
-                scan_time=self.scan_parameter.scan_time,
-                p_kink=self.scan_parameter.p_kink,
-                e_kink=self.scan_parameter.e_kink,
-            )
-            self.set_trig_settings(
-                enable_low=False,
-                enable_high=False,
-                break_time_low=0,
-                break_time_high=0,
-                cycle_low=0,
-                cycle_high=0,
-                exp_time=0,
-                n_of_trigger=0,
-            )
-            self.set_scan_control_settings(
-                mode=ScanControlMode.ADVANCED, scan_duration=self.scan_parameter.scan_duration
-            )
-        elif scan_name == "xas_advanced_scan_with_xrd":
-            self.set_advanced_xas_settings(
-                low=self.scan_parameter.start,
-                high=self.scan_parameter.stop,
-                scan_time=self.scan_parameter.scan_time,
-                p_kink=self.scan_parameter.p_kink,
-                e_kink=self.scan_parameter.e_kink,
-            )
-            self.set_trig_settings(
-                enable_low=self.scan_parameter.break_enable_low,
-                enable_high=self.scan_parameter.break_enable_high,
-                break_time_low=self.scan_parameter.break_time_low,
-                break_time_high=self.scan_parameter.break_time_high,
-                cycle_low=self.scan_parameter.cycle_low,
-                cycle_high=self.scan_parameter.cycle_high,
-                exp_time=self.scan_parameter.exp_time,
-                n_of_trigger=self.scan_parameter.n_of_trigger,
-            )
-            self.set_scan_control_settings(
-                mode=ScanControlMode.ADVANCED, scan_duration=self.scan_parameter.scan_duration
+        scan_name = self.scan_parameters.scan_name
+        if self._check_if_scan_name_is_valid(self.scan_parameters):
+            if self.scan_parameters.positions is not None:
+                start, stop = (
+                    self.scan_parameters.positions
+                    if len(self.scan_parameters.positions) == 2
+                    else (None, None)
+                )
+            else:
+                start, stop = (None, None)
+            scan_time = self.scan_parameters.additional_scan_parameters.get("scan_time", None)
+            scan_duration = self.scan_parameters.additional_scan_parameters.get(
+                "scan_duration", None
             )
+            if scan_name == "xas_simple_scan":
+                if any(param is None for param in [start, stop, scan_time, scan_duration]):
+                    raise Mo1BraggError(
+                        f"Missing scan parameters for xas_simple_scan. Required parameters: start, stop, scan_time, scan_duration in additional_scan_parameters dict {self.scan_parameters.additional_scan_parameters}"
+                    )
+                self.set_xas_settings(low=start, high=stop, scan_time=scan_time)
+                self.set_trig_settings(
+                    enable_low=False,
+                    enable_high=False,
+                    break_time_low=0,
+                    break_time_high=0,
+                    cycle_low=0,
+                    cycle_high=0,
+                    exp_time=0,
+                    n_of_trigger=0,
+                )
+                self.set_scan_control_settings(
+                    mode=ScanControlMode.SIMPLE, scan_duration=scan_duration
+                )
+            elif scan_name == "xas_simple_scan_with_xrd":
+                break_enable_low = self.scan_parameters.additional_scan_parameters.get(
+                    "break_enable_low", None
+                )
+                break_enable_high = self.scan_parameters.additional_scan_parameters.get(
+                    "break_enable_high", None
+                )
+                break_time_low = self.scan_parameters.additional_scan_parameters.get(
+                    "break_time_low", None
+                )
+                break_time_high = self.scan_parameters.additional_scan_parameters.get(
+                    "break_time_high", None
+                )
+                cycle_low = self.scan_parameters.additional_scan_parameters.get("cycle_low", None)
+                cycle_high = self.scan_parameters.additional_scan_parameters.get("cycle_high", None)
+                exp_time = self.scan_parameters.exp_time
+                n_of_trigger = self.scan_parameters.additional_scan_parameters.get(
+                    "n_of_trigger", None
+                )
+                if any(
+                    param is None
+                    for param in [
+                        start,
+                        stop,
+                        scan_time,
+                        scan_duration,
+                        break_enable_low,
+                        break_enable_high,
+                        break_time_low,
+                        break_time_high,
+                        cycle_low,
+                        cycle_high,
+                        exp_time,
+                        n_of_trigger,
+                    ]
+                ):
+                    raise Mo1BraggError(
+                        f"Missing scan parameters for xas_simple_scan_with_xrd. Required parameters: start, stop, scan_time, scan_duration, break_enable_low, break_enable_high, break_time_low, break_time_high, cycle_low, cycle_high, exp_time, n_of_trigger in additional_scan_parameters dict {self.scan_parameters.additional_scan_parameters}"
+                    )
+                self.set_xas_settings(low=start, high=stop, scan_time=scan_time)
+                self.set_trig_settings(
+                    enable_low=break_enable_low,
+                    enable_high=break_enable_high,
+                    break_time_low=break_time_low,
+                    break_time_high=break_time_high,
+                    cycle_low=cycle_low,
+                    cycle_high=cycle_high,
+                    exp_time=exp_time,
+                    n_of_trigger=n_of_trigger,
+                )
+                self.set_scan_control_settings(
+                    mode=ScanControlMode.SIMPLE, scan_duration=scan_duration
+                )
+            elif scan_name == "xas_advanced_scan":
+                p_kink = self.scan_parameters.additional_scan_parameters.get("p_kink", None)
+                e_kink = self.scan_parameters.additional_scan_parameters.get("e_kink", None)
+                if any(
+                    param is None
+                    for param in [start, stop, scan_time, scan_duration, p_kink, e_kink]
+                ):
+                    raise Mo1BraggError(
+                        f"Missing scan parameters for xas_advanced_scan. Required parameters: start, stop, scan_time, scan_duration, p_kink, e_kink in additional_scan_parameters dict {self.scan_parameters.additional_scan_parameters}"
+                    )
+                self.set_advanced_xas_settings(
+                    low=start, high=stop, scan_time=scan_time, p_kink=p_kink, e_kink=e_kink
+                )
+                self.set_trig_settings(
+                    enable_low=False,
+                    enable_high=False,
+                    break_time_low=0,
+                    break_time_high=0,
+                    cycle_low=0,
+                    cycle_high=0,
+                    exp_time=0,
+                    n_of_trigger=0,
+                )
+                self.set_scan_control_settings(
+                    mode=ScanControlMode.ADVANCED, scan_duration=scan_duration
+                )
+            elif scan_name == "xas_advanced_scan_with_xrd":
+                p_kink = self.scan_parameters.additional_scan_parameters.get("p_kink", None)
+                e_kink = self.scan_parameters.additional_scan_parameters.get("e_kink", None)
+                break_enable_low = self.scan_parameters.additional_scan_parameters.get(
+                    "break_enable_low", None
+                )
+                break_enable_high = self.scan_parameters.additional_scan_parameters.get(
+                    "break_enable_high", None
+                )
+                break_time_low = self.scan_parameters.additional_scan_parameters.get(
+                    "break_time_low", None
+                )
+                break_time_high = self.scan_parameters.additional_scan_parameters.get(
+                    "break_time_high", None
+                )
+                cycle_low = self.scan_parameters.additional_scan_parameters.get("cycle_low", None)
+                cycle_high = self.scan_parameters.additional_scan_parameters.get("cycle_high", None)
+                exp_time = self.scan_parameters.exp_time
+                n_of_trigger = self.scan_parameters.additional_scan_parameters.get(
+                    "n_of_trigger", None
+                )
+                if any(
+                    param is None
+                    for param in [
+                        start,
+                        stop,
+                        scan_time,
+                        scan_duration,
+                        p_kink,
+                        e_kink,
+                        break_enable_low,
+                        break_enable_high,
+                        break_time_low,
+                        break_time_high,
+                        cycle_low,
+                        cycle_high,
+                        exp_time,
+                        n_of_trigger,
+                    ]
+                ):
+                    raise Mo1BraggError(
+                        f"Missing scan parameters for xas_advanced_scan_with_xrd. Required parameters: start, stop, scan_time, scan_duration, p_kink, e_kink, break_enable_low, break_enable_high, break_time_low, break_time_high, cycle_low, cycle_high, exp_time, n_of_trigger in additional_scan_parameters dict {self.scan_parameters.additional_scan_parameters}"
+                    )
+
+                self.set_advanced_xas_settings(
+                    low=start, high=stop, scan_time=scan_time, p_kink=p_kink, e_kink=e_kink
+                )
+                self.set_trig_settings(
+                    enable_low=break_enable_low,
+                    enable_high=break_enable_high,
+                    break_time_low=break_time_low,
+                    break_time_high=break_time_high,
+                    cycle_low=cycle_low,
+                    cycle_high=cycle_high,
+                    exp_time=exp_time,
+                    n_of_trigger=n_of_trigger,
+                )
+                self.set_scan_control_settings(
+                    mode=ScanControlMode.ADVANCED, scan_duration=scan_duration
+                )
+            else:
+                return  # Should never happen.
         else:
             return
         # Setting scan duration seems to lag behind slightly in the backend, include small sleep
@@ -281,7 +352,7 @@ class Mo1Bragg(PSIDeviceBase, Mo1BraggPositioner):
             self.scan_control.scan_status,
             transitions=[ScanControlScanStatus.READY, ScanControlScanStatus.RUNNING],
             strict=True,
-            raise_states=[ScanControlScanStatus.PARAMETER_WRONG],
+            failure_states=[ScanControlScanStatus.PARAMETER_WRONG],
         )
         self.cancel_on_stop(status)
         start_func(1)
@@ -292,6 +363,13 @@ class Mo1Bragg(PSIDeviceBase, Mo1BraggPositioner):
         self.stopped = True  # Needs to be set to stop motion
 
     ######### Utility Methods #########
+
+    def _check_if_scan_name_is_valid(self, scan_parameters: ScanServerScanInfo) -> bool:
+        """Check if the scan is within the list of scans for which the backend is working"""
+        if scan_parameters.scan_name in self.valid_scan_names:
+            return True
+        return False
+
     def _progress_update(self, value, **kwargs) -> None:
         """Callback method to update the scan progress, runs a callback
         to SUB_PROGRESS subscribers, i.e. BEC.
@@ -338,25 +416,31 @@ class Mo1Bragg(PSIDeviceBase, Mo1BraggPositioner):
         Returns:
             output (float): Converted angle or energy
         """
+
         self.calculator.calc_reset.put(0)
         self.calculator.calc_reset.put(1)
         status = CompareStatus(self.calculator.calc_done, 0)
         self.cancel_on_stop(status)
         status.wait(self.timeout_for_pvwait)
+        self.calculator.calc_reset.put(0)
 
         if mode == "AngleToEnergy":
-            self.calculator.calc_angle.put(inp)
+            in_signal = self.calculator.calc_angle
+            out_signal = self.calculator.calc_energy
         elif mode == "EnergyToAngle":
-            self.calculator.calc_energy.put(inp)
+            in_signal = self.calculator.calc_energy
+            out_signal = self.calculator.calc_angle
+        else:
+            raise Mo1BraggError(f'Unknown mode {mode}')
 
+        in_signal.put(inp)
         status = CompareStatus(self.calculator.calc_done, 1)
         self.cancel_on_stop(status)
         status.wait(self.timeout_for_pvwait)
-        time.sleep(0.25)  # TODO needed still? Needed due to update frequency of softIOC
-        if mode == "AngleToEnergy":
-            return self.calculator.calc_energy.get()
-        elif mode == "EnergyToAngle":
-            return self.calculator.calc_angle.get()
+        status = CompareStatus(out_signal, 0, operation_success='>')
+        self.cancel_on_stop(status)
+        status.wait(self.timeout_for_pvwait)
+        return out_signal.get()
 
     def set_advanced_xas_settings(
         self, low: float, high: float, scan_time: float, p_kink: float, e_kink: float
@@ -469,12 +553,3 @@ class Mo1Bragg(PSIDeviceBase, Mo1BraggPositioner):
 
         for s in status_list:
             s.wait(timeout=self.timeout_for_pvwait)
-
-    def _update_scan_parameter(self):
-        """Get the scan_info parameters for the scan."""
-        for key, value in self.scan_info.msg.request_inputs["inputs"].items():
-            if hasattr(self.scan_parameter, key):
-                setattr(self.scan_parameter, key, value)
-        for key, value in self.scan_info.msg.request_inputs["kwargs"].items():
-            if hasattr(self.scan_parameter, key):
-                setattr(self.scan_parameter, key, value)

debye_bec/devices/mo1_bragg/mo1_bragg_devices.py

@@ -76,8 +76,14 @@ class Mo1BraggEncoder(Device):
 class Mo1BraggCrystal(Device):
     """Mo1 Bragg PVs to set the crystal parameters"""
 
-    offset_si111 = Cpt(EpicsSignalWithRBV, suffix="offset_si111", kind="config")
-    offset_si311 = Cpt(EpicsSignalWithRBV, suffix="offset_si311", kind="config")
+    bragg_off_si111 = Cpt(EpicsSignalWithRBV, suffix="bragg_off_si111", kind="config")
+    bragg_off_si311 = Cpt(EpicsSignalWithRBV, suffix="bragg_off_si311", kind="config")
+    phi_off_si111 = Cpt(EpicsSignalWithRBV, suffix="phi_off_si111", kind="config")
+    phi_off_si311 = Cpt(EpicsSignalWithRBV, suffix="phi_off_si311", kind="config")
+    azm_off_si111 = Cpt(EpicsSignalWithRBV, suffix="azm_off_si111", kind="config")
+    azm_off_si311 = Cpt(EpicsSignalWithRBV, suffix="azm_off_si311", kind="config")
+    miscut_si111 = Cpt(EpicsSignalWithRBV, suffix="miscut_si111", kind="config")
+    miscut_si311 = Cpt(EpicsSignalWithRBV, suffix="miscut_si311", kind="config")
     xtal_enum = Cpt(EpicsSignalWithRBV, suffix="xtal_ENUM", kind="config")
     d_spacing_si111 = Cpt(EpicsSignalWithRBV, suffix="d_spacing_si111", kind="config")
     d_spacing_si311 = Cpt(EpicsSignalWithRBV, suffix="d_spacing_si311", kind="config")
@@ -85,13 +91,21 @@ class Mo1BraggCrystal(Device):
     current_d_spacing = Cpt(
         EpicsSignalRO, suffix="current_d_spacing_RBV", kind="normal", auto_monitor=True
     )
-    current_offset = Cpt(
-        EpicsSignalRO, suffix="current_offset_RBV", kind="normal", auto_monitor=True
+    current_bragg_off = Cpt(
+        EpicsSignalRO, suffix="current_bragg_off_RBV", kind="normal", auto_monitor=True
+    )
+    current_phi_off = Cpt(
+        EpicsSignalRO, suffix="current_phi_off_RBV", kind="normal", auto_monitor=True
+    )
+    current_azm_off = Cpt(
+        EpicsSignalRO, suffix="current_azm_off_RBV", kind="normal", auto_monitor=True
+    )
+    current_miscut = Cpt(
+        EpicsSignalRO, suffix="current_miscut_RBV", kind="normal", auto_monitor=True
     )
     current_xtal = Cpt(
         EpicsSignalRO, suffix="current_xtal_ENUM_RBV", kind="normal", auto_monitor=True
     )
-
     current_xtal_string = Cpt(
         EpicsSignalRO, suffix="current_xtal_ENUM_RBV", kind="normal", auto_monitor=True, string=True
     )
@@ -168,7 +182,7 @@ class Mo1TriggerSettings(Device):
 class Mo1BraggCalculator(Device):
     """Mo1 Bragg PVs to convert angle to energy or vice-versa."""
 
-    calc_reset = Cpt(EpicsSignal, suffix="calc_reset", kind="config", put_complete=True)
+    calc_reset = Cpt(EpicsSignalWithRBV, suffix="calc_reset", kind="config", put_complete=True)
     calc_done = Cpt(EpicsSignalRO, suffix="calc_done_RBV", kind="config")
     calc_energy = Cpt(EpicsSignalWithRBV, suffix="calc_energy", kind="config")
     calc_angle = Cpt(EpicsSignalWithRBV, suffix="calc_angle", kind="config")
@@ -240,6 +254,8 @@ class Mo1BraggPositioner(Device, PositionerBase):
     high_lim = Cpt(EpicsSignalRO, suffix="hi_lim_pos_energy_RBV", kind="config", auto_monitor=True)
     velocity = Cpt(EpicsSignalWithRBV, suffix="move_velocity", kind="config", auto_monitor=True)
 
+    angle = Cpt(EpicsSignalRO, suffix="feedback_pos_angle_RBV", kind="normal", auto_monitor=True)
+
     ########## Move Command PVs   ##########
 
     move_abs = Cpt(EpicsSignal, suffix="move_abs", kind="config", put_complete=True)
@@ -392,8 +408,8 @@ class Mo1BraggPositioner(Device, PositionerBase):
     def set_xtal(
         self,
         xtal_enum: Literal["111", "311"],
-        offset_si111: float = None,
-        offset_si311: float = None,
+        bragg_off_si111: float = None,
+        bragg_off_si311: float = None,
         d_spacing_si111: float = None,
         d_spacing_si311: float = None,
     ) -> None:
@@ -401,15 +417,15 @@ class Mo1BraggPositioner(Device, PositionerBase):
 
         Args:
             xtal_enum (Literal["111", "311"]) : Enum to set the crystal orientation
-            offset_si111 (float) : Offset for the 111 crystal
-            offset_si311 (float) : Offset for the 311 crystal
+            bragg_off_si111 (float) : Offset for the 111 crystal
+            bragg_off_si311 (float) : Offset for the 311 crystal
             d_spacing_si111 (float) : d-spacing for the 111 crystal
             d_spacing_si311 (float) : d-spacing for the 311 crystal
         """
-        if offset_si111 is not None:
-            self.crystal.offset_si111.put(offset_si111)
-        if offset_si311 is not None:
-            self.crystal.offset_si311.put(offset_si311)
+        if bragg_off_si111 is not None:
+            self.crystal.bragg_off_si111.put(bragg_off_si111)
+        if bragg_off_si311 is not None:
+            self.crystal.bragg_off_si311.put(bragg_off_si311)
         if d_spacing_si111 is not None:
             self.crystal.d_spacing_si111.put(d_spacing_si111)
         if d_spacing_si311 is not None:

debye_bec/devices/nidaq/nidaq.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
 
-import time
-from typing import TYPE_CHECKING, Literal, cast
+from typing import TYPE_CHECKING, Literal
 
 from bec_lib.logger import bec_logger
+from bec_server.scan_server.scans.scan_base import ScanInfo as ScanServerScanInfo
 from ophyd import Component as Cpt
 from ophyd import Device, DeviceStatus, EpicsSignal, EpicsSignalRO, Kind, StatusBase
-from ophyd.status import SubscriptionStatus, WaitTimeoutError
+from ophyd.status import WaitTimeoutError
 from ophyd_devices import CompareStatus, ProgressSignal, TransitionStatus
 from ophyd_devices.interfaces.base_classes.psi_device_base import PSIDeviceBase
 from ophyd_devices.sim.sim_signals import SetableSignal
@@ -19,6 +19,7 @@ from debye_bec.devices.nidaq.nidaq_enums import (
     ScanRates,
     ScanType,
 )
+from debye_bec.devices.utils.utils import fetch_scan_info
 
 if TYPE_CHECKING:  # pragma: no cover
     from bec_lib.devicemanager import ScanInfo
@@ -33,280 +34,23 @@ class NidaqError(Exception):
 class NidaqControl(Device):
     """Nidaq control class with all PVs"""
 
+    # fmt: off
     ### Readback PVs for EpicsEmitter ###
-    ai0 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-AI0",
-        kind=Kind.normal,
-        doc="EPICS analog input 0",
-        auto_monitor=True,
-    )
-    ai1 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-AI1",
-        kind=Kind.normal,
-        doc="EPICS analog input 1",
-        auto_monitor=True,
-    )
-    ai2 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-AI2",
-        kind=Kind.normal,
-        doc="EPICS analog input 2",
-        auto_monitor=True,
-    )
-    ai3 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-AI3",
-        kind=Kind.normal,
-        doc="EPICS analog input 3",
-        auto_monitor=True,
-    )
-    ai4 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-AI4",
-        kind=Kind.normal,
-        doc="EPICS analog input 4",
-        auto_monitor=True,
-    )
-    ai5 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-AI5",
-        kind=Kind.normal,
-        doc="EPICS analog input 5",
-        auto_monitor=True,
-    )
-    ai6 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-AI6",
-        kind=Kind.normal,
-        doc="EPICS analog input 6",
-        auto_monitor=True,
-    )
-    ai7 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-AI7",
-        kind=Kind.normal,
-        doc="EPICS analog input 7",
-        auto_monitor=True,
-    )
+    energy = Cpt(SetableSignal, value=0, kind=Kind.normal)
 
-    ci0 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-CI0",
-        kind=Kind.normal,
-        doc="EPICS counter input 0",
-        auto_monitor=True,
-    )
-    ci1 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-CI1",
-        kind=Kind.normal,
-        doc="EPICS counter input 1",
-        auto_monitor=True,
-    )
-    ci2 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-CI2",
-        kind=Kind.normal,
-        doc="EPICS counter input 2",
-        auto_monitor=True,
-    )
-    ci3 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-CI3",
-        kind=Kind.normal,
-        doc="EPICS counter input 3",
-        auto_monitor=True,
-    )
-    ci4 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-CI4",
-        kind=Kind.normal,
-        doc="EPICS counter input 4",
-        auto_monitor=True,
-    )
-    ci5 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-CI5",
-        kind=Kind.normal,
-        doc="EPICS counter input 5",
-        auto_monitor=True,
-    )
-    ci6 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-CI6",
-        kind=Kind.normal,
-        doc="EPICS counter input 6",
-        auto_monitor=True,
-    )
-    ci7 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-CI7",
-        kind=Kind.normal,
-        doc="EPICS counter input 7",
-        auto_monitor=True,
-    )
+    smpl_abs = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream sample absorption")
+    smpl_fluo = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream sample fluorescence")
+    ref_abs = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream reference absorption")
+    cisum = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter sum")
 
-    di0 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-DI0",
-        kind=Kind.normal,
-        doc="EPICS digital input 0",
-        auto_monitor=True,
-    )
-    di1 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-DI1",
-        kind=Kind.normal,
-        doc="EPICS digital input 1",
-        auto_monitor=True,
-    )
-    di2 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-DI2",
-        kind=Kind.normal,
-        doc="EPICS digital input 2",
-        auto_monitor=True,
-    )
-    di3 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-DI3",
-        kind=Kind.normal,
-        doc="EPICS digital input 3",
-        auto_monitor=True,
-    )
-    di4 = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-DI4",
-        kind=Kind.normal,
-        doc="EPICS digital input 4",
-        auto_monitor=True,
-    )
-
-    enc_epics = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-ENC",
-        kind=Kind.normal,
-        doc="EPICS Encoder reading",
-        auto_monitor=True,
-    )
-
-    energy_epics = Cpt(
-        EpicsSignalRO,
-        suffix="NIDAQ-ENERGY",
-        kind=Kind.normal,
-        doc="EPICS Energy reading",
-        auto_monitor=True,
-    )
-
-    ### Readback for BEC emitter ###
-
-    ai0_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 0, MEAN"
-    )
-    ai1_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 1, MEAN"
-    )
-    ai2_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 2, MEAN"
-    )
-    ai3_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 3, MEAN"
-    )
-    ai4_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 4, MEAN"
-    )
-    ai5_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 5, MEAN"
-    )
-    ai6_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 6, MEAN"
-    )
-    ai7_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 7, MEAN"
-    )
-
-    ai0_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 0, STD"
-    )
-    ai1_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 1, STD"
-    )
-    ai2_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 2, STD"
-    )
-    ai3_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 3, STD"
-    )
-    ai4_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 4, STD"
-    )
-    ai5_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 5, STD"
-    )
-    ai6_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 6, STD"
-    )
-    ai7_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 7, STD"
-    )
-
-    ci0_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 0, MEAN"
-    )
-    ci1_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 1, MEAN"
-    )
-    ci2_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 2, MEAN"
-    )
-    ci3_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 3, MEAN"
-    )
-    ci4_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 4, MEAN"
-    )
-    ci5_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 5, MEAN"
-    )
-    ci6_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 6, MEAN"
-    )
-    ci7_mean = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 7, MEAN"
-    )
-
-    ci0_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 0. STD"
-    )
-    ci1_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 1. STD"
-    )
-    ci2_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 2. STD"
-    )
-    ci3_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 3. STD"
-    )
-    ci4_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 4. STD"
-    )
-    ci5_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 5. STD"
-    )
-    ci6_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 6. STD"
-    )
-    ci7_std_dev = Cpt(
-        SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 7. STD"
-    )
-
-    xas_timestamp = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream XAS timestamp")
-
-    xrd_timestamp = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream XRD timestamp")
-
-    xrd_energy = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream XRD energy")
+    ai0_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 0, MEAN")
+    ai1_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 1, MEAN")
+    ai2_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 2, MEAN")
+    ai3_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 3, MEAN")
+    ai4_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 4, MEAN")
+    ai5_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 5, MEAN")
+    ai6_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 6, MEAN")
+    ai7_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 7, MEAN")
 
     di0_max = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream digital input 0, MAX")
     di1_max = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream digital input 1, MAX")
@@ -314,31 +58,152 @@ class NidaqControl(Device):
     di3_max = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream digital input 3, MAX")
     di4_max = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream digital input 4, MAX")
 
+    ci0_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 0, MEAN")
+    ci1_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 1, MEAN")
+    ci2_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 2, MEAN")
+    ci3_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 3, MEAN")
+    ci4_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 4, MEAN")
+    ci5_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 5, MEAN")
+    ci6_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 6, MEAN")
+    ci7_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 7, MEAN")
+    ci8_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 8, MEAN")
+    ci9_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 9, MEAN")
+    ci10_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 10, MEAN")
+    ci11_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 11, MEAN")
+    ci12_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 12, MEAN")
+    ci13_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 13, MEAN")
+    ci14_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 14, MEAN")
+    ci15_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 15, MEAN")
+    ci16_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 16, MEAN")
+    ci17_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 17, MEAN")
+
+    ai0 = Cpt(EpicsSignalRO, suffix="NIDAQ-AI0", kind=Kind.normal, doc="EPICS analog input 0", auto_monitor=True)
+    ai1 = Cpt(EpicsSignalRO, suffix="NIDAQ-AI1", kind=Kind.normal, doc="EPICS analog input 1", auto_monitor=True)
+    ai2 = Cpt(EpicsSignalRO, suffix="NIDAQ-AI2", kind=Kind.normal, doc="EPICS analog input 2", auto_monitor=True)
+    ai3 = Cpt(EpicsSignalRO, suffix="NIDAQ-AI3", kind=Kind.normal, doc="EPICS analog input 3", auto_monitor=True)
+    ai4 = Cpt(EpicsSignalRO, suffix="NIDAQ-AI4", kind=Kind.normal, doc="EPICS analog input 4", auto_monitor=True)
+    ai5 = Cpt(EpicsSignalRO, suffix="NIDAQ-AI5", kind=Kind.normal, doc="EPICS analog input 5", auto_monitor=True)
+    ai6 = Cpt(EpicsSignalRO, suffix="NIDAQ-AI6", kind=Kind.normal, doc="EPICS analog input 6", auto_monitor=True)
+    ai7 = Cpt(EpicsSignalRO, suffix="NIDAQ-AI7", kind=Kind.normal, doc="EPICS analog input 7", auto_monitor=True)
+
+    ci0 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI0", kind=Kind.normal, doc="EPICS counter input 0", auto_monitor=True)
+    ci1 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI1", kind=Kind.normal, doc="EPICS counter input 1", auto_monitor=True)
+    ci2 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI2", kind=Kind.normal, doc="EPICS counter input 2", auto_monitor=True)
+    ci3 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI3", kind=Kind.normal, doc="EPICS counter input 3", auto_monitor=True)
+    ci4 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI4", kind=Kind.normal, doc="EPICS counter input 4", auto_monitor=True)
+    ci5 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI5", kind=Kind.normal, doc="EPICS counter input 5", auto_monitor=True)
+    ci6 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI6", kind=Kind.normal, doc="EPICS counter input 6", auto_monitor=True)
+    ci7 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI7", kind=Kind.normal, doc="EPICS counter input 7", auto_monitor=True)
+    ci8 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI8", kind=Kind.normal, doc="EPICS counter input 8", auto_monitor=True)
+    ci9 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI9", kind=Kind.normal, doc="EPICS counter input 9", auto_monitor=True)
+    ci10 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI10", kind=Kind.normal, doc="EPICS counter input 0", auto_monitor=True)
+    ci11 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI11", kind=Kind.normal, doc="EPICS counter input 1", auto_monitor=True)
+    ci12 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI12", kind=Kind.normal, doc="EPICS counter input 2", auto_monitor=True)
+    ci13 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI13", kind=Kind.normal, doc="EPICS counter input 3", auto_monitor=True)
+    ci14 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI14", kind=Kind.normal, doc="EPICS counter input 4", auto_monitor=True)
+    ci15 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI15", kind=Kind.normal, doc="EPICS counter input 5", auto_monitor=True)
+    ci16 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI16", kind=Kind.normal, doc="EPICS counter input 6", auto_monitor=True)
+    ci17 = Cpt(EpicsSignalRO, suffix="NIDAQ-CI17", kind=Kind.normal, doc="EPICS counter input 7", auto_monitor=True)
+
+    di0 = Cpt(EpicsSignalRO, suffix="NIDAQ-DI0", kind=Kind.normal, doc="EPICS digital input 0", auto_monitor=True)
+    di1 = Cpt(EpicsSignalRO, suffix="NIDAQ-DI1", kind=Kind.normal, doc="EPICS digital input 1", auto_monitor=True)
+    di2 = Cpt(EpicsSignalRO, suffix="NIDAQ-DI2", kind=Kind.normal, doc="EPICS digital input 2", auto_monitor=True)
+    di3 = Cpt(EpicsSignalRO, suffix="NIDAQ-DI3", kind=Kind.normal, doc="EPICS digital input 3", auto_monitor=True)
+    di4 = Cpt(EpicsSignalRO, suffix="NIDAQ-DI4", kind=Kind.normal, doc="EPICS digital input 4", auto_monitor=True)
+
+    enc_epics = Cpt(EpicsSignalRO, suffix="NIDAQ-ENC", kind=Kind.normal, doc="EPICS Encoder reading", auto_monitor=True)
+
+    energy_epics = Cpt(EpicsSignalRO, suffix="NIDAQ-ENERGY", kind=Kind.normal, doc="EPICS Energy reading", auto_monitor=True)
+
+    ### Readback for BEC emitter ###
+    ai0_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 0, STD")
+    ai1_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 1, STD")
+    ai2_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 2, STD")
+    ai3_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 3, STD")
+    ai4_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 4, STD")
+    ai5_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 5, STD")
+    ai6_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 6, STD")
+    ai7_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream analog input 7, STD")
+
+    ci0_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 0. STD")
+    ci1_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 1. STD")
+    ci2_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 2. STD")
+    ci3_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 3. STD")
+    ci4_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 4. STD")
+    ci5_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 5. STD")
+    ci6_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 6. STD")
+    ci7_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 7. STD")
+    ci8_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 8. STD")
+    ci9_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 9. STD")
+    ci10_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 10. STD")
+    ci11_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 11. STD")
+    ci12_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 12. STD")
+    ci13_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 13. STD")
+    ci14_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 14. STD")
+    ci15_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 15. STD")
+    ci16_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 16. STD")
+    ci17_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream counter input 17. STD")
+
+    xas_timestamp = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream XAS timestamp")
+    xrd_timestamp = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream XRD timestamp")
+    xrd_angle = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream XRD angle")
+    xrd_energy = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream XRD energy")
+    xrd_ai0_mean = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream XRD ai0 mean")
+    xrd_ai0_std_dev = Cpt(SetableSignal, value=0, kind=Kind.normal, doc="NIDAQ stream XRD ai0 std dev")
+
     enc = Cpt(SetableSignal, value=0, kind=Kind.normal)
-    energy = Cpt(SetableSignal, value=0, kind=Kind.normal)
     rle = Cpt(SetableSignal, value=0, kind=Kind.normal)
 
     ### Control PVs ###
 
-    enable_compression = Cpt(EpicsSignal, suffix="NIDAQ-EnableRLE", kind=Kind.config)
+    enable_compression = Cpt(EpicsSignal, suffix="NIDAQ-EnableRLE", kind=Kind.config, auto_monitor=True)
+    enable_dead_time_correction = Cpt(EpicsSignal, suffix="NIDAQ-EnableDTC", kind=Kind.config, auto_monitor=True)
     kickoff_call = Cpt(EpicsSignal, suffix="NIDAQ-Kickoff", kind=Kind.config)
     stage_call = Cpt(EpicsSignal, suffix="NIDAQ-Stage", kind=Kind.config)
     state = Cpt(EpicsSignal, suffix="NIDAQ-FSMState", kind=Kind.config, auto_monitor=True)
     server_status = Cpt(EpicsSignalRO, suffix="NIDAQ-ServerStatus", kind=Kind.config)
     compression_ratio = Cpt(EpicsSignalRO, suffix="NIDAQ-CompressionRatio", kind=Kind.config)
     scan_type = Cpt(EpicsSignal, suffix="NIDAQ-ScanType", kind=Kind.config)
-    sampling_rate = Cpt(EpicsSignal, suffix="NIDAQ-SamplingRateRequested", kind=Kind.config)
+    scan_type_string = Cpt(EpicsSignal, suffix="NIDAQ-ScanType", kind=Kind.config, string=True)
+    sampling_rate = Cpt(EpicsSignal, suffix="NIDAQ-SamplingRateRequested", kind=Kind.config, auto_monitor=True)
+    sampling_rate_string = Cpt(EpicsSignal, suffix="NIDAQ-SamplingRateRequested", kind=Kind.config, string=True, auto_monitor=True)
     scan_duration = Cpt(EpicsSignal, suffix="NIDAQ-SamplingDuration", kind=Kind.config)
-    readout_range = Cpt(EpicsSignal, suffix="NIDAQ-ReadoutRange", kind=Kind.config)
-    encoder_factor = Cpt(EpicsSignal, suffix="NIDAQ-EncoderFactor", kind=Kind.config)
+    readout_range = Cpt(EpicsSignal, suffix="NIDAQ-ReadoutRange", kind=Kind.config, auto_monitor=True)
+    readout_range_string = Cpt(EpicsSignal, suffix="NIDAQ-ReadoutRange", kind=Kind.config, string=True, auto_monitor=True)
+    encoder_factor = Cpt(EpicsSignal, suffix="NIDAQ-EncoderFactor", kind=Kind.config, auto_monitor=True)
+    encoder_factor_string = Cpt(EpicsSignal, suffix="NIDAQ-EncoderFactor", kind=Kind.config, string=True, auto_monitor=True)
     stop_call = Cpt(EpicsSignal, suffix="NIDAQ-Stop", kind=Kind.config)
     power = Cpt(EpicsSignal, suffix="NIDAQ-Power", kind=Kind.config)
     heartbeat = Cpt(EpicsSignal, suffix="NIDAQ-Heartbeat", kind=Kind.config, auto_monitor=True)
     time_left = Cpt(EpicsSignalRO, suffix="NIDAQ-TimeLeft", kind=Kind.config, auto_monitor=True)
 
-    ai_chans = Cpt(EpicsSignal, suffix="NIDAQ-AIChans", kind=Kind.config)
-    ci_chans = Cpt(EpicsSignal, suffix="NIDAQ-CIChans", kind=Kind.config)
-    di_chans = Cpt(EpicsSignal, suffix="NIDAQ-DIChans", kind=Kind.config)
+    ai_chans = Cpt(EpicsSignal, suffix="NIDAQ-AIChans", kind=Kind.config, auto_monitor=True)
+    ci_chans = Cpt(EpicsSignal, suffix="NIDAQ-CIChans", kind=Kind.config, auto_monitor=True)
+    di_chans = Cpt(EpicsSignal, suffix="NIDAQ-DIChans", kind=Kind.config, auto_monitor=True)
+    add_chans = Cpt(EpicsSignal, suffix="NIDAQ-AddChans", kind=Kind.config, auto_monitor=True)
+
+    smpl_abs_ln = Cpt(EpicsSignal, suffix="NIDAQ-smpl_abs_ln", kind=Kind.config, auto_monitor=True)
+    ref_abs_ln = Cpt(EpicsSignal, suffix="NIDAQ-ref_abs_ln", kind=Kind.config, auto_monitor=True)
+
+    smpl_abs_no = Cpt(EpicsSignal, suffix="NIDAQ-smpl_abs_no", kind=Kind.config, auto_monitor=True)
+    smpl_abs_no_string = Cpt(EpicsSignal, suffix="NIDAQ-smpl_abs_no", kind=Kind.config, string=True, auto_monitor=True)
+
+    smpl_abs_de = Cpt(EpicsSignal, suffix="NIDAQ-smpl_abs_de", kind=Kind.config, auto_monitor=True)
+    smpl_abs_de_string = Cpt(EpicsSignal, suffix="NIDAQ-smpl_abs_de", kind=Kind.config, string=True, auto_monitor=True)
+
+    smpl_fluo_no = Cpt(EpicsSignal, suffix="NIDAQ-smpl_fluo_no", kind=Kind.config, auto_monitor=True)
+    smpl_fluo_no_string = Cpt(EpicsSignal, suffix="NIDAQ-smpl_fluo_no", kind=Kind.config, string=True, auto_monitor=True)
+
+    smpl_fluo_de = Cpt(EpicsSignal, suffix="NIDAQ-smpl_fluo_de", kind=Kind.config, auto_monitor=True)
+    smpl_fluo_de_string = Cpt(EpicsSignal, suffix="NIDAQ-smpl_fluo_de", kind=Kind.config, string=True, auto_monitor=True)
+
+    ref_abs_no = Cpt(EpicsSignal, suffix="NIDAQ-ref_abs_no", kind=Kind.config, auto_monitor=True)
+    ref_abs_no_string = Cpt(EpicsSignal, suffix="NIDAQ-ref_abs_no", kind=Kind.config, string=True, auto_monitor=True)
+
+    ref_abs_de = Cpt(EpicsSignal, suffix="NIDAQ-ref_abs_de", kind=Kind.config, auto_monitor=True)
+    ref_abs_de_string = Cpt(EpicsSignal, suffix="NIDAQ-ref_abs_de", kind=Kind.config, string=True, auto_monitor=True)
+
+    # fmt: on
 
 
 class Nidaq(PSIDeviceBase, NidaqControl):
@@ -356,9 +221,11 @@ class Nidaq(PSIDeviceBase, NidaqControl):
 
     def __init__(self, prefix: str = "", *, name: str, scan_info: ScanInfo = None, **kwargs):
         super().__init__(name=name, prefix=prefix, scan_info=scan_info, **kwargs)
-        self.scan_info: ScanInfo
+        self.scan_parameters: ScanServerScanInfo = None
         self.timeout_wait_for_signal = 5  # put 5s firsts
-        self._timeout_wait_for_pv = 3  # 3s timeout for pv calls
+        self._timeout_wait_for_pv = (
+            5  # 5s timeout for pv calls. editted due to timeout issues persisting
+        )
         self.valid_scan_names = [
             "xas_simple_scan",
             "xas_simple_scan_with_xrd",
@@ -371,10 +238,9 @@ class Nidaq(PSIDeviceBase, NidaqControl):
     #           Beamline Methods           #
     ########################################
 
-    def _check_if_scan_name_is_valid(self) -> bool:
+    def _check_if_scan_name_is_valid(self, scan_parameters: ScanServerScanInfo) -> bool:
         """Check if the scan is within the list of scans for which the backend is working"""
-        scan_name = self.scan_info.msg.scan_name
-        if scan_name in self.valid_scan_names:
+        if scan_parameters.scan_name in self.valid_scan_names:
             return True
         return False
 
@@ -531,7 +397,8 @@ class Nidaq(PSIDeviceBase, NidaqControl):
         Information about the upcoming scan can be accessed from the scan_info (self.scan_info.msg) object.
         If the upcoming scan is not in the list of valid scans, return immediately.
         """
-        if not self._check_if_scan_name_is_valid():
+        self.scan_parameters = fetch_scan_info(self.scan_info)
+        if not self._check_if_scan_name_is_valid(self.scan_parameters):
             return None
 
         if self.state.get() != NidaqState.STANDBY:
@@ -541,25 +408,29 @@ class Nidaq(PSIDeviceBase, NidaqControl):
             status.wait(timeout=self.timeout_wait_for_signal)
 
         # If scan is not part of the valid_scan_names,
-        if self.scan_info.msg.scan_name != "nidaq_continuous_scan":
+        if self.scan_parameters.scan_name != "nidaq_continuous_scan":  # what is the new v4 scan
             self.scan_type.set(ScanType.TRIGGERED).wait(timeout=self._timeout_wait_for_pv)
             self.scan_duration.set(0).wait(timeout=self._timeout_wait_for_pv)
             self.enable_compression.set(1).wait(timeout=self._timeout_wait_for_pv)
         else:
             self.scan_type.set(ScanType.CONTINUOUS).wait(timeout=self._timeout_wait_for_pv)
-            self.scan_duration.set(self.scan_info.msg.scan_parameters["scan_duration"]).wait(
-                timeout=self._timeout_wait_for_pv
-            )
-            self.enable_compression.set(self.scan_info.msg.scan_parameters["compression"]).wait(
-                timeout=self._timeout_wait_for_pv
-            )
+            self.scan_duration.set(
+                self.scan_parameters.additional_scan_parameters["scan_duration"]
+            ).wait(timeout=self._timeout_wait_for_pv)
+            self.enable_compression.set(
+                self.scan_parameters.additional_scan_parameters["compression"]
+            ).wait(timeout=self._timeout_wait_for_pv)
 
         # Stage call to IOC
         status = CompareStatus(self.state, NidaqState.STAGE)
         self.cancel_on_stop(status)
-        self.stage_call.set(1).wait(timeout=self._timeout_wait_for_pv)
+        # TODO 11.11.25/HS64
+        # Switched from set to put in the hope to get rid of the rare event where nidaq is stopped at the start of a scan
+        # Problems consistently persisting, testing changing back to set, unconvinced this is the actual cause 14.11.25/AHC
+        # self.stage_call.set(1).wait(timeout=self._timeout_wait_for_pv)
+        self.stage_call.put(1)
         status.wait(timeout=self.timeout_wait_for_signal)
-        if self.scan_info.msg.scan_name != "nidaq_continuous_scan":
+        if self.scan_parameters.scan_name != "nidaq_continuous_scan":
             status = self.on_kickoff()
             self.cancel_on_stop(status)
             status.wait(timeout=self._timeout_wait_for_pv)
@@ -590,10 +461,10 @@ class Nidaq(PSIDeviceBase, NidaqControl):
         before the motor starts its oscillation. This is needed for being properly homed.
         The NIDAQ should go into Acquiring mode.
         """
-        if not self._check_if_scan_name_is_valid():
+        if not self._check_if_scan_name_is_valid(self.scan_parameters):
             return None
 
-        if self.scan_info.msg.scan_name == "nidaq_continuous_scan":
+        if self.scan_parameters.scan_name == "nidaq_continuous_scan":
             logger.info(f"Device {self.name} ready to be kicked off for nidaq_continuous_scan")
             return None
 
@@ -614,12 +485,12 @@ class Nidaq(PSIDeviceBase, NidaqControl):
         For the NIDAQ we use this method to stop the backend since it
         would not stop by itself in its current implementation since the number of points are not predefined.
         """
-        if not self._check_if_scan_name_is_valid():
+        if not self._check_if_scan_name_is_valid(self.scan_parameters):
             return None
 
         status = CompareStatus(self.state, NidaqState.STANDBY)
         self.cancel_on_stop(status)
-        if self.scan_info.msg.scan_name != "nidaq_continuous_scan":
+        if self.scan_parameters.scan_name != "nidaq_continuous_scan":
             self.on_stop()
         return status
 
@@ -630,7 +501,9 @@ class Nidaq(PSIDeviceBase, NidaqControl):
         Args:
             value (int) : current progress value
         """
-        scan_duration = self.scan_info.msg.scan_parameters.get("scan_duration", None)
+        if self.scan_parameters is None:
+            return
+        scan_duration = self.scan_parameters.additional_scan_parameters.get("scan_duration", None)
         if not isinstance(scan_duration, (int, float)):
             return
         value = scan_duration - value

debye_bec/devices/pilatus/pilatus.py

@@ -6,22 +6,30 @@ import enum
 import threading
 import time
 import traceback
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Tuple
 
 import numpy as np
 from bec_lib.file_utils import get_full_path
 from bec_lib.logger import bec_logger
+from bec_server.scan_server.scans.scan_base import ScanInfo as ScanServerScanInfo
 from ophyd import Component as Cpt
-from ophyd import EpicsSignal, Kind
+from ophyd import EpicsSignal, EpicsSignalRO, Kind
 from ophyd.areadetector.cam import ADBase, PilatusDetectorCam
 from ophyd.areadetector.plugins import HDF5Plugin_V22 as HDF5Plugin
 from ophyd.areadetector.plugins import ImagePlugin_V22 as ImagePlugin
 from ophyd.status import WaitTimeoutError
-from ophyd_devices import CompareStatus, DeviceStatus, FileEventSignal, PreviewSignal
+from ophyd_devices import (
+    AndStatus,
+    CompareStatus,
+    DeviceStatus,
+    ExceptionStatus,
+    FileEventSignal,
+    PreviewSignal,
+)
 from ophyd_devices.interfaces.base_classes.psi_device_base import PSIDeviceBase
 from pydantic import BaseModel, Field
 
-from debye_bec.devices.pilatus.utils import AndStatusWithList
+from debye_bec.devices.utils.utils import fetch_scan_info
 
 if TYPE_CHECKING:  # pragma: no cover
     from bec_lib.devicemanager import ScanInfo
@@ -147,6 +155,19 @@ class Pilatus(PSIDeviceBase, ADBase):
 
     # USER_ACCESS = ["start_live_mode", "stop_live_mode"]
 
+    cam_gain_menu_string = Cpt(EpicsSignalRO, suffix="cam1:GainMenu", string=True)
+
+    _default_configuration_attrs = [
+        "cam.threshold_energy",
+        "cam.threshold_auto_apply",
+        "cam.gain_menu",
+        "cam_gain_menu_string",
+        "cam.pixel_cut_off",
+        "cam.acquire_time",
+        "cam.num_exposures",
+        "cam.model",
+    ]
+
     cam = Cpt(PilatusDetectorCam, "cam1:")
     hdf = Cpt(HDF5Plugin, "HDF1:")
     image1 = Cpt(ImagePlugin, "image1:")
@@ -205,22 +226,11 @@ class Pilatus(PSIDeviceBase, ADBase):
         PreviewSignal,
         name="preview",
         ndim=2,
-        num_rotation_90=0,  # Check this
+        num_rotation_90=3,
         doc="Preview signal for the Pilatus Detector",
     )
     file_event = Cpt(FileEventSignal, name="file_event")
 
-    @property
-    def baseline_signals(self):
-        """Define baseline signals"""
-        return [
-            self.cam.acquire_time,
-            self.cam.num_exposures,
-            self.cam.threshold_energy,
-            self.cam.gain_menu,
-            self.cam.pixel_cut_off,
-        ]
-
     def __init__(
         self,
         *,
@@ -233,7 +243,6 @@ class Pilatus(PSIDeviceBase, ADBase):
         super().__init__(
             name=name, prefix=prefix, scan_info=scan_info, device_manager=device_manager, **kwargs
         )
-        self.scan_parameter = ScanParameter()
         self.device_manager = device_manager
         self._readout_time = PILATUS_READOUT_TIME
         self._full_path = ""
@@ -251,6 +260,7 @@ class Pilatus(PSIDeviceBase, ADBase):
         # self._live_mode_run_event = threading.Event()
         # self._live_mode_stopped_event = threading.Event()
         # self._live_mode_stopped_event.set()  # Initial state is stopped
+        self.scan_parameters: ScanServerScanInfo = None
 
     ########################################
     #        Custom Beamline Methods       #
@@ -360,78 +370,70 @@ class Pilatus(PSIDeviceBase, ADBase):
     #             f"Live Mode on detector {self.name} did not stop: {content} after 10s."
     #         )
 
-    def check_detector_stop_running_acquisition(self) -> AndStatusWithList:
+    def check_detector_stop_running_acquisition(self) -> AndStatus:
         """Check if the detector is still running an acquisition."""
         status_acquire = CompareStatus(self.cam.acquire, ACQUIREMODE.DONE.value)
         status_writing = CompareStatus(self.hdf.capture, ACQUIREMODE.DONE.value)
         status_cam_server = CompareStatus(self.cam.armed, DETECTORSTATE.UNARMED.value)
-        status = AndStatusWithList(
-            device=self, status_list=[status_acquire, status_writing, status_cam_server]
-        )
+        status = status_acquire & status_writing & status_cam_server
         return status
 
-    def _calculate_trigger(self, scan_msg: ScanStatusMessage):
-        self._update_scan_parameter()
+    def _calculate_trigger(self, scan_parameters: ScanServerScanInfo) -> Tuple[float, float]:
         total_osc = 0
+        calc_duration = 0
         total_trig_lo = 0
         total_trig_hi = 0
-        calc_duration = 0
-        n_trig_lo = 1
-        n_trig_hi = 1
-        init_lo = 1
-        init_hi = 1
-        lo_done = 0
-        hi_done = 0
-        if not self.scan_parameter.break_enable_low:
-            lo_done = 1
-        if not self.scan_parameter.break_enable_high:
-            hi_done = 1
-        start_time = time.time()
-        while True:
-            # TODO, we should not use infinite loops, for now let's add the escape Timeout of 20s, but should eventually be reviewed.
-            if time.time() - start_time > 20:
-                raise RuntimeError(
-                    f"Calculating the number of triggers for scan {scan_msg.scan_name} took more than 20 seconds, aborting."
-                )
+        # Switching high/low is intended as angle is inverse to energy and settings in BEC are always in energy
+        loc_break_enable_low = scan_parameters.additional_scan_parameters.get(
+            "break_enable_high", False
+        )
+        loc_break_time_low = scan_parameters.additional_scan_parameters.get("break_time_high", 0)
+        loc_cycle_low = scan_parameters.additional_scan_parameters.get("cycle_high", 1)
+        loc_break_enable_high = scan_parameters.additional_scan_parameters.get(
+            "break_enable_low", False
+        )
+        loc_break_time_high = scan_parameters.additional_scan_parameters.get("break_time_low", 0)
+        loc_cycle_high = scan_parameters.additional_scan_parameters.get("cycle_low", 1)
+
+        if not loc_break_enable_low:
+            loc_break_time_low = 0
+            loc_cycle_low = 1
+        if not loc_break_enable_high:
+            loc_break_time_high = 0
+            loc_cycle_high = 1
+
+        total_osc = scan_parameters.additional_scan_parameters.get("scan_duration", 0) / (
+            scan_parameters.additional_scan_parameters.get("scan_time", 0)
+            + loc_break_time_low / (2 * loc_cycle_low)
+            + loc_break_time_high / (2 * loc_cycle_high)
+        )
+        total_osc = np.ceil(total_osc)
+        total_osc = total_osc + total_osc % 2  # round up to the next even number
+
+        if loc_break_enable_low:
+            total_trig_lo = np.floor(total_osc / (2 * loc_cycle_low))
+        if loc_break_enable_high:
+            total_trig_hi = np.floor(total_osc / (2 * loc_cycle_high))
+        calc_duration = (
+            total_osc * scan_parameters.additional_scan_parameters.get("scan_time", 0)
+            + total_trig_lo * loc_break_time_low
+            + total_trig_hi * loc_break_time_high
+        )
+
+        if calc_duration < scan_parameters.additional_scan_parameters.get("scan_duration", 0):
+            # Due to inaccuracy in formula, this can happen, we then need to manually add two oscillations and recalculate the triggers
             total_osc = total_osc + 2
-            calc_duration = calc_duration + 2 * self.scan_parameter.scan_time
+            if loc_break_enable_low:
+                total_trig_lo = np.floor(total_osc / (2 * loc_cycle_low))
+            if loc_break_enable_high:
+                total_trig_hi = np.floor(total_osc / (2 * loc_cycle_high))
+            calc_duration = (
+                total_osc * scan_parameters.additional_scan_parameters.get("scan_time", 0)
+                + total_trig_lo * loc_break_time_low
+                + total_trig_hi * loc_break_time_high
+            )
 
-            if self.scan_parameter.break_enable_low and n_trig_lo >= self.scan_parameter.cycle_low:
-                n_trig_lo = 1
-                calc_duration = calc_duration + self.scan_parameter.break_time_low
-                if init_lo:
-                    lo_done = 1
-                    init_lo = 0
-            else:
-                n_trig_lo += 1
-
-            if (
-                self.scan_parameter.break_enable_high
-                and n_trig_hi >= self.scan_parameter.cycle_high
-            ):
-                n_trig_hi = 1
-                calc_duration = calc_duration + self.scan_parameter.break_time_high
-                if init_hi:
-                    hi_done = 1
-                    init_hi = 0
-            else:
-                n_trig_hi += 1
-
-            if lo_done and hi_done:
-                n = np.floor(self.scan_parameter.scan_duration / calc_duration)
-                total_osc = total_osc * n
-                if self.scan_parameter.break_enable_low:
-                    total_trig_lo = n + 1
-                if self.scan_parameter.break_enable_high:
-                    total_trig_hi = n + 1
-                calc_duration = calc_duration * n
-                lo_done = 0
-                hi_done = 0
-
-            if calc_duration >= self.scan_parameter.scan_duration:
-                break
-
-        return total_trig_lo + total_trig_hi
+        return total_trig_lo, total_trig_hi
 
     ########################################
     #  Beamline Specific Implementations   #
@@ -483,26 +485,39 @@ class Pilatus(PSIDeviceBase, ADBase):
         (self.scan_info.msg) object.
         """
         # self.stop_live_mode()  # Make sure that live mode is stopped if scan runs
+        self.scan_parameters = fetch_scan_info(self.scan_info)
 
-        scan_msg: ScanStatusMessage = self.scan_info.msg
-        if scan_msg.scan_name in self.xas_xrd_scan_names:
-            self._update_scan_parameter()
+        # If user has activated alignment mode on qt panel, switch back to multitrigger and stop acquisition
+        if self.cam.trigger_mode.get() != TRIGGERMODE.MULT_TRIGGER.value:
+            self.cam.trigger_mode.set(TRIGGERMODE.MULT_TRIGGER.value).wait(5)
+        if self.cam.acquire.get() == ACQUIREMODE.ACQUIRING.value:
+            self.cam.acquire.put(0)
+            status_cam = CompareStatus(self.cam.acquire, ACQUIREMODE.DONE.value)
+            status_cam.wait(timeout=5)
+
+        if self.scan_parameters.scan_name in self.xas_xrd_scan_names:
             # Compute number of triggers
-            total_trig_lo, total_trig_hi = self._calculate_trigger(scan_msg)
+            total_trig_lo, total_trig_hi = self._calculate_trigger(self.scan_parameters)
             # Set the number of images, we may also set this to a higher values if preferred and stop the acquisition
             # TODO This logic is prone to errors, as we rely on the scans to nicely resolve to n_images. We should
             # use here instead a way of settings the n_images independently of the scan parameters to avoid running out of sync
             # with the complete method. Ideally we comput them in the scan itself.. This is much safer IMO!
-            self.n_images = (total_trig_lo + total_trig_hi) * self.scan_parameter.n_of_trigger
-            exp_time = self.scan_parameter.exp_time
+            self.n_images = (
+                total_trig_lo + total_trig_hi
+            ) * self.scan_parameters.additional_scan_parameters.get("n_of_trigger", 1)
+            exp_time = self.scan_parameters.exp_time
             self.trigger_source.set(MONOTRIGGERSOURCE.INPOS).wait(5)
-            self.trigger_n_of.set(self.scan_parameter.n_of_trigger).wait(5)
-
-        elif scan_msg.scan_type == "step":
-            self.n_images = scan_msg.num_points * scan_msg.scan_parameters.get(
-                "frames_per_trigger", 1
+            self.trigger_n_of.set(
+                self.scan_parameters.additional_scan_parameters.get("n_of_trigger", 1)
+            ).wait(5)
+        # TODO migrate logic to v4 once old scans are deprecated,
+        # TODO if num_points=None and no logic from scan_name applies, can't measure with this detector..
+        elif self.scan_parameters.scan_type == "software_triggered":
+            self.n_images = (
+                self.scan_parameters.num_monitored_readouts
+                * self.scan_parameters.frames_per_trigger
             )
-            exp_time = scan_msg.scan_parameters.get("exp_time")
+            exp_time = self.scan_parameters.exp_time
             self.trigger_source.set(MONOTRIGGERSOURCE.EPICS).wait(5)
             self.trigger_n_of.set(1).wait(5)  # BEC will trigger each acquisition
         else:
@@ -523,18 +538,18 @@ class Pilatus(PSIDeviceBase, ADBase):
                 )
             )
         detector_exp_time = exp_time - self._readout_time
-        self._full_path = get_full_path(scan_msg, name="pilatus")
+        self._full_path = get_full_path(self.scan_info.msg, name="pilatus")
         file_path = "/".join(self._full_path.split("/")[:-1])
         file_name = self._full_path.split("/")[-1]
         # Prepare detector and backend
         self.cam.array_callbacks.set(1).wait(5)  # Enable array callbacks
         self.hdf.enable.set(1).wait(5)  # Enable HDF5 plugin
         # Camera settings
-        self.cam.num_exposures.set(1).wait(5)
-        self.cam.num_images.set(self.n_images).wait(5)
-        self.cam.acquire_time.set(detector_exp_time).wait(5)  # let's try this
-        self.cam.acquire_period.set(exp_time).wait(5)
-        self.filter_number.set(0).wait(5)
+        self.cam.num_exposures.set(1, timeout=5).wait()
+        self.cam.num_images.set(self.n_images, timeout=5).wait()
+        self.cam.acquire_time.set(detector_exp_time, timeout=5).wait()  # let's try this
+        self.cam.acquire_period.set(exp_time, timeout=5).wait()
+        self.filter_number.set(0, timeout=5).wait()
         # HDF5 settings
         logger.debug(
             f"Setting HDF5 file path to {file_path} and file name to {file_name}. full_path is {self._full_path}"
@@ -555,16 +570,14 @@ class Pilatus(PSIDeviceBase, ADBase):
 
     def on_pre_scan(self) -> DeviceStatus | None:
         """Called right before the scan starts on all devices automatically."""
-        scan_msg: ScanStatusMessage = self.scan_info.msg
         if (
-            scan_msg.scan_name in self.xas_xrd_scan_names or scan_msg.scan_type == "step"
+            self.scan_parameters.scan_name in self.xas_xrd_scan_names
+            or self.scan_parameters.scan_type == "software_triggered"
         ):  # TODO how to deal with fly scans?
             status_hdf = CompareStatus(self.hdf.capture, ACQUIREMODE.ACQUIRING.value)
             status_cam = CompareStatus(self.cam.acquire, ACQUIREMODE.ACQUIRING.value)
             status_cam_server = CompareStatus(self.cam.armed, DETECTORSTATE.ARMED.value)
-            status = AndStatusWithList(
-                device=self, status_list=[status_hdf, status_cam, status_cam_server]
-            )
+            status = status_hdf & status_cam & status_cam_server
             self.cam.acquire.put(1)
             self.hdf.capture.put(1)
             self.cancel_on_stop(status)
@@ -574,8 +587,7 @@ class Pilatus(PSIDeviceBase, ADBase):
 
     def on_trigger(self) -> DeviceStatus | None:
         """Called when the device is triggered."""
-        scan_msg: ScanStatusMessage = self.scan_info.msg
-        if not scan_msg.scan_type == "step":
+        if not self.scan_parameters.scan_type == "software_triggered":
             return None
         start_time = time.time()
         img_counter = self.hdf.num_captured.get()
@@ -588,20 +600,20 @@ class Pilatus(PSIDeviceBase, ADBase):
 
     def _complete_callback(self, status: DeviceStatus):
         """Callback for when the device completes a scan."""
-        scan_msg: ScanStatusMessage = self.scan_info.msg
         if (
-            scan_msg.scan_name in self.xas_xrd_scan_names or scan_msg.scan_type == "step"
+            self.scan_parameters.scan_name in self.xas_xrd_scan_names
+            or self.scan_parameters.scan_type == "software_triggered"
         ):  # TODO how to deal with fly scans?
             if status.success:
-                status.device.file_event.put(
-                    file_path=status.device._full_path,  # pylint: disable:protected-access
+                self.file_event.put(
+                    file_path=self._full_path,
                     done=True,
                     successful=True,
                     hinted_h5_entries={"data": "/entry/data/data"},
                 )
             else:
-                status.device.file_event.put(
-                    file_path=status.device._full_path,  # pylint: disable:protected-access
+                self.file_event.put(
+                    file_path=self._full_path,
                     done=True,
                     successful=False,
                     hinted_h5_entries={"data": "/entry/data/data"},
@@ -611,26 +623,26 @@ class Pilatus(PSIDeviceBase, ADBase):
 
     def on_complete(self) -> DeviceStatus | None:
         """Called to inquire if a device has completed a scans."""
-        scan_msg: ScanStatusMessage = self.scan_info.msg
         if (
-            scan_msg.scan_name in self.xas_xrd_scan_names or scan_msg.scan_type == "step"
+            self.scan_parameters.scan_name in self.xas_xrd_scan_names
+            or self.scan_parameters.scan_type == "software_triggered"
         ):  # TODO how to deal with fly scans?
             status_hdf = CompareStatus(self.hdf.capture, ACQUIREMODE.DONE.value)
             status_cam = CompareStatus(self.cam.acquire, ACQUIREMODE.DONE.value)
             status_cam_server = CompareStatus(self.cam.armed, DETECTORSTATE.UNARMED.value)
-            if self.scan_info.msg.scan_name in self.xas_xrd_scan_names:
+            # status_write_error = ExceptionStatus(self.hdf.write_status, 0, operation="!=")
+            if self.scan_parameters.scan_name in self.xas_xrd_scan_names:
                 # For long scans, it can be that the mono will execute one cycle more,
                 # meaning a few more XRD triggers will be sent
                 status_img_written = CompareStatus(
-                    self.hdf.num_captured, self.n_images, operation=">="
+                    self.hdf.num_captured, self.n_images, operation_success=">="
                 )
             else:
                 status_img_written = CompareStatus(self.hdf.num_captured, self.n_images)
             status_img_written = CompareStatus(self.hdf.num_captured, self.n_images)
-            status = AndStatusWithList(
-                device=self,
-                status_list=[status_hdf, status_cam, status_img_written, status_cam_server],
-            )
+            status = (
+                status_hdf & status_cam & status_img_written & status_cam_server
+            )  # & status_write_error
             status.add_callback(self._complete_callback)  # Callback that writing was successful
             self.cancel_on_stop(status)
             return status
@@ -651,15 +663,6 @@ class Pilatus(PSIDeviceBase, ADBase):
         # TODO do we need to clean the poll thread ourselves?
         self.on_stop()
 
-    def _update_scan_parameter(self):
-        """Get the scan_info parameters for the scan."""
-        for key, value in self.scan_info.msg.request_inputs["inputs"].items():
-            if hasattr(self.scan_parameter, key):
-                setattr(self.scan_parameter, key, value)
-        for key, value in self.scan_info.msg.request_inputs["kwargs"].items():
-            if hasattr(self.scan_parameter, key):
-                setattr(self.scan_parameter, key, value)
-
 
 if __name__ == "__main__":
     try:

debye_bec/devices/pilatus/utils.py

@@ -1,238 +0,0 @@
-"""Temporary utility module for Status Object implementations."""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-from ophyd import Device, DeviceStatus, StatusBase
-
-
-class AndStatusWithList(DeviceStatus):
-    """
-    Custom implementation of the AndStatus that combines the
-    option to add multiple statuses as a list, and in addition
-    allows for adding the Device as an object to access its
-    methods.
-
-    Args"""
-
-    def __init__(
-        self,
-        device: Device,
-        status_list: StatusBase | DeviceStatus | list[StatusBase | DeviceStatus],
-        **kwargs,
-    ):
-        self.all_statuses = status_list if isinstance(status_list, list) else [status_list]
-        super().__init__(device=device, **kwargs)
-        self._trace_attributes["all"] = [st._trace_attributes for st in self.all_statuses]
-
-        def inner(status):
-            with self._lock:
-                if self._externally_initiated_completion:
-                    return
-                if self.done:  # Return if status is already done.. It must be resolved already
-                    return
-
-                for st in self.all_statuses:
-                    with st._lock:
-                        if st.done and not st.success:
-                            self.set_exception(st.exception())  # st._exception
-                            return
-
-                if all(st.done for st in self.all_statuses) and all(
-                    st.success for st in self.all_statuses
-                ):
-                    self.set_finished()
-
-        for st in self.all_statuses:
-            with st._lock:
-                st.add_callback(inner)
-
-    # TODO improve __repr__ and __str__
-    def __repr__(self):
-        return "<AndStatusWithList({self.all_statuses!r})>".format(self=self)
-
-    def __str__(self):
-        return "<AndStatusWithList(done={self.done}, success={self.success})>".format(self=self)
-
-    def __contains__(self, status: StatusBase | DeviceStatus) -> bool:
-        for child in self.all_statuses:
-            if child == status:
-                return True
-            if isinstance(child, AndStatusWithList):
-                if status in child:
-                    return True
-
-        return False
-
-    # TODO Check if this actually works....
-    def set_exception(self, exc):
-        super().set_exception(exc)
-        # Propagate the exception to all sub-statuses that are not done yet.
-        with self._lock:
-            for st in self.all_statuses:
-                with st._lock:
-                    if not st.done:
-                        st.set_exception(exc)
-
-    def _run_callbacks(self):
-        """
-        Set the Event and run the callbacks.
-        """
-        if self.timeout is None:
-            timeout = None
-        else:
-            timeout = self.timeout + self.settle_time
-        if not self._settled_event.wait(timeout):
-            self.log.warning("%r has timed out", self)
-            with self._externally_initiated_completion_lock:
-                if self._exception is None:
-                    exc = TimeoutError(
-                        f"AndStatus from device {self.device.name} failed to complete in specified timeout of {self.timeout + self.settle_time}."
-                    )
-                    self._exception = exc
-        # Mark this as "settled".
-        try:
-            self._settled()
-        except Exception:
-            self.log.exception("%r encountered error during _settled()", self)
-        with self._lock:
-            self._event.set()
-        if self._exception is not None:
-            try:
-                self._handle_failure()
-            except Exception:
-                self.log.exception("%r encountered an error during _handle_failure()", self)
-        for cb in self._callbacks:
-            try:
-                cb(self)
-            except Exception:
-                self.log.exception(
-                    "An error was raised on a background thread while "
-                    "running the callback %r(%r).",
-                    cb,
-                    self,
-                )
-        self._callbacks.clear()
-
-
-class AndStatus(StatusBase):
-    """Custom AndStatus for TimePix detector."""
-
-    def __init__(
-        self,
-        left: StatusBase | DeviceStatus | list[StatusBase | DeviceStatus] | None,
-        name: str | Device | None = None,
-        right: StatusBase | DeviceStatus | list[StatusBase | DeviceStatus] | None = None,
-        **kwargs,
-    ):
-        self.left = left if isinstance(left, list) else [left]
-        if right is not None:
-            self.right = right if isinstance(right, list) else [right]
-        else:
-            self.right = []
-        self.all_statuses = self.left + self.right
-        if name is None:
-            name = "unname_status"
-        elif isinstance(name, Device):
-            name = name.name
-        else:
-            name = name
-        self.name = name
-        super().__init__(**kwargs)
-        self._trace_attributes["left"] = [st._trace_attributes for st in self.left]
-        self._trace_attributes["right"] = [st._trace_attributes for st in self.right]
-
-        def inner(status):
-            with self._lock:
-                if self._externally_initiated_completion:
-                    return
-                if self.done:  # Return if status is already done.. It must be resolved already
-                    return
-
-                for st in self.all_statuses:
-                    with st._lock:
-                        if st.done and not st.success:
-                            self.set_exception(st.exception())  # st._exception
-                            return
-
-                if all(st.done for st in self.all_statuses) and all(
-                    st.success for st in self.all_statuses
-                ):
-                    self.set_finished()
-
-        for st in self.all_statuses:
-            with st._lock:
-                st.add_callback(inner)
-
-    def __repr__(self):
-        return "({self.left!r} & {self.right!r})".format(self=self)
-
-    def __str__(self):
-        return "{0}(done={1.done}, " "success={1.success})" "".format(self.__class__.__name__, self)
-
-    def __contains__(self, status: StatusBase) -> bool:
-        for child in [self.left, self.right]:
-            if child == status:
-                return True
-            if isinstance(child, AndStatus):
-                if status in child:
-                    return True
-
-        return False
-
-    def _run_callbacks(self):
-        """
-        Set the Event and run the callbacks.
-        """
-        if self.timeout is None:
-            timeout = None
-        else:
-            timeout = self.timeout + self.settle_time
-        if not self._settled_event.wait(timeout):
-            # We have timed out. It's possible that set_finished() has already
-            # been called but we got here before the settle_time timer expired.
-            # And it's possible that in this space be between the above
-            # statement timing out grabbing the lock just below,
-            # set_exception(exc) has been called. Both of these possibilties
-            # are accounted for.
-            self.log.warning("%r has timed out", self)
-            with self._externally_initiated_completion_lock:
-                # Set the exception and mark the Status as done, unless
-                # set_exception(exc) was called externally before we grabbed
-                # the lock.
-                if self._exception is None:
-                    exc = TimeoutError(
-                        f"Status with name {self.name} failed to complete in specified timeout of {self.timeout + self.settle_time}."
-                    )
-                    self._exception = exc
-        # Mark this as "settled".
-        try:
-            self._settled()
-        except Exception:
-            # No alternative but to log this. We can't supersede set_exception,
-            # and we have to continue and run the callbacks.
-            self.log.exception("%r encountered error during _settled()", self)
-        # Now we know whether or not we have succeed or failed, either by
-        # timeout above or by set_exception(exc), so we can set the Event that
-        # will mark this Status as done.
-        with self._lock:
-            self._event.set()
-        if self._exception is not None:
-            try:
-                self._handle_failure()
-            except Exception:
-                self.log.exception("%r encountered an error during _handle_failure()", self)
-        # The callbacks have access to self, from which they can distinguish
-        # success or failure.
-        for cb in self._callbacks:
-            try:
-                cb(self)
-            except Exception:
-                self.log.exception(
-                    "An error was raised on a background thread while "
-                    "running the callback %r(%r).",
-                    cb,
-                    self,
-                )
-        self._callbacks.clear()

debye_bec/devices/pilatus_curtain.py

@@ -10,8 +10,6 @@ from ophyd import EpicsSignal, EpicsSignalRO
 from ophyd_devices import CompareStatus, DeviceStatus
 from ophyd_devices.interfaces.base_classes.psi_device_base import PSIDeviceBase
 
-from debye_bec.devices.pilatus.utils import AndStatusWithList
-
 if TYPE_CHECKING:
     from bec_lib.devicemanager import ScanInfo
 
@@ -71,11 +69,11 @@ class PilatusCurtain(PSIDeviceBase):
 
     def on_unstage(self) -> DeviceStatus | None:
         """Called while unstaging the device."""
-        return self.close()
+        # return self.close()
 
     def on_stop(self) -> DeviceStatus | None:
         """Called when the device is stopped."""
-        return self.close()
+        # return self.close()
 
     def open(self) -> DeviceStatus | None:
         """Open the cover"""
@@ -83,8 +81,8 @@ class PilatusCurtain(PSIDeviceBase):
             self.open_cover.put(1)
             # TODO timeout ok?
             status_open = CompareStatus(self.cover_is_open, COVER.OPEN, timeout=5)
-            status_error = CompareStatus(self.cover_error, COVER.ERROR, operation="!=")
-            status = AndStatusWithList(device=self, status_list=[status_open, status_error])
+            status_error = CompareStatus(self.cover_error, COVER.ERROR, operation_success="!=")
+            status = status_open & status_error
             return status
         else:
             return None
@@ -95,8 +93,8 @@ class PilatusCurtain(PSIDeviceBase):
             self.close_cover.put(1)
             # TODO timeout ok?
             status_close = CompareStatus(self.cover_is_closed, COVER.CLOSED, timeout=5)
-            status_error = CompareStatus(self.cover_error, COVER.ERROR, operation="!=")
-            status = AndStatusWithList(device=self, status_list=[status_close, status_error])
+            status_error = CompareStatus(self.cover_error, COVER.ERROR, operation_success="!=")
+            status = status_close & status_error
             return status
         else:
             return None

debye_bec/devices/reffoilchanger.py

@@ -52,9 +52,15 @@ class Reffoilchanger(PSIDeviceBase):
     status = Cpt(
         EpicsSignal, suffix="ES2-REF:SELN-FilterState-ENUM_RBV", kind="config", doc="Status"
     )
+    status_string = Cpt(
+        EpicsSignal, suffix="ES2-REF:SELN-FilterState-ENUM_RBV", kind="config", doc="Status", string=True
+    )
     op_mode = Cpt(
         EpicsSignalWithRBV, suffix="ES2-REF:SELN-OpMode-ENUM", kind="config", doc="Status"
     )
+    op_mode_string = Cpt(
+        EpicsSignalWithRBV, suffix="ES2-REF:SELN-OpMode-ENUM", kind="config", doc="Status", string=True
+    )
     ref_set = Cpt(EpicsSignal, suffix="ES2-REF:SELN-SET", kind="config", doc="Requested reference")
     ref_rb = Cpt(
         EpicsSignalRO, suffix="ES2-REF:SELN-RB", kind="config", doc="Currently set reference"

debye_bec/devices/utils/utils.py

@@ -0,0 +1,26 @@
+"""Utility functions for the devices."""
+
+from copy import deepcopy
+
+import numpy as np
+from bec_lib.devicemanager import ScanInfo
+from bec_server.scan_server.scans.scan_base import ScanInfo as ScanServerScanInfo
+from pydantic import ValidationError
+
+
+def fetch_scan_info(scan_info: ScanInfo) -> ScanServerScanInfo:
+    """Fetch the scan parameters from the scan_info object and return them as a ScanServerScanInfo object."""
+    info = scan_info.msg.info
+    if isinstance(info["positions"], list):
+        info["positions"] = np.array(info["positions"])
+    try:
+        msg = ScanServerScanInfo.model_validate(info)
+    except ValidationError:  # This means we have an old scan_info object.
+        info = deepcopy(info)
+        # We need to convert a few parameters manually.
+        info["scan_type"] = (
+            "hardware_triggered" if info["scan_type"] == "fly" else "software_triggered"
+        )
+        msg = ScanServerScanInfo.model_validate(info)
+
+    return msg

debye_bec/file_writer/debye_nexus_structure.py

@@ -1,5 +1,7 @@
 from bec_server.file_writer.default_writer import DefaultFormat
 
+import debye_bec.bec_widgets.widgets.digital_twin.x01da_parameters as bl
+
 
 class DebyeNexusStructure(DefaultFormat):
     """Nexus Structure for Debye"""
@@ -12,102 +14,6 @@ class DebyeNexusStructure(DefaultFormat):
         instrument = entry.create_group(name="instrument")
         instrument.attrs["NX_class"] = "NXinstrument"
 
-        ###################
-        ## mo1_bragg specific information
-        ###################
-
-        # Logic if device exist
-        if "mo1_bragg" in self.device_manager.devices:
-
-            monochromator = instrument.create_group(name="monochromator")
-            monochromator.attrs["NX_class"] = "NXmonochromator"
-            crystal = monochromator.create_group(name="crystal")
-            crystal.attrs["NX_class"] = "NXcrystal"
-
-            # Create a dataset
-            chemical_formular = crystal.create_dataset(name="chemical_formular", data="Si")
-            chemical_formular.attrs["NX_class"] = "NX_CHAR"
-
-            # Create a softlink
-            d_spacing = crystal.create_soft_link(
-                name="d_spacing",
-                target="/entry/collection/devices/mo1_bragg/mo1_bragg_crystal_current_d_spacing/value",
-            )
-            d_spacing.attrs["NX_class"] = "NX_FLOAT"
-
-            offset = crystal.create_soft_link(
-                name="offset",
-                target="/entry/collection/devices/mo1_bragg/mo1_bragg_crystal_current_offset/value",
-            )
-            offset.attrs["NX_class"] = "NX_FLOAT"
-
-            reflection = crystal.create_soft_link(
-                name="reflection",
-                target="/entry/collection/devices/mo1_bragg/mo1_bragg_crystal_current_xtal_string/value",
-            )
-            reflection.attrs["NX_class"] = "NX_CHAR"
-
-        ##################
-        ## cm mirror specific information
-        ###################
-
-        collimating_mirror = instrument.create_group(name="collimating_mirror")
-        collimating_mirror.attrs["NX_class"] = "NXmirror"
-
-        cm_substrate_material = collimating_mirror.create_dataset(
-            name="substrate_material", data="Si"
-        )
-        cm_substrate_material.attrs["NX_class"] = "NX_CHAR"
-
-        cm_bending_radius = collimating_mirror.create_soft_link(
-            name="sagittal radius",
-            target="/entry/collection/devices/cm_bnd_radius/cm_bnd_radius/value",
-        )
-        cm_bending_radius.attrs["NX_class"] = "NX_FLOAT"
-        cm_bending_radius.attrs["units"] = "km"
-
-        cm_incidence_angle = collimating_mirror.create_soft_link(
-            name="incidence angle", target="/entry/collection/devices/cm_rotx/cm_rotx/value"
-        )
-        cm_incidence_angle.attrs["NX_class"] = "NX_FLOAT"
-
-        cm_yaw_angle = collimating_mirror.create_soft_link(
-            name="incident angle", target="/entry/collection/devices/cm_roty/cm_roty/value"
-        )
-        cm_yaw_angle.attrs["NX_class"] = "NX_FLOAT"
-
-        ##################
-        ## fm mirror specific information
-        ###################
-
-        focusing_mirror = instrument.create_group(name="focusing_mirror")
-        focusing_mirror.attrs["NX_class"] = "NXmirror"
-
-        fm_substrate_material = focusing_mirror.create_dataset(name="substrate_material", data="Si")
-        fm_substrate_material.attrs["NX_class"] = "NX_CHAR"
-
-        fm_bending_radius = focusing_mirror.create_soft_link(
-            name="sagittal radius",
-            target="/entry/collection/devices/fm_bnd_radius/fm_bnd_radius/value",
-        )
-        fm_bending_radius.attrs["NX_class"] = "NX_FLOAT"
-
-        fm_incidence_angle = focusing_mirror.create_soft_link(
-            name="incidence angle",
-            target="/entry/collection/devices/fm_incidence_angle/fm_incidence_angle/value",
-        )
-        fm_incidence_angle.attrs["NX_class"] = "NX_FLOAT"
-
-        fm_yaw_angle = focusing_mirror.create_soft_link(
-            name="yaw angle", target="/entry/collection/devices/fm_roty/fm_roty/value"
-        )
-        fm_yaw_angle.attrs["NX_class"] = "NX_FLOAT"
-
-        fm_roll_angle = focusing_mirror.create_soft_link(
-            name="roll angle", target="/entry/collection/devices/fm_rotz/fm_rotz/value"
-        )
-        fm_roll_angle.attrs["NX_class"] = "NX_FLOAT"
-
         ##################
         ## source specific information
         ###################
@@ -123,3 +29,375 @@ class DebyeNexusStructure(DefaultFormat):
 
         probe = source.create_dataset(name="probe", data="X-ray")
         probe.attrs["NX_class"] = "NX_CHAR"
+
+        if "curr" in self.device_manager.devices:
+            ring_current = source.create_soft_link(
+                name="ring_current", target="/entry/collection/devices/curr/curr/value"
+            )
+            ring_current.attrs["NX_class"] = "NX_FLOAT"
+            ring_current.attrs["units"] = "mA"
+
+        ###################
+        ## mo1_bragg specific information
+        ###################
+
+        ## Logic if device exist
+        if "mo1_bragg" in self.device_manager.devices:
+
+            monochromator = instrument.create_group(name="monochromator")
+            monochromator.attrs["NX_class"] = "NXmonochromator"
+            crystal = monochromator.create_group(name="crystal")
+            crystal.attrs["NX_class"] = "NXcrystal"
+
+            # Create a dataset
+            chemical_formular = crystal.create_dataset(name="chemical_formular", data="Si")
+            chemical_formular.attrs["NX_class"] = "NX_CHAR"
+
+            reflection = crystal.create_soft_link(
+                name="reflection",
+                target="/entry/collection/devices/mo1_bragg/mo1_bragg_crystal_current_xtal_string/value",
+            )
+            reflection.attrs["NX_class"] = "NX_CHAR"
+
+            # Create a softlink
+            d_spacing = crystal.create_soft_link(
+                name="d_spacing",
+                target="/entry/collection/devices/mo1_bragg/mo1_bragg_crystal_current_d_spacing/value",
+            )
+            d_spacing.attrs["NX_class"] = "NX_FLOAT"
+            d_spacing.attrs["units"] = "angstrom"
+
+            bragg_offset = crystal.create_soft_link(
+                name="bragg_offset",
+                target="/entry/collection/devices/mo1_bragg/mo1_bragg_crystal_current_bragg_off/value",
+            )
+            bragg_offset.attrs["NX_class"] = "NX_FLOAT"
+            bragg_offset.attrs["units"] = "degree"
+
+            phi_offset = crystal.create_soft_link(
+                name="phi_offset",
+                target="/entry/collection/devices/mo1_bragg/mo1_bragg_crystal_current_phi_off/value",
+            )
+            phi_offset.attrs["NX_class"] = "NX_FLOAT"
+            phi_offset.attrs["units"] = "degree"
+
+            ## Logic if device exist
+            if "mo1_roty" in self.device_manager.devices:
+
+                # Create a softlink
+                azimuthal_angle = crystal.create_soft_link(
+                    name="azimuthal_angle",
+                    target="/entry/collection/devices/mo1_roty/mo1_roty/value",
+                )
+                azimuthal_angle.attrs["NX_class"] = "NX_FLOAT"
+                azimuthal_angle.attrs["units"] = "degree"
+
+            azm_offset = crystal.create_soft_link(
+                name="azm_offset",
+                target="/entry/collection/devices/mo1_bragg/mo1_bragg_crystal_current_azm_off/value",
+            )
+            azm_offset.attrs["NX_class"] = "NX_FLOAT"
+            azm_offset.attrs["units"] = "degree"
+
+            miscut = crystal.create_soft_link(
+                name="miscut",
+                target="/entry/collection/devices/mo1_bragg/mo1_bragg_crystal_current_miscut/value",
+            )
+            miscut.attrs["NX_class"] = "NX_FLOAT"
+            miscut.attrs["units"] = "degree"
+
+        ###################
+        ### cm mirror specific information
+        ####################
+
+        collimating_mirror = instrument.create_group(name="collimating_mirror")
+        collimating_mirror.attrs["NX_class"] = "NXmirror"
+
+        cm_substrate_material = collimating_mirror.create_dataset(
+            name="substrate_material", data="Si"
+        )
+        cm_substrate_material.attrs["NX_class"] = "NX_CHAR"
+
+        # previous error due to space in name field
+
+        if "cm_bnd_radius" in self.device_manager.devices:
+            cm_bending_radius = collimating_mirror.create_soft_link(
+                name="sagittal_radius",
+                target="/entry/collection/devices/cm_bnd_radius/cm_bnd_radius/value",
+            )
+            cm_bending_radius.attrs["NX_class"] = "NX_FLOAT"
+            cm_bending_radius.attrs["units"] = "km"
+
+        if "cm_rotx" in self.device_manager.devices:
+            cm_incidence_angle = collimating_mirror.create_soft_link(
+                name="incidence_angle", target="/entry/collection/devices/cm_rotx/cm_rotx/value"
+            )
+            cm_incidence_angle.attrs["NX_class"] = "NX_FLOAT"
+            cm_incidence_angle.attrs["units"] = "mrad"
+
+        if "cm_roty" in self.device_manager.devices:
+            cm_yaw_angle = collimating_mirror.create_soft_link(
+                name="yaw_angle", target="/entry/collection/devices/cm_roty/cm_roty/value"
+            )
+            cm_yaw_angle.attrs["NX_class"] = "NX_FLOAT"
+            cm_yaw_angle.attrs["units"] = "mrad"
+
+        if "cm_rotz" in self.device_manager.devices:
+            cm_roll_angle = collimating_mirror.create_soft_link(
+                name="roll_angle", target="/entry/collection/devices/cm_rotz/cm_rotz/value"
+            )
+            cm_roll_angle.attrs["NX_class"] = "NX_FLOAT"
+            cm_roll_angle.attrs["units"] = "mrad"
+
+        if "cm_trx" in self.device_manager.devices:
+            cm_trx = (
+                -self.device_manager.devices.cm_trx.read(cached=True).get("cm_trx").get("value")
+            )
+            stripe = "Unknown"
+            for name, low, high in zip(bl.cm.surface, bl.cm.limOptX[0], bl.cm.limOptX[1]):
+                if low <= cm_trx <= high:
+                    stripe = name
+            cm_stripe = collimating_mirror.create_dataset(name="stripe", data=stripe)
+            cm_stripe.attrs["NX_class"] = "NX_CHAR"
+
+        ###################
+        ### fm mirror specific information
+        ####################
+
+        focusing_mirror = instrument.create_group(name="focusing_mirror")
+        focusing_mirror.attrs["NX_class"] = "NXmirror"
+
+        fm_substrate_material = focusing_mirror.create_dataset(name="substrate_material", data="Si")
+        fm_substrate_material.attrs["NX_class"] = "NX_CHAR"
+
+        if "fm_bnd_radius" in self.device_manager.devices:
+            fm_bending_radius = focusing_mirror.create_soft_link(
+                name="sagittal_radius",
+                target="/entry/collection/devices/fm_bnd_radius/fm_bnd_radius/value",
+            )
+            fm_bending_radius.attrs["NX_class"] = "NX_FLOAT"
+            fm_bending_radius.attrs["units"] = "km"
+
+        if "fm_rotx" in self.device_manager.devices:
+            fm_incidence_angle = focusing_mirror.create_soft_link(
+                name="incidence_angle", target="/entry/collection/devices/fm_rotx/fm_rotx/value"
+            )
+            fm_incidence_angle.attrs["NX_class"] = "NX_FLOAT"
+            fm_incidence_angle.attrs["units"] = "mrad"
+
+        if "fm_roty" in self.device_manager.devices:
+            fm_yaw_angle = focusing_mirror.create_soft_link(
+                name="yaw_angle", target="/entry/collection/devices/fm_roty/fm_roty/value"
+            )
+            fm_yaw_angle.attrs["NX_class"] = "NX_FLOAT"
+            fm_yaw_angle.attrs["units"] = "mrad"
+
+        if "fm_rotz" in self.device_manager.devices:
+            fm_roll_angle = focusing_mirror.create_soft_link(
+                name="roll_angle", target="/entry/collection/devices/fm_rotz/fm_rotz/value"
+            )
+            fm_roll_angle.attrs["NX_class"] = "NX_FLOAT"
+            fm_roll_angle.attrs["units"] = "mrad"
+
+        if "fm_trx" in self.device_manager.devices:
+            fm_trx = (
+                -self.device_manager.devices.fm_trx.read(cached=True).get("fm_trx").get("value")
+            )
+            stripe = "Unknown"
+            for name, low, high in zip(
+                bl.fm.surfaceFlat, bl.fm.limOptXFlat[1], bl.fm.limOptXFlat[0]
+            ):
+                if low <= fm_trx <= high:
+                    stripe = name + " (flat)"
+            for name, low, high in zip(
+                bl.fm.surfaceToroid, bl.fm.limOptXToroid[1], bl.fm.limOptXToroid[0]
+            ):
+                if low <= fm_trx <= high:
+                    stripe = name + " (toroid)"
+            fm_stripe = focusing_mirror.create_dataset(name="stripe", data=stripe)
+            fm_stripe.attrs["NX_class"] = "NX_CHAR"
+
+        ###################
+        ## nidaq specific information
+        ###################
+
+        ## Logic if device exist
+        if "nidaq" in self.device_manager.devices:
+
+            # ai_chans_bits = self.device_manager.devices.nidaq.ai_chans.read(cached=True).get("nidaq_ai_chans").get("value")
+            ai_chans_bits = (
+                self.configuration.get("nidaq", {}).get("nidaq_ai_chans", {}).get("value")
+            )
+            ci_chans_bits = (
+                self.configuration.get("nidaq", {}).get("nidaq_ci_chans", {}).get("value")
+            )
+            # add_chans_bits = self.device_manager.devices.nidaq.add_chans.read(cached=True).get("nidaq_add_chans").get("value")
+            add_chans_bits = (
+                self.configuration.get("nidaq", {}).get("nidaq_add_chans", {}).get("value")
+            )
+
+            rle = self.configuration.get("nidaq", {}).get("nidaq_rle", {}).get("value")
+
+            measurement_mode = entry.create_group(name="mode")
+            measurement_mode.attrs["NX_class"] = "NX_CHAR"
+
+            if ci_chans_bits is not None:
+                if (int(ci_chans_bits) & 0x7F) != 0:
+                    # Create a dataset
+                    rayspec_sdd_active = measurement_mode.create_group(
+                        name="Multi_Element_Partial_Fluorescence_Yield"
+                    )
+                    me_sdd = rayspec_sdd_active.create_dataset(
+                        name="Detector", data="Rayspec 7 element Silicon Drift Detector"
+                    )
+                    me_sdd.attrs["NX_class"] = "NX_CHAR"
+
+                if (int(ci_chans_bits) & (1 << 8)) != 0:
+                    # Create a dataset
+                    ketek_sdd_active = measurement_mode.create_group(
+                        name="Single_Element_Partial_Fluorescence_Yield"
+                    )
+                    se_sdd = ketek_sdd_active.create_dataset(
+                        name="Detector", data="Ketex mini single element Silicon Drift Detector"
+                    )
+                    se_sdd.attrs["NX_class"] = "NX_CHAR"
+
+            if ai_chans_bits is not None:
+                if (int(ai_chans_bits) & (1 << 6)) != 0:
+                    # Create a dataset
+                    pips_active = measurement_mode.create_group(name="Total_Flourescence_Yield")
+                    tfy = pips_active.create_dataset(
+                        name="Detector",
+                        data="Mirion Technologies Partially Depeleted PIPS Detector",
+                    )
+                    tfy.attrs["NX_class"] = "NX_CHAR"
+
+                if ((int(ai_chans_bits) & (1 << 0)) != 0) & ((int(ai_chans_bits) & (1 << 2)) != 0):
+                    # Create a dataset
+                    ai0ai2_active = measurement_mode.create_group(name="Sample_Transmission")
+                    sam_trans = ai0ai2_active.create_dataset(
+                        name="Detector", data="Ionitec 15 cm gas filled Ionisation Chambers"
+                    )
+                    sam_trans.attrs["NX_class"] = "NX_CHAR"
+
+                if ((int(ai_chans_bits) & (1 << 2)) != 0) & ((int(ai_chans_bits) & (1 << 4)) != 0):
+                    # Create a dataset
+                    ai2ai4_active = measurement_mode.create_group(name="Reference_Transmission")
+                    ref_trans = ai2ai4_active.create_dataset(
+                        name="Detector", data="Ionitec 15 cm gas filled Ionisation Chambers"
+                    )
+                    ref_trans.attrs["NX_class"] = "NX_CHAR"
+
+            main_data = entry.create_group(name="data")
+            main_data.attrs["NX_class"] = "NXdata"
+
+            ##################
+            ## energy, test whether the signal exists. how to check from config?
+            ###################
+
+            energy = main_data.create_group(name="energy")
+            energy.attrs["NX_class"] = "NXdata"
+            energy.attrs["units"] = "eV"
+
+            main_data.create_soft_link(
+                name="energy",
+                target="/entry/collection/readout_groups/async/nidaq/nidaq_energy/value",
+            )
+
+            ##################
+            ## i0
+            ###################
+
+            if (int(ai_chans_bits) & (1 << 0)) != 0:
+                i0 = main_data.create_group(name="i0")
+                i0.attrs["NX_class"] = "NXdata"
+                i0.attrs["units"] = "V"
+
+                if rle:
+                    target = "/entry/collection/readout_groups/async/nidaq/nidaq_ai0_mean/value"
+                else:
+                    target = "/entry/collection/readout_groups/async/nidaq/nidaq_ai0/value"
+                main_data.create_soft_link(name="i0", target=target)
+
+            ##################
+            ## i1
+            ###################
+
+            if (int(ai_chans_bits) & (1 << 2)) != 0:
+                i1 = main_data.create_group(name="i1")
+                i1.attrs["NX_class"] = "NXdata"
+                i1.attrs["units"] = "V"
+
+                if rle:
+                    target = "/entry/collection/readout_groups/async/nidaq/nidaq_ai2_mean/value"
+                else:
+                    target = "/entry/collection/readout_groups/async/nidaq/nidaq_ai2/value"
+                main_data.create_soft_link(name="i1", target=target)
+
+            ##################
+            ## i2
+            ###################
+
+            if (int(ai_chans_bits) & (1 << 4)) != 0:
+                i2 = main_data.create_group(name="i2")
+                i2.attrs["NX_class"] = "NXdata"
+                i2.attrs["units"] = "V"
+
+                if rle:
+                    target = "/entry/collection/readout_groups/async/nidaq/nidaq_ai4_mean/value"
+                else:
+                    target = "/entry/collection/readout_groups/async/nidaq/nidaq_ai4/value"
+                main_data.create_soft_link(name="i2", target=target)
+
+            ##################
+            ## ci sum
+            ###################
+
+            if int(ci_chans_bits) > 0:
+                ci_sum = main_data.create_group(name="Fluorescence_Sum")
+                ci_sum.attrs["NX_class"] = "NXdata"
+                ci_sum.attrs["units"] = "counts"
+
+                main_data.create_soft_link(
+                    name="Fluorescence_Sum",
+                    target="/entry/collection/readout_groups/async/nidaq/nidaq_cisum/value",
+                )
+
+            ##################
+            ## mu sample, test whether the signal exists. how to check from config?
+            ###################
+
+            if (int(add_chans_bits) & (1 << 0)) != 0:
+                mu_sample = main_data.create_group(name="mu_sample")
+                mu_sample.attrs["NX_class"] = "NXdata"
+
+                main_data.create_soft_link(
+                    name="mu_sample",
+                    target="/entry/collection/readout_groups/async/nidaq/nidaq_smpl_abs/value",
+                )
+
+            ##################
+            ## fluo sample, test whether the signal exists. how to check from config?
+            ###################
+
+            if (int(add_chans_bits) & (1 << 1)) != 0:
+                mu_sample = main_data.create_group(name="fluo_sample")
+                mu_sample.attrs["NX_class"] = "NXdata"
+
+                main_data.create_soft_link(
+                    name="fluo_sample",
+                    target="/entry/collection/readout_groups/async/nidaq/nidaq_smpl_fluo/value",
+                )
+
+            ##################
+            ## mu reference, test whether the signal exists. how to check from config?
+            ###################
+
+            if (int(add_chans_bits) & (1 << 2)) != 0:
+                mu_reference = main_data.create_group(name="mu_reference")
+                mu_reference.attrs["NX_class"] = "NXdata"
+
+                main_data.create_soft_link(
+                    name="mu_reference",
+                    target="/entry/collection/readout_groups/async/nidaq/nidaq_ref_abs/value",
+                )

debye_bec/scans/__init__.py

@@ -1,7 +1,7 @@
-from .mono_bragg_scans import (
-    XASAdvancedScan,
-    XASAdvancedScanWithXRD,
-    XASSimpleScan,
-    XASSimpleScanWithXRD,
+from .nidaq_continuous_scan import NidaqContinuousScan
+from .xas_simple_scan import (
+    XasAdvancedScan,
+    XasAdvancedScanWithXrd,
+    XasSimpleScan,
+    XasSimpleScanWithXrd,
 )
-from .nidaq_cont_scan import NIDAQContinuousScan

debye_bec/scans/mono_bragg_scans.py

@@ -1,310 +0,0 @@
-"""This module contains the scan classes for the mono bragg motor of the Debye beamline."""
-
-import time
-from typing import Literal
-
-import numpy as np
-from bec_lib.device import DeviceBase
-from bec_lib.logger import bec_logger
-from bec_server.scan_server.scans import AsyncFlyScanBase
-
-logger = bec_logger.logger
-
-
-class XASSimpleScan(AsyncFlyScanBase):
-    """Class for the XAS simple scan"""
-
-    scan_name = "xas_simple_scan"
-    scan_type = "fly"
-    scan_report_hint = "device_progress"
-    required_kwargs = []
-    use_scan_progress_report = False
-    pre_move = False
-    gui_config = {
-        "Movement Parameters": ["start", "stop"],
-        "Scan Parameters": ["scan_time", "scan_duration"],
-    }
-
-    def __init__(
-        self,
-        start: float,
-        stop: float,
-        scan_time: float,
-        scan_duration: float,
-        motor: DeviceBase = "mo1_bragg",
-        **kwargs,
-    ):
-        """The xas_simple_scan is used to start a simple oscillating scan on the mono bragg motor.
-        Start and Stop define the energy range for the scan, scan_time is the time for one scan
-        cycle and scan_duration is the duration of the scan. If scan duration is set to 0, the
-        scan will run infinitely.
-
-        Args:
-            start (float): Start energy for the scan.
-            stop (float): Stop energy for the scan.
-            scan_time (float): Time for one scan cycle.
-            scan_duration (float): Duration of the scan.
-            motor (DeviceBase, optional): Motor device to be used for the scan.
-                                          Defaults to "mo1_bragg".
-        Examples:
-            >>> scans.xas_simple_scan(start=8000, stop=9000, scan_time=1, scan_duration=10)
-        """
-        super().__init__(**kwargs)
-        self.motor = motor
-        self.start = start
-        self.stop = stop
-        self.scan_time = scan_time
-        self.scan_duration = scan_duration
-        self.primary_readout_cycle = 1
-
-    def update_readout_priority(self):
-        """Ensure that NIDAQ is not monitored for any quick EXAFS."""
-        super().update_readout_priority()
-        self.readout_priority["async"].append("nidaq")
-
-    def prepare_positions(self):
-        """Prepare the positions for the scan.
-
-        Use here only start and end energy defining the range for the scan.
-        """
-        self.positions = np.array([self.start, self.stop], dtype=float)
-        self.num_pos = None
-        yield None
-
-    def pre_scan(self):
-        """Pre Scan action."""
-
-        self._check_limits()
-        # Ensure parent class pre_scan actions to be called.
-        yield from super().pre_scan()
-
-    def scan_report_instructions(self):
-        """
-        Return the instructions for the scan report.
-        """
-        yield from self.stubs.scan_report_instruction({"device_progress": [self.motor]})
-
-    def scan_core(self):
-        """Run the scan core.
-        Kickoff the oscillation on the Bragg motor and wait for the completion of the motion.
-        """
-        # Start the oscillation on the Bragg motor.
-        yield from self.stubs.kickoff(device=self.motor)
-        complete_status = yield from self.stubs.complete(device=self.motor, wait=False)
-
-        while not complete_status.done:
-            # Readout monitored devices
-            yield from self.stubs.read(group="monitored", point_id=self.point_id)
-            time.sleep(self.primary_readout_cycle)
-            self.point_id += 1
-
-        self.num_pos = self.point_id
-
-
-class XASSimpleScanWithXRD(XASSimpleScan):
-    """Class for the XAS simple scan with XRD"""
-
-    scan_name = "xas_simple_scan_with_xrd"
-    gui_config = {
-        "Movement Parameters": ["start", "stop"],
-        "Scan Parameters": ["scan_time", "scan_duration"],
-        "Low Energy Break": ["break_enable_low", "break_time_low", "cycle_low"],
-        "High Energy Break": ["break_enable_high", "break_time_high", "cycle_high"],
-        "XRD Triggers": ["exp_time", "n_of_trigger"],
-    }
-
-    def __init__(
-        self,
-        start: float,
-        stop: float,
-        scan_time: float,
-        scan_duration: float,
-        break_enable_low: bool,
-        break_time_low: float,
-        cycle_low: int,
-        break_enable_high: bool,
-        break_time_high: float,
-        cycle_high: float,
-        exp_time: float,
-        n_of_trigger: int,
-        motor: DeviceBase = "mo1_bragg",
-        **kwargs,
-    ):
-        """The xas_simple_scan_with_xrd is an oscillation motion on the mono motor
-        with XRD triggering at low and high energy ranges.
-        If scan duration is set to 0, the scan will run infinitely.
-
-        Args:
-            start (float): Start energy for the scan.
-            stop (float): Stop energy for the scan.
-            scan_time (float): Time for one oscillation .
-            scan_duration (float): Total duration of the scan.
-            break_enable_low (bool): Enable breaks for the low energy range.
-            break_time_low (float): Break time for the low energy range.
-            cycle_low (int): Specify how often the triggers should be considered,
-                             every nth cycle for low
-            break_enable_high (bool): Enable breaks for the high energy range.
-            break_time_high (float): Break time for the high energy range.
-            cycle_high (int): Specify how often the triggers should be considered,
-                              every nth cycle for high
-            exp_time (float): Length of 1 trigger period in seconds
-            n_of_trigger (int): Amount of triggers to be fired during break
-            motor (DeviceBase, optional): Motor device to be used for the scan.
-                                          Defaults to "mo1_bragg".
-
-        Examples:
-            >>> scans.xas_simple_scan_with_xrd(start=8000, stop=9000, scan_time=1, scan_duration=10, xrd_enable_low=True, num_trigger_low=5, cycle_low=2, exp_time_low=100, xrd_enable_high=False, num_trigger_high=3, cycle_high=1, exp_time_high=1000)
-        """
-        super().__init__(
-            start=start,
-            stop=stop,
-            scan_time=scan_time,
-            scan_duration=scan_duration,
-            motor=motor,
-            **kwargs,
-        )
-        self.break_enable_low = break_enable_low
-        self.break_time_low = break_time_low
-        self.cycle_low = cycle_low
-        self.break_enable_high = break_enable_high
-        self.break_time_high = break_time_high
-        self.cycle_high = cycle_high
-        self.exp_time = exp_time
-        self.n_of_trigger = n_of_trigger
-
-
-class XASAdvancedScan(XASSimpleScan):
-    """Class for the XAS advanced scan"""
-
-    scan_name = "xas_advanced_scan"
-    gui_config = {
-        "Movement Parameters": ["start", "stop"],
-        "Scan Parameters": ["scan_time", "scan_duration"],
-        "Spline Parameters": ["p_kink", "e_kink"],
-    }
-
-    def __init__(
-        self,
-        start: float,
-        stop: float,
-        scan_time: float,
-        scan_duration: float,
-        p_kink: float,
-        e_kink: float,
-        motor: DeviceBase = "mo1_bragg",
-        **kwargs,
-    ):
-        """The xas_advanced_scan is an oscillation motion on the mono motor.
-        Start and Stop define the energy range for the scan, scan_time is the
-        time for one scan cycle and scan_duration is the duration of the scan.
-        If scan duration is set to 0, the scan will run infinitely.
-        p_kink and e_kink add a kink to the motion profile to slow down in the
-        exafs region of the scan.
-
-        Args:
-            start (float): Start angle for the scan.
-            stop (float): Stop angle for the scan.
-            scan_time (float): Time for one oscillation .
-            scan_duration (float): Total duration of the scan.
-            p_kink (float): Position of the kink.
-            e_kink (float): Energy of the kink.
-            motor (DeviceBase, optional): Motor device to be used for the scan.
-                                          Defaults to "mo1_bragg".
-
-        Examples:
-            >>> scans.xas_advanced_scan(start=10000, stop=12000, scan_time=0.5, scan_duration=10, p_kink=50, e_kink=10500)
-        """
-        super().__init__(
-            start=start,
-            stop=stop,
-            scan_time=scan_time,
-            scan_duration=scan_duration,
-            motor=motor,
-            **kwargs,
-        )
-        self.p_kink = p_kink
-        self.e_kink = e_kink
-
-
-class XASAdvancedScanWithXRD(XASAdvancedScan):
-    """Class for the XAS advanced scan with XRD"""
-
-    scan_name = "xas_advanced_scan_with_xrd"
-    gui_config = {
-        "Movement Parameters": ["start", "stop"],
-        "Scan Parameters": ["scan_time", "scan_duration"],
-        "Spline Parameters": ["p_kink", "e_kink"],
-        "Low Energy Break": ["break_enable_low", "break_time_low", "cycle_low"],
-        "High Energy Break": ["break_enable_high", "break_time_high", "cycle_high"],
-        "XRD Triggers": ["exp_time", "n_of_trigger"],
-    }
-
-    def __init__(
-        self,
-        start: float,
-        stop: float,
-        scan_time: float,
-        scan_duration: float,
-        p_kink: float,
-        e_kink: float,
-        break_enable_low: bool,
-        break_time_low: float,
-        cycle_low: int,
-        break_enable_high: bool,
-        break_time_high: float,
-        cycle_high: float,
-        exp_time: float,
-        n_of_trigger: int,
-        motor: DeviceBase = "mo1_bragg",
-        **kwargs,
-    ):
-        """The xas_advanced_scan is an oscillation motion on the mono motor
-        with XRD triggering at low and high energy ranges.
-        Start and Stop define the energy range for the scan, scan_time is the time for
-        one scan cycle and scan_duration is the duration of the scan. If scan duration
-        is set to 0, the scan will run infinitely. p_kink and e_kink add a kink to the
-        motion profile to slow down in the exafs region of the scan.
-
-        Args:
-            start (float): Start angle for the scan.
-            stop (float): Stop angle for the scan.
-            scan_time (float): Time for one oscillation .
-            scan_duration (float): Total duration of the scan.
-            p_kink (float): Position of kink.
-            e_kink (float): Energy of the kink.
-            break_enable_low (bool): Enable breaks for the low energy range.
-            break_time_low (float): Break time for the low energy range.
-            cycle_low (int): Specify how often the triggers should be considered,
-                             every nth cycle for low
-            break_enable_high (bool): Enable breaks for the high energy range.
-            break_time_high (float): Break time for the high energy range.
-            cycle_high (int): Specify how often the triggers should be considered,
-                              every nth cycle for high
-            exp_time (float): Length of 1 trigger period in seconds
-            n_of_trigger (int): Amount of triggers to be fired during break
-            motor (DeviceBase, optional): Motor device to be used for the scan.
-                                          Defaults to "mo1_bragg".
-
-        Examples:
-            >>> scans.xas_advanced_scan_with_xrd(start=10000, stop=12000, scan_time=0.5, scan_duration=10, p_kink=50, e_kink=10500, xrd_enable_low=True, num_trigger_low=5, cycle_low=2, exp_time_low=100, xrd_enable_high=False, num_trigger_high=3, cycle_high=1, exp_time_high=1000)
-        """
-        super().__init__(
-            start=start,
-            stop=stop,
-            scan_time=scan_time,
-            scan_duration=scan_duration,
-            p_kink=p_kink,
-            e_kink=e_kink,
-            motor=motor,
-            **kwargs,
-        )
-        self.p_kink = p_kink
-        self.e_kink = e_kink
-        self.break_enable_low = break_enable_low
-        self.break_time_low = break_time_low
-        self.cycle_low = cycle_low
-        self.break_enable_high = break_enable_high
-        self.break_time_high = break_time_high
-        self.cycle_high = cycle_high
-        self.exp_time = exp_time
-        self.n_of_trigger = n_of_trigger

debye_bec/scans/nidaq_cont_scan.py

@@ -1,84 +0,0 @@
-"""This module contains the scan class for the nidaq of the Debye beamline for use in continuous mode."""
-
-import time
-from typing import Literal
-
-import numpy as np
-from bec_lib.device import DeviceBase
-from bec_lib.logger import bec_logger
-from bec_server.scan_server.scans import AsyncFlyScanBase
-
-logger = bec_logger.logger
-
-
-class NIDAQContinuousScan(AsyncFlyScanBase):
-    """Class for the nidaq continuous scan (without mono)"""
-
-    scan_name = "nidaq_continuous_scan"
-    scan_type = "fly"
-    scan_report_hint = "device_progress"
-    required_kwargs = []
-    use_scan_progress_report = False
-    pre_move = False
-    gui_config = {"Scan Parameters": ["scan_duration"], "Data Compression": ["compression"]}
-
-    def __init__(
-        self, scan_duration: float, daq: DeviceBase = "nidaq", compression: bool = False, **kwargs
-    ):
-        """The NIDAQ continuous scan is used to measure with the NIDAQ without moving the
-        monochromator or any other motor. The NIDAQ thus runs in continuous mode, with a
-        set scan_duration.
-
-        Args:
-            scan_duration (float): Duration of the scan.
-            daq (DeviceBase, optional): DAQ device to be used for the scan.
-                                          Defaults to "nidaq".
-        Examples:
-            >>> scans.nidaq_continuous_scan(scan_duration=10)
-        """
-        super().__init__(**kwargs)
-        self.scan_duration = scan_duration
-        self.daq = daq
-        self.start_time = 0
-        self.primary_readout_cycle = 1
-        self.scan_parameters["scan_duration"] = scan_duration
-        self.scan_parameters["compression"] = compression
-
-    def update_readout_priority(self):
-        """Ensure that NIDAQ is not monitored for any quick EXAFS."""
-        super().update_readout_priority()
-        self.readout_priority["async"].append("nidaq")
-
-    def prepare_positions(self):
-        """Prepare the positions for the scan."""
-        yield None
-
-    def pre_scan(self):
-        """Pre Scan action."""
-
-        self.start_time = time.time()
-        # Ensure parent class pre_scan actions to be called.
-        yield from super().pre_scan()
-
-    def scan_report_instructions(self):
-        """
-        Return the instructions for the scan report.
-        """
-        yield from self.stubs.scan_report_instruction({"device_progress": [self.daq]})
-
-    def scan_core(self):
-        """Run the scan core.
-        Kickoff the acquisition of the NIDAQ wait for the completion of the scan.
-        """
-        kickoff_status = yield from self.stubs.kickoff(device=self.daq)
-        kickoff_status.wait(timeout=5)  # wait for proper kickoff of device
-
-        complete_status = yield from self.stubs.complete(device=self.daq, wait=False)
-
-        while not complete_status.done:
-            # Readout monitored devices
-            yield from self.stubs.read(group="monitored", point_id=self.point_id)
-            time.sleep(self.primary_readout_cycle)
-            self.point_id += 1
-
-        self.num_pos = self.point_id

debye_bec/scans/nidaq_continuous_scan.py

@@ -0,0 +1,174 @@
+"""
+The NIDAQ continuous scan is used to measure with the NIDAQ without moving the monochromator or any other motor.
+
+Scan procedure:
+    - prepare_scan
+    - open_scan
+    - stage
+    - pre_scan
+    - scan_core
+        - at_each_point (optionally called by scan_core)
+    - post_scan
+    - unstage
+    - close_scan
+    - on_exception (called if any exception is raised during the scan)
+"""
+
+from __future__ import annotations
+
+import time
+from typing import Annotated
+
+from bec_lib.device import DeviceBase
+from bec_lib.scan_args import ScanArgument, Units
+from bec_server.scan_server.scans.scan_base import ScanBase, ScanType
+from bec_server.scan_server.scans.scan_modifier import scan_hook
+
+
+class NidaqContinuousScan(ScanBase):
+    # Scan Type: Hardware triggered or software triggered?
+    # If the main trigger and readout logic is done within the at_each_point method in scan_core, choose SOFTWARE_TRIGGERED.
+    # If the main trigger and readout logic is implemented on a device that is simply kicked off in this scan, choose HARDWARE_TRIGGERED.
+    # This primarily serves as information for devices: The device may need to react differently if a software trigger is expected
+    # for every point.
+    scan_type = ScanType.HARDWARE_TRIGGERED
+
+    # Scan name: This is the name of the scan, e.g. "line_scan". This is used for display purposes and to identify the scan type in user interfaces.
+    # Choose a descriptive name that does not conflict with existing scan names.
+    # It must be a valid Python identifier, that is, it can only contain letters, numbers, and underscores, and must not start with a number.
+    scan_name = "nidaq_continuous_scan"
+
+    gui_config = {"Scan Parameters": ["scan_duration", "daq", "compression"]}
+
+    def __init__(
+        self,
+        #fmt: off
+        scan_duration: Annotated[float, ScanArgument(display_name="Scan Duration", description="Duration of the scan", units=Units.s)],
+        daq: Annotated[DeviceBase | None, ScanArgument(display_name="Daq", description="DAQ device to be used for the scan")] = None,
+        compression: Annotated[bool, ScanArgument(display_name="Compression", description="Whether to compress the data")]= False,
+        #fmt: on
+        **kwargs,
+    ):
+        """
+        The NIDAQ continuous scan is used to measure with the NIDAQ without moving the
+        monochromator or any other motor. The NIDAQ thus runs in continuous mode, with a
+        set scan_duration.
+
+        Args:
+            scan_duration (float): Duration of the scan
+            daq (DeviceBase): DAQ device to be used for the scan
+            compression (bool): Whether to compress the data
+
+        Returns:
+            ScanReport
+        """
+        super().__init__(**kwargs)
+        self._baseline_readout_status = None
+        self.scan_duration = scan_duration
+        self.daq = daq or self.dev["nidaq"]
+        self.compression = compression
+        self.monitored_readout_cycle = 1.0  # seconds
+        self.motors = [self.daq]
+
+        self.update_scan_info(scan_duration=scan_duration, compression=compression)
+        self.actions.set_device_readout_priority([self.daq], priority="async")
+
+    @scan_hook
+    def prepare_scan(self):
+        """
+        Prepare the scan. This can include any steps that need to be executed
+        before the scan is opened, such as preparing the positions (if not done already)
+        or setting up the devices.
+        """
+
+        self.actions.add_scan_report_instruction_device_progress(self.daq)
+        self._baseline_readout_status = self.actions.read_baseline_devices(wait=False)
+
+    @scan_hook
+    def open_scan(self):
+        """
+        Open the scan.
+        This step must call self.actions.open_scan() to ensure that a new scan is
+        opened. Make sure to prepare the scan metadata before, either in
+        prepare_scan() or in open_scan() itself and call self.update_scan_info(...)
+        to update the scan metadata if needed.
+        """
+        self.actions.open_scan()
+
+    @scan_hook
+    def stage(self):
+        """
+        Stage the devices for the upcoming scan. The stage logic is typically
+        implemented on the device itself (i.e. by the device's stage method).
+        However, if there are any additional steps that need to be executed before
+        staging the devices, they can be implemented here.
+        """
+        self.actions.stage_all_devices()
+
+    @scan_hook
+    def pre_scan(self):
+        """
+        Pre-scan steps to be executed before the main scan logic.
+        This is typically the last chance to prepare the devices before the core scan
+        logic is executed. For example, this is a good place to initialize time-criticial
+        devices, e.g. devices that have a short timeout.
+        The pre-scan logic is typically implemented on the device itself.
+        """
+        self.actions.pre_scan_all_devices()
+
+    @scan_hook
+    def scan_core(self):
+        """
+        Core scan logic to be executed during the scan.
+        This is where the main scan logic should be implemented.
+        """
+
+        kickoff_status = self.actions.kickoff(device=self.daq, wait=False)
+        kickoff_status.wait(timeout=5)  # wait for proper kickoff of device
+
+        complete_status = self.actions.complete(device=self.daq, wait=False)
+
+        while not complete_status.done:
+            self.at_each_point()
+            time.sleep(self.monitored_readout_cycle)
+
+    @scan_hook
+    def at_each_point(self):
+        """
+        Logic to be executed at each acquisition point during the scan.
+        """
+        self.actions.read_monitored_devices()
+
+    @scan_hook
+    def post_scan(self):
+        """
+        Post-scan steps to be executed after the main scan logic.
+        """
+        self.actions.complete_all_devices()
+
+    @scan_hook
+    def unstage(self):
+        """Unstage the scan by executing post-scan steps."""
+        self.actions.unstage_all_devices()
+
+    @scan_hook
+    def close_scan(self):
+        """Close the scan."""
+        if self._baseline_readout_status is not None:
+            self._baseline_readout_status.wait()
+        self.actions.close_scan()
+        self.actions.check_for_unchecked_statuses()
+
+    @scan_hook
+    def on_exception(self, exception: Exception):
+        """
+        Handle exceptions that occur during the scan.
+        This is a good place to implement any cleanup logic that needs to be executed in case of an exception,
+        such as returning the devices to a safe state or moving the motors back to their starting position.
+        """
+
+    #######################################################
+    ######### Helper methods for the scan logic ###########
+    #######################################################
+
+    # Implement scan-specific helper methods below.

debye_bec/scans/scan_customization/scan_components.py

@@ -0,0 +1,12 @@
+"""
+Scan components for debye_bec.
+
+The scan components module allows you to define custom components that can be used in your scans.
+These components can be used to encapsulate reusable logic, interact with devices, or perform specific actions during the scan lifecycle.
+"""
+
+from bec_server.scan_server.scans.scan_components import ScanComponents
+
+
+class DebyeBecScanComponents(ScanComponents):
+    """Scan components for debye_bec."""

debye_bec/scans/scan_customization/scan_modifier.py

@@ -0,0 +1,33 @@
+"""
+Scan modifier plugin for debye_bec.
+
+The scan modifier allows you to modify the scan lifecycle and run custom actions before or after the scan hook or replace the scan hook entirely.
+Note that the scan_modifier module must be registered as a plugin in the pyproject.toml file for it to be recognized by the BEC framework and that
+there can only be one scan_modifier plugin registered at a time. If you need to run multiple scan modifiers, you can create a single scan
+modifier plugin that runs multiple actions in sequence with conditional logic to determine which actions to run based on the scan context.
+"""
+
+from bec_server.scan_server.scans.scan_modifier import ScanModifier, scan_hook_impl
+
+
+class DebyeBecScanModifier(ScanModifier):
+    """
+    Scan modifier for debye_bec.
+    
+    By inheriting from the ScanModifier base class, you get access to currently running scan (self.scan), the devices (self.dev), the scan info (self.scan_info),
+    the scan components (self.components) and the scan actions (self.actions).
+    """
+
+    def __init__(self, **kwargs):
+        """Initialize the scan modifier."""
+        super().__init__(**kwargs)
+
+    # Example of running code before the scan stage for a specific scan
+    # @scan_hook_impl("stage", "before")
+    # def before_stage(self):
+    #     """Run before the stage hook."""
+    #     self.actions.send_client_info("Custom stage logic executed by ScanModifier.")
+    #     if self.scan_info.scan_name == "example_scan":
+    #         self.dev.samx.set(20)
+
+    

debye_bec/scans/xas_simple_scan.py

@@ -0,0 +1,326 @@
+"""
+V4 implementation of the Debye XAS simple scan.
+
+Scan procedure:
+    - prepare_scan
+    - open_scan
+    - stage
+    - pre_scan
+    - scan_core
+        - at_each_point (optionally called by scan_core)
+    - post_scan
+    - unstage
+    - close_scan
+    - on_exception (called if any exception is raised during the scan)
+"""
+
+from __future__ import annotations
+
+import time
+from typing import Annotated
+
+import numpy as np
+from bec_lib.device import DeviceBase
+from bec_lib.scan_args import ScanArgument, Units
+from bec_server.scan_server.scans.scan_base import ScanBase, ScanType
+from bec_server.scan_server.scans.scan_modifier import scan_hook
+
+
+class XasSimpleScan(ScanBase):
+    scan_type = ScanType.HARDWARE_TRIGGERED
+    scan_name = "xas_simple_scan"
+
+    gui_config = {
+        "Movement Parameters": ["start", "stop"],
+        "Scan Parameters": ["scan_time", "scan_duration", "monitored_readout_cycle"],
+    }
+
+    def __init__(
+        self,
+        # fmt: off
+        start: Annotated[float, ScanArgument(display_name="Start Energy", description="Start energy.", units=Units.eV, ge=4500, le=64000)],
+        stop: Annotated[float, ScanArgument(display_name="Stop Energy", description="Stop energy.", units=Units.eV, ge=4500, le=64000)],
+        scan_time: Annotated[float, ScanArgument(display_name="Scan Time", description="Time for one scan cycle.", units=Units.s, ge=0.05)],
+        scan_duration: Annotated[float, ScanArgument(display_name="Scan Duration", description="Total scan duration.", units=Units.s, ge=0.05)],
+        motor: Annotated[DeviceBase | None, ScanArgument(display_name="Motor", description="Bragg motor device.")] = None,
+        daq: Annotated[DeviceBase | None, ScanArgument(display_name="DAQ", description="NIDAQ device.")] = None,
+        monitored_readout_cycle: Annotated[float, ScanArgument(display_name="Monitored Readout Cycle", description="Delay between monitored readouts.",units=Units.s, gt=0)] = 1,
+        # fmt: on
+        **kwargs,
+    ):
+        """
+        Start a simple oscillating scan on the mono bragg motor.
+
+        Args:
+            start (float): Start energy.
+            stop (float): Stop energy.
+            scan_time (float): Time for one scan cycle.
+            scan_duration (float): Total scan duration.
+            motor (DeviceBase | None): Bragg motor device.
+            daq (DeviceBase | None): NIDAQ device.
+            monitored_readout_cycle (float): Delay between monitored readouts.
+
+        Returns:
+            ScanReport
+        """
+        super().__init__(**kwargs)
+        self.start = start
+        self.stop = stop
+        self.scan_time = scan_time
+        self.scan_duration = scan_duration
+        self.motor = motor if motor is not None else self.dev["mo1_bragg"]
+        self.daq = daq if daq is not None else self.dev["nidaq"]
+        self.monitored_readout_cycle = monitored_readout_cycle
+        self.positions = np.array([self.start, self.stop], dtype=float)
+
+        # We pass on the arguments as "additional_scan_parameters" in the scan info
+        self.update_scan_info(
+            positions=self.positions,
+            scan_time=scan_time,
+            scan_duration=scan_duration,
+            monitored_readout_cycle=monitored_readout_cycle,
+        )
+        self.actions.set_device_readout_priority([self.daq], priority="async")
+
+    @scan_hook
+    def prepare_scan(self):
+        """
+        Prepare the scan. This can include any steps that need to be executed
+        before the scan is opened, such as preparing the positions (if not done already)
+        or setting up the devices.
+        """
+        self.actions.add_scan_report_instruction_device_progress(self.motor)
+        self._baseline_readout_status = self.actions.read_baseline_devices(wait=False)
+
+    @scan_hook
+    def open_scan(self):
+        """
+        Open the scan.
+        This step must call self.actions.open_scan() to ensure that a new scan is
+        opened. Make sure to prepare the scan metadata before, either in
+        prepare_scan() or in open_scan() itself and call self.update_scan_info(...)
+        to update the scan metadata if needed.
+        """
+        self.actions.open_scan()
+
+    @scan_hook
+    def stage(self):
+        """
+        Stage the devices for the upcoming scan. The stage logic is typically
+        implemented on the device itself (i.e. by the device's stage method).
+        However, if there are any additional steps that need to be executed before
+        staging the devices, they can be implemented here.
+        """
+        self.actions.stage_all_devices()
+
+    @scan_hook
+    def pre_scan(self):
+        """
+        Pre-scan steps to be executed before the main scan logic.
+        This is typically the last chance to prepare the devices before the core scan
+        logic is executed. For example, this is a good place to initialize time-criticial
+        devices, e.g. devices that have a short timeout.
+        The pre-scan logic is typically implemented on the device itself.
+        """
+        self.actions.pre_scan_all_devices()
+
+    @scan_hook
+    def scan_core(self):
+        """
+        Core scan logic to be executed during the scan.
+        This is where the main scan logic should be implemented.
+        """
+        self.actions.kickoff(self.motor)
+        completion_status = self.actions.complete(self.motor, wait=False)
+
+        while not completion_status.done:
+            self.at_each_point()
+
+    @scan_hook
+    def at_each_point(self):
+        """
+        Logic to be executed at each acquisition point during the scan.
+        """
+        self.actions.read_monitored_devices()
+        time.sleep(self.monitored_readout_cycle)
+
+    @scan_hook
+    def post_scan(self):
+        """
+        Post-scan steps to be executed after the main scan logic.
+        """
+        self.actions.complete_all_devices()
+
+    @scan_hook
+    def unstage(self):
+        """Unstage the scan by executing post-scan steps."""
+        self.actions.unstage_all_devices()
+
+    @scan_hook
+    def close_scan(self):
+        """Close the scan."""
+        if self._baseline_readout_status is not None:
+            self._baseline_readout_status.wait()
+        self.actions.close_scan()
+        self.actions.check_for_unchecked_statuses()
+
+    @scan_hook
+    def on_exception(self, exception: Exception):
+        """
+        Handle exceptions that occur during the scan.
+        This is a good place to implement any cleanup logic that needs to be executed in case of an exception,
+        such as returning the devices to a safe state or moving the motors back to their starting position.
+        """
+        self.actions.complete_all_devices(wait=False)
+
+
+class XasSimpleScanWithXrd(XasSimpleScan):
+    scan_name = "xas_simple_scan_with_xrd"
+    gui_config = {
+        "Movement Parameters": ["start", "stop"],
+        "Scan Parameters": ["scan_time", "scan_duration", "monitored_readout_cycle"],
+        "Low Energy Break": ["break_enable_low", "break_time_low", "cycle_low"],
+        "High Energy Break": ["break_enable_high", "break_time_high", "cycle_high"],
+        "XRD Triggers": ["exp_time", "n_of_trigger"],
+    }
+
+    def __init__(
+        self,
+        # fmt: off
+        start: Annotated[float, ScanArgument(display_name="Start Energy", description="Start energy.", units=Units.eV)],
+        stop: Annotated[float, ScanArgument(display_name="Stop Energy", description="Stop energy.", units=Units.eV)],
+        scan_time: Annotated[float, ScanArgument(display_name="Scan Time", description="Time for one scan cycle.", units=Units.s, ge=0)],
+        scan_duration: Annotated[float, ScanArgument(display_name="Scan Duration", description="Total scan duration.", units=Units.s, ge=0)],
+        break_enable_low: Annotated[bool, ScanArgument(display_name="Break Enable Low", description="Enable breaks for the low energy range.")],
+        break_time_low: Annotated[float, ScanArgument(display_name="Break Time Low", description="Break time for the low energy range.", units=Units.s, ge=0)],
+        cycle_low: Annotated[int, ScanArgument(display_name="Cycle Low", description="Use triggers every nth low-energy cycle.", ge=0)],
+        break_enable_high: Annotated[bool, ScanArgument(display_name="Break Enable High", description="Enable breaks for the high energy range.")],
+        break_time_high: Annotated[float, ScanArgument(display_name="Break Time High", description="Break time for the high energy range.", units=Units.s, ge=0)],
+        cycle_high: Annotated[int, ScanArgument(display_name="Cycle High", description="Use triggers every nth high-energy cycle.", ge=0)],
+        exp_time: Annotated[float, ScanArgument(display_name="Exposure Time", description="Length of one trigger period.", units=Units.s, ge=0)],
+        n_of_trigger: Annotated[int, ScanArgument(display_name="Number Of Trigger", description="Amount of triggers fired during a break.", ge=0)],
+        motor: Annotated[DeviceBase | None, ScanArgument(display_name="Motor", description="Bragg motor device.")] = None,
+        daq: Annotated[DeviceBase | None, ScanArgument(display_name="DAQ", description="NIDAQ device.")] = None,
+        monitored_readout_cycle: Annotated[float, ScanArgument(display_name="Monitored Readout Cycle", description="Delay between monitored readouts.", units=Units.s, gt=0)] = 1,
+        **kwargs,
+        # fmt: on
+    ):
+        super().__init__(
+            start=start,
+            stop=stop,
+            scan_time=scan_time,
+            scan_duration=scan_duration,
+            motor=motor,
+            daq=daq,
+            monitored_readout_cycle=monitored_readout_cycle,
+            **kwargs,
+        )
+
+        # We pass on the arguments as "additional_scan_parameters" in the scan info
+        self.update_scan_info(
+            break_enable_low=break_enable_low,
+            break_time_low=break_time_low,
+            cycle_low=cycle_low,
+            break_enable_high=break_enable_high,
+            break_time_high=break_time_high,
+            cycle_high=cycle_high,
+            exp_time=exp_time,
+            n_of_trigger=n_of_trigger,
+        )
+
+
+class XasAdvancedScan(XasSimpleScan):
+    scan_name = "xas_advanced_scan"
+    gui_config = {
+        "Movement Parameters": ["start", "stop"],
+        "Scan Parameters": ["scan_time", "scan_duration", "monitored_readout_cycle"],
+        "Spline Parameters": ["p_kink", "e_kink"],
+    }
+
+    def __init__(
+        self,
+        # fmt: off
+        start: Annotated[float, ScanArgument(display_name="Start Energy", description="Start energy.", units=Units.eV)],
+        stop: Annotated[float, ScanArgument(display_name="Stop Energy", description="Stop energy.", units=Units.eV)],
+        scan_time: Annotated[float, ScanArgument(display_name="Scan Time", description="Time for one scan cycle.", units=Units.s, ge=0)],
+        scan_duration: Annotated[float, ScanArgument(display_name="Scan Duration", description="Total scan duration.", units=Units.s, ge=0)],
+        p_kink: Annotated[float, ScanArgument(display_name="P Kink", description="Position of the kink.", ge=0)],
+        e_kink: Annotated[float, ScanArgument(display_name="E Kink", description="Energy of the kink.", units=Units.eV)],
+        motor: Annotated[DeviceBase | None, ScanArgument(display_name="Motor", description="Bragg motor device.")] = None,
+        daq: Annotated[DeviceBase | None, ScanArgument(display_name="DAQ", description="NIDAQ device.")] = None,
+        monitored_readout_cycle: Annotated[float, ScanArgument(display_name="Monitored Readout Cycle", description="Delay between monitored readouts.", units=Units.s, gt=0)] = 1,
+        **kwargs,
+        # fmt: on
+    ):
+        super().__init__(
+            start=start,
+            stop=stop,
+            scan_time=scan_time,
+            scan_duration=scan_duration,
+            motor=motor,
+            daq=daq,
+            monitored_readout_cycle=monitored_readout_cycle,
+            **kwargs,
+        )
+        # We pass on the arguments as "additional_scan_parameters" in the scan info
+        self.update_scan_info(p_kink=p_kink, e_kink=e_kink)
+
+
+class XasAdvancedScanWithXrd(XasAdvancedScan):
+    scan_name = "xas_advanced_scan_with_xrd"
+    gui_config = {
+        "Movement Parameters": ["start", "stop"],
+        "Scan Parameters": ["scan_time", "scan_duration", "monitored_readout_cycle"],
+        "Spline Parameters": ["p_kink", "e_kink"],
+        "Low Energy Break": ["break_enable_low", "break_time_low", "cycle_low"],
+        "High Energy Break": ["break_enable_high", "break_time_high", "cycle_high"],
+        "XRD Triggers": ["exp_time", "n_of_trigger"],
+    }
+
+    def __init__(
+        self,
+        # fmt: off
+        start: Annotated[float, ScanArgument(display_name="Start Energy", description="Start energy.", units=Units.eV)],
+        stop: Annotated[float, ScanArgument(display_name="Stop Energy", description="Stop energy.", units=Units.eV)],
+        scan_time: Annotated[float, ScanArgument(display_name="Scan Time", description="Time for one scan cycle.", units=Units.s, ge=0)],
+        scan_duration: Annotated[float, ScanArgument(display_name="Scan Duration", description="Total scan duration.", units=Units.s, ge=0)],
+        p_kink: Annotated[float, ScanArgument(display_name="P Kink", description="Position of the kink.", ge=0)],
+        e_kink: Annotated[float, ScanArgument(display_name="E Kink", description="Energy of the kink.", units=Units.eV)],
+        break_enable_low: Annotated[bool, ScanArgument(display_name="Break Enable Low", description="Enable breaks for the low energy range.")],
+        break_time_low: Annotated[float, ScanArgument(display_name="Break Time Low", description="Break time for the low energy range.", units=Units.s, ge=0)],
+        cycle_low: Annotated[int, ScanArgument(display_name="Cycle Low", description="Use triggers every nth low-energy cycle.", ge=0)],
+        break_enable_high: Annotated[bool, ScanArgument(display_name="Break Enable High", description="Enable breaks for the high energy range.")],
+        break_time_high: Annotated[float, ScanArgument(display_name="Break Time High", description="Break time for the high energy range.", units=Units.s, ge=0)],
+        cycle_high: Annotated[int, ScanArgument(display_name="Cycle High", description="Use triggers every nth high-energy cycle.", ge=0)],
+        exp_time: Annotated[float, ScanArgument(display_name="Exposure Time", description="Length of one trigger period.", units=Units.s, ge=0)],
+        n_of_trigger: Annotated[int, ScanArgument(display_name="Number Of Trigger", description="Amount of triggers fired during a break.", ge=0)],
+        motor: Annotated[DeviceBase | None, ScanArgument(display_name="Motor", description="Bragg motor device.")] = None,
+        daq: Annotated[DeviceBase | None, ScanArgument(display_name="DAQ", description="NIDAQ device.")] = None,
+        monitored_readout_cycle: Annotated[float, ScanArgument(display_name="Monitored Readout Cycle", description="Delay between monitored readouts.", units=Units.s, gt=0)] = 1,
+        **kwargs,
+        # fmt: on
+    ):
+        super().__init__(
+            start=start,
+            stop=stop,
+            scan_time=scan_time,
+            scan_duration=scan_duration,
+            p_kink=p_kink,
+            e_kink=e_kink,
+            motor=motor,
+            daq=daq,
+            monitored_readout_cycle=monitored_readout_cycle,
+            **kwargs,
+        )
+
+        # We pass on the arguments as "additional_scan_parameters" in the scan info
+        self.update_scan_info(
+            break_enable_low=break_enable_low,
+            break_time_low=break_time_low,
+            cycle_low=cycle_low,
+            break_enable_high=break_enable_high,
+            break_time_high=break_time_high,
+            cycle_high=cycle_high,
+            exp_time=exp_time,
+            n_of_trigger=n_of_trigger,
+        )

pyproject.toml

@@ -6,13 +6,21 @@ build-backend = "hatchling.build"
 name = "debye_bec"
 version = "0.0.0"
 description = "A plugin repository for BEC"
-requires-python = ">=3.10"
+requires-python = ">=3.11"
 classifiers = [
     "Development Status :: 3 - Alpha",
     "Programming Language :: Python :: 3",
     "Topic :: Scientific/Engineering",
 ]
-dependencies = ["numpy", "scipy", "bec_lib", "h5py", "ophyd_devices"]
+dependencies = [
+    "numpy",
+    "scipy",
+    "bec_lib",
+    "h5py",
+    "ophyd_devices",
+    "opencv-python==4.11.0.86",
+    "xrt",
+]
 
 [project.optional-dependencies]
 dev = [
@@ -38,6 +46,9 @@ plugin_file_writer = "debye_bec.file_writer"
 [project.entry-points."bec.scans"]
 plugin_scans = "debye_bec.scans"
 
+[project.entry-points."bec.scans.scan_modifier"]
+plugin_scan_modifier = "debye_bec.scans.scan_customization.scan_modifier"
+
 [project.entry-points."bec.scans.metadata_schema"]
 plugin_metadata_schema = "debye_bec.scans.metadata_schema"
 

tests/tests_devices/test_mo1_bragg.py

@@ -52,7 +52,7 @@ def test_init(mock_bragg):
     dev = mock_bragg
     assert dev.name == "bragg"
     assert dev.prefix == "X01DA-OP-MO1:BRAGG:"
-    assert dev.crystal.offset_si111._read_pvname == "X01DA-OP-MO1:BRAGG:offset_si111_RBV"
+    assert dev.crystal.bragg_off_si111._read_pvname == "X01DA-OP-MO1:BRAGG:bragg_off_si111_RBV"
     assert dev.move_abs._read_pvname == "X01DA-OP-MO1:BRAGG:move_abs"
 
 
@@ -106,14 +106,14 @@ def test_set_xtal(mock_bragg):
     dev = mock_bragg
     dev.set_xtal("111")
     # Default values for mock
-    assert dev.crystal.offset_si111.get() == 0
-    assert dev.crystal.offset_si311.get() == 0
+    assert dev.crystal.bragg_off_si111.get() == 0
+    assert dev.crystal.bragg_off_si311.get() == 0
     assert dev.crystal.d_spacing_si111.get() == 0
     assert dev.crystal.d_spacing_si311.get() == 0
     assert dev.crystal.xtal_enum.get() == 0
-    dev.set_xtal("311", offset_si111=1, offset_si311=2, d_spacing_si111=3, d_spacing_si311=4)
-    assert dev.crystal.offset_si111.get() == 1
-    assert dev.crystal.offset_si311.get() == 2
+    dev.set_xtal("311", bragg_off_si111=1, bragg_off_si311=2, d_spacing_si111=3, d_spacing_si311=4)
+    assert dev.crystal.bragg_off_si111.get() == 1
+    assert dev.crystal.bragg_off_si311.get() == 2
     assert dev.crystal.d_spacing_si111.get() == 3
     assert dev.crystal.d_spacing_si311.get() == 4
     assert dev.crystal.xtal_enum.get() == 1
@@ -159,60 +159,6 @@ def test_set_control_settings(mock_bragg):
     assert dev.scan_control.scan_duration.get() == 5
 
 
-def test_update_scan_parameters(mock_bragg):
-    dev = mock_bragg
-    msg = ScanStatusMessage(
-        scan_id="my_scan_id",
-        status="closed",
-        request_inputs={
-            "inputs": {},
-            "kwargs": {
-                "start": 0,
-                "stop": 5,
-                "scan_time": 1,
-                "scan_duration": 10,
-                "xrd_enable_low": True,
-                "xrd_enable_high": False,
-                "num_trigger_low": 1,
-                "num_trigger_high": 7,
-                "exp_time_low": 1,
-                "exp_time_high": 3,
-                "cycle_low": 1,
-                "cycle_high": 5,
-                "p_kink": 50,
-                "e_kink": 8000,
-            },
-        },
-        info={
-            "kwargs": {
-                "start": 0,
-                "stop": 5,
-                "scan_time": 1,
-                "scan_duration": 10,
-                "xrd_enable_low": True,
-                "xrd_enable_high": False,
-                "num_trigger_low": 1,
-                "num_trigger_high": 7,
-                "exp_time_low": 1,
-                "exp_time_high": 3,
-                "cycle_low": 1,
-                "cycle_high": 5,
-                "p_kink": 50,
-                "e_kink": 8000,
-            }
-        },
-        metadata={},
-    )
-    mock_bragg.scan_info.msg = msg
-    scan_param = dev.scan_parameter.model_dump()
-    for _, v in scan_param.items():
-        assert v == None
-    dev._update_scan_parameter()
-    scan_param = dev.scan_parameter.model_dump()
-    for k, v in scan_param.items():
-        assert v == msg.content["request_inputs"]["kwargs"].get(k, None)
-
-
 def test_kickoff_scan(mock_bragg):
     dev = mock_bragg
     dev.scan_control.scan_status._read_pv.mock_data = ScanControlScanStatus.READY

tests/tests_devices/test_nidaq.py

@@ -6,6 +6,7 @@ from unittest import mock
 import ophyd
 import pytest
 from bec_server.scan_server.scan_worker import ScanWorker
+from bec_server.scan_server.scans.scan_base import ScanInfo as ScanServerScanInfo
 from ophyd.status import WaitTimeoutError
 from ophyd_devices.interfaces.base_classes.psi_device_base import DeviceStoppedError
 from ophyd_devices.tests.utils import MockPV
@@ -15,6 +16,13 @@ from debye_bec.devices.nidaq.nidaq import Nidaq, NidaqError
 
 # TODO move this function to ophyd_devices, it is duplicated in csaxs_bec and needed for other pluging repositories
 from debye_bec.devices.test_utils.utils import patch_dual_pvs
+from debye_bec.devices.utils.utils import fetch_scan_info
+
+
+@pytest.fixture(scope="function")
+def scan_info_mock():
+    """Fixture for the ScanInfo object."""
+    return ScanServerScanInfo(scan_name="xas_simple_scan", scan_id="test")
 
 
 @pytest.fixture(scope="function")
@@ -52,13 +60,17 @@ def test_init(mock_nidaq):
     ]
 
 
-def test_check_if_scan_name_is_valid(mock_nidaq):
+def test_check_if_scan_name_is_valid(mock_nidaq, scan_info_mock):
     """Test the check_if_scan_name_is_valid method."""
     dev = mock_nidaq
-    dev.scan_info.msg.scan_name = "xas_simple_scan"
-    assert dev._check_if_scan_name_is_valid()
-    dev.scan_info.msg.scan_name = "invalid_scan_name"
-    assert not dev._check_if_scan_name_is_valid()
+    scan_info_mock.scan_name = "xas_simple_scan"
+    dev.scan_info.msg.info.update(scan_info_mock.model_dump())
+    scan_parameters = fetch_scan_info(dev.scan_info)
+    assert dev._check_if_scan_name_is_valid(scan_parameters)
+    scan_info_mock.scan_name = "invalid_scan_name"
+    dev.scan_info.msg.info.update(scan_info_mock.model_dump())
+    scan_parameters = fetch_scan_info(dev.scan_info)
+    assert not dev._check_if_scan_name_is_valid(scan_parameters)
 
 
 def test_set_config(mock_nidaq):
@@ -120,11 +132,13 @@ def test_on_unstage(mock_nidaq):
         ("nidaq_continuous_scan", False, 0),
     ],
 )
-def test_on_pre_scan(mock_nidaq, scan_name, raise_error, nidaq_state):
+def test_on_pre_scan(mock_nidaq, scan_name, raise_error, nidaq_state, scan_info_mock):
     """Test the on_pre_scan method of the Nidaq device."""
     dev = mock_nidaq
     dev.state.put(nidaq_state)
-    dev.scan_info.msg.scan_name = scan_name
+    scan_info_mock.scan_name = scan_name
+    dev.scan_info.msg.info.update(scan_info_mock.model_dump())
+    dev.scan_parameters = fetch_scan_info(dev.scan_info)
     dev._timeout_wait_for_pv = 0.1  # Set a short timeout for testing
     if not raise_error:
         dev.pre_scan()
@@ -133,11 +147,13 @@ def test_on_pre_scan(mock_nidaq, scan_name, raise_error, nidaq_state):
             dev.pre_scan()
 
 
-def test_on_complete(mock_nidaq):
+def test_on_complete(mock_nidaq, scan_info_mock):
     """Test the on_complete method of the Nidaq device."""
     dev = mock_nidaq
+    scan_info_mock.scan_name = "nidaq_continuous_scan"
+    dev.scan_info.msg.info.update(scan_info_mock.model_dump())
+    dev.scan_parameters = fetch_scan_info(dev.scan_info)
     # Check for nidaq_continuous_scan
-    dev.scan_info.msg.scan_name = "nidaq_continuous_scan"
     dev.state.put(0)  # Set state to DISABLED
     status = dev.complete()
     assert status.done is False
@@ -147,7 +163,9 @@ def test_on_complete(mock_nidaq):
     assert status.done is True
 
     # Check for XAS simple scan
-    dev.scan_info.msg.scan_name = "xas_simple_scan"
+    scan_info_mock.scan_name = "xas_simple_scan"
+    dev.scan_info.msg.info.update(scan_info_mock.model_dump())
+    dev.scan_parameters = fetch_scan_info(dev.scan_info)
     dev.state.put(0)  # Set state to ACQUIRE
     dev.stop_call.put(0)
     dev._timeout_wait_for_pv = 5

tests/tests_devices/test_pilatus.py

@@ -1,14 +1,15 @@
 # pylint: skip-file
-import os
 import threading
 from typing import TYPE_CHECKING, Generator
 from unittest import mock
 
-import numpy as np
 import ophyd
 import pytest
 from bec_lib.messages import ScanStatusMessage
 from bec_server.scan_server.scan_worker import ScanWorker
+from bec_server.scan_server.scans.scan_base import ScanInfo as ScanServerScanInfo
+from bec_server.scan_server.tests.scan_fixtures import *
+from bec_server.scan_server.tests.scan_fixtures import _MockDevice
 from ophyd_devices import CompareStatus, DeviceStatus
 from ophyd_devices.interfaces.base_classes.psi_device_base import DeviceStoppedError
 from ophyd_devices.tests.utils import MockPV, patch_dual_pvs
@@ -22,6 +23,7 @@ from debye_bec.devices.pilatus.pilatus import (
     TRIGGERMODE,
     Pilatus,
 )
+from debye_bec.devices.utils.utils import fetch_scan_info
 
 if TYPE_CHECKING:  # pragma no cover
     from bec_lib.messages import FileMessage
@@ -36,32 +38,38 @@ if TYPE_CHECKING:  # pragma no cover
 @pytest.fixture(
     scope="function",
     params=[
-        (0.1, 1, 1, "line_scan", "step"),
-        (0.2, 2, 2, "time_scan", "step"),
-        (0.5, 5, 5, "xas_advanced_scan", "fly"),
+        (("samx", 0.1, 1, 5, "samy", 0, 1, 5), {"relative": True}, "_v4_hexagonal_scan"),
+        ((1, 0.2), {}, "_v4_time_scan"),
+        ((9000, 10000, 1, 20, 0.1, 9500), {}, "xas_advanced_scan"),
     ],
 )
-def mock_scan_info(request, tmpdir):
-    exp_time, frames_per_trigger, num_points, scan_name, scan_type = request.param
-    scan_info = ScanStatusMessage(
-        scan_id="test_id",
-        status="open",
-        scan_type=scan_type,
-        scan_number=1,
-        scan_parameters={
-            "exp_time": exp_time,
-            "frames_per_trigger": frames_per_trigger,
-            "system_config": {},
-        },
-        info={"file_components": (f"{tmpdir}/data/S00000/S000001", "h5")},
-        num_points=num_points,
-        scan_name=scan_name,
-    )
-    yield scan_info
+def mock_scan_info(request, tmpdir, v4_scan_assembler, device_manager):
+    args, kwargs, scan_name = request.param
+    mo1_bragg = _MockDevice(name="mo1_bragg")
+    nidaq = _MockDevice(name="nidaq")
+    device_manager.add_device(mo1_bragg)
+    device_manager.add_device(nidaq)
+    scan = v4_scan_assembler(scan_name, *args, **kwargs)
+    yield scan.scan_info
 
 
 @pytest.fixture(scope="function")
-def pilatus(mock_scan_info) -> Generator[Pilatus, None, None]:
+def mock_scan_status_message(mock_scan_info, tmpdir) -> ScanStatusMessage:
+    info = mock_scan_info.model_dump()
+    info.update({"file_components": (f"{tmpdir}/data/S00000/S000001", "h5")})
+    return ScanStatusMessage(
+        scan_id=mock_scan_info.scan_id,
+        status="open",
+        scan_number=1,
+        scan_name=mock_scan_info.scan_name,
+        scan_type="fly" if mock_scan_info.scan_type == "hardware_triggered" else "step",
+        num_points=mock_scan_info.num_points,
+        info=info,
+    )
+
+
+@pytest.fixture(scope="function")
+def pilatus(mock_scan_status_message) -> Generator[Pilatus, None, None]:
     name = "pilatus"
     prefix = "X01DA-OP-MO1:PILATUS:"
     with mock.patch.object(ophyd, "cl") as mock_cl:
@@ -72,8 +80,9 @@ def pilatus(mock_scan_info) -> Generator[Pilatus, None, None]:
         # dev.image1 = mock.MagicMock()
         # with mock.patch.object(dev, "image1"):
         with mock.patch.object(dev, "task_handler"):
-            dev.scan_info.msg = mock_scan_info
+            dev.scan_info.msg = mock_scan_status_message
             try:
+                dev.scan_parameters = fetch_scan_info(dev.scan_info)
                 yield dev
             finally:
                 try:
@@ -177,9 +186,8 @@ def test_pilatus_on_trigger_cancel_on_stop(pilatus):
         status.wait(timeout=5)
 
 
-def test_pilatus_on_complete(pilatus):
+def test_pilatus_on_complete(pilatus: Pilatus):
     """Test the on_complete logic of the Pilatus detector."""
-
     if pilatus.scan_info.msg.scan_name.startswith("xas"):
         # TODO add test cases for xas scans
         # status = pilatus.complete()
@@ -198,8 +206,9 @@ def test_pilatus_on_complete(pilatus):
         pilatus.cam.acquire._read_pv.mock_data = ACQUIREMODE.ACQUIRING.value
         pilatus.hdf.capture._read_pv.mock_data = ACQUIREMODE.ACQUIRING.value
         pilatus.cam.armed._read_pv.mock_data = DETECTORSTATE.ARMED.value
-        num_images = pilatus.scan_info.msg.num_points * pilatus.scan_info.msg.scan_parameters.get(
-            "frames_per_trigger", 1
+        num_images = (
+            pilatus.scan_parameters.num_points
+            * pilatus.scan_parameters.additional_scan_parameters.get("frames_per_trigger", 1)
         )
         pilatus.hdf.num_captured._read_pv.mock_data = num_images - 1
         # Call on complete
@@ -277,9 +286,12 @@ def test_pilatus_on_complete(pilatus):
 
 def test_pilatus_on_stage_raises_low_exp_time(pilatus):
     """Test that on_stage raises a ValueError if the exposure time is too low."""
-    pilatus.scan_info.msg.scan_parameters["exp_time"] = 0.09
-    scan_msg = pilatus.scan_info.msg
-    if scan_msg.scan_type != "step" and scan_msg.scan_name not in pilatus.xas_xrd_scan_names:
+    pilatus.scan_info.msg.info["exp_time"] = 0.09
+    pilatus.scan_parameters = fetch_scan_info(pilatus.scan_info)
+    if (
+        pilatus.scan_parameters.scan_type != "software_triggered"
+        and pilatus.scan_parameters.scan_name not in pilatus.xas_xrd_scan_names
+    ):
         return
     with pytest.raises(ValueError):
         pilatus.on_stage()

tests/tests_scans/conftest.py

@@ -3,10 +3,10 @@ from functools import partial
 
 import pytest
 from bec_server.device_server.tests.utils import DeviceMockType, DMMock
-from bec_server.scan_server.tests.fixtures import (
-    ScanStubStatusMock,
-    connector_mock,
-    instruction_handler_mock,
+from bec_server.scan_server.tests.scan_fixtures import (
+    nth_done_status_mock,
+    readout_priority,
+    v4_scan_assembler,
 )
 
 

tests/tests_scans/test_mono_bragg_scans.py

@@ -1,433 +0,0 @@
-# pylint: skip-file
-from unittest import mock
-
-from bec_lib.messages import DeviceInstructionMessage
-from bec_server.device_server.tests.utils import DMMock
-
-from debye_bec.scans import (
-    XASAdvancedScan,
-    XASAdvancedScanWithXRD,
-    XASSimpleScan,
-    XASSimpleScanWithXRD,
-)
-
-
-def get_instructions(request, ScanStubStatusMock):
-    request.metadata["RID"] = "my_test_request_id"
-
-    def fake_done():
-        """
-        Fake done function for ScanStubStatusMock. Upon each call, it returns the next value from the generator.
-        This is used to simulate the completion of the scan.
-        """
-        yield False
-        yield False
-        yield True
-
-    def fake_complete(*args, **kwargs):
-        yield "fake_complete"
-        return ScanStubStatusMock(done_func=fake_done)
-
-    with (
-        mock.patch.object(request.stubs, "complete", side_effect=fake_complete),
-        mock.patch.object(request.stubs, "_get_result_from_status", return_value=None),
-    ):
-        reference_commands = list(request.run())
-
-        for cmd in reference_commands:
-            if not cmd or isinstance(cmd, str):
-                continue
-            if "RID" in cmd.metadata:
-                cmd.metadata["RID"] = "my_test_request_id"
-            if "rpc_id" in cmd.parameter:
-                cmd.parameter["rpc_id"] = "my_test_rpc_id"
-            cmd.metadata.pop("device_instr_id", None)
-
-        return reference_commands
-
-
-def test_xas_simple_scan(scan_assembler, ScanStubStatusMock):
-
-    request = scan_assembler(XASSimpleScan, start=0, stop=5, scan_time=1, scan_duration=10)
-    request.device_manager.add_device("nidaq")
-    reference_commands = get_instructions(request, ScanStubStatusMock)
-
-    assert reference_commands == [
-        None,
-        None,
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="scan_report_instruction",
-            parameter={"device_progress": ["mo1_bragg"]},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="open_scan",
-            parameter={
-                "scan_motors": [],
-                "readout_priority": {
-                    "monitored": [],
-                    "baseline": [],
-                    "on_request": [],
-                    "async": ["nidaq"],
-                },
-                "num_points": None,
-                "positions": [0.0, 5.0],
-                "scan_name": "xas_simple_scan",
-                "scan_type": "fly",
-            },
-        ),
-        DeviceInstructionMessage(metadata={}, device="nidaq", action="stage", parameter={}),
-        DeviceInstructionMessage(
-            metadata={},
-            device=["bpm4i", "eiger", "mo1_bragg", "samx"],
-            action="stage",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "baseline", "RID": "my_test_request_id"},
-            device=["samx"],
-            action="read",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=["bpm4i", "eiger", "mo1_bragg", "nidaq", "samx"],
-            action="pre_scan",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device="mo1_bragg",
-            action="kickoff",
-            parameter={"configure": {}},
-        ),
-        "fake_complete",
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id", "point_id": 0},
-            device=["bpm4i", "eiger", "mo1_bragg"],
-            action="read",
-            parameter={"group": "monitored"},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id", "point_id": 1},
-            device=["bpm4i", "eiger", "mo1_bragg"],
-            action="read",
-            parameter={"group": "monitored"},
-        ),
-        "fake_complete",
-        DeviceInstructionMessage(
-            metadata={},
-            device=["bpm4i", "eiger", "mo1_bragg", "nidaq", "samx"],
-            action="unstage",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="close_scan",
-            parameter={},
-        ),
-    ]
-
-
-def test_xas_simple_scan_with_xrd(scan_assembler, ScanStubStatusMock):
-
-    request = scan_assembler(
-        XASSimpleScanWithXRD,
-        start=0,
-        stop=5,
-        scan_time=1,
-        scan_duration=10,
-        break_enable_low=True,
-        break_time_low=1,
-        cycle_low=1,
-        break_enable_high=True,
-        break_time_high=2,
-        exp_time=1,
-        n_of_trigger=1,
-        cycle_high=4,
-    )
-    request.device_manager.add_device("nidaq")
-    reference_commands = get_instructions(request, ScanStubStatusMock)
-    # TODO #64 based on creating this ScanStatusMessage, we should test the logic of stage/kickoff/complete/unstage in Pilatus and mo1Bragg
-
-    assert reference_commands == [
-        None,
-        None,
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="scan_report_instruction",
-            parameter={"device_progress": ["mo1_bragg"]},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="open_scan",
-            parameter={
-                "scan_motors": [],
-                "readout_priority": {
-                    "monitored": [],
-                    "baseline": [],
-                    "on_request": [],
-                    "async": ["nidaq"],
-                },
-                "num_points": None,
-                "positions": [0.0, 5.0],
-                "scan_name": "xas_simple_scan_with_xrd",
-                "scan_type": "fly",
-            },
-        ),
-        DeviceInstructionMessage(metadata={}, device="nidaq", action="stage", parameter={}),
-        DeviceInstructionMessage(
-            metadata={},
-            device=["bpm4i", "eiger", "mo1_bragg", "samx"],
-            action="stage",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "baseline", "RID": "my_test_request_id"},
-            device=["samx"],
-            action="read",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=["bpm4i", "eiger", "mo1_bragg", "nidaq", "samx"],
-            action="pre_scan",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device="mo1_bragg",
-            action="kickoff",
-            parameter={"configure": {}},
-        ),
-        "fake_complete",
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id", "point_id": 0},
-            device=["bpm4i", "eiger", "mo1_bragg"],
-            action="read",
-            parameter={"group": "monitored"},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id", "point_id": 1},
-            device=["bpm4i", "eiger", "mo1_bragg"],
-            action="read",
-            parameter={"group": "monitored"},
-        ),
-        "fake_complete",
-        DeviceInstructionMessage(
-            metadata={},
-            device=["bpm4i", "eiger", "mo1_bragg", "nidaq", "samx"],
-            action="unstage",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="close_scan",
-            parameter={},
-        ),
-    ]
-
-
-def test_xas_advanced_scan(scan_assembler, ScanStubStatusMock):
-
-    request = scan_assembler(
-        XASAdvancedScan,
-        start=8000,
-        stop=9000,
-        scan_time=1,
-        scan_duration=10,
-        p_kink=50,
-        e_kink=8500,
-    )
-    request.device_manager.add_device("nidaq")
-    reference_commands = get_instructions(request, ScanStubStatusMock)
-
-    assert reference_commands == [
-        None,
-        None,
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="scan_report_instruction",
-            parameter={"device_progress": ["mo1_bragg"]},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="open_scan",
-            parameter={
-                "scan_motors": [],
-                "readout_priority": {
-                    "monitored": [],
-                    "baseline": [],
-                    "on_request": [],
-                    "async": ["nidaq"],
-                },
-                "num_points": None,
-                "positions": [8000.0, 9000.0],
-                "scan_name": "xas_advanced_scan",
-                "scan_type": "fly",
-            },
-        ),
-        DeviceInstructionMessage(metadata={}, device="nidaq", action="stage", parameter={}),
-        DeviceInstructionMessage(
-            metadata={},
-            device=["bpm4i", "eiger", "mo1_bragg", "samx"],
-            action="stage",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "baseline", "RID": "my_test_request_id"},
-            device=["samx"],
-            action="read",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=["bpm4i", "eiger", "mo1_bragg", "nidaq", "samx"],
-            action="pre_scan",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device="mo1_bragg",
-            action="kickoff",
-            parameter={"configure": {}},
-        ),
-        "fake_complete",
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id", "point_id": 0},
-            device=["bpm4i", "eiger", "mo1_bragg"],
-            action="read",
-            parameter={"group": "monitored"},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id", "point_id": 1},
-            device=["bpm4i", "eiger", "mo1_bragg"],
-            action="read",
-            parameter={"group": "monitored"},
-        ),
-        "fake_complete",
-        DeviceInstructionMessage(
-            metadata={},
-            device=["bpm4i", "eiger", "mo1_bragg", "nidaq", "samx"],
-            action="unstage",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="close_scan",
-            parameter={},
-        ),
-    ]
-
-
-def test_xas_advanced_scan_with_xrd(scan_assembler, ScanStubStatusMock):
-
-    request = scan_assembler(
-        XASAdvancedScanWithXRD,
-        start=8000,
-        stop=9000,
-        scan_time=1,
-        scan_duration=10,
-        p_kink=50,
-        e_kink=8500,
-        break_enable_low=True,
-        break_time_low=1,
-        cycle_low=1,
-        break_enable_high=True,
-        break_time_high=2,
-        exp_time=1,
-        n_of_trigger=1,
-        cycle_high=4,
-    )
-    request.device_manager.add_device("nidaq")
-    reference_commands = get_instructions(request, ScanStubStatusMock)
-
-    assert reference_commands == [
-        None,
-        None,
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="scan_report_instruction",
-            parameter={"device_progress": ["mo1_bragg"]},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="open_scan",
-            parameter={
-                "scan_motors": [],
-                "readout_priority": {
-                    "monitored": [],
-                    "baseline": [],
-                    "on_request": [],
-                    "async": ["nidaq"],
-                },
-                "num_points": None,
-                "positions": [8000.0, 9000.0],
-                "scan_name": "xas_advanced_scan_with_xrd",
-                "scan_type": "fly",
-            },
-        ),
-        DeviceInstructionMessage(metadata={}, device="nidaq", action="stage", parameter={}),
-        DeviceInstructionMessage(
-            metadata={},
-            device=["bpm4i", "eiger", "mo1_bragg", "samx"],
-            action="stage",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "baseline", "RID": "my_test_request_id"},
-            device=["samx"],
-            action="read",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=["bpm4i", "eiger", "mo1_bragg", "nidaq", "samx"],
-            action="pre_scan",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device="mo1_bragg",
-            action="kickoff",
-            parameter={"configure": {}},
-        ),
-        "fake_complete",
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id", "point_id": 0},
-            device=["bpm4i", "eiger", "mo1_bragg"],
-            action="read",
-            parameter={"group": "monitored"},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id", "point_id": 1},
-            device=["bpm4i", "eiger", "mo1_bragg"],
-            action="read",
-            parameter={"group": "monitored"},
-        ),
-        "fake_complete",
-        DeviceInstructionMessage(
-            metadata={},
-            device=["bpm4i", "eiger", "mo1_bragg", "nidaq", "samx"],
-            action="unstage",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="close_scan",
-            parameter={},
-        ),
-    ]

tests/tests_scans/test_mono_bragg_scans_v4.py

@@ -0,0 +1,152 @@
+# pylint: skip-file
+from unittest import mock
+
+import numpy as np
+import pytest
+from bec_server.scan_server.tests.scan_fixtures import *
+from bec_server.scan_server.tests.scan_hook_tests import *
+
+XAS_SIMPLE_SCAN_DEFAULT_HOOK_TESTS = [
+    ("prepare_scan", [assert_prepare_scan_reads_baseline_devices]),
+    ("open_scan", [assert_scan_open_called]),
+    ("stage", [assert_stage_all_devices_called]),
+    ("pre_scan", [assert_pre_scan_called]),
+    ("unstage", [assert_unstage_all_devices_called]),
+    ("close_scan", [assert_close_scan_waits_for_baseline_and_closes]),
+]
+
+
+def _assemble_xas_simple_scan(v4_scan_assembler, **overrides):
+    params = {
+        "start": 8000.0,
+        "stop": 9000.0,
+        "scan_time": 1.0,
+        "scan_duration": 10.0,
+        "motor": "mo1_bragg",
+        "daq": "nidaq",
+        "monitored_readout_cycle": 1.0,
+    }
+    params.update(overrides)
+    return v4_scan_assembler("xas_simple_scan", **params)
+
+
+@pytest.mark.parametrize(("hook_name", "hook_tests"), XAS_SIMPLE_SCAN_DEFAULT_HOOK_TESTS)
+def test_xas_simple_scan_v4_default_hooks(
+    v4_scan_assembler, nth_done_status_mock, hook_name, hook_tests
+):
+    scan = _assemble_xas_simple_scan(v4_scan_assembler)
+
+    run_scan_tests(scan, [(hook_name, hook_tests)], nth_done_status_mock=nth_done_status_mock)
+
+
+def test_xas_simple_scan_v4_prepare_scan_updates_metadata(v4_scan_assembler):
+    scan = _assemble_xas_simple_scan(v4_scan_assembler)
+    scan.actions.add_scan_report_instruction_device_progress = mock.MagicMock()
+    baseline_status = mock.MagicMock()
+    scan.actions.read_baseline_devices = mock.MagicMock(return_value=baseline_status)
+
+    scan.prepare_scan()
+
+    scan.actions._build_scan_status_message("open")
+
+    np.testing.assert_array_equal(scan.scan_info.positions, np.array([8000.0, 9000.0]))
+    assert scan.scan_info.additional_scan_parameters["scan_time"] == 1.0
+    assert scan.scan_info.additional_scan_parameters["scan_duration"] == 10.0
+    assert scan.scan_info.readout_priority_modification["async"] == ["nidaq"]
+    scan.actions.add_scan_report_instruction_device_progress.assert_called_once_with(scan.motor)
+    scan.actions.read_baseline_devices.assert_called_once_with(wait=False)
+    assert scan._baseline_readout_status is baseline_status
+
+
+def test_xas_simple_scan_v4_scan_core_reads_until_complete(v4_scan_assembler, nth_done_status_mock):
+    scan = _assemble_xas_simple_scan(v4_scan_assembler)
+    completion_status = nth_done_status_mock(resolve_after=3)
+    scan.actions.kickoff = mock.MagicMock()
+    scan.actions.complete = mock.MagicMock(return_value=completion_status)
+    scan.actions.read_monitored_devices = mock.MagicMock()
+
+    with mock.patch("debye_bec.scans.xas_simple_scan.time.sleep"):
+        scan.scan_core()
+
+    scan.actions.kickoff.assert_called_once_with(scan.motor)
+    scan.actions.complete.assert_called_once_with(scan.motor, wait=False)
+    assert scan.actions.read_monitored_devices.call_count == 2
+
+
+def test_xas_simple_scan_v4_post_scan_completes_all_devices(v4_scan_assembler):
+    scan = _assemble_xas_simple_scan(v4_scan_assembler)
+    scan.actions.complete_all_devices = mock.MagicMock()
+
+    scan.post_scan()
+
+    scan.actions.complete_all_devices.assert_called_once_with()
+
+
+def test_xas_simple_scan_with_xrd_v4_updates_xrd_metadata(v4_scan_assembler):
+    scan = v4_scan_assembler(
+        "xas_simple_scan_with_xrd",
+        start=8000.0,
+        stop=9000.0,
+        scan_time=1.0,
+        scan_duration=10.0,
+        break_enable_low=True,
+        break_time_low=1.0,
+        cycle_low=2,
+        break_enable_high=False,
+        break_time_high=3.0,
+        cycle_high=4,
+        exp_time=0.5,
+        n_of_trigger=6,
+        motor="mo1_bragg",
+        daq="nidaq",
+    )
+
+    assert scan.scan_name == "xas_simple_scan_with_xrd"
+    assert scan.scan_info.additional_scan_parameters["break_enable_low"] is True
+    assert scan.scan_info.additional_scan_parameters["cycle_high"] == 4
+    assert scan.scan_info.additional_scan_parameters["n_of_trigger"] == 6
+
+
+def test_xas_advanced_scan_v4_updates_spline_metadata(v4_scan_assembler):
+    scan = v4_scan_assembler(
+        "xas_advanced_scan",
+        start=8000.0,
+        stop=9000.0,
+        scan_time=1.0,
+        scan_duration=10.0,
+        p_kink=50.0,
+        e_kink=8500.0,
+        motor="mo1_bragg",
+        daq="nidaq",
+    )
+
+    assert scan.scan_name == "xas_advanced_scan"
+    assert scan.scan_info.additional_scan_parameters["p_kink"] == 50.0
+    assert scan.scan_info.additional_scan_parameters["e_kink"] == 8500.0
+
+
+def test_xas_advanced_scan_with_xrd_v4_updates_all_metadata(v4_scan_assembler):
+    scan = v4_scan_assembler(
+        "xas_advanced_scan_with_xrd",
+        start=8000.0,
+        stop=9000.0,
+        scan_time=1.0,
+        scan_duration=10.0,
+        p_kink=55.0,
+        e_kink=8450.0,
+        break_enable_low=True,
+        break_time_low=1.5,
+        cycle_low=2,
+        break_enable_high=True,
+        break_time_high=2.5,
+        cycle_high=3,
+        exp_time=0.25,
+        n_of_trigger=8,
+        motor="mo1_bragg",
+        daq="nidaq",
+    )
+
+    assert scan.scan_name == "xas_advanced_scan_with_xrd"
+    assert scan.scan_info.additional_scan_parameters["p_kink"] == 55.0
+    assert scan.scan_info.additional_scan_parameters["break_enable_high"] is True
+    assert scan.scan_info.exp_time == 0.25

tests/tests_scans/test_nidaq_continous_scan.py

@@ -1,127 +1,75 @@
 # pylint: skip-file
 from unittest import mock
 
-from bec_lib.messages import DeviceInstructionMessage
-from bec_server.device_server.tests.utils import DMMock
+import pytest
+from bec_server.scan_server.tests.scan_fixtures import *
+from bec_server.scan_server.tests.scan_hook_tests import *
 
-from debye_bec.scans import NIDAQContinuousScan
+NIDAQ_CONTINUOUS_SCAN_DEFAULT_HOOK_TESTS = [
+    ("prepare_scan", [assert_prepare_scan_reads_baseline_devices]),
+    ("open_scan", [assert_scan_open_called]),
+    ("stage", [assert_stage_all_devices_called]),
+    ("pre_scan", [assert_pre_scan_called]),
+    ("unstage", [assert_unstage_all_devices_called]),
+    ("close_scan", [assert_close_scan_waits_for_baseline_and_closes]),
+]
 
 
-def get_instructions(request, ScanStubStatusMock):
-    request.metadata["RID"] = "my_test_request_id"
-
-    def fake_done():
-        """
-        Fake done function for ScanStubStatusMock. Upon each call, it returns the next value from the generator.
-        This is used to simulate the completion of the scan.
-        """
-        yield False
-        yield False
-        yield True
-
-    def fake_complete(*args, **kwargs):
-        yield "fake_complete"
-        return ScanStubStatusMock(done_func=fake_done)
-
-    with (
-        mock.patch.object(request.stubs, "complete", side_effect=fake_complete),
-        mock.patch.object(request.stubs, "_get_result_from_status", return_value=None),
-    ):
-        reference_commands = list(request.run())
-
-        for cmd in reference_commands:
-            if not cmd or isinstance(cmd, str):
-                continue
-            if "RID" in cmd.metadata:
-                cmd.metadata["RID"] = "my_test_request_id"
-            if "rpc_id" in cmd.parameter:
-                cmd.parameter["rpc_id"] = "my_test_rpc_id"
-            cmd.metadata.pop("device_instr_id", None)
-
-        return reference_commands
+def _assemble_nidaq_continuous_scan(v4_scan_assembler, **overrides):
+    params = {"scan_duration": 10.0, "daq": "nidaq", "compression": False}
+    params.update(overrides)
+    return v4_scan_assembler("nidaq_continuous_scan", **params)
 
 
-def test_xas_simple_scan(scan_assembler, ScanStubStatusMock):
+@pytest.mark.parametrize(("hook_name", "hook_tests"), NIDAQ_CONTINUOUS_SCAN_DEFAULT_HOOK_TESTS)
+def test_nidaq_continuous_scan_v4_default_hooks(
+    v4_scan_assembler, nth_done_status_mock, hook_name, hook_tests
+):
+    scan = _assemble_nidaq_continuous_scan(v4_scan_assembler)
 
-    request = scan_assembler(NIDAQContinuousScan, scan_duration=10)
-    request.device_manager.add_device("nidaq")
-    reference_commands = get_instructions(request, ScanStubStatusMock)
-    assert reference_commands == [
-        None,
-        None,
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="scan_report_instruction",
-            parameter={"device_progress": ["nidaq"]},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="open_scan",
-            parameter={
-                "scan_motors": [],
-                "readout_priority": {
-                    "monitored": [],
-                    "baseline": [],
-                    "on_request": [],
-                    "async": ["nidaq"],
-                },
-                "num_points": 0,
-                "positions": [],
-                "scan_name": "nidaq_continuous_scan",
-                "scan_type": "fly",
-            },
-        ),
-        DeviceInstructionMessage(metadata={}, device="nidaq", action="stage", parameter={}),
-        DeviceInstructionMessage(
-            metadata={},
-            device=["bpm4i", "eiger", "mo1_bragg", "samx"],
-            action="stage",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "baseline", "RID": "my_test_request_id"},
-            device=["samx"],
-            action="read",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=["bpm4i", "eiger", "mo1_bragg", "nidaq", "samx"],
-            action="pre_scan",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device="nidaq",
-            action="kickoff",
-            parameter={"configure": {}},
-        ),
-        "fake_complete",
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id", "point_id": 0},
-            device=["bpm4i", "eiger", "mo1_bragg"],
-            action="read",
-            parameter={"group": "monitored"},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id", "point_id": 1},
-            device=["bpm4i", "eiger", "mo1_bragg"],
-            action="read",
-            parameter={"group": "monitored"},
-        ),
-        "fake_complete",
-        DeviceInstructionMessage(
-            metadata={},
-            device=["bpm4i", "eiger", "mo1_bragg", "nidaq", "samx"],
-            action="unstage",
-            parameter={},
-        ),
-        DeviceInstructionMessage(
-            metadata={"readout_priority": "monitored", "RID": "my_test_request_id"},
-            device=None,
-            action="close_scan",
-            parameter={},
-        ),
-    ]
+    run_scan_tests(scan, [(hook_name, hook_tests)], nth_done_status_mock=nth_done_status_mock)
+
+
+def test_nidaq_continuous_scan_v4_prepare_scan_updates_metadata(v4_scan_assembler):
+    scan = _assemble_nidaq_continuous_scan(v4_scan_assembler)
+    scan.actions.add_scan_report_instruction_device_progress = mock.MagicMock()
+    baseline_status = mock.MagicMock()
+    scan.actions.read_baseline_devices = mock.MagicMock(return_value=baseline_status)
+
+    scan.prepare_scan()
+    scan.actions._build_scan_status_message("open")
+
+    assert scan.scan_info.additional_scan_parameters["scan_duration"] == 10.0
+    assert scan.scan_info.additional_scan_parameters["compression"] is False
+    assert scan.scan_info.readout_priority_modification["async"] == ["nidaq"]
+    scan.actions.add_scan_report_instruction_device_progress.assert_called_once_with(scan.daq)
+    scan.actions.read_baseline_devices.assert_called_once_with(wait=False)
+    assert scan._baseline_readout_status is baseline_status
+
+
+def test_nidaq_continuous_scan_v4_scan_core_reads_until_complete(
+    v4_scan_assembler, nth_done_status_mock
+):
+    scan = _assemble_nidaq_continuous_scan(v4_scan_assembler)
+    kickoff_status = mock.MagicMock()
+    completion_status = nth_done_status_mock(resolve_after=3)
+    scan.actions.kickoff = mock.MagicMock(return_value=kickoff_status)
+    scan.actions.complete = mock.MagicMock(return_value=completion_status)
+    scan.actions.read_monitored_devices = mock.MagicMock()
+
+    with mock.patch("debye_bec.scans.nidaq_continuous_scan.time.sleep"):
+        scan.scan_core()
+
+    scan.actions.kickoff.assert_called_once_with(device=scan.daq, wait=False)
+    kickoff_status.wait.assert_called_once_with(timeout=5)
+    scan.actions.complete.assert_called_once_with(device=scan.daq, wait=False)
+    assert scan.actions.read_monitored_devices.call_count == 2
+
+
+def test_nidaq_continuous_scan_v4_post_scan_completes_all_devices(v4_scan_assembler):
+    scan = _assemble_nidaq_continuous_scan(v4_scan_assembler)
+    scan.actions.complete_all_devices = mock.MagicMock()
+
+    scan.post_scan()
+
+    scan.actions.complete_all_devices.assert_called_once_with()