Badger2040: Improve clock

* Toggle into time-set mode with B
* Move to date/time parts with A/C
* Change values with up/down
* Centered clock so it looks nicer

micropython/examples/badger2040/clock.py

@@ -1,30 +1,140 @@
 import time
+import machine
 import badger2040
-from machine import RTC
 
-rtc = RTC()
+
+rtc = machine.RTC()
 screen = badger2040.Badger2040()
 screen.update_speed(badger2040.UPDATE_TURBO)
 screen.font("gothic")
 
-year, month, day, wd, hour, minute, second, _ = rtc.datetime()
+cursors = ["year", "month", "day", "hour", "minute"]
+set_clock = False
+cursor = 0
+last = 0
 
-if (year, month, day) == (2021, 1, 1):
-    rtc.datetime((2022, 2, 25, 3, 12, 0, 0, 0))
+# Set up the buttons
+button_down = machine.Pin(badger2040.BUTTON_DOWN, machine.Pin.IN, machine.Pin.PULL_DOWN)
+button_up = machine.Pin(badger2040.BUTTON_UP, machine.Pin.IN, machine.Pin.PULL_DOWN)
 
-while True:
-    year, month, day, wd, hour, minute, second, _ = rtc.datetime()
+button_a = machine.Pin(badger2040.BUTTON_A, machine.Pin.IN, machine.Pin.PULL_DOWN)
+button_b = machine.Pin(badger2040.BUTTON_B, machine.Pin.IN, machine.Pin.PULL_DOWN)
+button_c = machine.Pin(badger2040.BUTTON_C, machine.Pin.IN, machine.Pin.PULL_DOWN)
+
+
+# Button handling function
+def button(pin):
+    global last, set_clock, cursor, year, month, day, hour, minute
+
+    time.sleep(0.05)
+    if not pin.value():
+        return
+
+    adjust = 0
+    changed = False
+
+    if pin == button_b:
+        set_clock = not set_clock
+        changed = True
+        if not set_clock:
+            rtc.datetime((year, month, day, 0, hour, minute, second, 0))
+
+    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)
+        if cursors[cursor] == "month":
+            month += adjust
+            month = min(max(month, 1), 12)
+        if cursors[cursor] == "day":
+            day += adjust
+            day = min(max(day, 1), 31)
+        if cursors[cursor] == "hour":
+            hour += adjust
+            hour %= 60
+        if cursors[cursor] == "minute":
+            minute += adjust
+            minute %= 60
+
+    if set_clock or changed:
+        draw_clock()
+
+
+# Register the button handling function with the buttons
+button_down.irq(trigger=machine.Pin.IRQ_RISING, handler=button)
+button_up.irq(trigger=machine.Pin.IRQ_RISING, handler=button)
+button_a.irq(trigger=machine.Pin.IRQ_RISING, handler=button)
+button_b.irq(trigger=machine.Pin.IRQ_RISING, handler=button)
+button_c.irq(trigger=machine.Pin.IRQ_RISING, handler=button)
+
+
+def draw_clock():
     hms = "{:02}:{:02}:{:02}".format(hour, minute, second)
     ymd = "{:04}/{:02}/{:02}".format(year, month, day)
 
+    hms_width = screen.measure_text(hms, 1.8)
+    hms_offset = int((badger2040.WIDTH / 2) - (hms_width / 2))
+    h_width = screen.measure_text(hms[0:2], 1.8)
+    mi_width = screen.measure_text(hms[3:5], 1.8)
+    mi_offset = screen.measure_text(hms[0:3], 1.8)
+
+    ymd_width = screen.measure_text(ymd, 1.0)
+    ymd_offset = int((badger2040.WIDTH / 2) - (ymd_width / 2))
+    y_width = screen.measure_text(ymd[0:4], 1.0)
+    m_width = screen.measure_text(ymd[5:7], 1.0)
+    m_offset = screen.measure_text(ymd[0:5], 1.0)
+    d_width = screen.measure_text(ymd[8:10], 1.0)
+    d_offset = screen.measure_text(ymd[0:8], 1.0)
+
     screen.pen(15)
     screen.clear()
     screen.pen(0)
     screen.thickness(5)
-    screen.text(hms, 0, 40, 1.8)
+    screen.text(hms, hms_offset, 40, 1.8)
     screen.thickness(3)
-    screen.text(ymd, 0, 100, 1.0)
+    screen.text(ymd, ymd_offset, 100, 1.0)
+
+    if set_clock:
+        if cursors[cursor] == "year":
+            screen.line(ymd_offset, 120, ymd_offset + y_width, 120)
+        if cursors[cursor] == "month":
+            screen.line(ymd_offset + m_offset, 120, ymd_offset + m_offset + m_width, 120)
+        if cursors[cursor] == "day":
+            screen.line(ymd_offset + d_offset, 120, ymd_offset + d_offset + d_width, 120)
+
+        if cursors[cursor] == "hour":
+            screen.line(hms_offset, 70, hms_offset + h_width, 70)
+        if cursors[cursor] == "minute":
+            screen.line(hms_offset + mi_offset, 70, hms_offset + mi_offset + mi_width, 70)
 
     screen.update()
 
-    time.sleep(0.5)
+
+year, month, day, wd, hour, minute, second, _ = rtc.datetime()
+
+if (year, month, day) == (2021, 1, 1):
+    rtc.datetime((2022, 2, 29, 3, 12, 0, 0, 0))
+
+last_second = second
+
+while True:
+    if not set_clock:
+        year, month, day, wd, hour, minute, second, _ = rtc.datetime()
+        if second != last_second:
+            draw_clock()
+            last_second = second
+    time.sleep(0.1)