FPPrac.hs

   1 {-# LANGUAGE RecordWildCards, ExistentialQuantification #-}
   2 module FPPrac (
   3         Request(..),
   4         Response(..),
   5         TinaProgram(..),
   6         Color, Rect(..), Point(..),
   7         rgb, pt, point,
   8         runTina
   9 ) where
  10
  11 import qualified Graphics.UI.Gtk as Gtk
  12 import Graphics.UI.Gtk (AttrOp(..)) -- For the := constructor
  13 import qualified Graphics.Rendering.Cairo as Cairo
  14 import qualified Graphics.UI.Gtk.Gdk.EventM as EventM
  15
  16 import Control.Monad.Trans -- for liftIO
  17 import Data.IORef
  18 import Control.Applicative
  19 import Control.Monad
  20 import Char
  21 import System.Exit
  22
  23 -- | A rectangle in two dimensional space
  24 data Rect = Rect
  25         { rectLeft :: !Int
  26         , rectTop :: !Int
  27         , rectWidth :: !Int
  28         , rectHeight :: !Int
  29 } deriving (Show, Eq)
  30
  31 data Point = Point !Int !Int deriving (Show, Eq)
  32
  33 type Color = Gtk.Color
  34
  35 -- Create a Color from Red, Green and Blue values. The inputs should be
  36 -- between 0 and 255 (inclusive).
  37 rgb :: Int -> Int -> Int -> Color
  38 rgb r g b = Gtk.Color (conv r) (conv g) (conv b)
  39         where conv = fromInteger . toInteger . (*256)
  40
  41 -- | Some predefined colours
  42 red = rgb 0xff 0 0
  43 green = rgb 0 0xff 0
  44 blue = rgb 0 0 0xff
  45 white = rgb 0xff 0xff 0xff
  46
  47 -- | Helper functions for creating a Point
  48 point, pt :: Int -> Int -> Point
  49 point = Point
  50 pt = Point
  51
  52 data Request
  53         = GfxLines      Color   [Point]                 -- coloured line through a list of points
  54         | GfxPolygon    Color   [Point]                 -- filled polygon of given colour
  55         | GfxPicture    FilePath Point                  -- shows a picture
  56         | GfxText       Color    Point  String          -- coloured string on position Point
  57         | GfxRectangle  Color    Rect                   -- filled rectangle of given colour
  58         | GfxEllipse    Color    Rect                   -- ellipse within given rectangle
  59         | GfxDisc       Color    Rect                   -- filled ellipse within given rectangle
  60         | GfxClear                                      -- clears the graphical window
  61         -- | GfxInstance   Bool
  62         | GfxFont       String   Int                    -- changes to fontname of given size
  63         | WinPrompt     String   String String          -- pops up a window with an edit field
  64         -- | WinFilePrompt Bool
  65         | WinMenu       [(String,[String])]             -- adds a menu list to the graphical window
  66         | WinTitle      String                          -- gives a title to the graphical window
  67         | FRead         String                          -- read file with a given name
  68         | FWrite        String   String                 -- writes a text file with a given filename
  69         | ReqQuit                                       -- quits the graphical system
  70         deriving Show
  71
  72 data Response
  73         = KeyIn            Char                         -- touched key with given character
  74         | MouseDoubleClick Point                        -- mouse event on position Point
  75         | MouseDragged     Point                        -- ibid
  76         | MouseDown        Point                        -- ibid
  77         | MouseUp          Point                        -- ibid
  78         | MenuItem         String String                -- selected item from WinMenu with a given name
  79         | PromptResponse   String String String         -- response to WinPrompt request
  80         | FileContents     String String                -- response to FRead request
  81         deriving Show
  82
  83 type TinaStep s = s -> Response -> (s,[Request])
  84 data TinaProgram = forall s. Main
  85         { initialState    :: s
  86         , initialRequests :: [Request]
  87         , eventHandler    :: TinaStep s
  88         , windowWidth
  89         , windowHeight    :: Int
  90         }
  91
  92 testProg = Main
  93         { initialState    = 1
  94         , initialRequests = [GfxText red (pt 0 0) "foo", GfxText blue (pt 100 100) "bar"]
  95         , eventHandler    = \s e -> (s+1,[GfxText green (pt 50 50) $ show (s,e)])
  96         , windowWidth     = 200
  97         , windowHeight    = 200
  98         }
  99
 100 data IState = forall s. IS
 101         { window    :: Gtk.Window
 102         , buffer    :: IORef Cairo.Surface
 103         , postponed :: IORef [Request]
 104         , usrState  :: IORef s
 105         , usrProg   :: TinaStep s
 106         }
 107
 108 processPostponed :: IState -> IO ()
 109 processPostponed s@IS {..} = do
 110         ps <- readIORef postponed
 111         unless (null ps) $ do
 112                 writeIORef postponed (tail ps)
 113                 rs  <- handle s (head ps)
 114                 mapM (stepUserProgram s) rs
 115                 processPostponed s
 116
 117 post s r = stepUserProgram s r >> processPostponed s
 118
 119 stepUserProgram :: IState -> Response -> IO ()
 120 stepUserProgram IS {..} r = do
 121         state <- readIORef usrState
 122         let (state',reqs) = usrProg state r
 123         writeIORef usrState state'
 124         readIORef postponed >>= writeIORef postponed . (++ reqs)
 125
 126 handle :: IState -> Request -> IO [Response]
 127 handle s@IS {..} r = do
 128         resps <- maybe (fail $ "No handler for request " ++ show r) id $
 129                 fmap (>> return []) (gfxHandler s r <|> winHandler s r <|> miscHandler s r)
 130         return resps
 131
 132 runTina :: TinaProgram -> IO ()
 133 runTina Main {..} = do
 134         usrState  <- newIORef initialState
 135         postponed <- newIORef (GfxText (rgb 0 0 0) (pt 50 50) "foo" : GfxClear :initialRequests)
 136         let state = IS { window = undefined, buffer = undefined, usrProg = eventHandler, .. }
 137         runGUI windowWidth windowHeight state
 138
 139 runGUI :: Int -> Int -> IState -> IO ()
 140 runGUI w h (IS { .. }) = do
 141         -- Init GTK.
 142         Gtk.initGUI
 143
 144         -- Create a window, which will make the mainloop terminated when
 145         -- it is closed.
 146         window <- Gtk.windowNew
 147
 148         Gtk.set window [ Gtk.containerBorderWidth := 10
 149                        , Gtk.windowTitle := "FP Practicum"
 150                        , Gtk.windowDefaultWidth := w
 151                        , Gtk.windowDefaultHeight := h
 152                        ]
 153         Gtk.onDestroy window Gtk.mainQuit
 154
 155         -- Create a buffer to draw on (name the actual buffer buffer', so we
 156         -- can use IS { .. } syntax below to pack the state. Using a record update
 157         -- wouldn't work, probably because Cairo.Surface contains an existential
 158         -- type...
 159         -- We put the buffer in an IORef, so we can change it for a new one
 160         -- later on (on window resizes).
 161         buffer' <- Cairo.createImageSurface Cairo.FormatARGB32 w h
 162         buffer <- newIORef buffer'
 163
 164         -- Register the expose event
 165         Gtk.on window Gtk.exposeEvent $ onExpose buffer
 166
 167         -- Repack state
 168         let state = IS { .. }
 169
 170         -- Show the window and start the Gtk mainloop.
 171         Gtk.widgetShowAll window
 172         Gtk.mainGUI
 173
 174 -- | Copy the given surface to the exposed window on an expose event.
 175 onExpose :: IORef Cairo.Surface -> EventM.EventM EventM.EExpose Bool
 176 onExpose buffer = do
 177         current_buffer <- liftIO $ readIORef buffer
 178         dw <- EventM.eventWindow
 179         -- Copy the buffer to the window
 180         liftIO $ Gtk.renderWithDrawable dw $ do
 181                 Cairo.setSourceSurface current_buffer 0 0
 182                 Cairo.paint
 183         return True -- No clue what this means
 184
 185 {-
 186 runGUI s IS {..} = do
 187         sFrame <- frame
 188                 [ text       := "FP Practicum"
 189                 , size       := s
 190                 ]
 191         buffer <- memoryDCCreate
 192         bitmapCreateEmpty s 24 >>= memoryDCSelectObject buffer
 193         withBrushStyle (BrushStyle BrushSolid white) (dcSetBackground buffer)
 194         dcClear buffer
 195         buffer `set`
 196                 [ fontFace   := "Courier New"
 197                 , fontSize   := 10
 198                 , brushColor := rgb 0 0 0
 199                 , brushKind  := BrushSolid
 200                 , penColor   := rgb 0 0 0
 201                 , penKind    := PenSolid
 202                 ]
 203         sPanel <- panel sFrame [ size := s ]
 204         let state = IS {..}
 205         sPanel `set`
 206                 [ on paint       := onPaint state
 207                 , on doubleClick := post state . MouseDoubleClick
 208                 , on click       := post state . MouseDown
 209                 , on drag        := post state . MouseDragged
 210                 , on unclick     := post state . MouseUp
 211                 , on anyKey      := transKey (post state . KeyIn)
 212                 ]
 213         sFrame `set`
 214                 [ on closing := sFrame `set` [ visible := False ] >> wxcAppExit
 215                 , on anyKey  := transKey (post state . KeyIn)
 216                 , layout     := widget sPanel
 217                 ]
 218         windowSetFocus sFrame
 219         processPostponed state
 220 onPaint :: IState -> DC a -> Rect -> IO ()
 221 onPaint IS {..} dest va = do
 222         dcBlit dest va buffer (Point 0 0) wxCOPY False >> return ()
 223
 224 transKey :: (Char -> IO ()) -> Key -> IO ()
 225 transKey prod (KeyChar c) = prod c
 226 transKey prod  KeySpace   = prod ' '
 227 transKey prod  KeyEscape  = prod '\ESC'
 228 transKey prod  KeyReturn  = prod '\n'
 229 transKey prod  KeyTab     = prod '\t'
 230 transKey _ _ = return ()
 231
 232 -}
 233
 234 {-
 235 miscHandler s@IS {..} (FRead  fn     ) = Just $ readFile fn >>= post s . FileContents fn
 236 miscHandler   IS {..} (FWrite fn cnts) = Just $ writeFile fn cnts
 237 miscHandler   IS {..} ReqQuit = Just $ putStrLn "Quiting" >> wxcAppExit
 238 -}
 239 miscHandler   IS {..} _ = Nothing
 240
 241 {-
 242 winHandler s@IS {..} (WinPrompt st1 st2 st3) = Just $ textDialog sFrame st1 st2 st3 >>= post s . PromptResponse st1 st2
 243 winHandler   IS {..} (WinTitle     st) = Just $ sFrame `set` [text := st]
 244 winHandler s@IS {..} (WinMenu      ms) = Just $ mkMenu >>= \ms' -> sFrame `set` [menuBar := ms']
 245         where
 246         mkMenu = sequence
 247                 [ do
 248                         p  <- menuPane [ text := name ]
 249                         sequence
 250                                 [ do
 251                                         i <- menuItem p [ text := item ]
 252                                         sFrame `set` [on (menu i) := post s (MenuItem name item)]
 253                                  | item <- items ]
 254                         return p
 255                  | (name,items) <- ms ]
 256 -}
 257 winHandler _        _                = Nothing
 258
 259 {-
 260 gfxHandler IS {..} (GfxLines     col ps)    = Just $ polyline buffer ps [penColor := col] >> dirtyPts sPanel ps
 261 gfxHandler IS {..} (GfxPolygon   col ps)    = Just $ polygon  buffer ps [penColor := col, brushColor := col] >> dirtyPts sPanel ps
 262 gfxHandler IS {..} (GfxText      col xy st) = Just $ drawText buffer st xy [textColor := col] >> getTextExtent buffer st >>= dirtyRect' sPanel xy
 263 gfxHandler IS {..} (GfxRectangle col rt)    = Just $ drawRect buffer rt [penColor := col, brushColor := col] >> dirtyRect sPanel rt
 264 gfxHandler IS {..} (GfxEllipse   col rt)    = Just $ ellipse buffer rt [penColor := col, brushKind := BrushTransparent] >> dirtyRect sPanel rt
 265 gfxHandler IS {..} (GfxDisc      col rt)    = Just $ ellipse buffer rt [penColor := col, brushColor := col] >> dirtyRect sPanel rt
 266 gfxHandler IS {..} (GfxFont      st  sz)    = Just $ buffer `set` [ fontSize := sz, fontFace := st ]
 267 gfxHandler IS {..}  GfxClear                = Just $ dcClear buffer >> windowRefresh sPanel False
 268 gfxHandler IS {..} (GfxPicture   fd  pt)    = Just $ bitmapCreateFromFile fd >>= \bm -> drawBitmap buffer bm pt False [] >> bitmapGetSize bm >>= dirtyRect' sPanel pt
 269 -}
 270 gfxHandler _        _                       = Nothing
 271
 272 {-
 273 dirtyPts :: Window a -> [Point] -> IO ()
 274 dirtyPts dc ps = dirtyRect' dc (pt x y) Size {..}
 275         where
 276         xs     = map pointX ps
 277         ys     = map pointY ps
 278         x      = minimum xs
 279         y      = minimum ys
 280         sizeW  = maximum xs - x
 281         sizeH  = maximum ys - y
 282
 283 dirtyRect' :: Window a -> Point -> Size -> IO ()
 284 dirtyRect' dc Point {..} Size {..} = dirtyRect dc $ Rect pointX pointY sizeW sizeH
 285
 286 dirtyRect :: Window a -> Rect -> IO ()
 287 dirtyRect dc rect = windowRefreshRect dc False (grow 2 rect)
 288
 289 rectanglify :: Point -> Size -> Rect
 290 rectanglify Point {..} Size {..} = Rect pointX pointY sizeW sizeH
 291
 292 grow :: Int -> Rect -> Rect
 293 grow n Rect {..} = Rect
 294         { rectLeft   = rectLeft   - n
 295         , rectTop    = rectTop    - n
 296         , rectWidth  = rectWidth  + 2 * n
 297         , rectHeight = rectHeight + 2 * n }
 298
 299 -}