149 lines
6.5 KiB
Python
149 lines
6.5 KiB
Python
"""
|
|
uv-belichter-software - The main program run at the Pi Picos startup
|
|
Copyright (C) 2024 Benjamin Burkhardt
|
|
|
|
This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
"""
|
|
|
|
|
|
import utils
|
|
from lcdMenu import lcdMenu
|
|
from WelcomeScreen import WelcomeScreen
|
|
from time import sleep, time_ns
|
|
from gc import collect # garbage collector for better memory performance
|
|
|
|
config = utils.Config()
|
|
btn_mapping = {"ok_btn": config.PIN_IN_BTN_1, "next_btn": config.PIN_IN_BTN_2} # the btn mapping for all lcdMenus
|
|
|
|
# extra functions to access the garbage collector
|
|
def manual():
|
|
config.LCD.clear()
|
|
set_value = config.PIN_OUT_RELAIS.value()
|
|
config.LCD.putstr(f"---- MANUAL ---- State: {set_value} ")
|
|
while True:
|
|
if set_value != config.PIN_IN_SWITCH.value():
|
|
config.PIN_OUT_RELAIS.value(config.PIN_IN_SWITCH.value())
|
|
set_value = config.PIN_OUT_RELAIS.value()
|
|
config.LCD.putstr(f"---- MANUAL ---- State: {set_value} ")
|
|
if config.PIN_IN_BTN_1.value() == 1 or config.PIN_IN_BTN_2.value() == 1:
|
|
return True # exit on press of these buttons; True to disable the Quitting message from lcdMenu
|
|
def timer():
|
|
# display WIP
|
|
config.LCD.clear()
|
|
config.LCD.putstr(" Still work-in-progress")
|
|
sleep(3)
|
|
return True # disable the "Quitting" message from lcdMenu
|
|
def uv_on():
|
|
config.PIN_OUT_RELAIS.value(1)
|
|
config.LCD.clear()
|
|
config.LCD.putstr("------ UV ------ turned on ")
|
|
sleep(1)
|
|
return True # disable the "Quitting" message from lcdMenu
|
|
def uv_off():
|
|
config.PIN_OUT_RELAIS.value(0)
|
|
config.LCD.clear()
|
|
config.LCD.putstr("------ UV ------ turned off ")
|
|
sleep(1)
|
|
return True # disable the "Quitting" message from lcdMenu
|
|
def lcd_big_hello():
|
|
import lcd_big_hello as lbh
|
|
lbh.run(config.LCD)
|
|
del lbh
|
|
collect()
|
|
return True
|
|
def input_tests():
|
|
import input_tests as input_tests
|
|
input_tests.run(serial_output=False)
|
|
collect()
|
|
return True
|
|
def settings():
|
|
settings_menu = lcdMenu(config.LCD, btn_mapping, scroll_direction=True, cycle=True, hide_menu_name=False, name="SETTINGS")
|
|
def swap_welcome():
|
|
config.STARTUP_WELCOME_SHOW = not config.STARTUP_WELCOME_SHOW
|
|
config.LCD.clear()
|
|
config.LCD.putstr(f" Welcome Screen ")
|
|
if config.STARTUP_WELCOME_SHOW:
|
|
config.LCD.putstr("--- Enabled! ---")
|
|
else:
|
|
config.LCD.putstr("--- Disabled ---")
|
|
sleep(1)
|
|
def welcome_cycles():
|
|
config.LCD.clear()
|
|
current_cycles = config.STARTUP_WELCOME_CYCLES
|
|
|
|
option_down = [" ", "v"][current_cycles>1]
|
|
config.LCD.putstr(f" Cycles \n{option_down} {str(current_cycles).center(12)} ^")
|
|
while True:
|
|
if config.PIN_IN_BTN_1.value() == 1:
|
|
time_ns_when_pressed = time_ns()
|
|
while config.PIN_IN_BTN_1.value() == 1:
|
|
if (time_ns() - time_ns_when_pressed) > 1000000000: # if the time passed is longer than a second
|
|
while config.PIN_IN_BTN_1.value() == 1:
|
|
pass # wait till release
|
|
config.STARTUP_WELCOME_CYCLES = current_cycles
|
|
return True
|
|
sleep(0.05)
|
|
current_cycles -= 1
|
|
if current_cycles < 1: current_cycles = 1
|
|
option_down = [" ", "v"][current_cycles>1]
|
|
config.LCD.putstr(f" Cycles \n{option_down} {str(current_cycles).center(12)} ^")
|
|
if config.PIN_IN_BTN_2.value() == 1:
|
|
time_ns_when_pressed = time_ns()
|
|
while config.PIN_IN_BTN_2.value() == 1:
|
|
if (time_ns() - time_ns_when_pressed) > 1000000000: # if the time passed is longer than a second
|
|
while config.PIN_IN_BTN_2.value() == 1:
|
|
pass # wait till release
|
|
config.STARTUP_WELCOME_CYCLES = current_cycles
|
|
return True
|
|
sleep(0.05)
|
|
current_cycles += 1
|
|
option_down = [" ", "v"][current_cycles>1]
|
|
config.LCD.putstr(f" Cycles \n{option_down} {str(current_cycles).center(12)} ^")
|
|
|
|
settings_programs = [("Show welcome", swap_welcome),
|
|
("Welcome cycles", welcome_cycles),
|
|
("Exit", settings_menu.stop)]
|
|
settings_menu.setup(settings_programs) # give it the callback list
|
|
settings_menu.run() # run the menu until it's closed
|
|
del settings_menu, settings_programs
|
|
collect()
|
|
return True
|
|
def run_demo_menu():
|
|
demo_menu = lcdMenu(config.LCD, btn_mapping, scroll_direction=True, cycle=True, hide_menu_name=False, name="DEMOS")
|
|
demo_programs = [("LCD Demo", lcd_big_hello),
|
|
("Input tests", input_tests),
|
|
("Exit", demo_menu.stop)]
|
|
demo_menu.setup(demo_programs) # give it the callback list
|
|
ret = demo_menu.run()
|
|
del demo_menu, demo_programs
|
|
collect()
|
|
return ret
|
|
|
|
if config.STARTUP_WELCOME_SHOW:
|
|
ws = WelcomeScreen(config.LCD,
|
|
interrupt_pins=[config.PIN_IN_BTN_1, config.PIN_IN_BTN_2, config.PIN_IN_SWITCH],
|
|
subtitle=config.STARTUP_PROJECT_NAME,
|
|
starting_msg=config.STARTUP_MESSAGE_STARTING,
|
|
started_msg=config.STARTUP_MESSAGE_FINISHED)
|
|
ws.show(cycles=config.STARTUP_WELCOME_CYCLES)
|
|
del ws
|
|
collect()
|
|
|
|
|
|
# create the menus
|
|
main_menu = lcdMenu(config.LCD, btn_mapping, scroll_direction=True, cycle=True, hide_menu_name=False, name="PROGRAMS")
|
|
main_programs = [("Timer", timer),
|
|
("Manual", manual),
|
|
("UV off", uv_off),
|
|
("UV on", uv_on),
|
|
("Demos", run_demo_menu),
|
|
("Settings", settings)]
|
|
main_menu.setup(main_programs) # give it the callback list
|
|
|
|
# and run the main menu (will be an endless loop)
|
|
main_menu.run()
|