#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

sshort RoachSprite = -1;
sshort 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;
				}
			}
		}
	}
}