From a00ed41bbe741a9a39afaa5b1d9003293eb7cc29 Mon Sep 17 00:00:00 2001
From: alex <alex@slab.org>
Date: Sat, 13 Apr 2024 12:29:43 +0100
Subject: [PATCH] separate stepfirstof and stepfirstofstep. need better names
 for these..

---
 src/Sound/Tidal/Stepwise.hs | 48 +++++++++++++++++++++++--------------
 src/Sound/Tidal/Utils.hs    | 11 ++++++---
 2 files changed, 38 insertions(+), 21 deletions(-)

diff --git a/src/Sound/Tidal/Stepwise.hs b/src/Sound/Tidal/Stepwise.hs
index ea178d8b9..703795609 100644
--- a/src/Sound/Tidal/Stepwise.hs
+++ b/src/Sound/Tidal/Stepwise.hs
@@ -25,6 +25,7 @@ import           Data.Maybe          (fromMaybe)
 import           Sound.Tidal.Core
 import           Sound.Tidal.Pattern
 import           Sound.Tidal.UI      (while)
+import           Sound.Tidal.Utils   (applyWhen)
 
 stepcat :: [Pattern a] -> Pattern a
 stepcat pats = timecat $ map (\pat -> (fromMaybe 1 $ tactus pat, pat)) pats
@@ -55,29 +56,28 @@ stepwhen patb f pat@(Pattern _ (Just t) _) = while (_steps t patb) f pat
 -- TODO raise exception?
 stepwhen _ _ pat                           = pat
 
--- _steplastof :: Int -> (Pattern a -> Pattern a) -> Pattern a -> Pattern a
--- _steplastof i f pat | i <= 1 = pat
---                     | otherwise = stepwhen (fastcat $ map pure $ (replicate (i-1) False) ++ [True]) f pat
-
-_steplastof :: Int -> (Pattern a -> Pattern a) -> Pattern a -> Pattern a
-_steplastof n f pat | n <= 1 = pat
-                    | otherwise = _fast t $ stepcat $ reverse $ (f $ head cycles):tail cycles
-  where cycles = reverse $ separateCycles n $ _slow t pat
+_stepevery :: Bool -> Bool -> Int -> (Pattern a -> Pattern a) -> Pattern a -> Pattern a
+_stepevery lastone stepwise n f pat
+  | n <= 1 = pat
+  | otherwise = applyWhen stepwise (_fast t) $ stepcat $ applyWhen lastone reverse $ (f $ head cycles):tail cycles
+  where cycles = applyWhen lastone reverse $ separateCycles n $ applyWhen stepwise (_slow t) pat
         t = fromMaybe 1 $ tactus pat
 
 steplastof :: Pattern Int -> (Pattern a -> Pattern a) -> Pattern a -> Pattern a
-steplastof (Pattern _ _ (Just i)) f pat = _steplastof i f pat
-steplastof tp f p = innerJoin $ (\t -> _steplastof t f p) <$> tp
-
-_stepfirstof :: Int -> (Pattern a -> Pattern a) -> Pattern a -> Pattern a
-_stepfirstof n f pat | n <= 1 = pat
-                    | otherwise = _fast t $ stepcat $ (f $ head cycles):tail cycles
-  where cycles = separateCycles n $ _slow t pat
-        t = fromMaybe 1 $ tactus pat
+steplastof (Pattern _ _ (Just i)) f pat = _stepevery True False i f pat
+steplastof tp f p = innerJoin $ (\t -> _stepevery True False t f p) <$> tp
 
 stepfirstof :: Pattern Int -> (Pattern a -> Pattern a) -> Pattern a -> Pattern a
-stepfirstof (Pattern _ _ (Just i)) f pat = _stepfirstof i f pat
-stepfirstof tp f p = innerJoin $ (\t -> _stepfirstof t f p) <$> tp
+stepfirstof (Pattern _ _ (Just i)) f pat = _stepevery False False i f pat
+stepfirstof tp f p = innerJoin $ (\t -> _stepevery False False t f p) <$> tp
+
+steplastofstep :: Pattern Int -> (Pattern a -> Pattern a) -> Pattern a -> Pattern a
+steplastofstep (Pattern _ _ (Just i)) f pat = _stepevery True True i f pat
+steplastofstep tp f p = innerJoin $ (\t -> _stepevery True True t f p) <$> tp
+
+stepfirstofstep :: Pattern Int -> (Pattern a -> Pattern a) -> Pattern a -> Pattern a
+stepfirstofstep (Pattern _ _ (Just i)) f pat = _stepevery False True i f pat
+stepfirstofstep tp f p = innerJoin $ (\t -> _stepevery False True t f p) <$> tp
 
 stepevery :: Pattern Int -> (Pattern a -> Pattern a) -> Pattern a -> Pattern a
 stepevery = stepfirstof
@@ -96,3 +96,15 @@ steptaper = stepcat . steptaperlist
 stepalt :: [[Pattern a]] -> Pattern a
 stepalt groups = stepcat $ concat $ take (c * length groups) $ transpose $ map cycle groups
   where c = foldl1 lcm $ map length groups
+
+_stepexpand :: Rational -> Pattern a -> Pattern a
+_stepexpand factor pat = withTactus (* factor) pat
+
+_stepcontract :: Rational -> Pattern a -> Pattern a
+_stepcontract factor pat = withTactus (/ factor) pat
+
+stepexpand :: Pattern Rational -> Pattern a -> Pattern a
+stepexpand = tParam _stepexpand
+
+stepcontract :: Pattern Rational -> Pattern a -> Pattern a
+stepcontract = tParam _stepcontract
diff --git a/src/Sound/Tidal/Utils.hs b/src/Sound/Tidal/Utils.hs
index d9955aa63..e2c7568a7 100644
--- a/src/Sound/Tidal/Utils.hs
+++ b/src/Sound/Tidal/Utils.hs
@@ -18,8 +18,8 @@ module Sound.Tidal.Utils where
     along with this library.  If not, see <http://www.gnu.org/licenses/>.
 -}
 
-import Data.List (delete)
-import System.IO (hPutStrLn, stderr)
+import           Data.List (delete)
+import           System.IO (hPutStrLn, stderr)
 
 writeError :: String -> IO ()
 writeError = hPutStrLn stderr
@@ -71,7 +71,7 @@ nth 0 (x : _)  = Just x
 nth n (_ : xs) = nth (n - 1) xs
 
 accumulate :: Num t => [t] -> [t]
-accumulate [] = []
+accumulate []     = []
 accumulate (x:xs) = scanl (+) x xs
 
 {- | enumerate a list of things
@@ -101,3 +101,8 @@ matchMaybe x       _ = x
 fromRight :: b -> Either a b -> b
 fromRight _ (Right b) = b
 fromRight b _         = b
+
+-- Available in Data.Function, but only since 4.18
+applyWhen :: Bool -> (a -> a) -> a -> a
+applyWhen True  f x = f x
+applyWhen False _ x = x