feat: move and rotate

Author: eigen-value

arduino_alvik.py

@@ -51,6 +51,7 @@ def __init__(self):
         self.bottom_tof = None
         self.linear_velocity = None
         self.angular_velocity = None
+        self.last_ack = ''
         self.version = [None, None, None]
 
     def begin(self) -> int:
@@ -86,6 +87,24 @@ def _stop_update_thread(cls):
         """
         cls._update_thread_running = False
 
+    def rotate(self, angle: float):
+        """
+        Rotates the robot by given angle
+        :param angle:
+        :return:
+        """
+        self.packeter.packetC1F(ord('R'), angle)
+        uart.write(self.packeter.msg[0:self.packeter.msg_size])
+
+    def move(self, distance: float):
+        """
+        Moves the robot by given distance
+        :param distance:
+        :return:
+        """
+        self.packeter.packetC1F(ord('G'), distance)
+        uart.write(self.packeter.msg[0:self.packeter.msg_size])
+
     def stop(self):
         """
         Stops all Alvik operations
@@ -193,6 +212,13 @@ def set_servo_positions(self, a_position: int, b_position: int):
         self.packeter.packetC2B(ord('S'), a_position & 0xFF, b_position & 0xFF)
         uart.write(self.packeter.msg[0:self.packeter.msg_size])
 
+    def get_ack(self):
+        """
+        Resets and returns last acknowledgement
+        :return:
+        """
+        return self.last_ack
+
     # def send_ack(self):
     #     self.packeter.packetC1B(ord('X'), ACK_)
     #     uart.write(self.packeter.msg[0:self.packeter.msg_size])
@@ -301,6 +327,9 @@ def _parse_message(self) -> int:
         elif code == ord('v'):
             # robot velocity
             _, self.linear_velocity, self.angular_velocity = self.packeter.unpacketC2F()
+        elif code == ord('x'):
+            # robot ack
+            _, self.last_ack = self.packeter.unpacketC1B()
         elif code == 0x7E:
             # firmware version
             _, *self.version = self.packeter.unpacketC3B()

examples/move_example.py

@@ -7,14 +7,18 @@
 
 while True:
     try:
-        alvik.set_wheels_speed(10, 10)
-        sleep_ms(1000)
+        alvik.move(100.0)
+        while (ack := alvik.get_ack()) != ord('M'):
+            print(f'moving... not on target yet ack={ack}')
+            sleep_ms(200)
+        print("on target after move")
 
-        alvik.set_wheels_speed(30, 60)
-        sleep_ms(1000)
+        alvik.rotate(90.0)
+        while (ack := alvik.get_ack()) != ord('R'):
+            print(f'rotating... not on target yet ack={ack}')
+            sleep_ms(200)
+        print("on target after rotation")
 
-        alvik.set_wheels_speed(60, 30)
-        sleep_ms(1000)
     except KeyboardInterrupt as e:
         print('over')
         alvik.stop()

examples/wheels_speed.py

@@ -0,0 +1,21 @@
+from arduino_alvik import ArduinoAlvik
+from time import sleep_ms
+import sys
+
+alvik = ArduinoAlvik()
+alvik.begin()
+
+while True:
+    try:
+        alvik.set_wheels_speed(10, 10)
+        sleep_ms(1000)
+
+        alvik.set_wheels_speed(30, 60)
+        sleep_ms(1000)
+
+        alvik.set_wheels_speed(60, 30)
+        sleep_ms(1000)
+    except KeyboardInterrupt as e:
+        print('over')
+        alvik.stop()
+        sys.exit()