Energy widgets

src/firefly/application.py

@@ -1,6 +1,7 @@
 import asyncio
 import logging
 import subprocess
+import warnings
 from collections import OrderedDict
 from functools import partial
 from pathlib import Path
@@ -551,11 +552,10 @@ def prepare_queue_client(self, client=None, api=None):
         if api is None:
             try:
                 api = queueserver_api()
-            except InvalidConfiguration:
-                log.error(
-                    "Could not load queueserver API "
-                    "configuration from iconfig.toml file."
-                )
+            except InvalidConfiguration as exc:
+                msg = f"Could not create queue client. Missing key {exc}"
+                log.warning(msg)
+                warnings.warn(msg)
                 return
         # Create the client object
         if client is None:

src/firefly/energy.py

@@ -1,33 +1,76 @@
 import logging
+import warnings
 
 import qtawesome as qta
 from bluesky_queueserver_api import BPlan
+from ophydregistry import ComponentNotFound
+from pydm.widgets.label import PyDMLabel
+from pydm.widgets.line_edit import PyDMLineEdit
 from qtpy import QtCore, QtWidgets
+from qtpy.QtWidgets import QDialogButtonBox, QFormLayout, QLineEdit, QVBoxLayout
 from xraydb.xraydb import XrayDB
 
 from firefly import display
-from haven import exceptions, load_config, registry
+from haven import load_config, registry
 
 log = logging.getLogger(__name__)
 
 
+class EnergyCalibrationDialog(QtWidgets.QDialog):
+    """A dialog box for calibrating the energy."""
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        self.setWindowTitle("Energy calibration")
+
+        self.layout = QVBoxLayout()
+        self.setLayout(self.layout)
+        # Widgets for inputting calibration parameters
+        self.form_layout = QFormLayout()
+        self.layout.addLayout(self.form_layout)
+        self.form_layout.addRow(
+            "Energy readback:", PyDMLabel(self, init_channel="haven://energy.readback")
+        )
+        self.form_layout.addRow(
+            "Energy setpoint:",
+            PyDMLineEdit(self, init_channel="haven://energy.setpoint"),
+        )
+        self.form_layout.addRow(
+            "Calibrated energy:",
+            QLineEdit(),
+        )
+        # Button for accept/close
+        buttons = QDialogButtonBox.Apply | QDialogButtonBox.Close
+        self.buttonBox = QDialogButtonBox(buttons)
+        self.buttonBox.accepted.connect(self.accept)
+        self.buttonBox.rejected.connect(self.reject)
+        self.layout.addWidget(self.buttonBox)
+
+
 class EnergyDisplay(display.FireflyDisplay):
     caqtdm_mono_ui_file = "/net/s25data/xorApps/ui/DCMControlCenter.ui"
     caqtdm_id_ui_file = (
         "/net/s25data/xorApps/epics/synApps_6_2/ioc/25ida/25idaApp/op/ui/IDControl.ui"
     )
-    min_energy = 4000
-    max_energy = 33000
     stylesheet_danger = (
         "background: rgb(220, 53, 69); color: white; border-color: rgb(220, 53, 69)"
     )
     stylesheet_normal = ""
+    energy_positioner = None
 
     def __init__(self, args=None, macros={}, **kwargs):
         # Load X-ray database for calculating edge energies
         self.xraydb = XrayDB()
         super().__init__(args=args, macros=macros, **kwargs)
 
+    def customize_device(self):
+        try:
+            self.energy_positioner = registry.find("energy")
+        except ComponentNotFound:
+            warnings.warn("Could not find energy positioner.")
+            log.warning("Could not find energy positioner.")
+
     def prepare_caqtdm_actions(self):
         """Create QActions for opening mono/ID caQtDM panels.
 
@@ -43,10 +86,6 @@ def prepare_caqtdm_actions(self):
         action.triggered.connect(self.launch_mono_caqtdm)
         action.setIcon(qta.icon("fa5s.wrench"))
         action.setToolTip("Launch the caQtDM panel for the monochromator.")
-        try:
-            registry.find(name="monochromator")
-        except exceptions.ComponentNotFound:
-            action.setDisabled(True)
         self.caqtdm_actions.append(action)
         # Create an action for launching the ID caQtDM file
         action = QtWidgets.QAction(self)
@@ -59,25 +98,26 @@ def prepare_caqtdm_actions(self):
 
     def launch_mono_caqtdm(self):
         config = load_config()
-        prefix = config["monochromator"]["ioc"] + ":"
-        mono = registry.find(name="monochromator")
-        ID = registry.find(name="undulator")
+        mono = self.energy_positioner.monochromator
+        ID = self.energy_positioner.undulator
+        prefix = mono.prefix
         caqtdm_macros = {
             "P": prefix,
             "MONO": config["monochromator"]["ioc_branch"],
             "BRAGG": mono.bragg.prefix.replace(prefix, ""),
             "GAP": mono.gap.prefix.replace(prefix, ""),
             "ENERGY": mono.energy.prefix.replace(prefix, ""),
             "OFFSET": mono.offset.prefix.replace(prefix, ""),
-            "IDENERGY": ID.energy.pvname,
+            "IDENERGY": ID.energy.prefix,
         }
         self.launch_caqtdm(macros=caqtdm_macros, ui_file=self.caqtdm_mono_ui_file)
 
     def launch_id_caqtdm(self):
         """Launch the pre-built caQtDM UI file for the ID."""
-        config = load_config()
-        prefix = config["undulator"]["ioc"]
-        # Strip leading "ID" from the mono IOC since caQtDM adds it
+        prefix = self.energy_positioner.undulator.prefix
+        # caQtDM doesn't expect the trailing ";"
+        prefix = prefix.rstrip(":")
+        # Strip leading "ID" from the ID IOC since caQtDM adds it
         prefix = prefix.strip("ID")
         caqtdm_macros = {
             # No idea what "M", and "D" do, they're not in the UI
@@ -105,16 +145,23 @@ def customize_ui(self):
         combo_box = self.ui.edge_combo_box
         ltab = self.xraydb.tables["xray_levels"]
         edges = self.xraydb.query(ltab)
+        min_energy, max_energy = self.energy_positioner.limits
         edges = edges.filter(
-            ltab.c.absorption_edge < self.max_energy,
-            ltab.c.absorption_edge > self.min_energy,
+            ltab.c.absorption_edge < max_energy,
+            ltab.c.absorption_edge > min_energy,
         )
         items = [
             f"{r.element} {r.iupac_symbol} ({int(r.absorption_edge)} eV)"
             for r in edges.all()
         ]
         combo_box.addItems(["Select edge…", *items])
         combo_box.activated.connect(self.select_edge)
+        # Respond to the "calibrate" button
+        self.ui.calibrate_button.clicked.connect(self.show_calibrate_dialog)
+
+    def show_calibrate_dialog(self):
+        dialog = EnergyCalibrationDialog(self)
+        dialog.exec()
 
     @QtCore.Slot(int)
     def select_edge(self, index):