Working speed control
This commit is contained in:
@@ -1,23 +1,19 @@
|
|||||||
import asyncio
|
import asyncio
|
||||||
import board
|
|
||||||
import busio
|
|
||||||
import robot
|
|
||||||
import time
|
import time
|
||||||
|
import robot
|
||||||
import pid_controller
|
import pid_controller
|
||||||
uart = busio.UART(board.GP12, board.GP13, baudrate=9600)
|
|
||||||
|
|
||||||
|
|
||||||
class Settings:
|
class Settings:
|
||||||
speed = 0.15
|
speed = 0.17
|
||||||
time_interval = 0.2
|
time_interval = 0.2
|
||||||
motors_enabled = False
|
motors_enabled = False
|
||||||
|
|
||||||
|
|
||||||
class SpeedController:
|
class SpeedController:
|
||||||
def __init__(self, encoder, motor_fn) -> None:
|
def __init__(self, encoder, motor_fn):
|
||||||
self.encoder = encoder
|
self.encoder = encoder
|
||||||
self.motor_fn = motor_fn
|
self.motor_fn = motor_fn
|
||||||
self.pid = pid_controller.PIDController(1, 6, 0)
|
self.pid = pid_controller.PIDController(0, 6, 0)
|
||||||
self.reset()
|
self.reset()
|
||||||
|
|
||||||
def reset(self):
|
def reset(self):
|
||||||
@@ -48,20 +44,21 @@ async def motor_speed_loop():
|
|||||||
await asyncio.sleep(Settings.time_interval)
|
await asyncio.sleep(Settings.time_interval)
|
||||||
current_time = time.monotonic()
|
current_time = time.monotonic()
|
||||||
dt = current_time - last_time
|
dt = current_time - last_time
|
||||||
|
last_time = current_time
|
||||||
left.update(dt)
|
left.update(dt)
|
||||||
right.update(dt)
|
right.update(dt)
|
||||||
last_time = current_time
|
robot.uart.write(f"0, {left.error:.2f},{right.error:.2f}\n".encode())
|
||||||
uart.write(f"0, {left.error:.2f},{right.error:.2f}\n".encode())
|
|
||||||
|
|
||||||
async def stop_motors_after(seconds):
|
async def stop_motors_after(seconds):
|
||||||
await asyncio.sleep(seconds)
|
await asyncio.sleep(seconds)
|
||||||
Settings.motors_enabled = False
|
Settings.motors_enabled = False
|
||||||
# robot.stop()
|
|
||||||
|
|
||||||
async def command_handler():
|
async def command_handler():
|
||||||
while True:
|
while True:
|
||||||
if uart.in_waiting:
|
if robot.uart.in_waiting:
|
||||||
command = uart.readline().decode().strip()
|
command = robot.uart.readline().decode().strip()
|
||||||
# PID settings
|
# PID settings
|
||||||
if command.startswith("P"):
|
if command.startswith("P"):
|
||||||
left.pid.kp = float(command[1:])
|
left.pid.kp = float(command[1:])
|
||||||
@@ -74,24 +71,25 @@ async def command_handler():
|
|||||||
elif command.startswith("D"):
|
elif command.startswith("D"):
|
||||||
left.pid.kd = float(command[1:])
|
left.pid.kd = float(command[1:])
|
||||||
right.pid.kd = float(command[1:])
|
right.pid.kd = float(command[1:])
|
||||||
|
elif command.startswith("T"):
|
||||||
|
Settings.time_interval = float(command[1:])
|
||||||
# Speed settings
|
# Speed settings
|
||||||
elif command.startswith("M"):
|
elif command.startswith("M"):
|
||||||
Settings.speed = float(command[1:])
|
Settings.speed = float(command[1:])
|
||||||
elif command.startswith("T"):
|
|
||||||
Settings.time_interval = float(command[1:])
|
|
||||||
# Start/stop commands
|
# Start/stop commands
|
||||||
elif command == "O":
|
elif command == "O":
|
||||||
Settings.motors_enabled = False
|
Settings.motors_enabled = False
|
||||||
elif command.startswith("O"):
|
elif command.startswith("O"):
|
||||||
|
await asyncio.sleep(5)
|
||||||
asyncio.create_task(stop_motors_after(float(command[1:])))
|
asyncio.create_task(stop_motors_after(float(command[1:])))
|
||||||
Settings.motors_enabled = True
|
Settings.motors_enabled = True
|
||||||
left.reset()
|
left.reset()
|
||||||
right.reset()
|
right.reset()
|
||||||
# Print settings
|
# Print settings
|
||||||
elif command.startswith("?"):
|
elif command.startswith("?"):
|
||||||
uart.write(f"M{Settings.speed:.1f}\n".encode())
|
robot.uart.write(f"M{Settings.speed:.1f}\n".encode())
|
||||||
uart.write(f"T{Settings.time_interval:.1f}\n".encode())
|
robot.uart.write(f"P{left.pid.kp:.2f}:I{left.pid.ki:.2f}:D{left.pid.kd:.2f}\n".encode())
|
||||||
uart.write(f"P{left.pid.kp:.2f}:I{left.pid.ki:.2f}:D{left.pid.kd:.2f}\n".encode())
|
robot.uart.write(f"T{Settings.time_interval:.1f}\n".encode())
|
||||||
await asyncio.sleep(3)
|
await asyncio.sleep(3)
|
||||||
await asyncio.sleep(0)
|
await asyncio.sleep(0)
|
||||||
|
|
||||||
@@ -4,6 +4,9 @@ import pio_encoder
|
|||||||
import busio
|
import busio
|
||||||
import adafruit_vl53l1x
|
import adafruit_vl53l1x
|
||||||
import math
|
import math
|
||||||
|
import busio
|
||||||
|
|
||||||
|
uart = busio.UART(board.GP12, board.GP13, baudrate=9600)
|
||||||
|
|
||||||
wheel_diameter_mm = 70
|
wheel_diameter_mm = 70
|
||||||
wheel_circumference_mm = math.pi * wheel_diameter_mm
|
wheel_circumference_mm = math.pi * wheel_diameter_mm
|
||||||
@@ -22,7 +25,6 @@ motor_B1 = pwmio.PWMOut(board.GP19, frequency=100)
|
|||||||
|
|
||||||
right_motor = motor_A1, motor_A2
|
right_motor = motor_A1, motor_A2
|
||||||
left_motor = motor_B1, motor_B2
|
left_motor = motor_B1, motor_B2
|
||||||
motor_dead_zone = 0.2
|
|
||||||
|
|
||||||
right_encoder = pio_encoder.QuadratureEncoder(board.GP20, board.GP21)
|
right_encoder = pio_encoder.QuadratureEncoder(board.GP20, board.GP21)
|
||||||
left_encoder = pio_encoder.QuadratureEncoder(board.GP26, board.GP27, reversed=True)
|
left_encoder = pio_encoder.QuadratureEncoder(board.GP26, board.GP27, reversed=True)
|
||||||
@@ -42,11 +44,6 @@ def stop():
|
|||||||
|
|
||||||
|
|
||||||
def set_speed(motor, speed):
|
def set_speed(motor, speed):
|
||||||
# stop completely if in the dead zone
|
|
||||||
if abs(speed) < motor_dead_zone:
|
|
||||||
motor[0].duty_cycle = 0
|
|
||||||
motor[1].duty_cycle = 0
|
|
||||||
return
|
|
||||||
# Swap motor pins if we reverse the speed
|
# Swap motor pins if we reverse the speed
|
||||||
if speed < 0:
|
if speed < 0:
|
||||||
direction = motor[1], motor[0]
|
direction = motor[1], motor[0]
|
||||||
Reference in New Issue
Block a user