{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE TypeFamilies #-}
{-# LANGUAGE TypeSynonymInstances #-}
module Diagrams.TwoD.Path.Turtle.Internal
(
TurtleState(..), TurtlePath(..), PenStyle(..)
, forward, backward, left, right
, setPenColor, setPenColour, setPenWidth
, startTurtle, setHeading, towards
, setPenPos
, penUp, penDown, penHop, closeCurrent
, getTurtleDiagram
, getTurtlePath
) where
import Diagrams.Prelude
data PenStyle n = PenStyle
{ forall n. PenStyle n -> Measure n
penWidth :: Measure n
, forall n. PenStyle n -> Colour Double
penColor :: Colour Double
}
data TurtlePath n = TurtlePath
{ forall n. TurtlePath n -> PenStyle n
penStyle :: PenStyle n
, forall n. TurtlePath n -> Located (Trail V2 n)
turtleTrail :: Located (Trail V2 n)
}
data TurtleState n = TurtleState
{
forall n. TurtleState n -> Bool
isPenDown :: Bool
, forall n. TurtleState n -> P2 n
penPos :: P2 n
, forall n. TurtleState n -> Angle n
heading :: Angle n
, forall n. TurtleState n -> Located (Trail' Line V2 n)
currTrail :: Located (Trail' Line V2 n)
, forall n. TurtleState n -> PenStyle n
currPenStyle :: PenStyle n
, forall n. TurtleState n -> [TurtlePath n]
paths :: [TurtlePath n]
}
defaultPenStyle :: (Floating n, Ord n) => PenStyle n
defaultPenStyle :: forall n. (Floating n, Ord n) => PenStyle n
defaultPenStyle = forall n. Measure n -> Colour Double -> PenStyle n
PenStyle (forall n. Num n => n -> Measure n
normalized n
0.004 forall n. Ord n => Measure n -> Measure n -> Measure n
`atLeast` forall n. n -> Measure n
output n
0.5) forall a. Num a => Colour a
black
startTurtle :: (Floating n, Ord n) => TurtleState n
startTurtle :: forall n. (Floating n, Ord n) => TurtleState n
startTurtle = forall n.
Bool
-> P2 n
-> Angle n
-> Located (Trail' Line V2 n)
-> PenStyle n
-> [TurtlePath n]
-> TurtleState n
TurtleState Bool
True forall (f :: * -> *) a. (Additive f, Num a) => Point f a
origin forall (f :: * -> *) a. (Additive f, Num a) => f a
zero (forall a. Monoid a => a
mempty forall a. a -> Point (V a) (N a) -> Located a
`at` forall (f :: * -> *) a. (Additive f, Num a) => Point f a
origin) forall n. (Floating n, Ord n) => PenStyle n
defaultPenStyle []
moveTurtle :: (Floating n, Ord n) => Segment Closed V2 n
-> TurtleState n
-> TurtleState n
moveTurtle :: forall n.
(Floating n, Ord n) =>
Segment Closed V2 n -> TurtleState n -> TurtleState n
moveTurtle Segment Closed V2 n
s t :: TurtleState n
t@(TurtleState Bool
pd P2 n
pos Angle n
h Located (Trail' Line V2 n)
tr PenStyle n
_ [TurtlePath n]
_) =
if Bool
pd
then TurtleState n
t { currTrail :: Located (Trail' Line V2 n)
currTrail = Located (Trail' Line V2 n)
newTrail
, penPos :: P2 n
penPos = P2 n
newPenPos
}
else TurtleState n
t { penPos :: P2 n
penPos = P2 n
newPenPos }
where
rotatedSeg :: Segment Closed V2 n
rotatedSeg = forall n t.
(InSpace V2 n t, Transformable t, Floating n) =>
Angle n -> t -> t
rotate Angle n
h Segment Closed V2 n
s
newTrail :: Located (Trail' Line V2 n)
newTrail = forall a b. SameSpace a b => (a -> b) -> Located a -> Located b
mapLoc (forall a. Semigroup a => a -> a -> a
<> forall t. TrailLike t => [Segment Closed (V t) (N t)] -> t
fromSegments [Segment Closed V2 n
rotatedSeg]) Located (Trail' Line V2 n)
tr
newPenPos :: P2 n
newPenPos = P2 n
pos forall (p :: * -> *) a. (Affine p, Num a) => p a -> Diff p a -> p a
.+^ forall (v :: * -> *) n. Segment Closed v n -> v n
segOffset Segment Closed V2 n
rotatedSeg
forward :: (Floating n, Ord n) => n
-> TurtleState n
-> TurtleState n
forward :: forall n.
(Floating n, Ord n) =>
n -> TurtleState n -> TurtleState n
forward n
x = forall n.
(Floating n, Ord n) =>
Segment Closed V2 n -> TurtleState n -> TurtleState n
moveTurtle (forall (v :: * -> *) n. v n -> Segment Closed v n
straight forall a b. (a -> b) -> a -> b
$ forall n. (n, n) -> V2 n
r2 (n
x,n
0))
backward :: (Floating n, Ord n) => n
-> TurtleState n
-> TurtleState n
backward :: forall n.
(Floating n, Ord n) =>
n -> TurtleState n -> TurtleState n
backward n
x = forall n.
(Floating n, Ord n) =>
Segment Closed V2 n -> TurtleState n -> TurtleState n
moveTurtle (forall (v :: * -> *) n. v n -> Segment Closed v n
straight forall a b. (a -> b) -> a -> b
$ forall n. (n, n) -> V2 n
r2 (forall a. Num a => a -> a
negate n
x, n
0))
turnTurtle :: (Angle n -> Angle n)
-> TurtleState n
-> TurtleState n
turnTurtle :: forall n. (Angle n -> Angle n) -> TurtleState n -> TurtleState n
turnTurtle Angle n -> Angle n
f t :: TurtleState n
t@(TurtleState Bool
_ P2 n
_ Angle n
h Located (Trail' Line V2 n)
_ PenStyle n
_ [TurtlePath n]
_) = TurtleState n
t { heading :: Angle n
heading = Angle n -> Angle n
f Angle n
h }
left :: Floating n
=> n
-> TurtleState n
-> TurtleState n
left :: forall n. Floating n => n -> TurtleState n -> TurtleState n
left n
d = forall n. (Angle n -> Angle n) -> TurtleState n -> TurtleState n
turnTurtle (forall (f :: * -> *) a. (Additive f, Num a) => f a -> f a -> f a
^+^ (n
d forall b a. b -> AReview a b -> a
@@ forall n. Floating n => Iso' (Angle n) n
deg))
right :: Floating n
=> n
-> TurtleState n
-> TurtleState n
right :: forall n. Floating n => n -> TurtleState n -> TurtleState n
right n
d = forall n. (Angle n -> Angle n) -> TurtleState n -> TurtleState n
turnTurtle (forall (f :: * -> *) a. (Additive f, Num a) => f a -> f a -> f a
^-^ (n
d forall b a. b -> AReview a b -> a
@@ forall n. Floating n => Iso' (Angle n) n
deg))
setHeading :: Floating n
=> n
-> TurtleState n
-> TurtleState n
setHeading :: forall n. Floating n => n -> TurtleState n -> TurtleState n
setHeading n
d = forall n. (Angle n -> Angle n) -> TurtleState n -> TurtleState n
turnTurtle (forall a b. a -> b -> a
const forall a b. (a -> b) -> a -> b
$ n
d forall b a. b -> AReview a b -> a
@@ forall n. Floating n => Iso' (Angle n) n
deg)
towards :: RealFloat n =>
P2 n
-> TurtleState n
-> TurtleState n
towards :: forall n. RealFloat n => P2 n -> TurtleState n -> TurtleState n
towards P2 n
p = forall n. Floating n => n -> TurtleState n -> TurtleState n
setHeading forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<< (n
360 forall a. Num a => a -> a -> a
*) forall b c a. (b -> c) -> (a -> b) -> a -> c
. (forall a. Fractional a => a -> a -> a
/ forall a. Floating a => a
tau) forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a b c. (a -> b -> c) -> (a, b) -> c
uncurry forall a. RealFloat a => a -> a -> a
atan2 forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall n. V2 n -> (n, n)
unr2 forall b c a. (b -> c) -> (a -> b) -> a -> c
. (P2 n
p forall (p :: * -> *) a. (Affine p, Num a) => p a -> p a -> Diff p a
.-.) forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall n. TurtleState n -> P2 n
penPos
penUp :: (Ord n, Floating n) => TurtleState n
-> TurtleState n
penUp :: forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
penUp TurtleState n
t
| forall n. TurtleState n -> Bool
isPenDown TurtleState n
t = TurtleState n
t forall a b. a -> (a -> b) -> b
# forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
makeNewTrail forall a b. a -> (a -> b) -> b
# \TurtleState n
t' -> TurtleState n
t' { isPenDown :: Bool
isPenDown = Bool
False }
| Bool
otherwise = TurtleState n
t
penDown :: (Ord n, Floating n) => TurtleState n
-> TurtleState n
penDown :: forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
penDown TurtleState n
t
| forall n. TurtleState n -> Bool
isPenDown TurtleState n
t = TurtleState n
t
| Bool
otherwise = TurtleState n
t forall a b. a -> (a -> b) -> b
# forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
makeNewTrail forall a b. a -> (a -> b) -> b
# \TurtleState n
t' -> TurtleState n
t' { isPenDown :: Bool
isPenDown = Bool
True }
penHop :: (Ord n, Floating n) => TurtleState n
-> TurtleState n
penHop :: forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
penHop TurtleState n
t = TurtleState n
t forall a b. a -> (a -> b) -> b
# forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
makeNewTrail
closeCurrent :: (Floating n, Ord n) => TurtleState n
-> TurtleState n
closeCurrent :: forall n. (Floating n, Ord n) => TurtleState n -> TurtleState n
closeCurrent TurtleState n
t
| forall n. TurtleState n -> Bool
isPenDown TurtleState n
t = TurtleState n
t forall a b. a -> (a -> b) -> b
# TurtleState n -> TurtleState n
closeTTrail
| Bool
otherwise = TurtleState n
t
where startPos :: P2 n
startPos = forall a. Located a -> Point (V a) (N a)
loc forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall n. TurtleState n -> Located (Trail' Line V2 n)
currTrail forall a b. (a -> b) -> a -> b
$ TurtleState n
t
closeTTrail :: TurtleState n -> TurtleState n
closeTTrail TurtleState n
t' = TurtleState n
t' { penPos :: P2 n
penPos = P2 n
startPos
, currTrail :: Located (Trail' Line V2 n)
currTrail = forall a. Monoid a => a
mempty forall a. a -> Point (V a) (N a) -> Located a
`at` P2 n
startPos
, paths :: [TurtlePath n]
paths = forall n.
(Ord n, Floating n) =>
TurtleState n -> Located (Trail V2 n) -> [TurtlePath n]
addTrailToPath TurtleState n
t'
(forall a b. SameSpace a b => (a -> b) -> Located a -> Located b
mapLoc (forall l (v :: * -> *) n. Trail' l v n -> Trail v n
wrapTrail forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (v :: * -> *) n. Trail' Line v n -> Trail' Loop v n
closeLine) forall a b. (a -> b) -> a -> b
$ forall n. TurtleState n -> Located (Trail' Line V2 n)
currTrail TurtleState n
t)
}
setPenPos :: (Ord n, Floating n) => P2 n
-> TurtleState n
-> TurtleState n
setPenPos :: forall n.
(Ord n, Floating n) =>
P2 n -> TurtleState n -> TurtleState n
setPenPos P2 n
newPos TurtleState n
t = TurtleState n
t {penPos :: P2 n
penPos = P2 n
newPos } forall a b. a -> (a -> b) -> b
# forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
makeNewTrail
setPenWidth :: (Ord n, Floating n) => Measure n
-> TurtleState n
-> TurtleState n
setPenWidth :: forall n.
(Ord n, Floating n) =>
Measure n -> TurtleState n -> TurtleState n
setPenWidth Measure n
w = forall n.
(Floating n, Ord n) =>
(PenStyle n -> PenStyle n) -> TurtleState n -> TurtleState n
modifyCurrStyle (\PenStyle n
s -> PenStyle n
s { penWidth :: Measure n
penWidth = Measure n
w })
setPenColour :: (Ord n, Floating n) => Colour Double
-> TurtleState n
-> TurtleState n
setPenColour :: forall n.
(Ord n, Floating n) =>
Colour Double -> TurtleState n -> TurtleState n
setPenColour Colour Double
c = forall n.
(Floating n, Ord n) =>
(PenStyle n -> PenStyle n) -> TurtleState n -> TurtleState n
modifyCurrStyle (\PenStyle n
s -> PenStyle n
s { penColor :: Colour Double
penColor = Colour Double
c })
setPenColor :: (Ord n, Floating n) => Colour Double
-> TurtleState n
-> TurtleState n
setPenColor :: forall n.
(Ord n, Floating n) =>
Colour Double -> TurtleState n -> TurtleState n
setPenColor = forall n.
(Ord n, Floating n) =>
Colour Double -> TurtleState n -> TurtleState n
setPenColour
getTurtleDiagram :: (Renderable (Path V2 n) b, TypeableFloat n)
=> TurtleState n
-> QDiagram b V2 n Any
getTurtleDiagram :: forall n b.
(Renderable (Path V2 n) b, TypeableFloat n) =>
TurtleState n -> QDiagram b V2 n Any
getTurtleDiagram TurtleState n
t =
forall a. Monoid a => [a] -> a
mconcat forall b c a. (b -> c) -> (a -> b) -> a -> c
.
forall a b. (a -> b) -> [a] -> [b]
map forall n b.
(Renderable (Path V2 n) b, TypeableFloat n) =>
TurtlePath n -> QDiagram b V2 n Any
turtlePathToStroke forall b c a. (b -> c) -> (a -> b) -> a -> c
.
forall n. TurtleState n -> [TurtlePath n]
paths forall a b. (a -> b) -> a -> b
$ TurtleState n
t forall a b. a -> (a -> b) -> b
# forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
penUp
getTurtlePath :: (Floating n, Ord n) => TurtleState n -> Path V2 n
getTurtlePath :: forall n. (Floating n, Ord n) => TurtleState n -> Path V2 n
getTurtlePath = forall a. Monoid a => [a] -> a
mconcat forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a b. (a -> b) -> [a] -> [b]
map forall t n. (V t ~ V2, N t ~ n, TrailLike t) => TurtlePath n -> t
turtlePathToTrailLike forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall n. TurtleState n -> [TurtlePath n]
paths forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
penUp
makeTurtlePath :: TurtleState n
-> Located (Trail V2 n)
-> TurtlePath n
makeTurtlePath :: forall n. TurtleState n -> Located (Trail V2 n) -> TurtlePath n
makeTurtlePath TurtleState n
t Located (Trail V2 n)
tr = forall n. PenStyle n -> Located (Trail V2 n) -> TurtlePath n
TurtlePath (forall n. TurtleState n -> PenStyle n
currPenStyle TurtleState n
t) Located (Trail V2 n)
tr
addTrailToPath :: (Ord n, Floating n) => TurtleState n
-> Located (Trail V2 n)
-> [TurtlePath n]
addTrailToPath :: forall n.
(Ord n, Floating n) =>
TurtleState n -> Located (Trail V2 n) -> [TurtlePath n]
addTrailToPath TurtleState n
t Located (Trail V2 n)
tr
| forall (v :: * -> *) n.
(Metric v, OrderedField n) =>
Trail v n -> Bool
isTrailEmpty (forall a. Located a -> a
unLoc Located (Trail V2 n)
tr) = forall n. TurtleState n -> [TurtlePath n]
paths TurtleState n
t
| Bool
otherwise = forall n. TurtleState n -> Located (Trail V2 n) -> TurtlePath n
makeTurtlePath TurtleState n
t Located (Trail V2 n)
tr forall a. a -> [a] -> [a]
: forall n. TurtleState n -> [TurtlePath n]
paths TurtleState n
t
makeNewTrail :: (Ord n, Floating n) => TurtleState n
-> TurtleState n
makeNewTrail :: forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
makeNewTrail TurtleState n
t = TurtleState n
t { currTrail :: Located (Trail' Line V2 n)
currTrail = forall a. Monoid a => a
mempty forall a. a -> Point (V a) (N a) -> Located a
`at` forall n. TurtleState n -> P2 n
penPos TurtleState n
t
, paths :: [TurtlePath n]
paths = forall n.
(Ord n, Floating n) =>
TurtleState n -> Located (Trail V2 n) -> [TurtlePath n]
addTrailToPath TurtleState n
t (forall a b. SameSpace a b => (a -> b) -> Located a -> Located b
mapLoc forall l (v :: * -> *) n. Trail' l v n -> Trail v n
wrapTrail (forall n. TurtleState n -> Located (Trail' Line V2 n)
currTrail TurtleState n
t))
}
modifyCurrStyle :: (Floating n, Ord n) =>
(PenStyle n -> PenStyle n)
-> TurtleState n
-> TurtleState n
modifyCurrStyle :: forall n.
(Floating n, Ord n) =>
(PenStyle n -> PenStyle n) -> TurtleState n -> TurtleState n
modifyCurrStyle PenStyle n -> PenStyle n
f TurtleState n
t = TurtleState n
t forall a b. a -> (a -> b) -> b
# forall n. (Ord n, Floating n) => TurtleState n -> TurtleState n
makeNewTrail forall a b. a -> (a -> b) -> b
# \TurtleState n
t' -> TurtleState n
t' { currPenStyle :: PenStyle n
currPenStyle = (PenStyle n -> PenStyle n
f forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall n. TurtleState n -> PenStyle n
currPenStyle) TurtleState n
t' }
turtlePathToTrailLike :: (V t ~ V2, N t ~ n, TrailLike t) => TurtlePath n -> t
turtlePathToTrailLike :: forall t n. (V t ~ V2, N t ~ n, TrailLike t) => TurtlePath n -> t
turtlePathToTrailLike (TurtlePath PenStyle n
_ Located (Trail V2 n)
t) = forall t. TrailLike t => Located (Trail (V t) (N t)) -> t
trailLike Located (Trail V2 n)
t
turtlePathToStroke :: (Renderable (Path V2 n) b, TypeableFloat n) =>
TurtlePath n
-> QDiagram b V2 n Any
turtlePathToStroke :: forall n b.
(Renderable (Path V2 n) b, TypeableFloat n) =>
TurtlePath n -> QDiagram b V2 n Any
turtlePathToStroke t :: TurtlePath n
t@(TurtlePath (PenStyle Measure n
lineWidth_ Colour Double
lineColor_) Located (Trail V2 n)
_) = QDiagram b V2 n Any
d
where d :: QDiagram b V2 n Any
d = forall n a.
(InSpace V2 n a, Typeable n, Floating n, HasStyle a) =>
Colour Double -> a -> a
lc Colour Double
lineColor_ forall b c a. (b -> c) -> (a -> b) -> a -> c
.
forall a n.
(N a ~ n, HasStyle a, Typeable n) =>
Measure n -> a -> a
lw Measure n
lineWidth_ forall b c a. (b -> c) -> (a -> b) -> a -> c
.
forall n b.
(TypeableFloat n, Renderable (Path V2 n) b) =>
Located (Trail V2 n) -> QDiagram b V2 n Any
strokeLocTrail forall a b. (a -> b) -> a -> b
$ forall t n. (V t ~ V2, N t ~ n, TrailLike t) => TurtlePath n -> t
turtlePathToTrailLike TurtlePath n
t