mirror of
https://github.com/pimoroni/grow-python
synced 2025-10-25 15:19:23 +00:00
Improved dose test
This commit is contained in:
Phil Howard
119
examples/dose.py
119
examples/dose.py
@@ -1,4 +1,5 @@
|
||||
import time
|
||||
import logging
|
||||
import RPi.GPIO as GPIO
|
||||
|
||||
from PIL import Image, ImageDraw, ImageFont
|
||||
@@ -8,28 +9,33 @@ import ST7735
|
||||
from grow.moisture import Moisture
|
||||
from grow.pump import Pump
|
||||
|
||||
# Channel settings
|
||||
pump_channel = 3
|
||||
moisture_channel = 3
|
||||
|
||||
# Default watering settings
|
||||
dry_level = 0.7 # Saturation level considered dry
|
||||
dose_speed = 0.63 # Pump speed for water dose
|
||||
dose_time = 0.96 # Time (in seconds) for water dose
|
||||
dry_level = 0.7 # Saturation level considered dry
|
||||
dose_speed = 0.63 # Pump speed for water dose
|
||||
dose_time = 0.96 # Time (in seconds) for water dose
|
||||
|
||||
|
||||
p = Pump(3)
|
||||
m = Moisture(3)
|
||||
|
||||
GPIO.setmode(GPIO.BCM)
|
||||
GPIO.setwarnings(False)
|
||||
|
||||
NUM_SAMPLES = 10
|
||||
# Here be dragons!
|
||||
FPS = 15 # Display framerate
|
||||
NUM_SAMPLES = 10 # Number of saturation level samples to average over
|
||||
DOSE_FREQUENCY = 30.0 # Minimum time between automatic waterings (in seconds)
|
||||
|
||||
BUTTONS = [5, 6, 16, 24]
|
||||
LABELS = ['A', 'B', 'X', 'Y']
|
||||
|
||||
p = Pump(pump_channel)
|
||||
m = Moisture(moisture_channel)
|
||||
|
||||
GPIO.setmode(GPIO.BCM)
|
||||
GPIO.setwarnings(False)
|
||||
GPIO.setup(BUTTONS, GPIO.IN, pull_up_down=GPIO.PUD_UP)
|
||||
|
||||
mode = 0
|
||||
|
||||
font = ImageFont.truetype(UserFont, 14)
|
||||
last_dose = time.time()
|
||||
saturation = [1.0 for _ in range(NUM_SAMPLES)]
|
||||
|
||||
display = ST7735.ST7735(
|
||||
port=0,
|
||||
@@ -42,64 +48,105 @@ display = ST7735.ST7735(
|
||||
|
||||
display.begin()
|
||||
|
||||
font = ImageFont.truetype(UserFont, 12)
|
||||
image = Image.new("RGBA", (display.width, display.height), color=(0, 0, 0))
|
||||
draw = ImageDraw.Draw(image)
|
||||
|
||||
logging.basicConfig(
|
||||
format='%(asctime)s.%(msecs)03d %(levelname)-8s %(message)s',
|
||||
level=logging.INFO,
|
||||
datefmt='%Y-%m-%d %H:%M:%S')
|
||||
|
||||
|
||||
def handle_button(pin):
|
||||
global mode, dose_time, dose_speed, dry_level
|
||||
global mode, last_dose, dose_time, dose_speed, dry_level
|
||||
index = BUTTONS.index(pin)
|
||||
label = LABELS[index]
|
||||
if label == 'A':
|
||||
if label == 'A': # Test
|
||||
logging.info("Manual watering triggered.")
|
||||
p.dose(dose_speed, dose_time, blocking=False)
|
||||
if label == 'B':
|
||||
last_dose = time.time()
|
||||
|
||||
if label == 'B': # Switch setting
|
||||
mode += 1
|
||||
mode %= 2
|
||||
if label == 'Y':
|
||||
mode %= 3 # Wrap 0, 1, 2 (Time, Speed, Dry level)
|
||||
|
||||
if label == 'Y': # Inc. setting
|
||||
if mode == 0:
|
||||
dose_time += 0.01
|
||||
logging.info("Dose time increased to: {:.2f}".format(dose_time))
|
||||
elif mode == 1:
|
||||
dose_speed += 0.01
|
||||
logging.info("Dose speed increased to: {:.2f}".format(dose_speed))
|
||||
elif mode == 2:
|
||||
dry_level += 0.01
|
||||
if label == 'X':
|
||||
logging.info("Dry level increased to: {:.2f}".format(dry_level))
|
||||
|
||||
if label == 'X': # Dec. setting
|
||||
if mode == 0:
|
||||
dose_time -= 0.01
|
||||
logging.info("Dose time decreased to: {:.2f}".format(dose_time))
|
||||
elif mode == 1:
|
||||
dose_speed -= 0.01
|
||||
logging.info("Dose speed decreased to: {:.2f}".format(dose_speed))
|
||||
elif mode == 2:
|
||||
dry_level -= 0.01
|
||||
print("Button {} pressed! Dose time: {} Dose speed: {} Dry level: {}".format(label, dose_time, dose_speed, dry_level))
|
||||
logging.info("Dry level decreased to: {:.2f}".format(dry_level))
|
||||
|
||||
|
||||
# Bind the button handler (above) to all buttons
|
||||
for pin in BUTTONS:
|
||||
GPIO.add_event_detect(pin, GPIO.FALLING, handle_button, bouncetime=150)
|
||||
|
||||
saturation = [1.0 for _ in range(NUM_SAMPLES)]
|
||||
|
||||
current_saturation = 0
|
||||
|
||||
try:
|
||||
while True:
|
||||
s = m.saturation
|
||||
print(time.time(), s)
|
||||
saturation.append(s)
|
||||
saturation = saturation[-NUM_SAMPLES:]
|
||||
current = sum(saturation) / float(NUM_SAMPLES)
|
||||
if current < dry_level:
|
||||
# New moisture readings are available approximately 1/sec
|
||||
# push them into the list for averagering
|
||||
if m.new_data:
|
||||
current_saturation = m.saturation
|
||||
saturation.append(current_saturation)
|
||||
saturation = saturation[-NUM_SAMPLES:]
|
||||
|
||||
avg_saturation = sum(saturation) / float(NUM_SAMPLES)
|
||||
|
||||
# Trigger a dose of water if the average saturation is less than the specified dry level
|
||||
# dose frequency is rate limited, so this doesn't re-trigger before the moistrure sensor
|
||||
# has had the opportunity to catch up.
|
||||
if avg_saturation < dry_level and (time.time() - last_dose) > DOSE_FREQUENCY:
|
||||
p.dose(dose_speed, dose_time)
|
||||
print("Auto dosing. Saturation: {}".format(current))
|
||||
time.sleep(2.0)
|
||||
logging.info("Auto watering. Saturation: {:.2f} (Dry: {:.2f})".format(avg_saturation, dry_level))
|
||||
last_dose = time.time()
|
||||
|
||||
draw.rectangle((0, 0, display.width, display.height), (0, 0, 0))
|
||||
|
||||
draw.text((10, 0), "Sat: {} AVG: {}".format(s, current), font=font, fill=(255, 255, 255))
|
||||
# Current and average saturation
|
||||
draw.text((5 + display.width // 2, 16), "Sat: {:.3f}".format(current_saturation), font=font, fill=(255, 255, 255))
|
||||
draw.text((5 + display.width // 2, 32), "AVG: {:.3f}".format(avg_saturation), font=font, fill=(255, 255, 255))
|
||||
|
||||
draw.text((10, 16), "Time: {:.2f}".format(dose_time), font=font, fill=(255, 255, 255))
|
||||
draw.text((10, 32), "Speed: {:.2f}".format(dose_speed), font=font, fill=(255, 255, 255))
|
||||
draw.text((10, 48), "Dry: {:.2f}".format(dry_level), font=font, fill=(255, 255, 255))
|
||||
# Selected setting box
|
||||
draw.rectangle((0, 16 + (16 * mode), display.width // 2, 31 + (16 * mode)), (30, 30, 30))
|
||||
|
||||
draw.text((0, 16 + (16 * mode)), ">", font=font, fill=(255, 255, 255))
|
||||
draw.text((5, 16), "Time: {:.2f}".format(dose_time), font=font, fill=(255, 255, 255) if mode == 0 else (128, 128, 128))
|
||||
draw.text((5, 32), "Speed: {:.2f}".format(dose_speed), font=font, fill=(255, 255, 255) if mode == 1 else (128, 128, 128))
|
||||
draw.text((5, 48), "Dry lvl: {:.2f}".format(dry_level), font=font, fill=(255, 255, 255) if mode == 2 else (128, 128, 128))
|
||||
|
||||
# Button lavel backgrounds
|
||||
draw.rectangle((0, 0, 42, 14), (255, 255, 255))
|
||||
draw.rectangle((display.width - 15, 0, display.width, 14), (255, 255, 255))
|
||||
draw.rectangle((display.width - 15, display.height - 14, display.width, display.height), (255, 255, 255))
|
||||
draw.rectangle((0, display.height - 14, 42, display.height), (255, 255, 255))
|
||||
|
||||
# Button labels
|
||||
draw.text((5, 0), "test", font=font, fill=(0, 0, 0))
|
||||
draw.text((5, display.height - 16), "next", font=font, fill=(0, 0, 0))
|
||||
draw.text((display.width - 10, 0), "-", font=font, fill=(0, 0, 0))
|
||||
draw.text((display.width - 12, display.height - 15), "+", font=font, fill=(0, 0, 0))
|
||||
|
||||
display.display(image)
|
||||
time.sleep(1.0)
|
||||
time.sleep(1.0 / FPS)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
print("Dose Time: {} Dose Speed: {}, Dry Level: {}".format(dose_time, dose_speed, dry_level))
|
||||
print("Dose Time: {:.2f} Dose Speed: {:.2f}, Dry Level: {:.2f}".format(dose_time, dose_speed, dry_level))
|
||||
|
||||
Reference in New Issue
Block a user