meerstetterTEC: Implement setpoint and enable commands

meerstetterTEC.py

@@ -134,6 +134,42 @@ class MeCom:
         return self.recv(crc)
 
 
+class Status(enum.IntEnum):
+    OFF = 0
+    ON = 1
+    LIVE = 2
+    HWENABLE = 3
+
+    def __str__(self):
+        return self.name
+
+    def __repr__(self):
+        return self.name
+
+    @classmethod
+    def loads(cls, data):
+        return cls(int32.loads(data))
+
+
+class DeviceStatus(enum.IntEnum):
+    INIT = 0
+    READY = 1
+    RUN = 2
+    ERROR = 3
+    BOOTL = 4
+    RESET = 5
+
+    def __str__(self):
+        return self.name
+
+    def __repr__(self):
+        return self.name
+
+    @classmethod
+    def loads(cls, data):
+        return cls(int32.loads(data))
+
+
 class MeerstetterTEC(MeCom):
     """Meerstetter TEC Controller."""
 
@@ -146,40 +182,6 @@ class MeerstetterTEC(MeCom):
         if idn != b'8065-TEC SW G01':
             raise NotImplementedError(idn)
 
-    class Status(enum.IntEnum):
-        OFF = 0
-        ON = 1
-        LIVE = 2
-        HWENABLE = 3
-
-        def __str__(self):
-            return self.name
-
-        def __repr__(self):
-            return self.name
-
-        @classmethod
-        def loads(cls, data):
-            return cls(int32.loads(data))
-
-    class DeviceStatus(enum.IntEnum):
-        INIT = 0
-        READY = 1
-        RUN = 2
-        ERROR = 3
-        BOOTL = 4
-        RESET = 5
-
-        def __str__(self):
-            return self.name
-
-        def __repr__(self):
-            return self.name
-
-        @classmethod
-        def loads(cls, data):
-            return cls(int32.loads(data))
-
     def read(self, *args):
         return super().read(*args)
 
@@ -249,6 +251,24 @@ def status(obj):
         print('{:4d}#{:05d} {:s}  {!s:>5s}  {!s:>3s}  {:9.6f}  {:9.6f}  {: 9.6f}'.format(*row))
 
 
+@main.command
+@click.argument('value', type=float)
+@click.pass_obj
+def setpoint(obj, value):
+    """Set target temperature."""
+    obj.write(3000, 1, float32, value)
+    print('  '.join(f'{x:.6f}' for x in obj.read(3000, 1, float32)))
+
+
+@main.command
+@click.argument('status', type=bool, required=False, default=True)
+@click.pass_obj
+def enable(obj, status):
+    """Set temperature control status."""
+    obj.set_status(Status.ON if status else Status.OFF)
+    print('  '.join(str(x) for x in obj.get_status()))
+
+
 @main.command
 @click.pass_obj
 @click.option('--datadir', type=click.Path(), help='Data folder for datalogger mode.')