From 9af7dbda56e2f15cd779f829092f3018d511a940 Mon Sep 17 00:00:00 2001
From: jgk <justin.g.kruger@gmail.com>
Date: Sat, 24 Apr 2021 18:25:09 +0200
Subject: [PATCH] Randomise enemies and floor items

---
 src/Dodge/Room/Procedural.hs | 27 ++++++++++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/src/Dodge/Room/Procedural.hs b/src/Dodge/Room/Procedural.hs
index 47a62b523..e2a07d5eb 100644
--- a/src/Dodge/Room/Procedural.hs
+++ b/src/Dodge/Room/Procedural.hs
@@ -6,13 +6,17 @@ module Dodge.Room.Procedural
     , roomRectAutoLinks
     , testRoom
     ) where
+import Dodge.Data
 import Dodge.Room.Data
 import Dodge.Room.Placement
 import Dodge.Room.Link
+import Dodge.Item.Consumable
+import Dodge.Item.Weapon
 import Dodge.RandomHelp
 import Dodge.LevelGen
 import Dodge.LevelGen.Data
 import Dodge.Creature
+import Dodge.Default
 import Geometry
 
 import Data.List (nub,nubBy,sortBy)
@@ -199,12 +203,29 @@ fillNothingPlacement pst r =
     replaceNothingWith x (ps:pss) = ps : replaceNothingWith x pss
     replaceNothingWith _ [] = []
 
+fillNothingPlacements :: [PSType] -> Room -> Room
+fillNothingPlacements pst r = foldr fillNothingPlacement r pst
+
+shufflePlacements :: RandomGen g => Room -> State g Room
+shufflePlacements r = do
+    newPSs <- shuffle $ _rmPS r
+    return $ r & rmPS .~ newPSs
+
+putDefaultFlIt :: Item -> PSType
+putDefaultFlIt itm = PutFlIt $ defaultFlIt { _flItRot = 0, _flIt = itm }
+
 testRoom :: RandomGen g => State g Room
 testRoom = do
     corners <- replicateM 4 . join $ takeOne [fourthWall 100, fourthCornerWall 100]
---    corners <- replicateM 4 (fourthCornerWall 100)
-    randomiseAllLinks 
-        . fillNothingPlacement (PutCrit chaseCrit)
+    nItms <- state $ randomR (1,2)
+    itms <- takeN nItms $ fmap putDefaultFlIt $ [autoRadar,autoSonar,remoteLauncher,jetPack,blinkGun] ++ replicate 5 (medkit 500)
+    nCrits <- state $ randomR (1,3)
+    crits <- takeN nCrits $ fmap PutCrit $ [spreadGunCrit,pistolCrit,autoCrit,armourChaseCrit]
+        ++ replicate 20 chaseCrit
+    randomiseAllLinks =<< 
+        ( fmap (fillNothingPlacements $ crits ++ itms)
+        . shufflePlacements
         . foldr1 combineRooms 
         $ zipWith (\r a -> shiftRoomBy ((0,0),a) r) corners [0,pi/2,pi,3*pi/2]
+        )