Skip to content

Commit

Permalink
Clock: Add manual time set back.
Browse files Browse the repository at this point in the history 
Fix "import ntptime" breaking clock on Badger 2040 (non W).

Use "is_wireless()" and don't squash unexpected errors.
  • Loading branch information
@Gadgetoid
Gadgetoid committed Mar 21, 2023
1 parent 0123e6b commit 5d72e59
Showing 1 changed file with 142 additions and 20 deletions.
162 changes: 142 additions & 20 deletions badger_os/examples/clock.py
View file
Original file line number Diff line number Diff line change
@@ -1,26 +1,114 @@
import time
import machine
import ntptime
import badger2040


display = badger2040.Badger2040()
display.set_update_speed(2)
display.set_thickness(4)

TIME_OFFSET = 8
WIDTH, HEIGHT = display.get_bounds()

if badger2040.is_wireless():
import ntptime
try:
display.connect()
if display.isconnected():
ntptime.settime(timezone=TIME_OFFSET)
except (RuntimeError, OSError) as e:
print(f"Wireless Error: {e.value}")

# Thonny overwrites the Pico RTC so re-sync from the physical RTC if we can
try:
display.connect()
if display.isconnected():
ntptime.settime()
except (RuntimeError, OSError):
pass # no WiFI
badger2040.pcf_to_pico_rtc()
except RuntimeError:
pass

rtc = machine.RTC()

display.set_font("gothic")

cursors = ["year", "month", "day", "hour", "minute"]
set_clock = False
toggle_set_clock = False
cursor = 0
last = 0

button_a = badger2040.BUTTONS[badger2040.BUTTON_A]
button_b = badger2040.BUTTONS[badger2040.BUTTON_B]
button_c = badger2040.BUTTONS[badger2040.BUTTON_C]

button_up = badger2040.BUTTONS[badger2040.BUTTON_UP]
button_down = badger2040.BUTTONS[badger2040.BUTTON_DOWN]


# Button handling function
def button(pin):
global last, set_clock, toggle_set_clock, cursor, year, month, day, hour, minute

time.sleep(0.01)
if not pin.value():
return

if button_a.value() and button_c.value():
machine.reset()

adjust = 0

if pin == button_b:
toggle_set_clock = True
if set_clock:
rtc.datetime((year, month, day, 0, hour, minute, second, 0))
if badger2040.is_wireless():
badger2040.pico_rtc_to_pcf()
return

if set_clock:
if pin == button_c:
cursor += 1
cursor %= len(cursors)

if pin == button_a:
cursor -= 1
cursor %= len(cursors)

if pin == button_up:
adjust = 1

if pin == button_down:
adjust = -1

if cursors[cursor] == "year":
year += adjust
year = max(year, 2022)
day = min(day, days_in_month(month, year))

if cursors[cursor] == "month":
month += adjust
month = min(max(month, 1), 12)
day = min(day, days_in_month(month, year))

if cursors[cursor] == "day":
day += adjust
day = min(max(day, 1), days_in_month(month, year))

if cursors[cursor] == "hour":
hour += adjust
hour %= 24

if cursors[cursor] == "minute":
minute += adjust
minute %= 60

draw_clock()


def days_in_month(month, year):
if month == 2 and ((year % 4 == 0 and year % 100 != 0) or year % 400 == 0):
return 29
return (31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31)[month - 1]


def draw_clock():
global second_offset, second_unit_offset
Expand All @@ -29,10 +117,18 @@ def draw_clock():
ymd = "{:04}/{:02}/{:02}".format(year, month, day)

hms_width = display.measure_text(hms, 1.8)
hms_offset = int((WIDTH / 2) - (hms_width / 2))
hms_offset = int((badger2040.WIDTH / 2) - (hms_width / 2))
h_width = display.measure_text(hms[0:2], 1.8)
mi_width = display.measure_text(hms[3:5], 1.8)
mi_offset = display.measure_text(hms[0:3], 1.8)

ymd_width = display.measure_text(ymd, 1.0)
ymd_offset = int((WIDTH / 2) - (ymd_width / 2))
ymd_offset = int((badger2040.WIDTH / 2) - (ymd_width / 2))
y_width = display.measure_text(ymd[0:4], 1.0)
m_width = display.measure_text(ymd[5:7], 1.0)
m_offset = display.measure_text(ymd[0:5], 1.0)
d_width = display.measure_text(ymd[8:10], 1.0)
d_offset = display.measure_text(ymd[0:8], 1.0)

display.set_pen(15)
display.clear()
Expand All @@ -41,15 +137,29 @@ def draw_clock():
display.text(hms, hms_offset, 40, 0, 1.8)
display.text(ymd, ymd_offset, 100, 0, 1.0)

display.set_update_speed(2)
display.update()
display.set_update_speed(3)

hms = "{:02}:{:02}:".format(hour, minute)
second_offset = hms_offset + display.measure_text(hms, 1.8)
hms = "{:02}:{:02}:{}".format(hour, minute, second // 10)
second_unit_offset = hms_offset + display.measure_text(hms, 1.8)

if set_clock:
display.set_pen(0)
if cursors[cursor] == "year":
display.line(ymd_offset, 120, ymd_offset + y_width, 120, 4)
if cursors[cursor] == "month":
display.line(ymd_offset + m_offset, 120, ymd_offset + m_offset + m_width, 120, 4)
if cursors[cursor] == "day":
display.line(ymd_offset + d_offset, 120, ymd_offset + d_offset + d_width, 120, 4)

if cursors[cursor] == "hour":
display.line(hms_offset, 70, hms_offset + h_width, 70, 4)
if cursors[cursor] == "minute":
display.line(hms_offset + mi_offset, 70, hms_offset + mi_offset + mi_width, 70, 4)

display.set_update_speed(2)
display.update()
display.set_update_speed(3)


def draw_second():
global second_offset, second_unit_offset
Expand All @@ -70,8 +180,12 @@ def draw_second():
s = "{}".format(second % 10)
display.text(s, second_unit_offset, 40, 0, 1.8)
display.partial_update(second_unit_offset, 8, 75 - (second_unit_offset - second_offset), 56)
time.sleep(0.9)


for b in badger2040.BUTTONS.values():
b.irq(trigger=machine.Pin.IRQ_RISING, handler=button)

year, month, day, wd, hour, minute, second, _ = rtc.datetime()

if (year, month, day) == (2021, 1, 1):
Expand All @@ -83,12 +197,20 @@ def draw_second():


while True:
year, month, day, wd, hour, minute, second, _ = rtc.datetime()
if second != last_second:
if minute != last_minute:
draw_clock()
last_minute = minute
else:
draw_second()
last_second = second
if not set_clock:
year, month, day, wd, hour, minute, second, _ = rtc.datetime()
if second != last_second:
if minute != last_minute:
draw_clock()
last_minute = minute
else:
draw_second()
last_second = second

if toggle_set_clock:
set_clock = not set_clock
print(f"Set clock changed to: {set_clock}")
toggle_set_clock = False
draw_clock()

time.sleep(0.01)

0 comments on commit 5d72e59

Please sign in to comment.