//floradinn

freeslot("SPR_FLDN",
		 "S_FLORA1",
		 "S_FLORA2",
		 "S_FLORA3",
		 "S_FLORA4")

states[S_FLORA1] = {
	sprite = SPR_FLDN,
	frame = A,
	tics = 8,
	nextstate = S_FLORA2
}

states[S_FLORA2] = {
	sprite = SPR_FLDN,
	frame = B,
	tics = 8,
	nextstate = S_FLORA3
}

states[S_FLORA3] = {
	sprite = SPR_FLDN,
	frame = C,
	tics = 8,
	nextstate = S_FLORA4
}

states[S_FLORA4] = {
	sprite = SPR_FLDN,
	frame = D,
	tics = 8,
	nextstate = S_FLORA1
}

local FLORATRANSFORM_SFX = sfx_ssxt21 -- transform sound
local UNFLORA_SFX = sfx_ssxt22 -- untransform sound
local SPRITE = SPR_FLDN -- sprite
local STATE = S_FLORA1 -- state

///// Previous functions /////
local function FLORAifyPlayer(mo)
    if mo.sflori then return end
	if mo.weegee then return end
	if mo.kubik then return end
	if mo.goomb then return end
	if mo.goomb2 then return end
	if mo.sanic then return end
	if mo.roark then return end 
	if mo.bear5 then return end
	if mo.flora then return end 
	if mo.fredd then return end -- We already FLORA'd this player, why do it again?

    S_StartSound(mo, FLORATRANSFORM_SFX)
    mo.flora = true
    mo.alpha = 0

    if not (mo.overlay and mo.overlay.valid) then
        mo.overlay = P_SpawnMobjFromMobj(mo, mo.x, mo.y, mo.z, MT_THOK)
        mo.overlay.sprite = SPRITE
		mo.overlay.state = STATE
        mo.overlay.flags = MF_NOCLIPHEIGHT|MF_NOBLOCKMAP|MF_NOGRAVITY|MF_NOCLIP
        P_SetOrigin(mo.overlay, mo.x, mo.y, mo.z)
    end
end

local function UnFLORAify(mo)
    if not mo.flora then return end -- We already removed FLORA from this player, why do it again?

    S_StartSound(mo, UNFLORA_SFX)
    mo.flora = false
    mo.alpha = FU

    if mo.overlay and mo.overlay.valid then
        P_RemoveMobj(mo.overlay)
    end
end

local function ZCollide(mo1, mo2) --Z Collision checks
	if mo1.z > mo2.height+mo2.z then return false end
	if mo2.z > mo1.height+mo1.z then return false end
	return true
end

//// Linedef Execute ////

addHook("LinedefExecute", function(line, mo) --Transform via linedef execute
    if mo.valid and mo.health and mo.player and not mo.flora then
        FLORAifyPlayer(mo)
    end
end, "SSXTRAFL")

addHook("LinedefExecute", function(line, mo) --Untransform via linedef execute
    if mo and mo.valid and mo.health and mo.player and mo.flora then
        UnFLORAify(mo)
    end
end, "UNFLORA")

addHook("PostThinkFrame", function() --Weeegeeeeeee! Using PostThinkFrame to have exact consistency with the player.
    for p in players.iterate() do
        local mo = p.mo

		if not mo then continue end
		if not mo.valid then continue end
        if not mo.flora then continue end

        -- Ugh since some custom characters gets fancy when setting a sprite
        -- we're using a separated object instead.
        if mo.health then
            if mo.overlay and mo.overlay.valid then
                P_MoveOrigin(mo.overlay, mo.x, mo.y, mo.z)
                mo.overlay.scale = mo.scale
                mo.overlay.spritexscale = mo.spritexscale
                mo.overlay.spriteyscale = mo.spriteyscale
                mo.overlay.color = mo.color
            end
        elseif mo.overlay then
            UnFLORAify(mo)
        end
    end
end)

addHook("TeamSwitch", function(player, team)
	if team then return end
	if not player.mo then return end
	if not player.mo.valid then return end
	if not player.mo.flora then return end

	UnFLORAify(player.mo)
end)