conways-game-of-life/Graphics/Display.hs

39 lines
1.3 KiB
Haskell
Raw Normal View History

2021-12-28 00:46:39 -05:00
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TypeApplications #-}
module Graphics.Display where
import Control.Monad.Representable.Reader
import Data.Maybe (mapMaybe)
import GOL.Space
import Graphics.Gloss
2021-12-28 11:21:26 -05:00
-- * Drawing the display grid
2021-12-29 19:57:29 -05:00
-- | Draw a square with the given size and position.
2021-12-28 00:46:39 -05:00
square :: Float -> Point -> Picture
square size (x, y) = translate (x + size / 2) y (rectangleUpperSolid size size)
squareAtIndex :: (Int, Int) -> Float -> Picture
squareAtIndex (x, y) size = square size (fromIntegral x * size, fromIntegral y * size)
drawCell :: DisplayableSpace f => f Bool -> (Int, Int) -> Float -> Maybe Picture
drawCell xs pos size =
if index xs pos
then Just $ squareAtIndex pos size
else Nothing
2021-12-28 11:21:26 -05:00
drawCells :: forall f. DisplayableSpace f => f Bool -> Float -> [Picture]
drawCells xs size =
2021-12-28 00:46:39 -05:00
let poss = (,) <$> [0 .. sizex @f -1] <*> [0 .. sizey @f -1]
2021-12-28 11:21:26 -05:00
in mapMaybe (\pos -> drawCell xs pos size) poss
2021-12-29 19:57:29 -05:00
-- | Draw a grid of a displayable space given the window size and cell color.
2021-12-29 18:51:48 -05:00
drawGrid :: forall f. DisplayableSpace f => (Int, Int) -> Color -> f Bool -> Picture
drawGrid (w, h) c xs =
2021-12-29 20:07:11 -05:00
let size =
if w < h
then fromIntegral w / fromIntegral (sizex @f)
else fromIntegral h / fromIntegral (sizey @f)
in translate (-fromIntegral w / 2) (-fromIntegral h / 2) $
color c $ pictures $ drawCells xs size