pituicat/examples/example01.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 = withAllAffection $
--   withDefaultWindow "test" $ do
--     changeColor $ RGBA 255 255 255 255
--     clear
--     present
--     liftIO $ delaySec 2

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

data UserData = UserData
  { nodeGraph   :: M.Map String G.GeglNode
  , foreground  :: G.GeglBuffer
  , elapsedTime :: Double
  }

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 20 20) =<<
    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)
    ]
  let roi = G.GeglRectangle 0 0 20 20
  G.iterateOver
    buffer
    roi
    (B.PixelFormat B.RGBA B.CFfloat)
    G.GeglAccessReadWrite
    G.GeglAbyssNone
    (\(G.Pixel px py pc) ->
      let dsqr = (((10 - px) ^ 2) + ((10 - py) ^ 2))
          (G.CVfloat (CFloat pr), G.CVfloat (CFloat pg), G.CVfloat (CFloat pb), G.CVfloat (CFloat pa)) = pc
          dist = (sqrt (fromIntegral dsqr :: Float))
      in if dsqr < 100
        then
          if dist < fromIntegral 9
            then
              G.Pixel px py
                ( G.CVfloat $ CFloat 0
                , G.CVfloat $ CFloat 0
                , G.CVfloat $ CFloat 0
                , G.CVfloat $ CFloat $ if pa < 1 then 1 else pa
                )
            else
              let alpha = fromIntegral 10 - dist
                  dst_a = pa
                  a = alpha + dst_a * (1 - alpha)
                  a_term = dst_a * (1 - alpha)
                  red = 0 * alpha + pr * a_term
                  gre = 0 * alpha + pg * a_term
                  blu = 0 * alpha + pb * a_term
              in
                G.Pixel px py
                  ( G.CVfloat $ CFloat $ red / a
                  , G.CVfloat $ CFloat $ gre / a
                  , G.CVfloat $ CFloat $ blu / a
                  , G.CVfloat $ CFloat $ if pa < alpha then alpha else pa
                  )
      else
        G.Pixel px py pc
    )
  traceM "loading complete"
  return $ UserData
    { nodeGraph = myMap
    , foreground = buffer
    , elapsedTime = 0
    }

draw :: AffectionState (AffectionData UserData) IO ()
draw = do
  traceM "drawing"
  AffectionData{..} <- get
  let UserData{..} = userState
  liftIO $ SDL.lockSurface drawSurface
  pixels <- liftIO $ SDL.surfacePixels drawSurface
  let SDL.Surface rawSurfacePtr _ = drawSurface
  rawSurface <- liftIO $ peek rawSurfacePtr
  pixelFormat <- liftIO $ peek $ Raw.surfaceFormat rawSurface
  format <- liftIO $ (B.babl_format $ B.PixelFormat B.RGBA B.CFu8)
  SDL.V2 (CInt rw) (CInt rh) <- SDL.surfaceDimensions drawSurface
  let (w, h) = (fromIntegral rw, fromIntegral rh)
  liftIO $ G.gegl_node_blit
    (nodeGraph M.! "over" :: G.GeglNode)
    1
    (G.GeglRectangle 0 0 w h)
    format
    pixels
    (fromIntegral (Raw.pixelFormatBytesPerPixel pixelFormat) * w)
    [G.GeglBlitDefault]
  liftIO $ SDL.unlockSurface drawSurface
  liftIO $ SDL.updateWindowSurface drawWindow

update :: Double -> AffectionState (AffectionData UserData) IO ()
update sec = do
  traceM "updating"
  -- liftIO $ delaySec 5
  ad@AffectionData{..} <- get
  let ud@UserData{..} = userState
  traceM $ show elapsedTime
  if elapsedTime < 5
  then do
    fg <- liftIO $ G.gegl_buffer_get_extent foreground
    _ <- liftIO $ G.gegl_buffer_set_extent foreground $ fg
      { G.rectangleX = G.rectangleX fg + 1
      , G.rectangleY = G.rectangleY fg + 1
      }
    put $ ad
      { userState = ud
        { elapsedTime = elapsedTime + sec
        }
      }
  else
    put $ ad
      { quitEvent = True
      }

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