Add initial devsta generation prototype

Author: OCopping

example-synoptic/b23-services/synoptic/create_gui.yaml

@@ -133,11 +133,16 @@ components:
 
   M2:
     prefix: FE23B-OP-MR-02
+    devsta:
+      - FE23B-OP-MR-02:MOTORSTA
 
   M3:
     prefix: BL23B-OP-MR-03
     extras:
       - BL23B-MO-PMAC-01
+    devsta:
+      - BL23B-OP-MR-03:MOTORSTA
+      - BL23B-MO-PMAC-01:MOTORSTA
 
   # ----- MOD -----
 

pyproject.toml

@@ -26,6 +26,7 @@ dependencies = [
     "lxml>=5.4.0",
     "typer>=0.16.0",
     "rich>=14.1.0",
+    "softioc>=4.6.1",
 ]
 scripts = { create-gui = "techui_builder.__main__:app" }
 

src/techui_builder/__main__.py

@@ -152,6 +152,7 @@ def main(
 
     gui.setup()
     gui.generate_screens()
+    gui.write_devsta_pvs()
 
     LOGGER.info(f"Screens generated for {gui.beamline.dom}.")
 

src/techui_builder/builder.py

@@ -6,8 +6,10 @@
 from typing import Any
 
 import yaml
+from epicsdbbuilder.recordbase import Record
 from lxml import etree, objectify
 from lxml.objectify import ObjectifiedElement
+from softioc.builder import records
 
 from techui_builder.generate import Generator
 from techui_builder.objects import Beamline, Component, Entity
@@ -45,6 +47,7 @@ class Builder:
     entities: defaultdict[str, list[Entity]] = field(
         default_factory=lambda: defaultdict(list), init=False
     )
+    devsta_pvs: dict[str, Record] = field(default_factory=dict, init=False)
     _services_dir: Path = field(init=False, repr=False)
     _gui_map: dict = field(init=False, repr=False)
     _write_directory: Path = field(default=Path("data"), init=False, repr=False)
@@ -79,6 +82,39 @@ def setup(self):
         """Run intial setup, e.g. extracting entries from service ioc.yaml."""
         self._extract_services()
 
+    def _create_devsta_pv(self, prefix: str, inputs: list[str]):
+        # Extract all input PVs, provided a default "" if not provided
+        values = [(inputs[i] if i < len(inputs) else "") for i in range(12)]
+        inpa, inpb, inpc, inpd, inpe, inpf, inpg, inph, inpi, inpj, inpk, inpl = values
+
+        devsta_pv = records.calc(  # pyright: ignore[reportAttributeAccessIssue]
+            f"{prefix}:DEVSTA",
+            CALC="(A|B|C|D|E|F|G|H|I|J|K|L)>0?1:0",
+            SCAN="1 second",
+            ACKT="NO",
+            INPA=inpa,
+            INPB=inpb,
+            INPC=inpc,
+            INPD=inpd,
+            INPE=inpe,
+            INPF=inpf,
+            INPG=inpg,
+            INPH=inph,
+            INPI=inpi,
+            INPJ=inpj,
+            INPK=inpk,
+            INPL=inpl,
+        )
+
+        self.devsta_pvs[prefix] = devsta_pv
+
+    def write_devsta_pvs(self):
+        devsta_file = self._write_directory.parent.joinpath("config/ioc.db")
+        with open(devsta_file, "w") as f:
+            for dpv in self.devsta_pvs.values():
+                dpv.Print(f)
+                # f.write(str(dpv))
+
     def _extract_services(self):
         """
         Finds the services folders in the services directory
@@ -134,6 +170,10 @@ def generate_screens(self):
         # any extras defined
         for component in self.components:
             screen_entities: list[Entity] = []
+
+            if component.devsta is not None:
+                self._create_devsta_pv(component.prefix, component.devsta)
+
             # ONLY IF there is a matching component and entity, generate a screen
             if component.prefix in self.entities.keys():
                 screen_entities.extend(self.entities[component.prefix])

src/techui_builder/objects.py

@@ -23,6 +23,7 @@ class Component:
     prefix: str
     desc: str | None = field(default=None)
     file: str | None = field(default=None)
+    devsta: list[str] | None = field(default=None)
     extras: list[str] | None = field(default=None)
 
     def __post_init__(self):