Tweak tree level generation

src/Dodge/Floor.hs

@@ -34,11 +34,19 @@ import System.Random
 roomTreex :: RandomGen g => State g (Maybe [Room])
 roomTreex = do
     struct <- aTreeStrut
+    -- t <- randomPadCorridors $ fmap (const []) struct
     let t = padCorridors $ fmap (const []) struct
     fmap shiftExpandTree $ mapM annoToRoom t
 
+roomTreex' :: RandomGen g => State g (Maybe [Room])
+roomTreex' = do
+    struct <- aTreeStrut
+    let t = padCorridors $ fmap (const []) struct
+    return . shiftExpandTree $ fmap annoToRoom' t
+
+
 levx :: RandomGen g => State g [Room]
-levx = fmap fromJust $ iterateWhile isNothing roomTreex
+levx = untilJust roomTreex
 
 lev1 :: RandomGen g => State g (Tree Room)
 lev1 = do