Fix handling of Balance

1 files changed, 4 insertions, 5 deletions

diff --git a/trivmix/Trivmix.hs b/trivmix/Trivmix.hs
index 7743246..2e05b13 100644
--- a/trivmix/Trivmix.hs
+++ b/trivmix/Trivmix.hs
@@ -42,12 +42,11 @@ import Data.Function
 import Control.Monad
 
 import Text.Heredoc (str)
-import Refined
+
+import Refined (refine)
   
 import Trivmix.Types
 
-type Balance = Refined ZeroToOne Float
-  
 data Options = Options
                { input :: String
                , output :: String
@@ -86,7 +85,7 @@ optionParser = Options
                <*> option auto ( long "initial-balance"
                               <> metavar "BALANCE"
                               <> help "Initial value for balance"
-                              <> value ($$(refineTH 1.0) :: Balance)
+                              <> value def
                               <> showDefault
                                )
                <*> many ( strOption ( long "run"
@@ -161,7 +160,7 @@ trivmix Options{..} = do
       delay = round $ recip fps * 1e6
       linInt x a b = a * (1 - x) + b * x
       linInt' x a b = either error id $ asFloat (linInt x) a b
-      mulBalance (unrefine -> b) x = either error id $ asFloat (*) (Lin . either error id $ refine b) x
+      mulBalance (bToFloat -> b) x = either error id $ asFloat (*) (Lin . either error id $ refine b) x
     newLevel <- mulBalance <$> readMVar balance <*> readMVar level
     currentLevel <- readMVar level'
     mapM_ (\x -> swapMVar level' (linInt' x currentLevel newLevel) >> threadDelay delay) ([0,recip frames..1] :: [Float])