diff options
author | Griffin Smith <root@gws.fyi> | 2019-12-25T00·40-0500 |
---|---|---|
committer | Griffin Smith <root@gws.fyi> | 2019-12-30T16·24-0500 |
commit | 6f427fe4d6ba9a03f122d15839298040a7cfb925 (patch) | |
tree | 5ab6c11d7b1f4109e1d196f9d1135cbb4bd828f6 /src/Xanthous/Generators/Util.hs | |
parent | 13516911366a484ee5484166520133e056010515 (diff) |
Fix circle rendering, add filled circle
Make raster circle rendering use the Rasterific package instead of attempting desperately to hand-roll it, and add a method for generating filled circles.
Diffstat (limited to 'src/Xanthous/Generators/Util.hs')
-rw-r--r-- | src/Xanthous/Generators/Util.hs | 49 |
1 files changed, 40 insertions, 9 deletions
diff --git a/src/Xanthous/Generators/Util.hs b/src/Xanthous/Generators/Util.hs index 8fd04c0b9326..2c041149d900 100644 --- a/src/Xanthous/Generators/Util.hs +++ b/src/Xanthous/Generators/Util.hs @@ -1,4 +1,6 @@ {-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE QuantifiedConstraints #-} +{-# LANGUAGE AllowAmbiguousTypes #-} -------------------------------------------------------------------------------- module Xanthous.Generators.Util ( MCells @@ -13,18 +15,22 @@ module Xanthous.Generators.Util , regions , fillAll , fillAllM + , fromPoints + , fromPointsM ) where -------------------------------------------------------------------------------- -import Xanthous.Prelude hiding (Foldable, toList, for_) -import Data.Array.ST -import Data.Array.Unboxed -import Control.Monad.ST -import Control.Monad.Random -import Data.Monoid -import Data.Foldable (Foldable, toList, for_) +import Xanthous.Prelude hiding (Foldable, toList, for_) +import Data.Array.ST +import Data.Array.Unboxed +import Control.Monad.ST +import Control.Monad.Random +import Data.Monoid +import Data.Foldable (Foldable, toList, for_) +import qualified Data.Set as Set +import Data.Semigroup.Foldable -------------------------------------------------------------------------------- -import Xanthous.Util (foldlMapM') -import Xanthous.Data (Dimensions, width, height) +import Xanthous.Util (foldlMapM', maximum1, minimum1) +import Xanthous.Data (Dimensions, width, height) -------------------------------------------------------------------------------- type MCells s = STUArray s (Word, Word) Bool @@ -184,3 +190,28 @@ fillAll ixes a = accum (const fst) a $ (, (False, ())) <$> toList ixes fillAllM :: (MArray a Bool m, Ix i, Foldable f) => f i -> a i Bool -> m () fillAllM ixes a = for_ ixes $ \i -> writeArray a i False + +fromPoints + :: forall a f i. + ( IArray a Bool + , Ix i + , Functor f + , Foldable1 f + ) + => f (i, i) + -> a (i, i) Bool +fromPoints points = + let pts = Set.fromList $ toList points + dims = ( (minimum1 $ fst <$> points, minimum1 $ snd <$> points) + , (maximum1 $ fst <$> points, maximum1 $ snd <$> points) + ) + in array dims $ range dims <&> \i -> (i, i `member` pts) + +fromPointsM + :: (MArray a Bool m, Ix i, Element f ~ i, MonoFoldable f) + => NonNull f + -> m (a i Bool) +fromPointsM points = do + arr <- newArray (minimum points, maximum points) False + fillAllM (otoList points) arr + pure arr |