66 lines
1.6 KiB
Python
Executable File
66 lines
1.6 KiB
Python
Executable File
import board
|
|
import pwmio
|
|
import pio_encoder
|
|
import busio
|
|
import adafruit_vl53l1x
|
|
import math
|
|
import busio
|
|
|
|
uart = busio.UART(board.GP12, board.GP13, baudrate=9600)
|
|
|
|
wheel_diameter_mm = 70
|
|
wheel_circumference_mm = math.pi * wheel_diameter_mm
|
|
gear_ratio = 298
|
|
encoder_poles = 28
|
|
ticks_per_revolution = encoder_poles * gear_ratio
|
|
ticks_to_mm_const = wheel_circumference_mm / ticks_per_revolution
|
|
|
|
def ticks_to_mm(ticks):
|
|
return ticks_to_mm_const * ticks
|
|
|
|
motor_A2 = pwmio.PWMOut(board.GP17, frequency=100)
|
|
motor_A1 = pwmio.PWMOut(board.GP16, frequency=100)
|
|
motor_B2 = pwmio.PWMOut(board.GP18, frequency=100)
|
|
motor_B1 = pwmio.PWMOut(board.GP19, frequency=100)
|
|
|
|
right_motor = motor_A1, motor_A2
|
|
left_motor = motor_B1, motor_B2
|
|
|
|
right_encoder = pio_encoder.QuadratureEncoder(board.GP20, board.GP21)
|
|
left_encoder = pio_encoder.QuadratureEncoder(board.GP26, board.GP27, reversed=True)
|
|
|
|
i2c0 = busio.I2C(sda=board.GP0, scl=board.GP1)
|
|
i2c1 = busio.I2C(sda=board.GP2, scl=board.GP3)
|
|
|
|
left_distance = adafruit_vl53l1x.VL53L1X(i2c0)
|
|
right_distance = adafruit_vl53l1x.VL53L1X(i2c1)
|
|
|
|
|
|
def stop():
|
|
motor_A1.duty_cycle = 0
|
|
motor_A2.duty_cycle = 0
|
|
motor_B1.duty_cycle = 0
|
|
motor_B2.duty_cycle = 0
|
|
|
|
|
|
def set_speed(motor, speed):
|
|
# Swap motor pins if we reverse the speed
|
|
if speed < 0:
|
|
direction = motor[1], motor[0]
|
|
speed = -speed
|
|
else:
|
|
direction = motor
|
|
speed = min(speed, 1) # limit to 1.0
|
|
max_speed = 2 ** 16 - 1
|
|
|
|
direction[0].duty_cycle = int(max_speed * speed)
|
|
direction[1].duty_cycle = 0
|
|
|
|
|
|
def set_left(speed):
|
|
set_speed(left_motor, speed)
|
|
|
|
|
|
def set_right(speed):
|
|
set_speed(right_motor, speed)
|