From 3b6c2cf84b618b47dfca091370fae8b0d0ae4e6b Mon Sep 17 00:00:00 2001
From: Sander Roosendaal <roosendaalsander@gmail.com>
Date: Mon, 8 Feb 2021 08:23:46 +0100
Subject: [PATCH] hr pie chart label now with new hR zones

---
 rowers/interactiveplots.py | 24 ++++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/rowers/interactiveplots.py b/rowers/interactiveplots.py
index 9374b4eb..3887e29d 100644
--- a/rowers/interactiveplots.py
+++ b/rowers/interactiveplots.py
@@ -230,7 +230,7 @@ def interactive_hr_piechart(df,rower,title,totalseconds=0):
     if totalseconds == 0:
         totalseconds = sumtimehr
 
-
+    hrzones = rower.hrzones
 
     qry = 'hr < {ut2}'.format(ut2=rower.ut2)
     qrydata = df.query(qry)
@@ -253,12 +253,12 @@ def interactive_hr_piechart(df,rower,title,totalseconds=0):
     frac_an = totalseconds*df.query(qry)['deltat'].sum()/sumtimehr
 
     datadict = {
-        '<ut2':frac_lut2,
-        'ut2': frac_ut2,
-        'ut1': frac_ut1,
-        'at': frac_at,
-        'tr': frac_tr,
-        'an': frac_an,
+        '<{ut2}'.format(ut2=hrzones[1]):frac_lut2,
+        '{ut2}'.format(ut2=hrzones[1]): frac_ut2,
+        '{ut1}'.format(ut1=hrzones[2]): frac_ut1,
+        '{at}'.format(at=hrzones[3]): frac_at,
+        '{tr}'.format(tr=hrzones[4]): frac_tr,
+        '{an}'.format(an=hrzones[5]): frac_an,
     }
 
 
@@ -270,7 +270,15 @@ def interactive_hr_piechart(df,rower,title,totalseconds=0):
     data = pd.Series(datadict).reset_index(name='value').rename(columns={'index':'zone'})
     data['angle'] = data['value']/data['value'].sum() * 2*pi
     data['color'] = colors
-    data['zone'] = ['<ut2','ut2','ut1','at','tr','an']
+    data['zone'] = [
+        '<{ut2}'.format(ut2=hrzones[1]),
+        '{ut2}'.format(ut2=hrzones[1]),
+        '{ut1}'.format(ut1=hrzones[2]),
+        '{at}'.format(at=hrzones[3]),
+        '{tr}'.format(tr=hrzones[4]),
+        '{an}'.format(an=hrzones[5])
+    ]
+
     data['totaltime'] = pd.Series([pretty_timedelta(v) for v in data['value']])