hasufell
/
cga
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

DIAG: improve function naming and return value in kdSquares

This commit is contained in:
hasufell 2014-11-29 19:27:19 +01:00
parent 3b61d21549
commit fd4f135efa
No known key found for this signature in database
GPG Key ID: 220CD1C5BDEED020
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
Graphics/Diagram/Plotter.hs
View File

@ -188,25 +188,27 @@ kdSquares = Diag f
f _ (Object []) = mempty
f p (Object vt) =
mconcat
. fmap fromVertices
$ drawAll (kdTree vt Horizontal) (xDimension p, yDimension p)
. fmap (uncurry (~~))
$ kdLines (kdTree vt Horizontal) (xDimension p, yDimension p)
where
drawAll :: KDTree PT -> Square -> [[PT]]
drawAll (KTNode ln pt Horizontal rn) ((xmin, xmax), (ymin, ymax)) =
(\(x, _) -> [[p2 (x, ymin), p2 (x, ymax)]])
-- Gets all lines that make up the kdSquares. Every line is
-- described by two points, start and end respectively.
kdLines :: KDTree PT -> Square -> [(PT, PT)]
kdLines (KTNode ln pt Horizontal rn) ((xmin, xmax), (ymin, ymax)) =
(\(x, _) -> [(p2 (x, ymin), p2 (x, ymax))])
(unp2 pt)
++ drawAll ln ((xmin, x'), (ymin, ymax))
++ drawAll rn ((x', xmax), (ymin, ymax))
++ kdLines ln ((xmin, x'), (ymin, ymax))
++ kdLines rn ((x', xmax), (ymin, ymax))
where
(x', _) = unp2 pt
drawAll (KTNode ln pt Vertical rn) ((xmin, xmax), (ymin, ymax)) =
(\(_, y) -> [[p2 (xmin, y), p2 (xmax, y)]])
kdLines (KTNode ln pt Vertical rn) ((xmin, xmax), (ymin, ymax)) =
(\(_, y) -> [(p2 (xmin, y), p2 (xmax, y))])
(unp2 pt)
++ drawAll ln ((xmin, xmax), (ymin, y'))
++ drawAll rn ((xmin, xmax), (y', ymax))
++ kdLines ln ((xmin, xmax), (ymin, y'))
++ kdLines rn ((xmin, xmax), (y', ymax))
where
(_, y') = unp2 pt
drawAll _ _ = [[]]
kdLines _ _ = []
f _ _ = mempty