{-# LANGUAGE RecordWildCards #-} import Affection import qualified SDL import qualified GEGL as G import qualified BABL as B import qualified Data.Map.Strict as M import Foreign.C.Types import System.Random (randomRIO) 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 , partsys :: ParticleSystem } 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.GeglBuffer b) <- G.gegl_buffer_new (Just $ G.GeglRectangle 0 0 800 600) =<< B.babl_format (B.PixelFormat B.RGBA B.CFfloat) sink <- G.gegl_node_new_child root $ G.Operation "gegl:copy-buffer" [G.Property "buffer" $ G.PropertyBuffer buffer ] traceM "buffer-sink" nop <- G.gegl_node_new_child root $ G.Operation "nop" [] traceM "nop" crop <- G.gegl_node_new_child root $ G.Operation "gegl:crop" [ G.Property "width" $ G.PropertyDouble 800 , G.Property "height" $ G.PropertyDouble 600 ] traceM "crop" G.gegl_node_link_many [checkerboard, over, crop, sink] G.gegl_node_connect_to nop "output" over "aux" traceM "connections made" myMap <- return $ M.fromList [ ("root" , root) , ("over" , over) , ("background" , checkerboard) , ("sink" , sink) , ("nop" , nop) , ("crop" , crop) ] traceM "loading complete" return $ UserData { nodeGraph = myMap , foreground = buffer , partsys = ParticleSystem [] nop buffer } drawInit :: Affection UserData () drawInit = do UserData{..} <- getAffection present (GeglRectangle 0 0 800 600) foreground True draw :: Affection UserData () draw = do traceM "drawing" UserData{..} <- getAffection liftIO $ G.gegl_node_process $ nodeGraph M.! "sink" -- ad <- get -- ud <- getAffection -- drawParticles partDraw $ particles ud -- SDL.V2 (CInt rw) (CInt rh) <- SDL.surfaceDimensions $ drawSurface ad -- let (w, h) = (fromIntegral rw, fromIntegral rh) -- liftIO $ clearArea (foreground ud) (GeglRectangle 0 0 w h) -- maybe (return ()) (\(x, y) -> -- drawRect -- (foreground ud) -- (nodeGraph ud M.! "over") -- (G.RGBA 1 0 0 0.5) -- (Line 7) -- (G.GeglRectangle (x - 10) (y - 10) 20 20) -- ) $ coordinates ud update :: Double -> AffectionState (AffectionData UserData) IO () update sec = do traceM "updating" ad <- get ud <- getAffection -- let newPart = updateParticles sec partUpd $ particles ud -- putAffection $ ud { particles = newPart } traceM $ (show $ 1 / sec) ++ " FPS" ev <- liftIO $ SDL.pollEvents mapM_ (\e -> case SDL.eventPayload e of SDL.MouseMotionEvent dat -> if SDL.ButtonLeft `elem` SDL.mouseMotionEventState dat then do let (SDL.P (SDL.V2 x y)) = SDL.mouseMotionEventPos dat vx <- liftIO $ randomRIO (-20, 20) vy <- liftIO $ randomRIO (-20, 20) life <- liftIO $ randomRIO (1, 5) tempRoot <- liftIO $ G.gegl_node_new tempOver <- liftIO $ G.gegl_node_new_child tempRoot G.defaultOverOperation tempNop <- liftIO $ G.gegl_node_new_child tempRoot $ G.Operation "gegl:nop" [] liftIO $ G.gegl_node_connect_to tempNop "output" tempOver "aux" -- traceM $ "position is: " ++ show x ++ " " ++ show y -- traceM $ "velocity is: " ++ show vx ++ " " ++ show vy putAffection $ ud { partsys = (partsys ud) { psParts = (Particle { particleLife = life , particlePosition = (fromIntegral x, fromIntegral y) , particleRotation = Rad 0 , particleVelocity = (vx, vy) , particlePitchRate = Rad 0 , particleRootNode = tempRoot , particleNodeGraph = M.fromList [ ("root", tempRoot) , ("over", tempOver) , ("nop" , tempNop) ] , particleStackCont = tempOver , particleDrawFlange = tempNop }) : (psParts $ partsys ud) } } else return () SDL.WindowClosedEvent _ -> do traceM "seeya!" put $ ad { quitEvent = True } _ -> return () ) ev ud2 <- getAffection nps <- updateParticleSystem (partsys ud2) sec partUpd partDraw putAffection $ ud2 { partsys = nps } clean :: UserData -> IO () clean _ = return () partUpd :: Double -> Particle -> Particle partUpd sec p@Particle{..} = p { particlePosition = (newX, newY) } where newX = (fst particlePosition) + sec * (fromIntegral $ fst particleVelocity) newY = (snd particlePosition) + sec * (fromIntegral $ snd particleVelocity) partDraw :: G.GeglBuffer -> G.GeglNode -> Particle -> Affection UserData () partDraw buf node Particle{..} = do ud <- getAffection drawRect' particleDrawFlange (G.RGBA 1 0 0 0.5) (Fill) (G.GeglRectangle ((floor $ fst particlePosition) - 10) ((floor $ snd particlePosition) -10) 20 20) buf