From a60bba671e4d356c59c61afdade654edf9ee10ea Mon Sep 17 00:00:00 2001
From: Sander Roosendaal <roosendaalsander@gmail.com>
Date: Mon, 5 Feb 2018 13:49:59 +0100
Subject: [PATCH] working on low level plannedsession

---
 rowers/dataprep.py        | 39 ++++++++++++++++++++++++-
 rowers/metrics.py         |  2 ++
 rowers/models.py          |  7 +++--
 rowers/plannedsessions.py | 60 +++++++++++++++++++++++++++++++++++++--
 rowers/views.py           |  1 -
 5 files changed, 103 insertions(+), 6 deletions(-)

diff --git a/rowers/dataprep.py b/rowers/dataprep.py
index 032d51b9..9c891dd9 100644
--- a/rowers/dataprep.py
+++ b/rowers/dataprep.py
@@ -36,7 +36,7 @@ from rowingdata import (
     summarydata, get_file_type,
 )
 
-from rowers.metrics import axes
+from rowers.metrics import axes,calc_trimp
 from async_messages import messages as a_messages
 import os
 import zipfile
@@ -2212,3 +2212,40 @@ def dataprep(rowdatadf, id=0, bands=True, barchart=True, otwpower=True,
         conn.close()
         engine.dispose()
     return data
+
+def workout_trimp(workout):
+    r = workout.user
+    df,row = getrowdata_db(id=workout.id)
+    df = clean_df_stats(df)
+    if df.empty:
+        df,row = getrowdata_db(id=workout.id)
+        df = clean_df_stats(df,workstrokesonly=False)
+    trimp = calc_trimp(df,r.sex,r.max,r.rest)
+    trimp = int(trimp)
+
+    return trimp
+
+def workout_rscore(w):
+    r = workout.user
+    df,row = getrowdata_db(id=workout.id)
+    df = clean_df_stats(df)
+    if df.empty:
+        df,row = getrowdata_db(id=workout.id)
+        df = clean_df_stats(df,workstrokesonly=False)
+
+    duration = df['time'].max()-df['time'].min()
+    duration /= 1.0e3
+    pwr4 = df['power']**(4.0)
+    normp = (pwr4.mean())**(0.25)
+    if not np.isnan(normp):
+        ftp = float(r.ftp)
+        if w.workouttype in ('water','coastal'):
+            ftp = ftp*(100.-r.otwslack)/100.
+
+        intensityfactor = df['power'].mean()/float(ftp)
+        intensityfactor = normp/float(ftp)
+        tss = 100.*((duration*normp*intensityfactor)/(3600.*ftp))
+    else:
+        tss = 0
+
+    return tss
diff --git a/rowers/metrics.py b/rowers/metrics.py
index bfac6b1f..c70a7639 100644
--- a/rowers/metrics.py
+++ b/rowers/metrics.py
@@ -307,6 +307,7 @@ This value should be fairly constant across all stroke rates.""",
         },
     )
 
+
 def calc_trimp(df,sex,hrmax,hrmin):
     if sex == 'male':
         f = 1.92
@@ -321,6 +322,7 @@ def calc_trimp(df,sex,hrmax,hrmin):
 
     return trimp
 
+
 def getagegrouprecord(age,sex='male',weightcategory='hwt',
                       distance=2000,duration=None,indf=pd.DataFrame()):
 
diff --git a/rowers/models.py b/rowers/models.py
index 433c4dd3..a5e38523 100644
--- a/rowers/models.py
+++ b/rowers/models.py
@@ -761,7 +761,9 @@ class PlannedSession(models.Model):
 
     sessionmodechoices = (
         ('distance','Distance'),
-        ('time','Time')
+        ('time','Time'),
+        ('rScore','rScore'),
+        ('TRIMP','TRIMP'),
         )
 
     criteriumchoices = (
@@ -779,7 +781,8 @@ class PlannedSession(models.Model):
     sessionunitchoices = (
         ('min','minutes'),
         ('km','km'),
-        ('m','meters')
+        ('m','meters'),
+        ('None',None),
         )
     
     manager = models.ForeignKey(User)
diff --git a/rowers/plannedsessions.py b/rowers/plannedsessions.py
index 68ad496c..438e2e44 100644
--- a/rowers/plannedsessions.py
+++ b/rowers/plannedsessions.py
@@ -20,20 +20,76 @@ from rowers.models import (
     TrainingPlan,
     )
 
+import metrics
+import numpy as np
+import dataprep
+
 # Low Level functions - to be called by higher level methods
 
 # dummies for now
-def submit_workout(w,ps):
+def add_workouts_plannedsession(ws,ps):
+    for w in ws:
+        w.plannedsession = ps
+        w.save()
+
     return 1
+    
 
 def remove_workout_plannedsession(w,ps):
-    return 1
+    if w.plannedsession == ps:
+        w.plannedsession = None
+        w.save()
+        return 1
+
+    return 0
+
+def clone_planned_session(ps):
+    ps.save()
+    ps.pk = None # creates new instance
+    ps.save()
+
+def timefield_to_seconds_duration(t):
+    duration = t.hour*3600.
+    duration += t.minute * 60.
+    duration += t.second
+    duration += t.microsecond/1.e6
+
+    return duration
+    
+def is_session_complete(ps):
+    ws = Workout.objects.filter(plannedsession=ps)
+    score = 0
+    for w in ws:
+        if ps.sessionmode == 'distance':
+            score += w.distance
+        elif ps.sessionmode == 'time':
+            durationseconds = timefield_to_seconds_duration(w.duration)
+            score += durationseconds
+        elif ps.sessionmode == 'TRIMP':
+            trimp = dataprep.workout_trimp(w)
+            score += trimp
+        elif ps.sessionmode == 'rScore':
+            rscore = dataprep.workout_rscore(w)
+            score += rscore
+
+    ratio = score/float(ps.value)
+
+    if ratio>0.8 and ratio<1.2:
+        return True
+
+    return False
 
 def rank_results(ps):
     return 1
 
 def add_team(t,ps):
+    ps.team.add(t)
+    ps.save()
+
     return 1
 
 def add_rower(r,ps):
+    ps.rower.add(r)
+    ps.save()
+    
     return 1
diff --git a/rowers/views.py b/rowers/views.py
index 88dea3be..75d2600f 100644
--- a/rowers/views.py
+++ b/rowers/views.py
@@ -7496,7 +7496,6 @@ def workout_stats_view(request,id=0,message="",successmessage=""):
         if w.workouttype in ('water','coastal'):
             ftp = ftp*(100.-r.otwslack)/100.
             
-        intensityfactor = datadf['power'].mean()/float(ftp)
         intensityfactor = normp/float(ftp)
         tss = 100.*((duration*normp*intensityfactor)/(3600.*ftp))