show separate workout routines

TODO: don't use Set, since we need to preserve order
TODO: support iterating backward
TODO: support a title, display it along with week (e.g., "Week 1, Monday AM")

1 files changed, 18 insertions, 4 deletions

diff --git a/repgoal.hs b/repgoal.hs
index 94982f1..96cc534 100755
--- a/repgoal.hs
+++ b/repgoal.hs
@@ -16,6 +16,9 @@
 {-# language DerivingVia #-}
 import Rebase.Prelude hiding (toList, on, (<+>), Max)
 import qualified Rebase.Prelude as Prelude
+import qualified Data.Set as Set
+import qualified Data.List.NonEmpty as NonEmpty
+import Data.List.NonEmpty (NonEmpty)
 import Control.Lens hiding ((<|))
 import Data.Foldable (toList)
 import Data.Ratio
@@ -127,7 +130,8 @@ data WeekSelection = Week1 | Week2 | Week3 deriving (Enum, Bounded, Show, Eq)
 -- TODO: State contains performances
 data St = St {
   _lifts :: [LiftRecord],
-  _week :: WeekSelection
+  _week :: WeekSelection,
+  _routines :: NonEmpty (Set Text)
 }
 makeLenses ''St
 
@@ -165,9 +169,10 @@ annotatePosition (x:xs) = (FirstInList, x) : annotateRest xs
 drawUI :: St -> [Widget ()]
 drawUI st = [vCenter $ vBox [hCenter oneRepMaxTable, header, withVScrollBarHandles $ withVScrollBars OnRight $ viewport () Vertical $ hCenter lastSetTable]]
   where
-    lastSetTable = renderTable $ table $ map (padLeftRight 1 . str) ["Lift", "Set", "Goal"] : concatMap (toLiftRows (view week st)) (view lifts st)
+    lifts' = filter ((flip Set.member $ view routines st & NonEmpty.head) . liftName) (view lifts st)
+    lastSetTable = renderTable $ table $ map (padLeftRight 1 . str) ["Lift", "Set", "Goal"] : concatMap (toLiftRows (view week st)) lifts'
     header = str $ "Week " ++ case (view week st) of Week1 -> "1"; Week2 -> "2"; Week3 -> "3"
-    oneRepMaxTable = renderTable $ table $ map (padLeftRight 1 . str) ["Lift", "Achieved Best", "Computed 1RM"] : map toRow (view lifts st)
+    oneRepMaxTable = renderTable $ table $ map (padLeftRight 1 . str) ["Lift", "Achieved Best", "Computed 1RM"] : map toRow lifts'
     toRow LiftRecord{..} =
       let best@Achieved{..} = bestPerformance stats
       in
@@ -211,6 +216,7 @@ handleEvent st e = case e of
   VtyEvent (V.EvKey V.KDown _)         -> M.vScrollBy (M.viewportScroll ()) 5    >> continue st
   VtyEvent (V.EvKey V.KUp _)           -> M.vScrollBy (M.viewportScroll ()) (-5) >> continue st
   VtyEvent (V.EvKey (V.KChar 'w') _)   -> continue $ st & week %~ succ'
+  VtyEvent (V.EvKey (V.KChar 'n') _)   -> continue $ st & routines %~ (NonEmpty.fromList . NonEmpty.tail)
   VtyEvent _                           -> continue st
   AppEvent _                           -> continue st
   _ -> continue st
@@ -243,6 +249,14 @@ powerClean = clean * 80 / 100
 pushPress = press * 100 / 75
 jerk = pushPress / 100 * 85
 
+routine :: NonEmpty (Set Text)
+routine = NonEmpty.cycle $
+  Set.fromList ["Deadlift", "Press"] :|
+  [Set.fromList ["Front Squat", "Left-Arm Snatch", "Right-Arm Snatch"]
+  , Set.fromList ["Squat", "Bench"]
+  , Set.fromList ["Push Press", "Power Clean"]
+  ]
+
 main, main' :: IO ()
 main' = pPrint initial'
 main = do
@@ -251,4 +265,4 @@ main = do
     -- liftIO $ setMode (outputIface vty) Mouse True
     chan <- newBChan 10
     initial <- read <$> readFile "lifts.dat" :: IO [LiftRecord]
-    void $ customMain vty buildVty (Just chan) (theApp) (St initial Week1)
+    void $ customMain vty buildVty (Just chan) (theApp) (St initial Week1 routine)