mirror of
https://github.com/danth/stylix
synced 2024-11-26 06:00:23 +00:00
ba5565d698
Simplified a lot of code which was unnecessarily generic. Now using monads to manage the state of the random number generator rather than passing it around by hand. Also made some performance improvements, then increased the population size so more combinations are tried in a similar length of time.
39 lines
1.4 KiB
Haskell
39 lines
1.4 KiB
Haskell
import Ai.Evolutionary ( evolve )
|
|
import Codec.Picture ( DynamicImage, convertRGB8, readImage )
|
|
import Data.Colour ( lab2rgb )
|
|
import qualified Data.Vector as V
|
|
import Stylix.Output ( makeOutputTable )
|
|
import Stylix.Palette ( )
|
|
import System.Environment ( getArgs )
|
|
import System.Exit ( die )
|
|
import System.Random ( setStdGen, mkStdGen )
|
|
import Text.JSON ( encode )
|
|
|
|
-- | Load an image file.
|
|
loadImage :: String -- ^ Path to the file
|
|
-> IO DynamicImage
|
|
loadImage input = either error id <$> readImage input
|
|
|
|
mainProcess :: (String, String, String) -> IO ()
|
|
mainProcess (polarity, input, output) = do
|
|
putStrLn $ "Processing " ++ input
|
|
|
|
-- Random numbers must be deterministic when running inside Nix.
|
|
setStdGen $ mkStdGen 0
|
|
|
|
image <- loadImage input
|
|
palette <- evolve (polarity, convertRGB8 image)
|
|
let outputTable = makeOutputTable $ V.map lab2rgb palette
|
|
|
|
writeFile output $ encode outputTable
|
|
putStrLn $ "Saved to " ++ output
|
|
|
|
parseArguments :: [String] -> Either String (String, String, String)
|
|
parseArguments [polarity, input, output] = Right (polarity, input, output)
|
|
parseArguments [_, _] = Left "Please specify an output file"
|
|
parseArguments [_] = Left "Please specify an image"
|
|
parseArguments [] = Left "Please specify a polarity: either, light or dark"
|
|
parseArguments _ = Left "Too many arguments"
|
|
|
|
main :: IO ()
|
|
main = either die mainProcess . parseArguments =<< getArgs
|