111 lines
3 KiB
Python
Executable file
111 lines
3 KiB
Python
Executable file
# SPDX-FileCopyrightText: 2023 Trevor Beaton for Adafruit Industries
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
import time
|
|
import ssl
|
|
import board
|
|
import math
|
|
import wifi
|
|
import socketpool
|
|
import adafruit_minimqtt.adafruit_minimqtt as MQTT
|
|
from adafruit_io.adafruit_io import IO_MQTT
|
|
from adafruit_adxl34x import ADXL345
|
|
from adafruit_lc709203f import LC709203F, PackSize
|
|
|
|
|
|
try:
|
|
from secrets import secrets
|
|
except ImportError:
|
|
print("WiFi and Adafruit IO credentials are kept in secrets.py - please add them there!")
|
|
raise
|
|
|
|
aio_username = secrets["aio_username"]
|
|
aio_key = secrets["aio_key"]
|
|
|
|
|
|
# Wi-Fi
|
|
try:
|
|
print("Connecting to %s" % secrets["ssid"])
|
|
wifi.radio.connect(secrets["ssid"], secrets["password"])
|
|
print("Connected to %s!" % secrets["ssid"])
|
|
# Wi-Fi connectivity fails with error messages, not specific errors, so this except is broad.
|
|
except Exception as e: # pylint: disable=broad-except
|
|
print("Failed to connect to WiFi. Error:", e, "\nBoard will hard reset in 5 seconds.")
|
|
time.sleep(5)
|
|
microcontroller.reset()
|
|
|
|
|
|
|
|
# Create a socket pool
|
|
pool = socketpool.SocketPool(wifi.radio)
|
|
|
|
# Initialize a new MQTT Client object
|
|
mqtt_client = MQTT.MQTT(
|
|
broker="io.adafruit.com",
|
|
username=secrets["aio_username"],
|
|
password=secrets["aio_key"],
|
|
socket_pool=pool,
|
|
ssl_context=ssl.create_default_context(),
|
|
)
|
|
|
|
# Initialize Adafruit IO MQTT "helper"
|
|
io = IO_MQTT(mqtt_client)
|
|
|
|
def connected(client):
|
|
print("Connected to Adafruit IO!")
|
|
|
|
|
|
# Set up the callback methods above
|
|
io.on_connect = connected
|
|
|
|
try:
|
|
# If Adafruit IO is not connected...
|
|
if not io.is_connected:
|
|
# Connect the client to the MQTT broker.
|
|
print("Connecting to Adafruit IO...")
|
|
io.connect()
|
|
|
|
except Exception as e: # pylint: disable=broad-except
|
|
print("Failed to get or send data, or connect. Error:", e,
|
|
"\nBoard will hard reset in 30 seconds.")
|
|
time.sleep(30)
|
|
microcontroller.reset()
|
|
|
|
threshold = 25 # set threshold value here
|
|
time_interval = 0.1 # set the time interval in seconds
|
|
|
|
# create the I2C bus object
|
|
i2c = board.STEMMA_I2C()
|
|
# For ADXL345
|
|
accelerometer = ADXL345(i2c)
|
|
|
|
battery_monitor = LC709203F(i2c)
|
|
battery_monitor.pack_size = PackSize.MAH400
|
|
|
|
|
|
# initialize velocity variables to zero
|
|
velocity_x = 0
|
|
velocity_y = 0
|
|
velocity_z = 0
|
|
|
|
t0 = time.monotonic()
|
|
|
|
while True:
|
|
x, y, z = accelerometer.acceleration
|
|
t1 = time.monotonic()
|
|
dt = t1 - t0
|
|
velocity_x += x*dt
|
|
velocity_y += y*dt
|
|
velocity_z += z*dt
|
|
|
|
total_acceleration = math.sqrt(x**2 + y**2 + z**2)
|
|
if total_acceleration >= threshold:
|
|
print("Battery Percent: {:.2f} %".format(battery_monitor.cell_percent))
|
|
print("Collision strength: %.2f" % total_acceleration)
|
|
print("Velocity X: %.2f Y: %.2f Z: %.2f \n" % (velocity_x, velocity_y, velocity_z))
|
|
io.publish("punch-strength", total_acceleration)
|
|
io.publish("punch-velocity", velocity_x)
|
|
|
|
# add code here to trigger an event or alert the user
|
|
t0 = t1
|
|
time.sleep(time_interval)
|