Merge pull request #199 from SHA2017-badge/basvs-stay-awake-on-usb

add option to stay awake on usb; suppress debug logging

Author: raboof

esp32/modules/easydraw.py

@@ -39,15 +39,15 @@ def nickname(y = 25, font = "PermanentMarker36", color = ugfx.BLACK):
     nick = badge.nvs_get_str("owner", "name", 'Henk de Vries')
     ugfx.string_box(0,y,296,38, nick, font, color, ugfx.justifyCenter)
 
-def battery(vUsb, vBatt, charging):
+def battery(on_usb, vBatt, charging):
     vMin = badge.nvs_get_u16('batt', 'vmin', 3500) # mV
     vMax = badge.nvs_get_u16('batt', 'vmax', 4100) # mV
-    if charging and vUsb>4000:
+    if charging and on_usb:
         try:
             badge.eink_png(0,0,'/lib/resources/chrg.png')
         except:
             ugfx.string(0, 0, "CHRG",'Roboto_Regular12',ugfx.BLACK)
-    elif vUsb>4000:
+    elif on_usb:
         try:
             badge.eink_png(0,0,'/lib/resources/usb.png')
         except:

esp32/modules/splash.py

@@ -50,12 +50,12 @@ def disp_string_right(y, s):
 
         easydraw.nickname()
 
-        vUsb = badge.usb_volt_sense()
+        on_usb = pm.usb_attached()
         vBatt = badge.battery_volt_sense()
         vBatt += vDrop
         charging = badge.battery_charge_status()
 
-        easydraw.battery(vUsb, vBatt, charging)
+        easydraw.battery(on_usb, vBatt, charging)
 
         if vBatt>500:
             ugfx.string(52, 0, str(round(vBatt/1000, 1)) + 'v','Roboto_Regular12',ugfx.BLACK)
@@ -123,7 +123,7 @@ def onSleep(idleTime):
 ### PROGRAM
 
 # Calibrate battery voltage drop
-if badge.battery_charge_status() == False and badge.usb_volt_sense() > 4500 and badge.battery_volt_sense() > 2500:
+if badge.battery_charge_status() == False and pm.usb_attached() and badge.battery_volt_sense() > 2500:
     badge.nvs_set_u16('splash', 'bat.volt.drop', 5200 - badge.battery_volt_sense()) # mV
     print('Set vDrop to: ' + str(4200 - badge.battery_volt_sense()))
 vDrop = badge.nvs_get_u16('splash', 'bat.volt.drop', 1000) - 1000 # mV
@@ -174,7 +174,6 @@ def onSleep(idleTime):
 easywifi.disable()
 gc.collect()
 
-virtualtimers.debug(True)
 virtualtimers.activate(25)
 pm.callback(onSleep)
 pm.feed()

esp32/modules/tasks/powermanagement.py

@@ -11,14 +11,17 @@
 
 userResponseTime = badge.nvs_get_u16('splash', 'urt', 5000)
 
+def usb_attached():
+    return badge.usb_volt_sense() > 4500
+
 def pm_task():
     ''' The power management task [internal function] '''
     global requestedStandbyTime
 
     idleTime = virtualtimers.idle_time()
-    print("[Power management] Next task wants to run in "+str(idleTime)+" ms.")
-        
-    if idleTime>30000 and not badge.safe_mode():
+    #print("[Power management] Next task wants to run in "+str(idleTime)+" ms.")
+
+    if idleTime > 30000 and not badge.safe_mode() and not ( usb_attached() and badge.nvs_get_u8('badge', 'usb_stay_awake', 0) != 0 ):
         global onSleepCallback
         if not onSleepCallback==None:
             print("[Power management] Running onSleepCallback...")