vulkan-tutorial/examples/example02.hs

{-# LANGUAGE RecordWildCards #-}

import Affection
import qualified SDL
import qualified SDL.Raw as Raw
import qualified GEGL as G
import qualified BABL as B
import qualified Data.Map.Strict as M

import Foreign.Storable (peek)
import Foreign.C.Types

import Debug.Trace

main :: IO ()
main = do
  conf <- return $ AffectionConfig
    { initComponents = All
    , windowTitle    = "Affection: example00"
    , windowConfig   = SDL.defaultWindow
    , preLoop        = drawInit
    , drawLoop       = draw
    , updateLoop     = update
    , loadState      = load
    , cleanUp        = clean
    }
  withAffection conf

data UserData = UserData
  { nodeGraph   :: M.Map String G.GeglNode
  , foreground  :: G.GeglBuffer
  , coordinates :: Maybe (Int, Int)
  }

data Particle = Particle
  { particleCoords :: (Int, Int)
  , particleTTL    :: Double
  , particleVel    :: (Int, Int)
  }

load :: SDL.Surface -> IO UserData
load _ = do
  traceM "loading"
  root <- G.gegl_node_new
  traceM "new root node"
  checkerboard <- G.gegl_node_new_child root $ G.checkerboardOperation
    [ G.Property "color1" $ G.PropertyColor $ G.RGBA 0.4 0.4 0.4 1
    , G.Property "color2" $ G.PropertyColor $ G.RGBA 0.6 0.6 0.6 1
    ]
  traceM "checkerboard"
  over <- G.gegl_node_new_child root G.defaultOverOperation
  traceM "over"
  buffer <- G.gegl_buffer_new (G.GeglRectangle 0 0 800 600) =<<
    B.babl_format (B.PixelFormat B.RGBA B.CFfloat)
  bufsrc <- G.gegl_node_new_child root $ G.bufferSourceOperation
    [ G.Property "buffer" $ G.PropertyBuffer buffer
    ]
  traceM "buffer-source"
  G.gegl_node_link checkerboard over
  G.gegl_node_connect_to bufsrc "output" over "aux"
  traceM "connections made"
  myMap <- return $ M.fromList
    [ ("root"        , root)
    , ("over"        , over)
    , ("background"  , checkerboard)
    , ("foreground"  , bufsrc)
    ]
  traceM "loading complete"
  return $ UserData
    { nodeGraph = myMap
    , foreground = buffer
    , coordinates = Nothing
    }

drawInit :: Affection UserData ()
drawInit = do
  AffectionData{..} <- get
  let UserData{..} = userState
  present (nodeGraph M.! "over") foreground (GeglRectangle 0 0 800 600) True

draw :: Affection UserData ()
draw = do
  traceM "drawing"
  AffectionData{..} <- get
  let UserData{..} = userState
  let SDL.Surface rawSurfacePtr _ = drawSurface
  SDL.V2 (CInt rw) (CInt rh) <- SDL.surfaceDimensions drawSurface
  let (w, h) = (fromIntegral rw, fromIntegral rh)
  liftIO $ clearArea foreground (GeglRectangle 0 0 w h)
  maybe (return ()) (\(x, y) ->
    drawRect
      foreground
      (nodeGraph M.! "over")
      (G.RGBA 1 0 0 0.5)
      (Line 7)
      (G.GeglRectangle (x - 10) (y - 10) 20 20)
    ) coordinates

update :: Double -> AffectionState (AffectionData UserData) IO ()
update sec = do
  traceM "updating"
  ad@AffectionData{..} <- get
  let ud@UserData{..} = userState
  traceM $ (show $ 1 / sec) ++ " FPS"
  ev <- liftIO $ SDL.pollEvent
  maybe (return ()) (\e ->
    case SDL.eventPayload e of
      SDL.MouseMotionEvent dat -> do
        let (SDL.P (SDL.V2 x y)) = SDL.mouseMotionEventPos dat
        put $ ad 
          { userState = ud
            { coordinates = Just (fromIntegral x, fromIntegral y)
            }
          }
      SDL.WindowClosedEvent _ -> do
        traceM "seeya!"
        put $ ad
          { quitEvent = True
          }
      _ ->
        return ()
    ) ev

clean :: UserData -> IO ()
clean _ = return ()