Add proxy

catkit2/testbed/proxies/__init__.py

@@ -6,6 +6,7 @@
     'NewportPicomotorProxy',
     'NiDaqProxy',
     'NktSuperkProxy',
+    'ThorlabsCubeMotorKinesisProxy',
     'WebPowerSwitchProxy'
 ]
 
@@ -16,4 +17,5 @@
 from .newport_picomotor import *
 from .ni_daq import *
 from .nkt_superk import *
+from .thorlabs_cube_motor_kinesis import *
 from .web_power_switch import *

catkit2/testbed/proxies/thorlabs_cube_motor_kinesis.py

@@ -0,0 +1,27 @@
+from ..service_proxy import ServiceProxy
+
+import numpy as np
+
+
+@ServiceProxy.register_service_interface('thorlabs_cube_motor_kinesis')
+class ThorlabsCubeMotorKinesisProxy(ServiceProxy):
+    def move_absolute(self, position):
+        position = self.resolve_position(position)
+        self.command.submit_data(np.array([position], dtype='float64'))
+
+    def move_relative(self, distance):
+        current_position = self.current_position.get()[0]
+        new_position = current_position + distance
+        self.move_absolute(new_position)
+
+    def resolve_position(self, position_name):
+        if type(position_name) == str:
+            # The position is a named position.
+            position = self.positions[position_name]
+            # The position may still be a named position, so try to resolve deeper.
+            return self.resolve_position(position)
+        else:
+            return position_name
+
+    def positions(self):
+        return self.config['positions']