module Graphics.Rasterific
(
fill
, fillWithMethod
, withTexture
, withClipping
, withTransformation
, stroke
, dashedStroke
, dashedStrokeWithOffset
, printTextAt
, ModulablePixel
, RenderablePixel
, renderDrawing
, pathToPrimitives
, Texture
, Drawing
, Modulable
, V2( .. )
, Point
, Vector
, CubicBezier( .. )
, Line( .. )
, Bezier( .. )
, Primitive( .. )
, Path( .. )
, PathCommand( .. )
, Transformable( .. )
, PointFoldable( .. )
, PlaneBoundable( .. )
, PlaneBound( .. )
, line
, rectangle
, roundedRectangle
, circle
, ellipse
, polyline
, polygon
, drawImageAtSize
, drawImage
, clip
, bezierFromPath
, lineFromPath
, cubicBezierFromPath
, Join( .. )
, Cap( .. )
, SamplerRepeat( .. )
, FillMethod( .. )
, DashPattern
, dumpDrawing
) where
import qualified Data.Foldable as F
import Control.Applicative( (<$>) )
import Control.Monad( forM_ )
import Control.Monad.Free( Free( .. ), liftF )
import Control.Monad.Free.Church( F, fromF )
import Control.Monad.ST( ST, runST )
import Control.Monad.State( StateT, execStateT, get, lift )
import Data.Maybe( fromMaybe )
import Data.Monoid( Monoid( .. ), (<>) )
import Codec.Picture.Types( Image( .. )
, Pixel( .. )
, Pixel8
, PixelRGBA8
, MutableImage( .. )
, createMutableImage
, unsafeFreezeImage )
import qualified Data.Vector.Unboxed as VU
import Graphics.Rasterific.Compositor
import Graphics.Rasterific.Linear( V2( .. ), (^+^), (^*) )
import Graphics.Rasterific.Rasterize
import Graphics.Rasterific.Texture
import Graphics.Rasterific.Shading
import Graphics.Rasterific.Types
import Graphics.Rasterific.Line
import Graphics.Rasterific.QuadraticBezier
import Graphics.Rasterific.CubicBezier
import Graphics.Rasterific.StrokeInternal
import Graphics.Rasterific.Transformations
import Graphics.Rasterific.PlaneBoundable
import Graphics.Text.TrueType( Font, PointSize, getStringCurveAtPoint )
type DrawContext s px a =
StateT (MutableImage s px) (ST s) a
type Drawing px = F (DrawCommand px)
data DrawCommand px next
= Fill FillMethod [Primitive] next
| Stroke Float Join (Cap, Cap) [Primitive] next
| DashedStroke Float DashPattern Float Join (Cap, Cap) [Primitive] next
| TextFill Font PointSize Point String next
| SetTexture (Texture px)
(Drawing px ()) next
| WithCliping (forall innerPixel. Drawing innerPixel ())
(Drawing px ()) next
| WithTransform Transformation (Drawing px ()) next
dumpDrawing :: ( Show px
, Show (PixelBaseComponent px)
, PixelBaseComponent (PixelBaseComponent px)
~ (PixelBaseComponent px)
) => Drawing px () -> String
dumpDrawing = go . fromF where
go ::
( Show px
, Show (PixelBaseComponent px)
, PixelBaseComponent (PixelBaseComponent px)
~ (PixelBaseComponent px)
) => Free (DrawCommand px) () -> String
go (Pure ()) = "return ()"
go (Free (Fill _ prims next)) =
"fill " ++ show prims ++ " >>=\n" ++ go next
go (Free (TextFill _ _ _ text next)) =
"-- Text : " ++ text ++ "\n" ++ go next
go (Free (SetTexture tx drawing next)) =
"withTexture (" ++ dumpTexture tx ++ ") (" ++
go (fromF drawing) ++ ") >>=\n" ++ go next
go (Free (DashedStroke o pat w j cap prims next)) =
"dashedStrokeWithOffset "
++ show o ++ " "
++ show pat ++ " "
++ show w ++ " ("
++ show j ++ ") "
++ show cap ++ " "
++ show prims ++ " >>=\n" ++ go next
go (Free (Stroke w j cap prims next)) =
"stroke " ++ show w ++ " ("
++ show j ++ ") "
++ show cap ++ " "
++ show prims ++ " >>=\n" ++ go next
go (Free (WithTransform trans sub next)) =
"withTransform (" ++ show trans ++ ") ("
++ go (fromF sub) ++ ") >>=\n "
++ go next
go (Free (WithCliping clipping draw next)) =
"withClipping (" ++ go (fromF $ withTexture clipTexture clipping)
++ ")\n" ++
" (" ++ go (fromF draw) ++ ")\n >>= " ++
go next
where clipTexture = uniformTexture (0xFF :: Pixel8)
instance Functor (DrawCommand px) where
fmap f (TextFill font size pos str next) =
TextFill font size pos str $ f next
fmap f (Fill method prims next) = Fill method prims $ f next
fmap f (SetTexture t sub next) = SetTexture t sub $ f next
fmap f (WithCliping sub com next) =
WithCliping sub com $ f next
fmap f (Stroke w j caps prims next) =
Stroke w j caps prims $ f next
fmap f (DashedStroke st pat w j caps prims next) =
DashedStroke st pat w j caps prims $ f next
fmap f (WithTransform trans draw next) =
WithTransform trans draw $ f next
instance Monoid (Drawing px ()) where
mempty = return ()
mappend a b = a >> b
withTexture :: Texture px -> Drawing px () -> Drawing px ()
withTexture texture subActions =
liftF $ SetTexture texture subActions ()
withTransformation :: Transformation -> Drawing px () -> Drawing px ()
withTransformation trans sub =
liftF $ WithTransform trans sub ()
fill :: [Primitive] -> Drawing px ()
fill prims = liftF $ Fill FillWinding prims ()
fillWithMethod :: FillMethod -> [Primitive] -> Drawing px ()
fillWithMethod method prims =
liftF $ Fill method prims ()
withClipping
:: (forall innerPixel. Drawing innerPixel ())
-> Drawing px ()
-> Drawing px ()
withClipping clipPath drawing =
liftF $ WithCliping clipPath drawing ()
stroke :: Float
-> Join
-> (Cap, Cap)
-> [Primitive]
-> Drawing px ()
stroke width join caping prims =
liftF $ Stroke width join caping prims ()
printTextAt :: Font
-> Int
-> Point
-> String
-> Drawing px ()
printTextAt font pointSize point string =
liftF $ TextFill font pointSize point string ()
data RenderContext px = RenderContext
{ currentClip :: Maybe (Texture (PixelBaseComponent px))
, currentTexture :: Texture px
, currentTransformation :: Maybe (Transformation, Transformation)
}
renderDrawing
:: forall px . (RenderablePixel px)
=> Int
-> Int
-> px
-> Drawing px ()
-> Image px
renderDrawing width height background drawing = runST $
createMutableImage width height background
>>= execStateT (go initialContext $ fromF drawing)
>>= unsafeFreezeImage
where
initialContext = RenderContext Nothing stupidDefaultTexture Nothing
clipBackground = emptyValue :: PixelBaseComponent px
clipForeground = fullValue :: PixelBaseComponent px
stupidDefaultTexture =
uniformTexture $ colorMap (const clipBackground) background
clipRender =
renderDrawing width height clipBackground
. withTexture (uniformTexture clipForeground)
textureOf ctxt@RenderContext { currentTransformation = Just (_, t) } =
transformTexture t $ currentTexture ctxt
textureOf ctxt = currentTexture ctxt
geometryOf RenderContext { currentTransformation = Just (trans, _) } =
transform (applyTransformation trans)
geometryOf _ = id
go :: RenderContext px
-> Free (DrawCommand px) ()
-> DrawContext s px ()
go _ (Pure ()) = return ()
go ctxt (Free (WithTransform trans sub next)) = do
let trans'
| Just (t, _) <- currentTransformation ctxt = t <> trans
| otherwise = trans
invTrans =
fromMaybe mempty $ inverseTransformation trans'
go ctxt { currentTransformation =
Just (trans', invTrans) } $ fromF sub
go ctxt next
go ctxt@RenderContext { currentClip = Nothing }
(Free (Fill method prims next)) = do
fillWithTexture method (textureOf ctxt) $ geometryOf ctxt prims
go ctxt next
go ctxt@RenderContext { currentClip = Just moduler }
(Free (Fill method prims next)) = do
fillWithTextureAndMask method (textureOf ctxt)
moduler $ geometryOf ctxt prims
go ctxt next
go ctxt (Free (Stroke w j cap prims next)) =
go ctxt . Free $ Fill FillWinding prim' next
where prim' = listOfContainer $ strokize w j cap prims
go ctxt (Free (SetTexture tx sub next)) = do
go (ctxt { currentTexture = tx }) $ fromF sub
go ctxt next
go ctxt (Free (DashedStroke o d w j cap prims next)) = do
let recurse sub =
go ctxt . liftF $ Fill FillWinding sub ()
mapM_ recurse $ dashedStrokize o d w j cap prims
go ctxt next
go ctxt (Free (TextFill font size (V2 x y) str next)) = do
forM_ drawCalls (go ctxt)
go ctxt next
where
drawCalls =
beziersOfChar <$> getStringCurveAtPoint 90 (x, y)
[(font, size, str)]
beziersOfChar curves = liftF $ Fill FillWinding bezierCurves ()
where
bezierCurves = concat
[map BezierPrim . bezierFromPath . map (uncurry V2)
$ VU.toList c | c <- curves]
go ctxt (Free (WithCliping clipPath path next)) = do
go (ctxt { currentClip = newModuler }) $ fromF path
go ctxt next
where
modulationTexture :: Texture (PixelBaseComponent px)
modulationTexture = RawTexture $ clipRender clipPath
newModuler = Just . subModuler $ currentClip ctxt
subModuler Nothing = modulationTexture
subModuler (Just v) =
modulateTexture v modulationTexture
dashedStroke
:: DashPattern
-> Float
-> Join
-> (Cap, Cap)
-> [Primitive]
-> Drawing px ()
dashedStroke = dashedStrokeWithOffset 0.0
dashedStrokeWithOffset
:: Float
-> DashPattern
-> Float
-> Join
-> (Cap, Cap)
-> [Primitive]
-> Drawing px ()
dashedStrokeWithOffset _ [] width join caping prims =
stroke width join caping prims
dashedStrokeWithOffset offset dashing width join caping prims =
liftF $ DashedStroke offset dashing width join caping prims ()
clip :: Point
-> Point
-> Primitive
-> Container Primitive
clip mini maxi (LinePrim l) = clipLine mini maxi l
clip mini maxi (BezierPrim b) = clipBezier mini maxi b
clip mini maxi (CubicBezierPrim c) = clipCubicBezier mini maxi c
isCoverageDrawable :: MutableImage s px -> CoverageSpan -> Bool
isCoverageDrawable img coverage =
_coverageVal coverage > 0 && x >= 0 && y >= 0 && x < imgWidth && y < imgHeight
where
!imgWidth = fromIntegral $ mutableImageWidth img
!imgHeight = fromIntegral $ mutableImageHeight img
x = _coverageX coverage
y = _coverageY coverage
fillWithTexture :: RenderablePixel px
=> FillMethod
-> Texture px
-> [Primitive]
-> DrawContext s px ()
fillWithTexture fillMethod texture els = do
img@(MutableImage width height _) <- get
let !mini = V2 0 0
!maxi = V2 (fromIntegral width) (fromIntegral height)
!filler = transformTextureToFiller texture img
clipped = F.foldMap (clip mini maxi) els
spans = rasterize fillMethod clipped
lift . mapExec filler $ filter (isCoverageDrawable img) spans
mapExec :: Monad m => (a -> m ()) -> [a] -> m ()
mapExec f = go
where
go [] = return ()
go (x : xs) = f x >> go xs
fillWithTextureAndMask
:: RenderablePixel px
=> FillMethod
-> Texture px
-> Texture (PixelBaseComponent px)
-> [Primitive]
-> DrawContext s px ()
fillWithTextureAndMask fillMethod texture mask els = do
img@(MutableImage width height _) <- get
let !mini = V2 0 0
!maxi = V2 (fromIntegral width) (fromIntegral height)
spans = rasterize fillMethod $ F.foldMap (clip mini maxi) els
!shader = transformTextureToFiller (modulateTexture texture mask) img
lift . mapM_ shader $ filter (isCoverageDrawable img) spans
circle :: Point
-> Float
-> [Primitive]
circle center radius =
CubicBezierPrim . transform mv <$> cubicBezierCircle
where
mv p = (p ^* radius) ^+^ center
ellipse :: Point -> Float -> Float -> [Primitive]
ellipse center rx ry =
CubicBezierPrim . transform mv <$> cubicBezierCircle
where
mv (V2 x y) = V2 (x * rx) (y * ry) ^+^ center
polyline :: [Point] -> [Primitive]
polyline = map LinePrim . lineFromPath
polygon :: [Point] -> [Primitive]
polygon [] = []
polygon [_] = []
polygon [_,_] = []
polygon lst@(p:_) = polyline $ lst ++ [p]
rectangle :: Point
-> Float
-> Float
-> [Primitive]
rectangle p@(V2 px py) w h =
LinePrim <$> lineFromPath
[ p, V2 (px + w) py, V2 (px + w) (py + h), V2 px (py + h), p ]
drawImage :: ModulablePixel px
=> Image px
-> StrokeWidth
-> Point
-> Drawing px ()
drawImage img@Image { imageWidth = w, imageHeight = h } s p =
drawImageAtSize img s p (fromIntegral w) (fromIntegral h)
drawImageAtSize :: (Pixel px, Modulable (PixelBaseComponent px))
=> Image px
-> StrokeWidth
-> Point
-> Float
-> Float
-> Drawing px ()
drawImageAtSize img@Image { imageWidth = w, imageHeight = h } borderSize p
reqWidth reqHeight
| borderSize <= 0 =
withTransformation (translate p <> scale scaleX scaleY) .
withTexture (sampledImageTexture img) $ fill rect
| otherwise = do
withTransformation (translate p <> scale scaleX scaleY) $ do
withTexture (sampledImageTexture img) $ fill rect
stroke borderSize (JoinMiter 0)
(CapStraight 0, CapStraight 0) rect'
where
rect = rectangle (V2 0 0) rw rh
rect' = rectangle p reqWidth reqHeight
(rw, rh) = (fromIntegral w, fromIntegral h)
scaleX | reqWidth == 0 = 1
| otherwise = reqWidth / rw
scaleY | reqHeight == 0 = 1
| otherwise = reqHeight / rh
roundedRectangle :: Point
-> Float
-> Float
-> Float
-> Float
-> [Primitive]
roundedRectangle (V2 px py) w h rx ry =
[ CubicBezierPrim . transform (^+^ V2 xFar yNear) $ cornerTopR
, LinePrim $ Line (V2 xFar py) (V2 xNear py)
, CubicBezierPrim . transform (^+^ V2 (px + rx) (py + ry)) $ cornerTopL
, LinePrim $ Line (V2 px yNear) (V2 px yFar)
, CubicBezierPrim . transform (^+^ V2 (px + rx) yFar) $ cornerBottomL
, LinePrim $ Line (V2 xNear (py + h)) (V2 xFar (py + h))
, CubicBezierPrim . transform (^+^ V2 xFar yFar) $ cornerBottomR
, LinePrim $ Line (V2 (px + w) yFar) (V2 (px + w) yNear)
]
where
xNear = px + rx
xFar = px + w rx
yNear = py + ry
yFar = py + h ry
(cornerBottomR :
cornerTopR :
cornerTopL :
cornerBottomL:_) = transform (\(V2 x y) -> V2 (x * rx) (y * ry)) <$> cubicBezierCircle
line :: Point -> Point -> [Primitive]
line p1 p2 = [LinePrim $ Line p1 p2]