Skip to content
Snippets Groups Projects
Commit d7edf7e5 authored by Daniel Hoffend's avatar Daniel Hoffend Committed by rahix
Browse files

feat(ecg-app): Refactor ECG App 

New Button-Layout
=================

          /-----------------------\
          |   xxxxxxxxxxxxxxxxx   |
          |                       |
          |                      <> * Pause screen
          |         card10        |
 +------- <>                     <> * During pause: Scroll right.
 |        |                       | * Short press: Toggle between Finger
 |        \-----------------------/    and USB.
* During pause: Scroll left.        * Long press: Toggle bias on/off.
* Short press: Toggle LED mode.
* Long press: Start/stop logging.

Other Changes
=============
- Split the bar into negativ [10:5] and positive [5:0] scale and update
  them seperate to have a better visualization.
- Blink red with backside LEDs on detected pulse.
- Allow toggling the LED modes on and off (see new button layout).
- Lock mode & bias during logging.
parent d3935ffd
Branches
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
preload/apps/ecg/__init__.py
+ 154
− 47
View file @ d7edf7e5
...@@ -6,6 +6,7 @@ import buttons ...@@ -6,6 +6,7 @@ import buttons
import max30001 import max30001
import math import math
import struct import struct
import itertools
WIDTH = 160 WIDTH = 160
HEIGHT = 80 HEIGHT = 80
...@@ -26,6 +27,15 @@ COLOR_WRITE_FG = [255, 255, 255] ...@@ -26,6 +27,15 @@ COLOR_WRITE_FG = [255, 255, 255]
COLOR_WRITE_BG = [255, 0, 0] COLOR_WRITE_BG = [255, 0, 0]
current_mode = MODE_FINGER current_mode = MODE_FINGER
modes = itertools.cycle(
[
({"bar", "pulse"}, {"text": "Top + Pulse", "posx": 0}),
({}, {"text": "off", "posx": 55}),
({"bar"}, {"text": "Top Only", "posx": 25}),
({"pulse"}, {"text": "Pulse Only", "posx": 5}),
]
)
led_mode = next(modes)[0]
history = [] history = []
filebuffer = bytearray() filebuffer = bytearray()
write = 0 write = 0
...@@ -36,6 +46,7 @@ pause_histogram = False ...@@ -36,6 +46,7 @@ pause_histogram = False
histogram_offset = 0 histogram_offset = 0
sensor = 0 sensor = 0
disp = display.open() disp = display.open()
last_sample_count = 1
leds.dim_top(1) leds.dim_top(1)
COLORS = [((23 + (15 * i)) % 360, 1.0, 1.0) for i in range(11)] COLORS = [((23 + (15 * i)) % 360, 1.0, 1.0) for i in range(11)]
...@@ -48,7 +59,8 @@ beta = 3 ...@@ -48,7 +59,8 @@ beta = 3
def update_history(datasets): def update_history(datasets):
global history, moving_average, alpha, beta global history, moving_average, alpha, beta, last_sample_count
last_sample_count = len(datasets)
for val in datasets: for val in datasets:
history.append(val - moving_average) history.append(val - moving_average)
moving_average = (alpha * moving_average + beta * val) / (alpha + beta) moving_average = (alpha * moving_average + beta * val) / (alpha + beta)
...@@ -60,6 +72,7 @@ def update_history(datasets): ...@@ -60,6 +72,7 @@ def update_history(datasets):
# variables for pulse detection # variables for pulse detection
pulse = -1 pulse = -1
samples_since_last_pulse = 0 samples_since_last_pulse = 0
last_pulse_blink = 0
q_threshold = -1 q_threshold = -1
r_threshold = 1 r_threshold = 1
q_spike = -ECG_RATE q_spike = -ECG_RATE
...@@ -177,14 +190,28 @@ def close_sensor(): ...@@ -177,14 +190,28 @@ def close_sensor():
def toggle_mode(): def toggle_mode():
global current_mode global current_mode, disp, pause_screen
if write > 0:
pause_screen = utime.time_ms() + 500
disp.clear(COLOR_BACKGROUND)
disp.print("Locked", posx=30, posy=30, fg=COLOR_TEXT)
disp.update()
return
close_sensor() close_sensor()
current_mode = MODE_USB if current_mode == MODE_FINGER else MODE_FINGER current_mode = MODE_USB if current_mode == MODE_FINGER else MODE_FINGER
open_sensor() open_sensor()
def toggle_bias(): def toggle_bias():
global bias global bias, disp, pause_screen
if write > 0:
pause_screen = utime.time_ms() + 500
disp.clear(COLOR_BACKGROUND)
disp.print("Locked", posx=30, posy=30, fg=COLOR_TEXT)
disp.update()
return
close_sensor() close_sensor()
bias = not bias bias = not bias
open_sensor() open_sensor()
...@@ -209,19 +236,61 @@ def toggle_write(): ...@@ -209,19 +236,61 @@ def toggle_write():
def toggle_pause(): def toggle_pause():
global pause_histogram, histogram_offset, history global pause_histogram, histogram_offset, history, leds
if pause_histogram: if pause_histogram:
pause_histogram = False pause_histogram = False
history = [] history = []
else: else:
pause_histogram = True pause_histogram = True
histogram_offset = 0 histogram_offset = 0
leds.clear()
def toggle_leds():
global led_mode, disp, pause_screen, leds, modes
led_mode, display_args = next(modes)
pause_screen = utime.time_ms() + 250
disp.clear(COLOR_BACKGROUND)
disp.print("LEDs", posx=50, posy=20, fg=COLOR_TEXT)
disp.print(**display_args, posy=40, fg=COLOR_TEXT)
disp.update()
leds.clear()
def draw_leds(vmin, vmax):
# vmin should be in [0, -1]
# vmax should be in [0, 1]
global pulse, samples_since_last_pulse, last_pulse_blink
# stop blinking
if not bool(led_mode):
return
# update led bar
if "bar" in led_mode:
for i in reversed(range(6)):
leds.prep_hsv(
5 + i, COLORS[5 + i] if vmin <= 0 and i <= vmin * -6 else (0, 0, 0)
)
for i in reversed(range(6)):
leds.prep_hsv(
i, COLORS[i] if vmax >= 0 and 5 - i <= vmax * 6 else (0, 0, 0)
)
# blink red on pulse
if (
"pulse" in led_mode
and pulse > 0
and samples_since_last_pulse < last_pulse_blink
):
for i in range(4):
leds.prep(11 + i, (255, 0, 0))
elif "pulse" in led_mode:
for i in range(4):
leds.prep(11 + i, (0, 0, 0))
last_pulse_blink = samples_since_last_pulse
def draw_leds(val):
# val should be in [0, 11]
for i in range(11):
leds.prep_hsv(10 - i, COLORS[10 - i] if i < val else (0, 0, 0))
leds.update() leds.update()
...@@ -273,7 +342,10 @@ def draw_histogram(): ...@@ -273,7 +342,10 @@ def draw_histogram():
fg=COLOR_TEXT, fg=COLOR_TEXT,
) )
else: else:
draw_leds((max(history[-5:]) * scale + SCALE_FACTOR) * 11 / (SCALE_FACTOR * 2)) led_range = last_sample_count if last_sample_count > 5 else 5
draw_leds(
min(history[-led_range:]) / value_max, max(history[-led_range:]) / value_max
)
if pulse < 0: if pulse < 0:
disp.print( disp.print(
current_mode + ("+Bias" if bias else ""), current_mode + ("+Bias" if bias else ""),
...@@ -308,10 +380,17 @@ def main(): ...@@ -308,10 +380,17 @@ def main():
# show button layout # show button layout
disp.clear(COLOR_BACKGROUND) disp.clear(COLOR_BACKGROUND)
disp.print(" BUTTONS ", posx=0, posy=0, fg=COLOR_TEXT) disp.print(" BUTTONS ", posx=0, posy=0, fg=COLOR_TEXT, font=display.FONT20)
disp.print("Finger/USB>", posx=0, posy=20, fg=COLOR_MODE_FINGER) disp.line(0, 20, 159, 20, col=COLOR_LINE)
disp.print(" Bias >", posx=0, posy=40, fg=COLOR_MODE_USB) disp.print(
disp.print("< Write Log", posx=0, posy=60, fg=COLOR_WRITE_BG) " Pause >", posx=0, posy=28, fg=COLOR_MODE_FINGER, font=display.FONT16
)
disp.print(
" Mode/Bias >", posx=0, posy=44, fg=COLOR_MODE_USB, font=display.FONT16
)
disp.print(
"< LED/WriteLog", posx=0, posy=64, fg=COLOR_WRITE_BG, font=display.FONT16
)
disp.update() disp.update()
utime.sleep(3) utime.sleep(3)
...@@ -324,61 +403,89 @@ def main(): ...@@ -324,61 +403,89 @@ def main():
buttons.BOTTOM_LEFT | buttons.BOTTOM_RIGHT | buttons.TOP_RIGHT buttons.BOTTOM_LEFT | buttons.BOTTOM_RIGHT | buttons.TOP_RIGHT
) )
# TOP RIGHT
# down
if button_pressed["TOP_RIGHT"] == 0 and v & buttons.TOP_RIGHT != 0:
button_pressed["TOP_RIGHT"] = utime.time_ms()
toggle_pause()
# up
if button_pressed["TOP_RIGHT"] > 0 and v & buttons.TOP_RIGHT == 0:
duration = utime.time_ms() - button_pressed["TOP_RIGHT"]
button_pressed["TOP_RIGHT"] = 0
# BOTTOM LEFT # BOTTOM LEFT
#
# on pause = shift view left
# long = toggle write
# short = toggle leds
# down, and still pressed
if (
button_pressed["BOTTOM_LEFT"] > 0
and v & buttons.BOTTOM_LEFT != 0
and not pause_histogram
):
duration = utime.time_ms() - button_pressed["BOTTOM_LEFT"]
if duration > 1000:
button_pressed["BOTTOM_LEFT"] = -1
toggle_write()
if button_pressed["BOTTOM_LEFT"] == 0 and v & buttons.BOTTOM_LEFT != 0: # register down event
elif button_pressed["BOTTOM_LEFT"] == 0 and v & buttons.BOTTOM_LEFT != 0:
button_pressed["BOTTOM_LEFT"] = utime.time_ms() button_pressed["BOTTOM_LEFT"] = utime.time_ms()
# register up event but event already called
if button_pressed["BOTTOM_LEFT"] == -1 and v & buttons.BOTTOM_LEFT == 0:
button_pressed["BOTTOM_LEFT"] = 0
# register normal up event
elif button_pressed["BOTTOM_LEFT"] > 0 and v & buttons.BOTTOM_LEFT == 0:
duration = utime.time_ms() - button_pressed["BOTTOM_LEFT"]
button_pressed["BOTTOM_LEFT"] = 0
if not pause_histogram: if not pause_histogram:
toggle_write() toggle_leds()
else: else:
l = len(history) l = len(history)
histogram_offset += ECG_RATE / 2 histogram_offset += ECG_RATE / 2
if l - histogram_offset < WIDTH: if l - histogram_offset < WIDTH:
histogram_offset = l - WIDTH histogram_offset = l - WIDTH
if button_pressed["BOTTOM_LEFT"] > 0 and v & buttons.BOTTOM_LEFT == 0:
duration = utime.time_ms() - button_pressed["BOTTOM_LEFT"]
button_pressed["BOTTOM_LEFT"] = 0
# BOTTOM RIGHT # BOTTOM RIGHT
#
if button_pressed["BOTTOM_RIGHT"] == 0 and v & buttons.BOTTOM_RIGHT != 0: # on pause = shift view right
button_pressed["BOTTOM_RIGHT"] = utime.time_ms() # long = toggle bias
if not pause_histogram: # short = toggle mode (finger/usb)
toggle_bias()
else:
histogram_offset -= ECG_RATE / 2
histogram_offset -= histogram_offset % (ECG_RATE / 2)
if histogram_offset < 0:
histogram_offset = 0
if button_pressed["BOTTOM_RIGHT"] > 0 and v & buttons.BOTTOM_RIGHT == 0:
duration = utime.time_ms() - button_pressed["BOTTOM_RIGHT"]
button_pressed["BOTTOM_RIGHT"] = 0
# TOP RIGHT
# down, and still pressed # down, and still pressed
if button_pressed["TOP_RIGHT"] > 0 and v & buttons.TOP_RIGHT != 0: if (
duration = utime.time_ms() - button_pressed["TOP_RIGHT"] button_pressed["BOTTOM_RIGHT"] > 0
and v & buttons.BOTTOM_RIGHT != 0
and not pause_histogram
):
duration = utime.time_ms() - button_pressed["BOTTOM_RIGHT"]
if duration > 1000: if duration > 1000:
button_pressed["TOP_RIGHT"] = -1 button_pressed["BOTTOM_RIGHT"] = -1
toggle_pause() toggle_bias()
# register down event # register down event
elif button_pressed["TOP_RIGHT"] == 0 and v & buttons.TOP_RIGHT != 0: elif button_pressed["BOTTOM_RIGHT"] == 0 and v & buttons.BOTTOM_RIGHT != 0:
button_pressed["TOP_RIGHT"] = utime.time_ms() button_pressed["BOTTOM_RIGHT"] = utime.time_ms()
# register up event but event already called # register up event but event already called
if button_pressed["TOP_RIGHT"] == -1 and v & buttons.TOP_RIGHT == 0: if button_pressed["BOTTOM_RIGHT"] == -1 and v & buttons.BOTTOM_RIGHT == 0:
button_pressed["TOP_RIGHT"] = 0 button_pressed["BOTTOM_RIGHT"] = 0
# register normal up event # register normal up event
elif button_pressed["TOP_RIGHT"] > 0 and v & buttons.TOP_RIGHT == 0: elif button_pressed["BOTTOM_RIGHT"] > 0 and v & buttons.BOTTOM_RIGHT == 0:
duration = utime.time_ms() - button_pressed["TOP_RIGHT"] duration = utime.time_ms() - button_pressed["BOTTOM_RIGHT"]
button_pressed["TOP_RIGHT"] = 0 button_pressed["BOTTOM_RIGHT"] = 0
if pause_histogram: if pause_histogram:
toggle_pause() histogram_offset -= ECG_RATE / 2
histogram_offset -= histogram_offset % (ECG_RATE / 2)
if histogram_offset < 0:
histogram_offset = 0
else: else:
toggle_mode() toggle_mode()
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment