From 872ffb2c4aa84294e19e6e7eb9e9d1b555bdaa21 Mon Sep 17 00:00:00 2001
From: oldnapalm <38410858+oldnapalm@users.noreply.github.com>
Date: Sat, 25 Jul 2020 15:21:42 -0300
Subject: [PATCH] Add heart rate version

---
 README.md      |   1 +
 heartrate.py   | 115 +++++++++++++++++++++++++++++++++++++++++++++++++
 heartrate.spec |  39 +++++++++++++++++
 zones.csv      |   7 +++
 4 files changed, 162 insertions(+)
 create mode 100644 heartrate.py
 create mode 100644 heartrate.spec
 create mode 100644 zones.csv

diff --git a/README.md b/README.md
index 578b965..25acc06 100644
--- a/README.md
+++ b/README.md
@@ -26,6 +26,7 @@ Supported devices:
 
 * Download the [standalone executable](https://github.com/oldnapalm/vpower/releases)
   * The bot version is for testing purposes, you can set the power value in a slider
+  * The HR version broadcasts power based on heart rate, set the values in the file [zones.csv](https://github.com/oldnapalm/vpower/blob/master/zones.csv)
 * Configure speed sensor, power calculator and wheel circumference in **vpower.cfg**
   * You can leave `speed_sensor_id` as zero if there's only one sensor around
 * Install the libusb-win32 driver for the ANT+ device, it can be easily done using [Zadig](https://zadig.akeo.ie/)
diff --git a/heartrate.py b/heartrate.py
new file mode 100644
index 0000000..048d348
--- /dev/null
+++ b/heartrate.py
@@ -0,0 +1,115 @@
+#!/usr/bin/env python
+import os, sys
+import time
+import csv
+import win32api
+
+from ant.core import driver
+from ant.core import node
+from ant.plus.heartrate import *
+
+from usb.core import find
+
+from PowerMeterTx import PowerMeterTx
+from config import DEBUG, LOG, NETKEY, POWER_SENSOR_ID
+
+if getattr(sys, 'frozen', False):
+    # If we're running as a pyinstaller bundle
+    SCRIPT_DIR = os.path.dirname(sys.executable)
+else:
+    SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
+
+antnode = None
+hr_monitor = None
+power_meter = None
+xp = [0]
+yp = [0]
+last = 0
+
+zones_file = '%s/zones.csv' % SCRIPT_DIR
+if os.path.isfile(zones_file):
+    with open(zones_file, 'r') as fd:
+        reader = csv.reader(fd)
+        next(reader, None)
+        for line in reader:
+            xp.append(int(line[0]))
+            yp.append(int(line[1]))
+
+def on_exit(sig, func=None):
+    if hr_monitor:
+        print("Closing heart rate monitor")
+        hr_monitor.close()
+    if power_meter:
+        print("Closing power meter")
+        power_meter.close()
+        power_meter.unassign()
+    if antnode:
+        print("Stopping ANT node")
+        antnode.stop()
+
+win32api.SetConsoleCtrlHandler(on_exit, True)
+
+def interp(x_arr, y_arr, x):
+    for i, xi in enumerate(x_arr):
+        if xi >= x:
+            break
+    else:
+        return y_arr[-1]
+
+    x_min = x_arr[i - 1]
+    y_min = y_arr[i - 1]
+    y_max = y_arr[i]
+    factor = (x - x_min) / (xi - x_min)
+    return y_min + (y_max - y_min) * factor
+
+def heart_rate_data(computed_heartrate, event_time_ms, rr_interval_ms):
+    global last
+    t = int(time.time())
+    if t >= last + 1:
+        power = int(interp(xp, yp, computed_heartrate))
+        if power and power_meter:
+            power_meter.update(power)
+        last = t
+
+try:
+    devs = find(find_all=True)
+    for dev in devs:
+        if dev.idVendor == 0x0fcf and dev.idProduct in [0x1008, 0x1009]:
+            break
+
+    stick = driver.USB2Driver(log=LOG, debug=DEBUG, idProduct=dev.idProduct)
+    antnode = node.Node(stick)
+    print("Starting ANT node")
+    antnode.start()
+    network = node.Network(NETKEY, 'N:ANT+')
+    antnode.setNetworkKey(0, network)
+
+    print("Starting heart rate monitor")
+    try:
+        # Create the heart rate monitor object and open it
+        hr_monitor = HeartRate(antnode, network, {'onHeartRateData': heart_rate_data})
+        hr_monitor.open()
+    except Exception as e:
+        print("hr_monitor error: " + repr(e))
+        hr_monitor = None
+
+    print("Starting power meter with ANT+ ID " + repr(POWER_SENSOR_ID))
+    try:
+        # Create the power meter object and open it
+        power_meter = PowerMeterTx(antnode, POWER_SENSOR_ID)
+        power_meter.open()
+    except Exception as e:
+        print("power_meter error: " + repr(e))
+        power_meter = None
+
+    print("Main wait loop")
+    while True:
+        try:
+            time.sleep(1)
+        except (KeyboardInterrupt, SystemExit):
+            break
+
+except Exception as e:
+    print("Exception: " + repr(e))
+    if getattr(sys, 'frozen', False):
+        input()
diff --git a/heartrate.spec b/heartrate.spec
new file mode 100644
index 0000000..545aec9
--- /dev/null
+++ b/heartrate.spec
@@ -0,0 +1,39 @@
+# -*- mode: python -*-
+
+block_cipher = None
+
+import sys
+sys.modules['FixTk'] = None
+
+binaries = [
+    ('C:\\Windows\\System32\\libusb0.dll', '.'),
+]
+
+a = Analysis(['heartrate.py'],
+             pathex=[],
+             binaries=binaries,
+             datas=[],
+             hiddenimports=[],
+             hookspath=[],
+             runtime_hooks=[],
+             excludes=['FixTk', 'tcl', 'tk', '_tkinter', 'tkinter', 'Tkinter'],
+             win_no_prefer_redirects=False,
+             win_private_assemblies=False,
+             cipher=block_cipher,
+             noarchive=False)
+pyz = PYZ(a.pure, a.zipped_data,
+             cipher=block_cipher)
+exe = EXE(pyz,
+          a.scripts,
+          a.binaries,
+          a.zipfiles,
+          a.datas,
+          [],
+          name='heartrate',
+          debug=False,
+          bootloader_ignore_signals=False,
+          strip=False,
+          upx=True,
+          upx_exclude="vcruntime140.dll",
+          runtime_tmpdir=None,
+          console=True )
diff --git a/zones.csv b/zones.csv
new file mode 100644
index 0000000..e7b3df4
--- /dev/null
+++ b/zones.csv
@@ -0,0 +1,7 @@
+Heart Rate (BPM),Power (W)
+80,0
+100,110
+120,140
+140,170
+160,200
+180,230