#include "exhumed.h"
#include "engine.h"
#include "runlist.h"
#include "roach.h"
#include "typedefs.h"
#include "sequence.h"
#include "move.h"
#include "random.h"
#include "trigdat.h"
#include "bullet.h"
#include "items.h"
#include <assert.h>

#define kMaxRoach	100

int16_t RoachSprite = -1;
int16_t RoachCount = -1;

static actionSeq ActionSeq[] = {{ 24, 0 }, { 0, 0 }, { 0, 0 }, { 16, 0 }, { 8, 0 }, { 32, 1 }, { 42, 1 }};

struct Roach
{
    short nHealth;
    short field_2;
    short nType;
    short nSprite;
    short nTarget;
    short field_A;
    short field_C;
    short field_E;
};

Roach RoachList[kMaxRoach];


/* Kilmaat Sentry */

void InitRoachs()
{
    RoachCount = kMaxRoach;
    RoachSprite = 1;
}

// TODO - make EAX a bool type?
int BuildRoach(int eax, int nSprite, int x, int y, int z, short nSector, int angle)
{
    RoachCount--;
    if (RoachCount < 0) {
        return -1;
    }

    if (nSprite == -1)
    {
        nSprite = insertsprite(nSector, 105);
    }
    else
    {
        changespritestat(nSprite, 105);
        x = sprite[nSprite].x;
        y = sprite[nSprite].y;
        z = sector[sprite[nSprite].sectnum].floorz;
        angle = sprite[nSprite].ang;
    }

    assert(nSprite >= 0 && nSprite < kMaxSprites);

    sprite[nSprite].x = x;
    sprite[nSprite].y = y;
    sprite[nSprite].z = z;
    sprite[nSprite].shade = -12;
    sprite[nSprite].xoffset = 0;
    sprite[nSprite].yoffset = 0;
    sprite[nSprite].picnum = 1;
    sprite[nSprite].pal = sector[sprite[nSprite].sectnum].ceilingpal;
    sprite[nSprite].clipdist = 60;
    sprite[nSprite].ang = angle;
    sprite[nSprite].xrepeat = 40;
    sprite[nSprite].yrepeat = 40;
    sprite[nSprite].xvel = 0;
    sprite[nSprite].yvel = 0;
    sprite[nSprite].zvel = 0;
    sprite[nSprite].hitag = 0;
    sprite[nSprite].lotag = runlist_HeadRun() + 1;
    sprite[nSprite].extra = -1;

//	GrabTimeSlot(3);

    if (eax)
    {
        RoachList[RoachCount].nType = 0;
    }
    else
    {
        RoachList[RoachCount].nType = 1;
    }

    RoachList[RoachCount].nSprite = nSprite;
    RoachList[RoachCount].field_2 = 0;
    RoachList[RoachCount].field_C = 0;
    RoachList[RoachCount].nTarget = -1;
    RoachList[RoachCount].nHealth = 600;

    sprite[nSprite].owner = (sprite[nSprite].lotag - 1, RoachCount | 0x1C0000);
    RoachList[RoachCount].field_A = runlist_AddRunRec(NewRun, RoachCount | 0x1C0000);

    nCreaturesLeft++;

    return RoachCount | 0x1C0000;
}

void GoRoach(short nSprite)
{
    sprite[nSprite].xvel = (Sin(sprite[nSprite].ang + 512) >> 1) - (Sin(sprite[nSprite].ang + 512) >> 3);
    sprite[nSprite].yvel = (Sin(sprite[nSprite].ang) >> 1) - (Sin(sprite[nSprite].ang) >> 3);
}

void FuncRoach(int a, int b, int nRun)
{
    short nRoach = RunData[nRun].nVal;
    assert(nRoach >= 0 && nRoach < kMaxRoach);

    int var_24 = 0;

    short nSprite = RoachList[nRoach].nSprite;
    short nType = RoachList[nRoach].nType;

    int nDamage = b;

    int nMessage = a & 0x7F0000;

    switch (nMessage)
    {
        default:
        {
            DebugOut("unknown msg %d for Roach\n", a & 0x7F0000);
            return;
        }

        case 0x90000:
        {
            seq_PlotSequence(a & 0xFFFF, ActionSeq[nType].a + SeqOffsets[kSeqRoach], RoachList[nRoach].field_2, ActionSeq[nType].b);
            return;
        }

        case 0x0A0000: // fall through to next case
        {
            nDamage = runlist_CheckRadialDamage(nSprite);
        }
        case 0x80000:
        {
            if (nDamage)
            {
                if (RoachList[nRoach].nHealth <= 0) {
                    return;
                }

                RoachList[nRoach].nHealth -= nDamage;
                if (RoachList[nRoach].nHealth <= 0)
                {
                    sprite[nSprite].zvel = 0;
                    sprite[nSprite].yvel = 0;
                    sprite[nSprite].xvel = 0;
                    RoachList[nRoach].nHealth = 0;
                    sprite[nSprite].cstat &= 0x0FEFE;
                    nCreaturesLeft++;

                    if (nType < 5)
                    {
                        DropMagic(nSprite);
                        RoachList[nRoach].nType = 5;
                        RoachList[nRoach].field_2 = 0;
                    }
                }
                else
                {
                    short nSprite2 = a & 0xFFFF;
                    if (nSprite2 < 0) {
                        return;
                    }

                    if (sprite[nSprite2].statnum < 199) {
                        RoachList[nRoach].nTarget = nSprite2;
                    }

                    if (nType == 0)
                    {
                        RoachList[nRoach].nType = 2;
                        GoRoach(nSprite);
                        RoachList[nRoach].field_2 = 0;
                    }
                    else
                    {
                        if (!RandomSize(4))
                        {
                            RoachList[nRoach].nType = 4;
                            RoachList[nRoach].field_2 = 0;
                        }
                    }
                }
            }

            return;
        }

        case 0x20000:
        {
            Gravity(nSprite);

            int nSeq = SeqOffsets[kSeqRoach] + ActionSeq[RoachList[nRoach].nType].a;
            int var_28 = nSeq;

            sprite[nSprite].picnum = seq_GetSeqPicnum2(var_28, RoachList[nRoach].field_2);
            seq_MoveSequence(nSprite, var_28, RoachList[nRoach].field_2);

            RoachList[nRoach].field_2++;
            if (RoachList[nRoach].field_2 >= SeqSize[var_28])
            {
                var_24 = 1;
                RoachList[nRoach].field_2 = 0;
            }

            int nFlag = FrameFlag[SeqBase[nSeq] + RoachList[nRoach].field_2];
            short nTarget = RoachList[nRoach].nTarget;

            if (nType > 5) {
                return;
            }

            switch (nType)
            {
                case 0:
                {
                    if (RoachList[nRoach].field_2 == 1)
                    {
                        RoachList[nRoach].field_C--;
                        if (RoachList[nRoach].field_C <= 0)
                        {
                            RoachList[nRoach].field_C = RandomSize(6);
                        }
                        else
                        {
                            RoachList[nRoach].field_2 = 0;
                        }
                    }

                    if ((nRoach & 0xF) == (totalmoves & 0xF) && nTarget < 0)
                    {
                        short nTarget = FindPlayer(nSprite, 50);
                        if (nTarget >= 0)
                        {
                            RoachList[nRoach].nType = 2;
                            RoachList[nRoach].field_2 = 0;
                            RoachList[nRoach].nTarget = nTarget;
                            GoRoach(nSprite);
                        }
                    }

                    return;
                }

                case 1:
                {
                    // parltly the same as case 0...
                    if ((nRoach & 0xF) == (totalmoves & 0xF) && nTarget < 0)
                    {
                        short nTarget = FindPlayer(nSprite, 100);
                        if (nTarget >= 0)
                        {
                            RoachList[nRoach].nType = 2;
                            RoachList[nRoach].field_2 = 0;
                            RoachList[nRoach].nTarget = nTarget;
                            GoRoach(nSprite);
                        }
                    }

                    return;
                }

                case 2:
                {
                    if ((totalmoves & 0xF) == (nRoach & 0xF))
                    {
                        PlotCourseToSprite(nSprite, nTarget);
                        GoRoach(nSprite);
                    }

                    int nVal = MoveCreatureWithCaution(nSprite);

                    if ((nVal & 0x0C000) == 49152)
                    {
                        if ((nVal & 0x3FFF) == nTarget)
                        {
                            // repeated below
                            RoachList[nRoach].field_E = RandomSize(2) + 1;
                            RoachList[nRoach].nType = 3;

                            sprite[nSprite].xvel = sprite[nSprite].yvel = 0;
                            sprite[nSprite].ang = GetMyAngle(sprite[nTarget].x - sprite[nSprite].x, sprite[nTarget].y - sprite[nSprite].y);

                            RoachList[nRoach].field_2 = 0;
                        }
                    }
                    else if ((nVal & 0x0C000) == 32768)
                    {
                        sprite[nSprite].ang = (sprite[nSprite].ang + 256) & kAngleMask;
                        GoRoach(nSprite);
                    }
                    //else if ((nVal & 0x0C000) < 32768)
                    else
                    {
                        if (RoachList[nRoach].field_C)
                        {
                            RoachList[nRoach].field_C--;
                        }
                        else
                        {
                            // same as above
                            RoachList[nRoach].field_E = RandomSize(2) + 1;
                            RoachList[nRoach].nType = 3;

                            sprite[nSprite].xvel = sprite[nSprite].yvel = 0;
                            sprite[nSprite].ang = GetMyAngle(sprite[nTarget].x - sprite[nSprite].x, sprite[nTarget].y - sprite[nSprite].y);

                            RoachList[nRoach].field_2 = 0;
                        }
                    }

                    if (nTarget != -1 && !(sprite[nTarget].cstat & 0x101))
                    {
                        RoachList[nRoach].nType = 1;
                        RoachList[nRoach].field_2 = 0;
                        RoachList[nRoach].field_C = 100;
                        RoachList[nRoach].nTarget = -1;
                        sprite[nSprite].yvel = 0;
                        sprite[nSprite].xvel = 0;
                    }

                    return;
                }
                
                case 3:
                {
                    if (var_24)
                    {
                        RoachList[nRoach].field_E--;
                        if (RoachList[nRoach].field_E <= 0)
                        {
                            RoachList[nRoach].nType = 2;
                            GoRoach(nSprite);
                            RoachList[nRoach].field_2 = 0;
                            RoachList[nRoach].field_C = RandomSize(7);
                        }
                    }
                    else
                    {
                        if (nFlag & 0x80)
                        {
                            BuildBullet(nSprite, 13, 0, 0, -1, sprite[nSprite].ang, nTarget + 10000, 1);
                        }
                    }

                    return;
                }

                case 4:
                {
                    if (var_24)
                    {
                        RoachList[nRoach].nType = 2;
                        RoachList[nRoach].field_2 = 0;
                    }

                    return;
                }

                case 5:
                {
                    if (var_24)
                    {
                        sprite[nSprite].cstat = 0;
                        RoachList[nRoach].nType = 6;
                        RoachList[nRoach].field_2 = 0;
                    }

                    return;
                }
            }
        }
    }
}