haga/src/Main.hs

{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE NoImplicitPrelude #-}
{-# LANGUAGE OverloadedStrings #-}

import Pipes
import Pretty
import Protolude hiding (for)
import System.IO
import WS19

mkPop = population 100 (I prios [])

main :: IO ()
main = do
  args <- getArgs
  let t = fromMaybe 100 $ headMay args >>= readMaybe
  hSetBuffering stdout NoBuffering
  pop <- mkPop
  pop' <- runEffect $ for (run 2 1 (5 / 100) pop (steps t)) log
  (res, _) <- bests 5 pop'
  sequence_ $ format <$> res
  where
    format s = do
      f <- liftIO $ fitness s
      putErrText $ show f <> "\n" <> pretty s
    log = putText . csv
    csv (t, f) = show t <> " " <> show f