//-------------------------------------------------------------------------
/*
Copyright (C) 2010-2019 EDuke32 developers and contributors
Copyright (C) 2019 sirlemonhead, Nuke.YKT
This file is part of PCExhumed.
PCExhumed is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License version 2
as published by the Free Software Foundation.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/
//-------------------------------------------------------------------------

#include "ns.h"
#include "player.h"

BEGIN_PS_NS

//---------------------------------------------------------------------------
//
//
//
//---------------------------------------------------------------------------

static DExhumedActor* feebtag(const DVector3& pos, sectortype* pSector, int nVal2, double deflen)
{
    DExhumedActor* pPickupActor = nullptr;

    auto startwall = pSector->walls.Data();
    int nWalls = pSector->walls.Size();

    int var_20 = nVal2 & 2;
    int var_14 = nVal2 & 1;

    while (1)
    {
        if (pSector != nullptr)
        {
            ExhumedSectIterator it(pSector);
            while (auto itActor = it.Next())
            {
                int nStat = itActor->spr.statnum;

                if (nStat >= 900 && !(itActor->spr.cstat & CSTAT_SPRITE_INVISIBLE))
                {
                    auto diff = itActor->spr.pos - pos;

                    if (diff.Z < 20 && diff.Z > -100)
                    {
                        double len = diff.XY().Length();

                        if (len < deflen && ((nStat != 950 && nStat != 949) || !(var_14 & 1)) && ((nStat != 912 && nStat != 913) || !(var_20 & 2)))
                        {
                            deflen = len;
                            pPickupActor = itActor;
                        }
                    }
                }
            }
        }

        nWalls--;
        if (nWalls < 0)
            return pPickupActor;

        pSector = startwall->nextSector();
        startwall++;
    }

    return pPickupActor;
}

//---------------------------------------------------------------------------
//
//
//
//---------------------------------------------------------------------------

enum
{
    kPickupProcess = 1,
    kPickupOnConsole = 2,
    kPickupDefaults = kPickupProcess | kPickupOnConsole,
    kPickupHealth = 4,
};

void doPlayerItemPickups(Player* const pPlayer)
{
    const auto pPlayerActor = pPlayer->pActor;
    const bool mplevel = (currentLevel->gameflags & LEVEL_EX_MULTI);
    const auto nFlags = (pPlayer->nMagic >= 1000) + 2 * (pPlayer->nHealth >= 800);
    const auto pPickupActor = feebtag(pPlayerActor->spr.pos, pPlayerActor->sector(), nFlags, 48);

    if (pPickupActor != nullptr && pPickupActor->spr.statnum >= 900)
    {
        // item lotags start at 6 (1-5 reserved?) so 0-offset them
        const int statBase = pPickupActor->spr.statnum - 900;
        const int itemtype = statBase - 6;

        if (itemtype <= 54)
        {
            int tintRed = 0;
            int tintGreen = 16;
            int nSound = -1;
            int nHealth = 0;
            int pickFlag = 0;
            int var_40;

            switch (itemtype)
            {
            case 0: // Speed Loader
                if (AddAmmo(pPlayer->nPlayer, 1, pPickupActor->spr.hitag))
                {
                    nSound = StaticSound[kSoundAmmoPickup];
                    pickFlag |= kPickupDefaults;
                }
                break;

            case 1: // Fuel Canister
                if (AddAmmo(pPlayer->nPlayer, 3, pPickupActor->spr.hitag))
                {
                    nSound = StaticSound[kSoundAmmoPickup];
                    pickFlag |= kPickupDefaults;
                }
                break;

            case 2: // M - 60 Ammo Belt
                if (AddAmmo(pPlayer->nPlayer, 2, pPickupActor->spr.hitag))
                {
                    nSound = StaticSound[kSoundAmmoPickup];
                    CheckClip(pPlayer->nPlayer);
                    pickFlag |= kPickupDefaults;
                }
                break;

            case 3: // Grenade
            case 21:
            case 49:
                if (AddAmmo(pPlayer->nPlayer, 4, 1))
                {
                    nSound = StaticSound[kSoundAmmoPickup];

                    if (!(pPlayer->nPlayerWeapons & 0x10))
                    {
                        pPlayer->nPlayerWeapons |= 0x10;
                        SetNewWeaponIfBetter(pPlayer->nPlayer, 4);
                    }

                    if (statBase == 55)
                    {
                        pPickupActor->spr.cstat = CSTAT_SPRITE_INVISIBLE;
                        DestroyItemAnim(pPickupActor);
                        pickFlag |= kPickupOnConsole;
                    }
                    else
                    {
                        pickFlag |= kPickupDefaults;
                    }
                }
                break;

            case 4: // Pickable item
            case 9: // Pickable item
            case 10: // Reserved
            case 18:
            case 25:
            case 28:
            case 29:
            case 30:
            case 33:
            case 34:
            case 35:
            case 36:
            case 37:
            case 38:
            case 45:
            case 52:
                pickFlag |= kPickupDefaults;
                break;

            case 5: // Map
                GrabMap();
                pickFlag |= kPickupDefaults;
                break;

            case 6: // Berry Twig
                if (pPickupActor->spr.hitag != 0) 
                {
                    nSound = 20;
                    nHealth = 40;
                    pickFlag |= kPickupHealth;
                }
                break;

            case 7: // Blood Bowl
                nHealth = 160;
                pickFlag |= kPickupHealth;
                break;

            case 8: // Cobra Venom Bowl
                nHealth = -200;
                pickFlag |= kPickupHealth;
                break;

            case 11: // Bubble Nest
            {
                pPlayer->nAir += 10;
                if (pPlayer->nAir > 100) {
                    pPlayer->nAir = 100; // TODO - constant
                }

                if (pPlayer->nBreathTimer < 89)
                {
                    D3PlayFX(StaticSound[kSound13], pPlayerActor);
                }

                pPlayer->nBreathTimer = 90;
                break;
            }

            case 12: // Still Beating Heart
            {
                if (GrabItem(pPlayer->nPlayer, kItemHeart)) {
                    pickFlag |= kPickupDefaults;
                }

                break;
            }

            case 13: // Scarab amulet(Invicibility)
            {
                if (GrabItem(pPlayer->nPlayer, kItemInvincibility)) {
                    pickFlag |= kPickupDefaults;
                }

                break;
            }

            case 14: // Severed Slave Hand(double damage)
            {
                if (GrabItem(pPlayer->nPlayer, kItemDoubleDamage)) {
                    pickFlag |= kPickupDefaults;
                }

                break;
            }

            case 15: // Unseen eye(Invisibility)
            {
                if (GrabItem(pPlayer->nPlayer, kItemInvisibility)) {
                    pickFlag |= kPickupDefaults;
                }

                break;
            }

            case 16: // Torch
            {
                if (GrabItem(pPlayer->nPlayer, kItemTorch)) {
                    pickFlag |= kPickupDefaults;
                }

                break;
            }

            case 17: // Sobek Mask
            {
                if (GrabItem(pPlayer->nPlayer, kItemMask)) {
                    pickFlag |= kPickupDefaults;
                }

                break;
            }

            case 19: // Extra Life
            {
                nSound = -1;

                if (pPlayer->nLives >= kMaxPlayerLives) {
                    break;
                }

                pPlayer->nLives++;

                tintGreen = 32;
                tintRed = 32;
                pickFlag |= kPickupDefaults;
            }

            // FIXME - lots of repeated code from here down!!
            case 20: // sword pickup??
            {
                var_40 = 0;
                int ebx = 0;

                // loc_1B75D
                int var_18 = 1 << var_40;

                int weapons = pPlayer->nPlayerWeapons;

                if (weapons & var_18)
                {
                    if (mplevel)
                    {
                        AddAmmo(pPlayer->nPlayer, WeaponInfo[var_40].nAmmoType, ebx);
                    }
                }
                else
                {
                    weapons = var_40;

                    SetNewWeaponIfBetter(pPlayer->nPlayer, weapons);

                    pPlayer->nPlayerWeapons |= var_18;

                    AddAmmo(pPlayer->nPlayer, WeaponInfo[weapons].nAmmoType, ebx);

                    nSound = StaticSound[kSound72];
                }

                if (var_40 == 2) {
                    CheckClip(pPlayer->nPlayer);
                }

                if (statBase <= 50) {
                    pickFlag |= kPickupDefaults;
                }

                pPickupActor->spr.cstat = CSTAT_SPRITE_INVISIBLE;
                DestroyItemAnim(pPickupActor);
                ////
                        // loc_1BA74: - repeated block, see in default case
                if (pPlayer->nPlayer == nLocalPlayer)
                {
                    if (nItemText[statBase] > -1 && nTotalPlayers == 1)
                    {
                        pickupMessage(statBase);
                    }

                    TintPalette(tintRed * 4, tintGreen * 4, 0);

                    if (nSound > -1)
                    {
                        PlayLocalSound(nSound, 0);
                    }
                }

                break;
                /////
            }

            case 22: // .357 Magnum Revolver
            case 46:
            {
                var_40 = 1;
                int ebx = 6;

                // loc_1B75D
                int var_18 = 1 << var_40;

                int weapons = pPlayer->nPlayerWeapons;

                if (weapons & var_18)
                {
                    if (mplevel)
                    {
                        AddAmmo(pPlayer->nPlayer, WeaponInfo[var_40].nAmmoType, ebx);
                    }
                }
                else
                {
                    weapons = var_40;

                    SetNewWeaponIfBetter(pPlayer->nPlayer, weapons);

                    pPlayer->nPlayerWeapons |= var_18;

                    AddAmmo(pPlayer->nPlayer, WeaponInfo[weapons].nAmmoType, ebx);

                    nSound = StaticSound[kSound72];
                }

                if (var_40 == 2) {
                    CheckClip(pPlayer->nPlayer);
                }

                if (statBase <= 50) {
                    pickFlag |= kPickupDefaults;
                }

                pPickupActor->spr.cstat = CSTAT_SPRITE_INVISIBLE;
                DestroyItemAnim(pPickupActor);
                ////
                        // loc_1BA74: - repeated block, see in default case
                if (pPlayer->nPlayer == nLocalPlayer)
                {
                    if (nItemText[statBase] > -1 && nTotalPlayers == 1)
                    {
                        pickupMessage(statBase);
                    }

                    TintPalette(tintRed * 4, tintGreen * 4, 0);

                    if (nSound > -1)
                    {
                        PlayLocalSound(nSound, 0);
                    }
                }

                break;
                /////
            }

            case 23: // M - 60 Machine Gun
            case 47:
            {
                var_40 = 2;
                int ebx = 24;

                // loc_1B75D
                int var_18 = 1 << var_40;

                int weapons = pPlayer->nPlayerWeapons;

                if (weapons & var_18)
                {
                    if (mplevel)
                    {
                        AddAmmo(pPlayer->nPlayer, WeaponInfo[var_40].nAmmoType, ebx);
                    }
                }
                else
                {
                    weapons = var_40;

                    SetNewWeaponIfBetter(pPlayer->nPlayer, weapons);

                    pPlayer->nPlayerWeapons |= var_18;

                    AddAmmo(pPlayer->nPlayer, WeaponInfo[weapons].nAmmoType, ebx);

                    nSound = StaticSound[kSound72];
                }

                if (var_40 == 2) {
                    CheckClip(pPlayer->nPlayer);
                }

                if (statBase <= 50) {
                    pickFlag |= kPickupDefaults;
                }

                pPickupActor->spr.cstat = CSTAT_SPRITE_INVISIBLE;
                DestroyItemAnim(pPickupActor);
                ////
                        // loc_1BA74: - repeated block, see in default case
                if (pPlayer->nPlayer == nLocalPlayer)
                {
                    if (nItemText[statBase] > -1 && nTotalPlayers == 1)
                    {
                        pickupMessage(statBase);
                    }

                    TintPalette(tintRed * 4, tintGreen * 4, 0);

                    if (nSound > -1)
                    {
                        PlayLocalSound(nSound, 0);
                    }
                }

                break;
                /////
            }

            case 24: // Flame Thrower
            case 48:
            {
                var_40 = 3;
                int ebx = 100;

                // loc_1B75D
                int var_18 = 1 << var_40;

                int weapons = pPlayer->nPlayerWeapons;

                if (weapons & var_18)
                {
                    if (mplevel)
                    {
                        AddAmmo(pPlayer->nPlayer, WeaponInfo[var_40].nAmmoType, ebx);
                    }
                }
                else
                {
                    weapons = var_40;

                    SetNewWeaponIfBetter(pPlayer->nPlayer, weapons);

                    pPlayer->nPlayerWeapons |= var_18;

                    AddAmmo(pPlayer->nPlayer, WeaponInfo[weapons].nAmmoType, ebx);

                    nSound = StaticSound[kSound72];
                }

                if (var_40 == 2) {
                    CheckClip(pPlayer->nPlayer);
                }

                if (statBase <= 50) {
                    pickFlag |= kPickupDefaults;
                }

                pPickupActor->spr.cstat = CSTAT_SPRITE_INVISIBLE;
                DestroyItemAnim(pPickupActor);
                ////
                        // loc_1BA74: - repeated block, see in default case
                if (pPlayer->nPlayer == nLocalPlayer)
                {
                    if (nItemText[statBase] > -1 && nTotalPlayers == 1)
                    {
                        pickupMessage(statBase);
                    }

                    TintPalette(tintRed * 4, tintGreen * 4, 0);

                    if (nSound > -1)
                    {
                        PlayLocalSound(nSound, 0);
                    }
                }

                break;
                /////
            }

            case 26: // Cobra Staff
            case 50:
            {
                var_40 = 5;
                int ebx = 20;

                // loc_1B75D
                int var_18 = 1 << var_40;

                int weapons = pPlayer->nPlayerWeapons;

                if (weapons & var_18)
                {
                    if (mplevel)
                    {
                        AddAmmo(pPlayer->nPlayer, WeaponInfo[var_40].nAmmoType, ebx);
                    }
                }
                else
                {
                    weapons = var_40;

                    SetNewWeaponIfBetter(pPlayer->nPlayer, weapons);

                    pPlayer->nPlayerWeapons |= var_18;

                    AddAmmo(pPlayer->nPlayer, WeaponInfo[weapons].nAmmoType, ebx);

                    nSound = StaticSound[kSound72];
                }

                if (var_40 == 2) {
                    CheckClip(pPlayer->nPlayer);
                }

                if (statBase <= 50) {
                    pickFlag |= kPickupDefaults;
                }

                pPickupActor->spr.cstat = CSTAT_SPRITE_INVISIBLE;
                DestroyItemAnim(pPickupActor);
                ////
                        // loc_1BA74: - repeated block, see in default case
                if (pPlayer->nPlayer == nLocalPlayer)
                {
                    if (nItemText[statBase] > -1 && nTotalPlayers == 1)
                    {
                        pickupMessage(statBase);
                    }

                    TintPalette(tintRed * 4, tintGreen * 4, 0);

                    if (nSound > -1)
                    {
                        PlayLocalSound(nSound, 0);
                    }
                }

                break;
                /////
            }

            case 27: // Eye of Ra Gauntlet
            case 51:
            {
                var_40 = 6;
                int ebx = 2;

                // loc_1B75D
                int var_18 = 1 << var_40;

                int weapons = pPlayer->nPlayerWeapons;

                if (weapons & var_18)
                {
                    if (mplevel)
                    {
                        AddAmmo(pPlayer->nPlayer, WeaponInfo[var_40].nAmmoType, ebx);
                    }
                }
                else
                {
                    weapons = var_40;

                    SetNewWeaponIfBetter(pPlayer->nPlayer, weapons);

                    pPlayer->nPlayerWeapons |= var_18;

                    AddAmmo(pPlayer->nPlayer, WeaponInfo[weapons].nAmmoType, ebx);

                    nSound = StaticSound[kSound72];
                }

                if (var_40 == 2) {
                    CheckClip(pPlayer->nPlayer);
                }

                if (statBase <= 50) {
                    pickFlag |= kPickupDefaults;
                }

                pPickupActor->spr.cstat = CSTAT_SPRITE_INVISIBLE;
                DestroyItemAnim(pPickupActor);
                ////
                        // loc_1BA74: - repeated block, see in default case
                if (pPlayer->nPlayer == nLocalPlayer)
                {
                    if (nItemText[statBase] > -1 && nTotalPlayers == 1)
                    {
                        pickupMessage(statBase);
                    }

                    TintPalette(tintRed * 4, tintGreen * 4, 0);

                    if (nSound > -1)
                    {
                        PlayLocalSound(nSound, 0);
                    }
                }

                break;
                /////
            }

            case 31: // Cobra staff ammo
            {
                if (AddAmmo(pPlayer->nPlayer, 5, 1)) {
                    nSound = StaticSound[kSoundAmmoPickup];
                    pickFlag |= kPickupDefaults;
                }

                break;
            }

            case 32: // Raw Energy
            {
                if (AddAmmo(pPlayer->nPlayer, 6, pPickupActor->spr.hitag)) {
                    nSound = StaticSound[kSoundAmmoPickup];
                    pickFlag |= kPickupDefaults;
                }

                break;
            }

            case 39: // Power key
            case 40: // Time key
            case 41: // War key
            case 42: // Earth key
            {
                int keybit = 4096 << (itemtype - 39);

                nSound = -1;

                if (!(pPlayer->keys & keybit))
                {
                    pPlayer->keys |= keybit;

                    if (nTotalPlayers > 1)
                    {
                        pickFlag |= kPickupOnConsole;
                    }
                    else
                    {
                        pickFlag |= kPickupDefaults;
                    }
                }

                break;
            }

            case 43: // Magical Essence
            case 44: // ?
            {
                if (pPlayer->nMagic >= 1000) {
                    break;
                }

                nSound = StaticSound[kSoundMana1];

                pPlayer->nMagic += 100;
                if (pPlayer->nMagic >= 1000) {
                    pPlayer->nMagic = 1000;
                }

                pickFlag |= kPickupDefaults;
            }

            case 53: // Scarab (Checkpoint)
            {
                if (nLocalPlayer == pPlayer->nPlayer)
                {
                    pPickupActor->nIndex2++;
                    pPickupActor->nAction &= 0xEF;
                    pPickupActor->nIndex = 0;

                    ChangeActorStat(pPickupActor, 899);
                }

                SetSavePoint(pPlayer->nPlayer, pPlayerActor->spr.pos, pPlayerActor->sector(), pPlayerActor->spr.Angles.Yaw);
                break;
            }

            case 54: // Golden Sarcophagus (End Level)
            {
                if (!bInDemo)
                {
                    LevelFinished();
                }

                DestroyItemAnim(pPickupActor);
                DeleteActor(pPickupActor);
                break;
            }
            }

            if (pickFlag & kPickupHealth)
            {
                if (nHealth <= 0 || (!(nFlags & 2)))
                {
                    if (!pPlayer->invincibility || nHealth > 0)
                    {
                        pPlayer->nHealth += nHealth;

                        if (pPlayer->nHealth > 800)
                        {
                            pPlayer->nHealth = 800;
                        }
                        else if (pPlayer->nHealth < 0)
                        {
                            nSound = -1;
                            StartDeathSeq(pPlayer->nPlayer, 0);
                        }
                    }

                    if (statBase == 12)
                    {
                        pPickupActor->spr.hitag = 0;
                        pPickupActor->spr.picnum++;
                        ChangeActorStat(pPickupActor, 0);
                        pickFlag |= kPickupOnConsole;
                    }
                    else
                    {
                        if (statBase != 14)
                        {
                            nSound = 21;
                        }
                        else
                        {
                            tintRed = tintGreen;
                            nSound = 22;
                            tintGreen = 0;
                        }

                        pickFlag |= kPickupDefaults;
                    }
                }
            }

            if (pickFlag & kPickupProcess)
            {
                if (!mplevel || (statBase >= 25 && (statBase <= 25 || statBase == 50)))
                {
                    // If this is an anim we need to properly destroy it so we need to do some proper detection and not wild guesses.
                    if (pPickupActor->nRun == pPickupActor->nDamage && pPickupActor->nRun != 0 && pPickupActor->nPhase == ITEM_MAGIC)
                    {
                        DestroyAnim(pPickupActor);
                    }
                    else
                    {
                        DeleteActor(pPickupActor);
                    }
                }
                else
                {
                    StartRegenerate(pPickupActor);
                }
            }

            if (pickFlag & kPickupOnConsole)
            {
                if (pPlayer->nPlayer == nLocalPlayer)
                {
                    if (nItemText[statBase] > -1 && nTotalPlayers == 1)
                        pickupMessage(statBase);

                    if (nSound > -1)
                        PlayLocalSound(nSound, 0);

                    TintPalette(tintRed * 4, tintGreen * 4, 0);
                }
            }
        }
    }
}

END_PS_NS