prototype for a camera server client plus CLI control script

cameras/cameraclient.py

@@ -0,0 +1,138 @@
+from types import SimpleNamespace
+from .waiting_epics import PV, caput, caget
+from .utils import intify
+
+
+class CameraClient:
+
+    def __init__(self, name):
+        self.name = name
+        pv_roi_xmin = PV(name + ":REGIONX_START")
+        pv_roi_xmax = PV(name + ":REGIONX_END")
+        pv_roi_ymin = PV(name + ":REGIONY_START")
+        pv_roi_ymax = PV(name + ":REGIONY_END")
+
+        self.pvs = SimpleNamespace(
+            xmin=pv_roi_xmin,
+            xmax=pv_roi_xmax,
+            ymin=pv_roi_ymin,
+            ymax=pv_roi_ymax
+        )
+
+        self.max_roi = self.get_max_roi()
+
+    def __repr__(self):
+        tn = type(self).__name__
+        fn = self.name
+        return f"{tn}(\"{fn}\")"
+
+    @property
+    def status(self):
+        head = repr(self)
+        print(head)
+        print("-" * len(head))
+        channels = ("WARNCODE", "ERRCODE", "STATUSCODE", "BUSY", "CAMRATE", "FILERATE")
+        maxlen = max(len(ch) for ch in channels)
+        for ch in channels:
+            fch = self.name +":" + ch
+            val = caget(fch)
+
+            line = ch + ": "
+            line = line.ljust(maxlen + 2)
+            line += str(val)
+            print(line)
+
+        print("-" * len(head))
+        roi  = self.get_roi()
+        full = self.get_max_roi()
+        print(f"roi:  {roi}")
+        print(f"full: {full}")
+
+
+    def restart(self):
+        self.off()
+        self.offline()
+        self.init()
+        sleep(0.01)
+        self.running()
+
+    def offline(self):
+        caput(self.name + ":INIT", 0) # OFFLINE
+
+    def init(self):
+        caput(self.name + ":INIT", 1) # INIT
+
+    def off(self):
+        caput(self.name + ":CAMERA", 0) # OFF
+
+    def running(self):
+        caput(self.name + ":CAMERA", 1) # RUNNING
+
+    def reset_roi(self):
+        caput(self.name + ":RESETROI.PROC", 1) # Reset ROI
+
+    def set_parameters(self):
+        caput(self.name + ":SET_PARAM", 1) # Set Parameters
+
+    def clear_buffer(self):
+        caput(self.name + ":CLEARMEM", 1) # Clear Buffer
+
+    def get_max_roi(self):
+        current_roi = self.get_roi()
+        self.reset_roi()
+        max_roi = self.get_roi()
+        self._set_roi(*current_roi)
+        return max_roi
+
+    def get_roi(self):
+        xmin = self.pvs.xmin.get()
+        xmax = self.pvs.xmax.get()
+        ymin = self.pvs.ymin.get()
+        ymax = self.pvs.ymax.get()
+        return intify(xmin, xmax, ymin, ymax)
+
+
+    def set_roi(self, xmin, xmax, ymin, ymax, debug=False):
+        if debug:
+            asked = (xmin, xmax, ymin, ymax)
+
+        xminmin, xmaxmax, yminmin, ymaxmax = self.max_roi
+
+        xmin = max(xmin, xminmin)
+        xmax = min(xmax, xmaxmax)
+        ymin = max(ymin, yminmin)
+        ymax = min(ymax, ymaxmax)
+
+        ymindelta = ymin - yminmin
+        ymaxdelta = ymaxmax - ymax
+        ydelta = min(ymindelta, ymaxdelta)
+
+        ymin = yminmin + ydelta
+        ymax = ymaxmax - ydelta
+
+        if debug:
+            adjusted = (xmin, xmax, ymin, ymax)
+            print("   ", asked, "\n-> ", adjusted, ":", ydelta, ymindelta, ymaxdelta)
+
+        self._set_roi(xmin, xmax, ymin, ymax, debug=debug)
+
+
+    def _set_roi(self, xmin, xmax, ymin, ymax, debug=False):
+        xmin, xmax, ymin, ymax = intify(xmin, xmax, ymin, ymax)
+
+        self.off()
+
+        self.pvs.xmin.put(xmin)
+        self.pvs.xmax.put(xmax)
+        self.pvs.ymin.put(ymin)
+        self.pvs.ymax.put(ymax)
+
+        self.set_parameters()
+        self.clear_buffer()
+        self.running()
+
+        if debug:
+            print("-->", self.get_roi(), "\n")
+
+
+