pimoroni-pico/micropython/examples/breakout_encoder_wheel/gpio_pwm.py

68 wiersze
2.0 KiB
Python

import math
import time
from pimoroni_i2c import PimoroniI2C
from pimoroni import BREAKOUT_GARDEN_I2C_PINS # or PICO_EXPLORER_I2C_PINS or HEADER_I2C_PINS
from breakout_encoder_wheel import BreakoutEncoderWheel, CENTRE, GPIOS, NUM_GPIOS
from breakout_ioexpander import PWM
"""
Output a sine wave PWM sequence on the Encoder Wheel's side GPIO pins.
Press the centre button or Ctrl+C to stop the program.
"""
# Constants
SPEED = 5 # The speed that the PWM will cycle at
UPDATES = 50 # How many times to update LEDs and Servos per second
UPDATE_RATE_US = 1000000 // UPDATES
FREQUENCY = 1000 # The frequency to run the PWM at
# Create a new BreakoutEncoderWheel
i2c = PimoroniI2C(**BREAKOUT_GARDEN_I2C_PINS)
wheel = BreakoutEncoderWheel(i2c)
# Set the PWM frequency for the GPIOs
period = wheel.gpio_pwm_frequency(FREQUENCY)
# Set the GPIO pins to PWM outputs
for g in GPIOS:
wheel.gpio_pin_mode(g, PWM)
# Variables
offset = 0.0
# Sleep until a specific time in the future. Use this instead of time.sleep() to correct for
# inconsistent timings when dealing with complex operations or external communication
def sleep_until(end_time):
time_to_sleep = time.ticks_diff(end_time, time.ticks_us())
if time_to_sleep > 0:
time.sleep_us(time_to_sleep)
# Make PWM waves until the centre button is pressed
while not wheel.pressed(CENTRE):
# Record the start time of this loop
start_time = time.ticks_us()
offset += SPEED / 1000.0
# Update all the PWMs
for i in range(NUM_GPIOS):
angle = ((i / NUM_GPIOS) + offset) * math.pi
duty = int(((math.sin(angle) / 2) + 0.5) * period)
# Set the GPIO pin to the new duty cycle, but do not load it yet
wheel.gpio_pin_value(GPIOS[i], duty, load=False)
# Have all the PWMs load at once
wheel.gpio_pwm_load()
# Sleep until the next update, accounting for how long the above operations took to perform
sleep_until(time.ticks_add(start_time, UPDATE_RATE_US))
# Turn off the PWM outputs
for g in GPIOS:
wheel.gpio_pin_value(g, 0)