mirror of
https://github.com/reanimate/reanimate.git
synced 2026-09-14 09:32:22 +00:00
77 lines
5.7 KiB
HTML
77 lines
5.7 KiB
HTML
<html>
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
|
|
<style type="text/css">
|
|
span.lineno { color: white; background: #aaaaaa; border-right: solid white 12px }
|
|
span.nottickedoff { background: yellow}
|
|
span.istickedoff { background: white }
|
|
span.tickonlyfalse { margin: -1px; border: 1px solid #f20913; background: #f20913 }
|
|
span.tickonlytrue { margin: -1px; border: 1px solid #60de51; background: #60de51 }
|
|
span.funcount { font-size: small; color: orange; z-index: 2; position: absolute; right: 20 }
|
|
span.decl { font-weight: bold }
|
|
span.spaces { background: white }
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<pre>
|
|
<span class="decl"><span class="nottickedoff">never executed</span> <span class="tickonlytrue">always true</span> <span class="tickonlyfalse">always false</span></span>
|
|
</pre>
|
|
<pre>
|
|
<span class="lineno"> 1 </span>module Reanimate.Math.Triangulate
|
|
<span class="lineno"> 2 </span> ( Triangulation
|
|
<span class="lineno"> 3 </span> , edgesToTriangulation
|
|
<span class="lineno"> 4 </span> , edgesToTriangulationM
|
|
<span class="lineno"> 5 </span> , trianglesToTriangulation
|
|
<span class="lineno"> 6 </span> , trianglesToTriangulationM
|
|
<span class="lineno"> 7 </span> )
|
|
<span class="lineno"> 8 </span>where
|
|
<span class="lineno"> 9 </span>
|
|
<span class="lineno"> 10 </span>import Control.Monad
|
|
<span class="lineno"> 11 </span>import qualified Data.IntSet as ISet
|
|
<span class="lineno"> 12 </span>import qualified Data.Vector as V
|
|
<span class="lineno"> 13 </span>import qualified Data.Vector.Mutable as MV
|
|
<span class="lineno"> 14 </span>import Control.Monad.ST
|
|
<span class="lineno"> 15 </span>
|
|
<span class="lineno"> 16 </span>-- Max edges: n-2
|
|
<span class="lineno"> 17 </span>-- Each edge is represented twice: 2n-4
|
|
<span class="lineno"> 18 </span>-- Flat structure:
|
|
<span class="lineno"> 19 </span>-- edges :: V.Vector Int -- max length (2n-4)
|
|
<span class="lineno"> 20 </span>-- offsets :: V.Vector Int -- length n
|
|
<span class="lineno"> 21 </span>-- Combine the two vectors? < n => offsets, >= n => edges?
|
|
<span class="lineno"> 22 </span>type Triangulation = V.Vector [Int]
|
|
<span class="lineno"> 23 </span>
|
|
<span class="lineno"> 24 </span>-- FIXME: Move to Common or a Triangulation module
|
|
<span class="lineno"> 25 </span>-- O(n)
|
|
<span class="lineno"> 26 </span>edgesToTriangulation :: Int -> [(Int, Int)] -> Triangulation
|
|
<span class="lineno"> 27 </span><span class="decl"><span class="nottickedoff">edgesToTriangulation size edges = runST $ do</span>
|
|
<span class="lineno"> 28 </span><span class="spaces"> </span><span class="nottickedoff">v <- edgesToTriangulationM size edges</span>
|
|
<span class="lineno"> 29 </span><span class="spaces"> </span><span class="nottickedoff">V.unsafeFreeze v</span></span>
|
|
<span class="lineno"> 30 </span>
|
|
<span class="lineno"> 31 </span>edgesToTriangulationM :: Int -> [(Int, Int)] -> ST s (V.MVector s [Int])
|
|
<span class="lineno"> 32 </span><span class="decl"><span class="nottickedoff">edgesToTriangulationM size edges = do</span>
|
|
<span class="lineno"> 33 </span><span class="spaces"> </span><span class="nottickedoff">v <- MV.replicate size []</span>
|
|
<span class="lineno"> 34 </span><span class="spaces"> </span><span class="nottickedoff">forM_ edges $ \(e1, e2) -> do</span>
|
|
<span class="lineno"> 35 </span><span class="spaces"> </span><span class="nottickedoff">MV.modify v (e1 :) e2</span>
|
|
<span class="lineno"> 36 </span><span class="spaces"> </span><span class="nottickedoff">MV.modify v (e2 :) e1</span>
|
|
<span class="lineno"> 37 </span><span class="spaces"> </span><span class="nottickedoff">forM_ [0 .. size - 1] $ \i -> MV.modify v (ISet.toList . ISet.fromList) i</span>
|
|
<span class="lineno"> 38 </span><span class="spaces"> </span><span class="nottickedoff">return v</span></span>
|
|
<span class="lineno"> 39 </span>
|
|
<span class="lineno"> 40 </span>trianglesToTriangulation :: Int -> V.Vector (Int, Int, Int) -> Triangulation
|
|
<span class="lineno"> 41 </span><span class="decl"><span class="nottickedoff">trianglesToTriangulation size edges = runST $ do</span>
|
|
<span class="lineno"> 42 </span><span class="spaces"> </span><span class="nottickedoff">v <- trianglesToTriangulationM size edges</span>
|
|
<span class="lineno"> 43 </span><span class="spaces"> </span><span class="nottickedoff">V.unsafeFreeze v</span></span>
|
|
<span class="lineno"> 44 </span>
|
|
<span class="lineno"> 45 </span>trianglesToTriangulationM
|
|
<span class="lineno"> 46 </span> :: Int -> V.Vector (Int, Int, Int) -> ST s (V.MVector s [Int])
|
|
<span class="lineno"> 47 </span><span class="decl"><span class="nottickedoff">trianglesToTriangulationM size trigs = do</span>
|
|
<span class="lineno"> 48 </span><span class="spaces"> </span><span class="nottickedoff">v <- MV.replicate size []</span>
|
|
<span class="lineno"> 49 </span><span class="spaces"> </span><span class="nottickedoff">forM_ (V.toList trigs) $ \(a, b, c) -> do</span>
|
|
<span class="lineno"> 50 </span><span class="spaces"> </span><span class="nottickedoff">MV.modify v (\x -> b : c : x) a</span>
|
|
<span class="lineno"> 51 </span><span class="spaces"> </span><span class="nottickedoff">MV.modify v (\x -> a : c : x) b</span>
|
|
<span class="lineno"> 52 </span><span class="spaces"> </span><span class="nottickedoff">MV.modify v (\x -> a : b : x) c</span>
|
|
<span class="lineno"> 53 </span><span class="spaces"> </span><span class="nottickedoff">forM_ [0 .. size - 1] $ \i -> MV.modify v (ISet.toList . ISet.fromList) i</span>
|
|
<span class="lineno"> 54 </span><span class="spaces"> </span><span class="nottickedoff">return v</span></span>
|
|
|
|
</pre>
|
|
</body>
|
|
</html>
|