{-# LANGUAGE TemplateHaskell #-}

module Config
  where

import Prelude                                          as P hiding ((.), id, fst, snd)
import qualified Prelude                                as P

import ParseArgs
import Data.Label
import Control.Category

import Data.Array.Accelerate                            ( Array, DIM2 )
import Data.Array.Accelerate.Math.Complex               ( Complex )


-- | Types
--
type R          = Float
type C          = Complex R
type Matrix a   = Array DIM2 a

data SigmoidFunction
  = Hard
  | Smooth
  | Atan
  | Atancos
  | Overshoot
  | Linear
  | Hermite
  | Sin
  deriving (Eq, Show, Read, Enum, Bounded)


data ColourScheme
  = RedBlack
  | WhiteBlack
  | BlackWhite
  | BrownGreen
  | GoldBrown
  | Rainbow1
  | Rainbow2
  | Rainbow3
  deriving (Eq, Show, Read, Enum, Bounded)


data Config = Config
  {
    -- How to execute the simulation
    _configBackend              :: Backend

    -- How to present the output
  , _configWindowSize           :: Int
  , _configWindowZoom           :: Int
  , _configFramerate            :: Int
  , _configTimestep             :: Float
  , _configColourScheme         :: ColourScheme

    -- Initial conditions
  , _configRim                  :: R            -- b
  , _configDiscRadius           :: (R,R)        -- (ri, ra)
  , _configBirthInterval        :: (R,R)        -- (b1, b2)
  , _configDeathInterval        :: (R,R)        -- (d1, d2)
  , _configStep                 :: (R,R)        -- (alpha_n, alpha_m)

   -- Timestep and sigmoid functions
  , _configTimestepMode         :: Int
  , _configSigmode              :: Int
  , _configSigtype              :: SigmoidFunction
  , _configMixtype              :: SigmoidFunction

    -- Terminating conditions
  , _configMaxSteps             :: Maybe Int
  , _configBenchmark            :: Bool
  , _configHelp                 :: Bool
  }

$(mkLabels [''Config])


defaults :: Config
defaults = Config
  {
    _configBackend              = maxBound

  , _configWindowSize           = 256
  , _configWindowZoom           = 3
  , _configFramerate            = 30
  , _configTimestep             = 0.1
  , _configColourScheme         = WhiteBlack

  -- generic smooth glider
  --, _configRim                  = 1
  --, _configDiscRadius           = (4, 12)
  --, _configBirthInterval        = (0.278, 0.365)
  --, _configDeathInterval        = (0.267, 0.445)
  --, _configStep                 = (0.028, 0.147)

  --, _configTimestepMode         = 0
  --, _configSigmode              = 2
  --, _configSigtype              = Smooth
  --, _configMixtype              = Smooth

  -- SmoothLifeL
  , _configRim                  = 1
  , _configDiscRadius           = (10.0/3.0, 10)
  , _configBirthInterval        = (0.257, 0.336)
  , _configDeathInterval        = (0.365, 0.549)
  , _configStep                 = (0.028, 0.147)

  , _configTimestepMode         = 1
  , _configSigmode              = 2
  , _configSigtype              = Smooth
  , _configMixtype              = Smooth

  , _configMaxSteps             = Nothing
  , _configBenchmark            = False
  , _configHelp                 = False
  }


options :: [OptDescr (Config -> Config)]
options =
  [ Option  [] ["size"]
            (ReqArg (set configWindowSize . read) "INT")
            (describe configWindowSize "visualisation size")

  , Option  [] ["zoom"]
            (ReqArg (set configWindowZoom . read) "INT")
            (describe configWindowZoom "visualisation pixel replication factor")

  , Option  [] ["framerate"]
            (ReqArg (set configFramerate . read) "INT")
            (describe configFramerate "visualisation frame rate")

  , Option  [] ["timestep"]
            (ReqArg (set configTimestep . read) "FLOAT")
            (describe configTimestep "simulation timestep")

  , Option  [] ["colours"]
            (ReqArg (set configColourScheme . read) "COLOURS")
            (describe configColourScheme "visualisation colour scheme")

  , Option  [] ["b", "rim"]
            (ReqArg (set configRim . read) "FLOAT")
            (describe configRim "anti-aliasing zone around the rim")

  , Option  [] ["ri", "inner-radius"]
            (ReqArg (set (fst . configDiscRadius) . read) "FLOAT")
            (describe (fst . configDiscRadius) "inner radius")

  , Option  [] ["ra", "outer-radius"]
            (ReqArg (set (snd . configDiscRadius) . read) "FLOAT")
            (describe (snd . configDiscRadius) "outer radius")

  , Option  [] ["b1", "birth-low"]
            (ReqArg (set (fst . configBirthInterval) . read) "FLOAT")
            (describe (fst . configBirthInterval) "lower birth interval")

  , Option  [] ["b2", "birth-high"]
            (ReqArg (set (snd . configBirthInterval) . read) "FLOAT")
            (describe (snd . configBirthInterval) "upper birth interval")

  , Option  [] ["d1", "death-low"]
            (ReqArg (set (fst . configDeathInterval) . read) "FLOAT")
            (describe (fst . configDeathInterval) "lower death interval")

  , Option  [] ["d2", "death-high"]
            (ReqArg (set (snd . configDeathInterval) . read) "FLOAT")
            (describe (snd . configDeathInterval) "upper death interval")

  , Option  [] ["sn", "alpha_n"]
            (ReqArg (set (fst . configStep) . read) "FLOAT")
            (describe (fst . configStep) "lower step interval")

  , Option  [] ["sm", "alpha_m"]
            (ReqArg (set (snd . configStep) . read) "FLOAT")
            (describe (snd . configStep) "upper step interval")

  , Option  [] ["tsm", "timestep-mode"]
            (ReqArg (set configTimestepMode . read) "INT")
            (describe configTimestepMode "timestep mode 0..2")

  , Option  [] ["sigmode"]
            (ReqArg (set configSigmode . read) "INT")
            (describe configSigmode "sigmode 1..4")

  , Option  [] ["sigtype"]
            (ReqArg (set configSigtype . read) "SIGMOID")
            (describe configSigtype "sigtype")

  , Option  [] ["mixtype"]
            (ReqArg (set configMixtype . read) "SIGMOID")
            (describe configMixtype "mixtype")

  , Option  [] ["max-steps"]
            (ReqArg (set configMaxSteps . read) "INT")
            (describe configMaxSteps "exit simulation after this many steps")

  , Option  [] ["benchmark"]
            (NoArg (set configBenchmark True))
            (describe configBenchmark "benchmark instead of displaying animation")

  , Option  "h?" ["help"]
            (NoArg (set configHelp True))
            "show this help message"
  ]
  where
    describe f msg
      = msg ++ " (" ++ show (get f defaults) ++ ")"

    fst = lens P.fst (\a (_,b) -> (a,b))
    snd = lens P.snd (\b (a,_) -> (a,b))

header :: [String]
header =
  [ "accelerate-smoothlife (c) [2012..2013] The Accelerate Team"
  , ""
  , "Usage: accelerate-smoothlife [OPTIONS]"
  , ""
  ]

footer :: [String]
footer =
  [ ""
  , "Available sigmoid functions:"
  , "  " ++ unwords (map show [minBound .. maxBound :: SigmoidFunction])
  , ""
  , "Available colour schemes:"
  , "  " ++ unwords (map show [minBound .. maxBound :: ColourScheme])
  ]