{- |
The tree of rooms that make up a level. -}
module Dodge.Floor
    ( levx
    ) where
import Geometry
import Picture
import Dodge.Data
import Dodge.Room
import Dodge.Creature.State.Data
import Dodge.Room.Procedural
import Dodge.Room.RoadBlock
import Dodge.Room.Data
import Dodge.Room.Link
import Dodge.Room.Door
import Dodge.Room.Branch
import Dodge.Room.Boss
import Dodge.Room.Airlock
import Dodge.Room.LongDoor
import Dodge.Room.NoNeedWeapon
import Dodge.Base
import Dodge.Layout
import Dodge.Layout.Tree.Polymorphic
import Dodge.Layout.Tree.Either
import Dodge.Layout.Tree.GenerateStructure
import Dodge.Layout.Tree.Annotate
import Dodge.Layout.Tree.Shift
import Dodge.Creature
import Dodge.SoundLogic
import Dodge.RandomHelp
import Dodge.LightSources
import Dodge.LevelGen.Data
import Dodge.LevelGen.SwarmPlacement
import Dodge.Item.Weapon

import Data.Tree
import Data.Maybe (fromJust,isNothing)
import Control.Lens
import Control.Monad.State
import Control.Monad.Loops
import System.Random
{- |
A test level tree. -}
roomTreex :: RandomGen g => State g (Maybe [Room])
roomTreex = do
    struct' <- aTreeStrut
--    let struct = treeFromPost [[SpecificRoom $ fmap (pure . Right) pistolerRoom]] [EndRoom]
    let struct = treeFromPost [[Corridor,SpecificRoom $ fmap (pure . Right) randomFourCornerRoom]] [EndRoom]
    let t' = padCorridors struct
        t = treeFromTrunk 
            [[StartRoom]
            ,[Corridor]
            ,[SpecificRoom . pure . pure . Right $ roomGlassOctogon 400 
                & rmPS %~ ([swarmPS 0 (x,y) 0 smallChaseCrit | x <- [-20,-19.5.. 20] , y <- [200,200.5..202] ]++)
             ]
            ,[Corridor]
            ,[Corridor]
            ,[SpecificRoom . fmap (pure . Right) $ twinSlowDoorChasers 30]
            ,[Corridor]
            ,[DoorAno]
            ,[SpecificRoom $ fmap (pure . Right) 
                $ centerVaultExplosiveExit 50]
            ,[SpecificRoom blockedCorridor]
            ,[OrAno [[DoorAno]
                    ,[Corridor]
                    ,[DoorNumAno 0,AirlockAno]] 
                    ]
            ,[FirstWeapon]
            ,[Corridor]
            ,[SpecificRoom $ fmap (pure . Right) randomFourCornerRoom]
            ,[Corridor]
            ,[OrAno 
                [[SpecificRoom $ branchRectWith $ fmap (fmap Left) armouredChasers]
                ,[BossAno $ addArmour launcherCrit & crHP +~ 800
                    & crState . crDropsOnDeath .~ DropSpecific [0] ]
                ]
             ]
            ,[Corridor]
            ,[SpecificRoom $ fmap (pure . Right) armouredCorridor]
            ,[Corridor]
            ,[TreasureAno [addArmour autoCrit,addArmour autoCrit] [launcher]]
            ] 
            t'
    fmap (shiftExpandTree . expandTreeBy id) $ mapM annoToRoomTree t

levx :: RandomGen g => State g [Room]
levx = untilJust roomTreex