add battery level

Author: xiongyihui

keyboard/__init__.py

@@ -10,10 +10,11 @@
 import adafruit_ble
 from adafruit_ble.advertising import Advertisement
 from adafruit_ble.advertising.standard import ProvideServicesAdvertisement
+from adafruit_ble.services.standard import BatteryService
 from adafruit_ble.services.standard.hid import HIDService
 
 from .hid import HID
-from .model import Matrix, COORDS, Backlight
+from .model import Matrix, COORDS, Backlight, battery_level
 from .action_code import *
 
 
@@ -160,7 +161,10 @@ def __init__(self, keymap=KEYMAP, pairs=(), verbose=True):
         self.heatmap = memoryview(self.data)[4:]
 
         ble_hid = HIDService()
-        self.advertisement = ProvideServicesAdvertisement(ble_hid)
+        self.battery = BatteryService()
+        self.battery.level = battery_level()
+        self.battery_update_time = time.time() + 360
+        self.advertisement = ProvideServicesAdvertisement(ble_hid, self.battery)
         self.advertisement.appearance = 961
         self.ble = adafruit_ble.BLERadio()
         self.change_bt(self.ble_id)
@@ -216,6 +220,11 @@ def check(self):
             self.backlight.set_hid_leds(leds)
             self.log('keyboard leds {}'.format(bin(leds)))
         self.update_current_conn()
+
+        # update battery level
+        if time.time() > self.battery_update_time:
+            self.battery_update_time = time.time() + 360
+            self.battery.level = battery_level()
 
     def setup(self):
         convert = lambda a: array.array('H', (get_action_code(k) for k in a))

keyboard/model/__init__.py

@@ -2,7 +2,7 @@
 
 machine = os.uname().machine
 
-if machine.find('M60 Keyboard') >= 0:
-    from .m60 import Matrix, COORDS, Backlight
-elif machine.find('Pitaya Go') >= 0:
-    from .pitaya_go import Matrix, COORDS, Backlight
+if machine.find("M60 Keyboard") >= 0:
+    from .m60 import Matrix, COORDS, Backlight, battery_level
+elif machine.find("Pitaya Go") >= 0:
+    from .pitaya_go import Matrix, COORDS, Backlight, battery_level

keyboard/model/m60.py

@@ -1,4 +1,6 @@
 
+import analogio
+import microcontroller
 from .is32fl3733 import IS31FL3733
 
 try:
@@ -27,6 +29,32 @@
 )
 
 
+BATTERY_LIMIT = 3100  # Cutoff voltage [mV].
+BATTERY_FULLLIMIT = 4190  # Full charge definition [mV].
+BATTERY_DELTA = 10  # mV between each element in the SoC vector.
+
+BATTERY_VOLTAGE = (
+    0,  0,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  2,
+    2,  2,  2,  2,  2,  2,  2,  2,  3,  3,  3,  3,  3,  3,  3,  4,  4,  4,  4,  4,
+    4,  5,  5,  5,  6,  6,  7,  7,  8,  8,  9,  9, 10, 11, 12, 13, 13, 14, 15, 16,
+    18, 19, 22, 25, 28, 32, 36, 40, 44, 47, 51, 53, 56, 58, 60, 62, 64, 66, 67, 69,
+    71, 72, 74, 76, 77, 79, 81, 82, 84, 85, 85, 86, 86, 86, 87, 88, 88, 89, 90, 91,
+    91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 100
+)
+
+battery_in = analogio.AnalogIn(microcontroller.pin.P0_02)
+
+def battery_level():
+    # (3300 * 2 * battery.value) >> 16
+    voltage = (3300 * battery_in.value) >> 15
+    i = (voltage - BATTERY_LIMIT) // BATTERY_DELTA
+    if i >= len(BATTERY_VOLTAGE):
+        i = len(BATTERY_VOLTAGE) - 1
+    elif i < 0:
+        i = 0
+    return BATTERY_VOLTAGE[i]
+
+
 class Backlight:
     def __init__(self):
         self.dev = IS31FL3733()

keyboard/model/pitaya_go.py

@@ -19,3 +19,6 @@
     41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51,  0, 52,  0,
     53, 55, 54,  0,  0, 56,  0,  0, 57, 58, 59, 60,  0,  0
 )
+
+def battery_level():
+    return 100