222 lines
6.4 KiB
Python
222 lines
6.4 KiB
Python
"""
|
|
mixgo_soar onboard resources
|
|
|
|
Micropython library for the mixgo_soar onboard resources
|
|
=======================================================
|
|
@dahanzimin From the Mixly Team
|
|
"""
|
|
|
|
from ws2812 import NeoPixel
|
|
from machine import *
|
|
import time
|
|
import gc
|
|
import math
|
|
import st7789_bf
|
|
|
|
'''RTC'''
|
|
rtc_clock = RTC()
|
|
|
|
'''I2C-onboard'''
|
|
# onboard_i2c = I2C(0)
|
|
onboard_i2c = SoftI2C(scl=Pin(47), sda=Pin(48), freq=400000)
|
|
|
|
'''SPI-onboard'''
|
|
onboard_spi = SPI(1, baudrate=80000000, polarity=0, phase=0)
|
|
|
|
'''BOT035-Sensor'''
|
|
try:
|
|
import soar_bot
|
|
onboard_bot = soar_bot.BOT035(onboard_i2c)
|
|
except Exception as e:
|
|
print("Warning: Failed to communicate with BOT035 (Coprocessor) or", e)
|
|
|
|
'''BPS-Sensor'''
|
|
try:
|
|
import spl06_001
|
|
onboard_bps = spl06_001.SPL06(onboard_i2c)
|
|
except Exception as e:
|
|
print("Warning: Failed to communicate with SPL06-001 (BPS) or", e)
|
|
|
|
'''IMU-Sensor'''
|
|
try:
|
|
import qmi8658
|
|
onboard_imu = qmi8658.QMI8658(onboard_i2c)
|
|
except Exception as e:
|
|
print("Warning: Failed to communicate with QMI8658 (IMU) or", e)
|
|
|
|
'''ALS_PS-Sensor'''
|
|
try:
|
|
import ltr553als
|
|
onboard_als = ltr553als.LTR_553ALS(onboard_i2c)
|
|
except Exception as e:
|
|
print("Warning: Failed to communicate with TR_553ALS (ALS&PS) or", e)
|
|
|
|
'''MGS-Sensor'''
|
|
try:
|
|
import mmc5603
|
|
onboard_mgs = mmc5603.MMC5603(onboard_i2c)
|
|
except Exception as e:
|
|
print("Warning: Failed to communicate with MMC5603 (MGS) or", e)
|
|
|
|
'''TFT/240*240'''
|
|
onboard_tft = st7789_bf.ST7789(onboard_spi, 240, 240, dc_pin=46, cs_pin=45, bl_pin=onboard_bot.tft_brightness, brightness=0.6, font_address=0x700000)
|
|
|
|
'''2RGB_WS2812'''
|
|
onboard_rgb = NeoPixel(Pin(40), 4)
|
|
|
|
'''5KEY_Sensor'''
|
|
class KEYSensor:
|
|
def __init__(self, pin, range):
|
|
self.pin = pin
|
|
self.adc = ADC(Pin(pin), atten=ADC.ATTN_0DB)
|
|
self.range = range
|
|
self.flag = True
|
|
|
|
def _value(self):
|
|
values = []
|
|
for _ in range(50):
|
|
values.append(self.adc.read())
|
|
time.sleep_us(2)
|
|
return (self.range-200) < min(values) < (self.range+200)
|
|
|
|
def get_presses(self, delay=1):
|
|
last_time, presses = time.time(), 0
|
|
while time.time() < last_time + delay:
|
|
time.sleep_ms(50)
|
|
if self.was_pressed():
|
|
presses += 1
|
|
return presses
|
|
|
|
def is_pressed(self):
|
|
return self._value()
|
|
|
|
def was_pressed(self):
|
|
if (self._value() != self.flag):
|
|
self.flag = self._value()
|
|
if self.flag:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def irq(self, handler, trigger):
|
|
Pin(self.pin, Pin.IN).irq(handler=handler, trigger=trigger)
|
|
|
|
'''1KEY_Button'''
|
|
class Button(KEYSensor):
|
|
def __init__(self, pin):
|
|
self.pin = pin
|
|
self.key = Pin(pin, Pin.IN)
|
|
self.flag = True
|
|
|
|
def _value(self):
|
|
return not self.key.value()
|
|
|
|
B1key = Button(0)
|
|
B2key = KEYSensor(17, 0)
|
|
A1key = KEYSensor(17, 2300)
|
|
A2key = KEYSensor(17, 1600)
|
|
A3key = KEYSensor(17, 800)
|
|
A4key = KEYSensor(17, 2900)
|
|
|
|
'''2LED-Independent'''
|
|
class LED:
|
|
def __init__(self, pins=[]):
|
|
self._pins = [PWM(Pin(pin), duty_u16=0) for pin in pins]
|
|
self._brightness = [0 for _ in range(len(self._pins))]
|
|
|
|
def setbrightness(self, index, val):
|
|
if not 0 <= val <= 100:
|
|
raise ValueError("Brightness must be in the range: 0-100%")
|
|
self._brightness[index - 1] = val
|
|
self._pins[index - 1].duty_u16(val * 65535 // 100)
|
|
|
|
def getbrightness(self, index):
|
|
return self._brightness[index - 1]
|
|
|
|
def setonoff(self, index, val):
|
|
if val == -1:
|
|
self.setbrightness(index, 100) if self.getbrightness(
|
|
index) < 50 else self.setbrightness(index, 0)
|
|
elif val == 1:
|
|
self.setbrightness(index, 100)
|
|
elif val == 0:
|
|
self.setbrightness(index, 0)
|
|
|
|
def getonoff(self, index):
|
|
return True if self.getbrightness(index) > 50 else False
|
|
|
|
onboard_led = LED([38, 39])
|
|
|
|
class Clock:
|
|
def __init__(self, x, y, radius, color, oled=onboard_tft): # 定义时钟中心点和半径
|
|
self.display = oled
|
|
self.xc = x
|
|
self.yc = y
|
|
self.r = radius
|
|
self.color = color
|
|
self.hour = 0
|
|
self.min = 0
|
|
self.sec = 0
|
|
|
|
def set_time(self, h, m, s): # 设定时间
|
|
self.hour = h
|
|
self.min = m
|
|
self.sec = s
|
|
|
|
def set_rtctime(self): # 设定时间
|
|
t = rtc_clock.datetime()
|
|
self.hour = t[4]
|
|
self.min = t[5]
|
|
self.sec = t[6]
|
|
|
|
def drawDial(self, color): # 画钟表刻度
|
|
r_tic1 = self.r - 1
|
|
r_tic2 = self.r - 2
|
|
self.display.ellipse(self.xc, self.yc, self.r, self.r, color)
|
|
self.display.ellipse(self.xc, self.yc, 2, 2, color, True)
|
|
|
|
for h in range(12):
|
|
at = math.pi * 2.0 * h / 12.0
|
|
x1 = round(self.xc + r_tic1 * math.sin(at))
|
|
x2 = round(self.xc + r_tic2 * math.sin(at))
|
|
y1 = round(self.yc - r_tic1 * math.cos(at))
|
|
y2 = round(self.yc - r_tic2 * math.cos(at))
|
|
self.display.line(x1, y1, x2, y2, color)
|
|
|
|
def drawHour(self, color): # 画时针
|
|
r_hour = int(self.r / 10.0 * 5)
|
|
ah = math.pi * 2.0 * ((self.hour % 12) + self.min / 60.0) / 12.0
|
|
xh = int(self.xc + r_hour * math.sin(ah))
|
|
yh = int(self.yc - r_hour * math.cos(ah))
|
|
self.display.line(self.xc, self.yc, xh, yh, color)
|
|
|
|
def drawMin(self, color): # 画分针
|
|
r_min = int(self.r / 10.0 * 7)
|
|
am = math.pi * 2.0 * self.min / 60.0
|
|
xm = round(self.xc + r_min * math.sin(am))
|
|
ym = round(self.yc - r_min * math.cos(am))
|
|
self.display.line(self.xc, self.yc, xm, ym, color)
|
|
|
|
def drawSec(self, color): # 画秒针
|
|
r_sec = int(self.r / 10.0 * 9)
|
|
asec = math.pi * 2.0 * self.sec / 60.0
|
|
xs = round(self.xc + r_sec * math.sin(asec))
|
|
ys = round(self.yc - r_sec * math.cos(asec))
|
|
self.display.line(self.xc, self.yc, xs, ys, color)
|
|
|
|
def draw_clock(self, bg_color=0): # 画完整钟表
|
|
self.drawDial(self.color)
|
|
self.drawHour(self.color)
|
|
self.drawMin(self.color)
|
|
self.drawSec(self.color)
|
|
self.display.show()
|
|
self.drawHour(bg_color)
|
|
self.drawMin(bg_color)
|
|
self.drawSec(bg_color)
|
|
|
|
def clear(self, color=0): # 清除
|
|
self.display.ellipse(self.xc, self.yc, self.r, self.r, color, True)
|
|
|
|
'''Reclaim memory'''
|
|
gc.collect()
|