From 785dbba5ef8d1b0e97a507b0339fe5add4ab3abe Mon Sep 17 00:00:00 2001
From: justin <justin.g.kruger@gmail.com>
Date: Thu, 9 Jun 2022 12:30:34 +0100
Subject: [PATCH] Add RoomCluster ideas

---
 src/Dodge/Data.hs             |   3 +
 src/Dodge/Default/Room.hs     |   3 +
 src/Dodge/Floor.hs            |   3 -
 src/Dodge/RoomCluster.hs      |   1 +
 src/Dodge/RoomCluster/Data.hs |  15 +++++
 src/RandomHelp.hs             | 117 ++++++++++++++++++++++++++++++++++
 6 files changed, 139 insertions(+), 3 deletions(-)
 create mode 100644 src/Dodge/RoomCluster.hs
 create mode 100644 src/Dodge/RoomCluster/Data.hs
 create mode 100644 src/RandomHelp.hs

diff --git a/src/Dodge/Data.hs b/src/Dodge/Data.hs
index 5bc1b0118..9ad6d56be 100644
--- a/src/Dodge/Data.hs
+++ b/src/Dodge/Data.hs
@@ -23,7 +23,9 @@ module Dodge.Data
     , module Dodge.Equipment.Data
     , module MaybeHelp
     , module Dodge.Data.ItemAmount
+    , module Dodge.RoomCluster.Data
     ) where
+import Dodge.RoomCluster.Data
 import Dodge.Data.ItemAmount
 import Dodge.ShortShow
 import Dodge.Creature.State.Data
@@ -1469,6 +1471,7 @@ data Room = Room
     , _rmMParent    :: Maybe Int
     , _rmChildren   :: [Int]
     , _rmType       :: RoomType
+    , _rmClusterStatus :: ClusterStatus
     }
 data RoomLink = RoomLink
     { _rlType   :: S.Set RoomLinkType
diff --git a/src/Dodge/Default/Room.hs b/src/Dodge/Default/Room.hs
index 13c5363cb..69937d443 100644
--- a/src/Dodge/Default/Room.hs
+++ b/src/Dodge/Default/Room.hs
@@ -31,4 +31,7 @@ defaultRoom = Room
     , _rmMParent    = Nothing
     , _rmChildren   = []
     , _rmType       = DefaultRoomType
+    , _rmClusterStatus = defaultClusterStatus
     }
+defaultClusterStatus :: ClusterStatus
+defaultClusterStatus = ClusterStatus "defRoomClust" S.empty
diff --git a/src/Dodge/Floor.hs b/src/Dodge/Floor.hs
index cc3fefa1a..3807ee1f0 100644
--- a/src/Dodge/Floor.hs
+++ b/src/Dodge/Floor.hs
@@ -102,6 +102,3 @@ initialAnoTree = padSucWithDoors $ treeFromPost
 {- | A test level tree. -}
 initialRoomTree :: State StdGen (Tree (LabSubCompTree Room))
 initialRoomTree = mapM anoToRoomTree initialAnoTree
---initialRoomTree :: State StdGen (Tree Room)
---initialRoomTree = do
---    expandTree <$> mapM anoToRoomTree initialAnoTree
diff --git a/src/Dodge/RoomCluster.hs b/src/Dodge/RoomCluster.hs
new file mode 100644
index 000000000..2d455ed6e
--- /dev/null
+++ b/src/Dodge/RoomCluster.hs
@@ -0,0 +1 @@
+module Dodge.RoomCluster where
diff --git a/src/Dodge/RoomCluster/Data.hs b/src/Dodge/RoomCluster/Data.hs
new file mode 100644
index 000000000..567cd021b
--- /dev/null
+++ b/src/Dodge/RoomCluster/Data.hs
@@ -0,0 +1,15 @@
+{-# LANGUAGE TemplateHaskell #-}
+{-# LANGUAGE StrictData #-}
+module Dodge.RoomCluster.Data where
+import qualified Data.Set as S
+import Control.Lens
+
+data ClusterStatus = ClusterStatus
+    { _csName :: String
+    , _csLinks :: S.Set ClusterLink
+    }
+
+data ClusterLink = OnwardCluster | SideCluster
+  deriving (Ord,Eq,Enum,Show)
+
+makeLenses ''ClusterStatus
diff --git a/src/RandomHelp.hs b/src/RandomHelp.hs
new file mode 100644
index 000000000..a628e411d
--- /dev/null
+++ b/src/RandomHelp.hs
@@ -0,0 +1,117 @@
+{- Helpers for random generation.  -}
+module RandomHelp 
+    ( module System.Random
+    , module Control.Monad.State
+    , module RandomHelp
+    ) where
+import Geometry
+
+import System.Random
+import Control.Monad.State
+import Data.List
+
+randomRanges :: (Random a,RandomGen g) => [a] -> State g a
+randomRanges = join . takeOne . f
+  where 
+    f (x:y:ys) = state (randomR (x,y)) : f ys
+    f  _ = []
+
+takeOne :: RandomGen g => [a] -> State g a
+takeOne xs = state (randomR (0,length xs - 1)) >>= (\i -> return (xs !! i))
+
+takeOneWeighted :: (RandomGen g, Random b, Ord b, Num b) => [b] -> [a] -> State g a
+takeOneWeighted ws xs = state (randomR (0, sum ws)) >>= (\w -> return (xs !! i w ws))
+  where 
+    i y (z:zs) 
+        | y <= z = 0
+        | otherwise = 1 + i (y-z) zs
+    i _ _ = 0
+
+takeOneMore :: RandomGen g => ([a],[a]) -> State g ([a],[a])
+takeOneMore ( _,[]) = error "trying to takeOneMore from empty list"
+takeOneMore (xs,ys) = do
+    i <- state $ randomR (0,length ys - 1)
+    let (zs, w:ws) = splitAt i ys
+    return (w:xs, zs ++ ws)
+
+takeNMore :: RandomGen g => Int -> ([a],[a]) -> State g ([a],[a])
+takeNMore n p = foldl' (flip $ const (>>= takeOneMore)) (return p) [1..n]
+
+takeN :: RandomGen g => Int -> [a] -> State g [a]
+takeN 0  _ = return []
+takeN i xs = fst <$> takeNMore i ([],xs)
+
+-- | Randomly shuffle a list.
+shuffle :: RandomGen g => [a] -> State g [a]
+shuffle xs = do 
+    rands <- forM [0..length xs-1] $ \i -> state $ randomR (0,i)
+    let f ys rand = let (as,b:bs) = splitAt rand ys
+                    in  (as ++ bs, b)
+    let (_,zs) = mapAccumR f xs rands
+    return zs
+
+-- | Randomly shuffle the tail of a list, not safe.
+shuffleTail :: RandomGen g => [a] -> State g [a]
+shuffleTail (x:xs) = (x :) <$> shuffle xs
+shuffleTail _ = undefined
+
+-- select elements from a list randomly
+-- each element has the same independent chance of being selected
+randomSelectionFromList :: RandomGen g => Float -> [a] -> State g [a]
+randomSelectionFromList = filterM . const . randProb
+
+randProb :: RandomGen g => Float -> State g Bool
+randProb p = do 
+    p1 <- state $ randomR (0,1)
+    return (p1 < p)
+
+randInCirc :: RandomGen g => Float -> State g Point2
+randInCirc maxRad = do 
+    rad <- state $ randomR (0,maxRad)
+    ang <- state $ randomR (0,2*pi)
+    return $ rad *.* unitVectorAtAngle ang
+
+randOnUnitSphere :: RandomGen g => State g Point3
+randOnUnitSphere = do
+    z <- state $ randomR (negate 1,1)
+    longitude <- state $ randomR (0, 2*pi)
+    let (V2 x y) = sqrt (1 - z^(2::Int)) *.* unitVectorAtAngle longitude
+    return (V3 x y z)
+
+randOnHemisphere :: RandomGen g => State g Point3
+randOnHemisphere = do
+    z <- state $ randomR (0,1)
+    longitude <- state $ randomR (0, 2*pi)
+    let (V2 x y) = sqrt (1 - z^(2::Int)) *.* unitVectorAtAngle longitude
+    return (V3 x y z)
+
+randInHemisphere :: RandomGen g => State g Point3
+randInHemisphere = do
+    p <- randOnHemisphere
+    r <- state $ randomR (0,1)
+    return $ r *.*.* p
+
+randInRect :: RandomGen g => Float -> Float -> State g Point2
+randInRect w h = do     
+    x <- state $ randomR (0,w)
+    y <- state $ randomR (0,h)
+    return (V2 x y)
+
+maybeTakeOne :: RandomGen g => [a] -> State g (Maybe a)
+maybeTakeOne [] = return Nothing
+maybeTakeOne xs = state (randomR (0,length xs - 1)) >>= (\i -> return (Just (xs !! i)))
+
+randsSpread :: RandomGen g => (Float,Float) -> Int -> State g [Float]
+randsSpread (a,b) i
+    | i <= 0    = error "tried to take <= 0 randsSpread"
+    | otherwise = zipWith (+) [a+x,a+2*x..] <$> replicateM i (state $ randomR (0,x))
+  where
+    x = (b-a)/fromIntegral i
+
+randsOnCirc :: RandomGen g => Int -> State g [Float]
+randsOnCirc = randsSpread (0,2*pi)
+--randsOnCirc i
+--    | i <= 0    = error "tried to take <= 0 randsOnCirc"
+--    | otherwise = zipWith (+) [x,2*x..] <$> replicateM i (state $ randomR (0,x))
+--  where
+--    x = 2*pi/fromIntegral i