From 11615bd3358405fabb572a9d9813b3013666b089 Mon Sep 17 00:00:00 2001
From: Denys Bezmenov <me@hatomist.pw>
Date: Thu, 14 Nov 2024 16:00:03 -0800
Subject: [PATCH] Added commands for actuators

---
 pykos/pykos/services/actuator.py | 50 ++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/pykos/pykos/services/actuator.py b/pykos/pykos/services/actuator.py
index c297b00..b1be475 100644
--- a/pykos/pykos/services/actuator.py
+++ b/pykos/pykos/services/actuator.py
@@ -48,3 +48,53 @@ def get_calibration_status(self, actuator_id: int):
         response = self.operations_stub.GetOperation(operations_pb2.GetOperationRequest(name=f"operations/calibrate_actuator/{actuator_id}"))
         metadata = CalibrationMetadata(response.metadata)
         return metadata.status
+
+    def command_actuators(self, commands: list[dict]):
+        """
+        Command multiple actuators at once.
+
+        Args:
+            commands: List of dictionaries containing actuator commands.
+                     Each dict should have 'actuator_id' and optionally 'position',
+                     'velocity', and 'torque'.
+
+        Returns:
+            List of ActionResult objects indicating success/failure for each command.
+        """
+        actuator_commands = [actuator_pb2.ActuatorCommand(**cmd) for cmd in commands]
+        request = actuator_pb2.CommandActuatorsRequest(commands=actuator_commands)
+        response = self.stub.CommandActuators(request)
+        return response.results
+
+    def configure_actuator(self, actuator_id: int, **kwargs):
+        """
+        Configure an actuator's parameters.
+
+        Args:
+            actuator_id: ID of the actuator to configure
+            **kwargs: Configuration parameters that may include:
+                     kp, kd, ki, max_torque, protective_torque,
+                     protection_time, torque_enabled, new_actuator_id
+
+        Returns:
+            ActionResponse indicating success/failure
+        """
+        config = {"actuator_id": actuator_id, **kwargs}
+        request = actuator_pb2.ConfigureActuatorRequest(**config)
+        return self.stub.ConfigureActuator(request)
+
+    def get_actuators_state(self, actuator_ids: list[int]):
+        """
+        Get the state of multiple actuators.
+
+        Args:
+            actuator_ids: List of actuator IDs to query
+
+        Returns:
+            List of ActuatorStateResponse objects containing the state information
+        """
+        request = actuator_pb2.GetActuatorsStateRequest(actuator_ids=actuator_ids)
+        response = self.stub.GetActuatorsState(request)
+        return response.states
+
+    
\ No newline at end of file