#include "../newobjects.h"
#include "../../Game/Box.h"
#include "../../Game/effects.h"

BITE_INFO yetiBiteR = { 12, 101, 19, 10 };
BITE_INFO yetiBiteL = { 12, 101, 19, 13 };

void InitialiseYeti(short itemNum)
{
	ITEM_INFO* item = &Items[itemNum];
	InitialiseCreature(itemNum);
	item->animNumber = Objects[item->objectNumber].animIndex + 19;
	item->frameNumber = Anims[item->animNumber].frameBase;
	item->currentAnimState = Anims[item->animNumber].currentAnimState;
}

void YetiControl(short itemNum)
{
	if (!CreatureActive(itemNum))
		return;

	ITEM_INFO* item = &Items[itemNum];
	CREATURE_INFO* yeti = (CREATURE_INFO*)item->data;
	AI_INFO info;
	short angle = 0, torso = 0, head = 0, tilt = 0;
	bool lara_alive;

	lara_alive = (LaraItem->hitPoints > 0);

	if (item->hitPoints <= 0)
	{
		if (item->currentAnimState != 8)
		{
			item->animNumber = Objects[item->objectNumber].animIndex + 31;
			item->frameNumber = Anims[item->animNumber].frameBase;
			item->currentAnimState = 8;
		}
	}
	else
	{
		CreatureAIInfo(item, &info);

		GetCreatureMood(item, &info, VIOLENT);
		CreatureMood(item, &info, VIOLENT);

		angle = CreatureTurn(item, yeti->maximumTurn);

		switch (item->currentAnimState)
		{
		case 2:
			if (info.ahead)
				head = info.angle;

			yeti->flags = 0;
			yeti->maximumTurn = 0;

			if (yeti->mood == ESCAPE_MOOD)
			{
				item->goalAnimState = 1;
			}
			else if (item->requiredAnimState)
			{
				item->goalAnimState = item->requiredAnimState;
			}
			else if (yeti->mood == BORED_MOOD)
			{
				if (GetRandomControl() < 0x100 || !lara_alive)
					item->goalAnimState = 7;
				else if (GetRandomControl() < 0x200)
					item->goalAnimState = 9;
				else if (GetRandomControl() < 0x300)
					item->goalAnimState = 3;
			}
			else if (info.ahead && info.distance < SQUARE(WALL_SIZE / 2) && GetRandomControl() < 0x4000)
			{
				item->goalAnimState = 4;
			}
			else if (info.ahead && info.distance < SQUARE(STEP_SIZE))
			{
				item->goalAnimState = 5;
			}
			else if (yeti->mood == STALK_MOOD)
			{
				item->goalAnimState = 3;
			}
			else
			{
				item->goalAnimState = 1;
			}
			break;
		case 7:
			if (info.ahead)
				head = info.angle;

			if (yeti->mood == ESCAPE_MOOD || item->hitStatus)
			{
				item->goalAnimState = 2;
			}
			else if (yeti->mood == BORED_MOOD)
			{
				if (lara_alive)
				{
					if (GetRandomControl() < 0x100)
					{
						item->goalAnimState = 2;
					}
					else if (GetRandomControl() < 0x200)
					{
						item->goalAnimState = 9;
					}
					else if (GetRandomControl() < 0x300)
					{
						item->goalAnimState = 2;
						item->requiredAnimState = 3;
					}
				}
			}
			else if (GetRandomControl() < 0x200)
			{
				item->goalAnimState = 2;
			}
			break;
		case 9:
			if (info.ahead)
				head = info.angle;

			if (yeti->mood == ESCAPE_MOOD || item->hitStatus)
			{
				item->goalAnimState = 2;
			}
			else if (yeti->mood == BORED_MOOD)
			{
				if (GetRandomControl() < 0x100 || !lara_alive)
				{
					item->goalAnimState = 7;
				}
				else if (GetRandomControl() < 0x200)
				{
					item->goalAnimState = 2;
				}
				else if (GetRandomControl() < 0x300)
				{
					item->goalAnimState = 2;
					item->requiredAnimState = 3;
				}
			}
			else if (GetRandomControl() < 0x200)
			{
				item->goalAnimState = 2;
			}
			break;
		case 3:
			if (info.ahead)
				head = info.angle;

			yeti->maximumTurn = ANGLE(4);

			if (yeti->mood == ESCAPE_MOOD)
			{
				item->goalAnimState = 1;
			}
			else if (yeti->mood == BORED_MOOD)
			{
				if (GetRandomControl() < 0x100 || !lara_alive)
				{
					item->goalAnimState = 2;
					item->requiredAnimState = 7;
				}
				else if (GetRandomControl() < 0x200)
				{
					item->goalAnimState = 2;
					item->requiredAnimState = 9;
				}
				else if (GetRandomControl() < 0x300)
				{
					item->goalAnimState = 2;
				}
			}
			else if (yeti->mood == ATTACK_MOOD)
			{
				if (info.ahead && info.distance < SQUARE(STEP_SIZE))
					item->goalAnimState = 2;
				else if (info.distance < SQUARE(WALL_SIZE * 2))
					item->goalAnimState = 1;
			}
			break;
		case 1:
			if (info.ahead)
				head = info.angle;

			yeti->flags = 0;
			yeti->maximumTurn = ANGLE(6);
			tilt = (angle / 4);

			if (yeti->mood == ESCAPE_MOOD)
			{
				break;
			}
			else if (yeti->mood == BORED_MOOD)
			{
				item->goalAnimState = 3;
			}
			else if (info.ahead && info.distance < SQUARE(STEP_SIZE))
			{
				item->goalAnimState = 2;
			}
			else if (info.ahead && info.distance < SQUARE(WALL_SIZE * 2))
			{
				item->goalAnimState = 6;
			}
			else if (yeti->mood == STALK_MOOD)
			{
				item->goalAnimState = 3;
			}
			break;
		case 4:
			if (info.ahead)
				torso = info.angle;

			if (!yeti->flags && (item->touchBits & 0x1400))
			{
				CreatureEffect(item, &yetiBiteR, DoBloodSplat);
				LaraItem->hitPoints -= 100;
				LaraItem->hitStatus = true;

				yeti->flags = 1;
			}
			break;
		case 5:
			if (info.ahead)
				torso = info.angle;
			yeti->maximumTurn = ANGLE(4);

			if (!yeti->flags && (item->touchBits & (0x0700 | 0x1400)))
			{
				if (item->touchBits & 0x0700)
					CreatureEffect(item, &yetiBiteL, DoBloodSplat);
				if (item->touchBits & 0x1400)
					CreatureEffect(item, &yetiBiteR, DoBloodSplat);

				LaraItem->hitPoints -= 150;
				LaraItem->hitStatus = true;

				yeti->flags = 1;
			}
			break;
		case 6:
			if (info.ahead)
				torso = info.angle;

			if (!yeti->flags && (item->touchBits & (0x0700 | 0x1400)))
			{
				if (item->touchBits & 0x0700)
					CreatureEffect(item, &yetiBiteL, DoBloodSplat);
				if (item->touchBits & 0x1400)
					CreatureEffect(item, &yetiBiteR, DoBloodSplat);

				LaraItem->hitPoints -= 200;
				LaraItem->hitStatus = true;

				yeti->flags = 1;
			}
			break;
		case 10:
		case 11:
		case 12:
		case 13:
			yeti->maximumTurn = 0; // stop yeti to rotate when climbing !
			break;
		}
	}

	// if lara dead, play special kill anim:
	if (!lara_alive)
	{
		yeti->maximumTurn = 0;
		CreatureKill(item, 31, 14, 103); // TODO: need to create a function for old extra anim use.
		return;
	}

	CreatureTilt(item, tilt);
	CreatureJoint(item, 0, torso);
	CreatureJoint(item, 1, head);

	if (item->currentAnimState < 10)
	{
		switch (CreatureVault(itemNum, angle, 2, 300))
		{
		case 2:
			item->animNumber = Objects[item->objectNumber].animIndex + 34;
			item->frameNumber = Anims[item->animNumber].frameBase;
			item->currentAnimState = 10;
			break;
		case 3:
			item->animNumber = Objects[item->objectNumber].animIndex + 33;
			item->frameNumber = Anims[item->animNumber].frameBase;
			item->currentAnimState = 11;
			break;
		case 4:
			item->animNumber = Objects[item->objectNumber].animIndex + 32;
			item->frameNumber = Anims[item->animNumber].frameBase;
			item->currentAnimState = 12;
			break;
		case -4:
			item->animNumber = Objects[item->objectNumber].animIndex + 35;
			item->frameNumber = Anims[item->animNumber].frameBase;
			item->currentAnimState = 13;
			break;
		}
	}
	else
	{
		CreatureAnimation(itemNum, angle, tilt);
	}
}