src/Main.hs

   1 module Main (main) where
   2
   3 import UI
   4 import Mine
   5 import Strategies
   6 import System.Random
   7 import Control.Monad
   8 import Control.Concurrent
   9 import Control.Exception
  10
  11 nGames :: Int
  12 nGames = 10000
  13
  14 config :: Config
  15 config = intermediate
  16
  17 strategy :: Strategy
  18 strategy = simpleStrat
  19
  20 singleGame :: Bool -> Int -> IO Int
  21 singleGame verbose n = do
  22     gen <- newStdGen
  23     gen2 <- newStdGen
  24     let (res, brd) = playGame config gen (sRun strategy config gen2)
  25     when verbose $ putStr (show res ++ show brd)
  26     if res == Won then return $! (n+1) else return n
  27
  28 summary :: Int -> IO ()
  29 summary i = do
  30     putStrLn ("Won " ++ show i ++ "/" ++ show nGames ++ " games")
  31
  32 mainSingle :: IO ()
  33 mainSingle = do
  34     singleGame True 0
  35     singleGame True 0
  36     foldM (\n _ -> singleGame False n) 0 (replicate nGames ()) >>= summary
  37
  38 mainUI' = do
  39     finish <- newEmptyMVar
  40     forkOS (finally (mainUI strategies) (putMVar finish ()))
  41     readMVar finish
  42
  43 main :: IO ()
  44 main = do
  45     mainUI strategies