Config.hs

module Gloss.Config
	( Config (..)
	, loadConfig)
where
import Gloss.MainArgs
import System.Console.ParseArgs
import Data.Maybe

-- | Program config.
data Config
	= Config {
	 
	-- How to present the output.
	  configWindowSize	:: Maybe Int
	, configShouldDrawTree	:: Bool
	, configRate		:: Int

	-- What solver to use
	, configSolverName	:: String

	-- System setup
	, configBodyCount	:: Int
	, configBodyMass	:: Double
	, configTimeStep	:: Double
	, configEpsilon		:: Double

	-- Initial conditions.
	, configStartDiscSize	:: Double
	, configStartSpeed	:: Double

	-- Terminating conditions.
	, configMaxSteps	:: Maybe Int
	
	-- dump points to file
	, configDumpFinal	:: Maybe FilePath }
	

-- | Load program config from its command line arguments.	
loadConfig :: Args MainArg -> Config
loadConfig args
 = let	mWindowSize	= getArgInt	args ArgGloss
	Just solverName	= getArgString	args ArgSolver
	shouldDrawTree	= gotArg  	args ArgDrawTree
	Just timeStep	= getArgDouble	args ArgTimeStep
	Just rate	= getArgInt	args ArgRate
	Just bodyCount	= getArgInt	args ArgBodyCount
	Just bodyMass	= getArgDouble  args ArgBodyMass
	Just epsilon	= getArgDouble	args ArgEpsilon
	Just discSize	= getArgDouble	args ArgDiscSize
	Just startSpeed	= getArgDouble	args ArgStartSpeed

	mMaxSteps	= getArgInt	args ArgMaxSteps
	mFilePath	= getArgString	args ArgDumpFinal

	checkMode x
	 = if not (isJust mWindowSize || isJust mMaxSteps)
		then error "you must specify either --max-steps or --gloss <window size :: Int>"
		else x	

   in	checkMode $
	Config
	{ configWindowSize	= mWindowSize
	, configShouldDrawTree	= shouldDrawTree
	, configRate		= rate
	, configSolverName	= solverName
	, configBodyCount	= bodyCount
	, configBodyMass	= bodyMass
	, configTimeStep	= timeStep
	, configEpsilon		= epsilon
	, configStartDiscSize	= discSize
	, configStartSpeed	= startSpeed
	, configMaxSteps	= mMaxSteps 
	, configDumpFinal	= mFilePath }