#include "framework.h"
#include "cog_switch.h"
#include "control.h"
#include "input.h"
#include "lara.h"
#include "generic_switch.h"
#include "door_data.h"
#include "Box.h"
#include "generic_doors.h"
#include "collide.h"

using namespace TEN::Entities::Doors;

namespace TEN::Entities::Switches
{
	OBJECT_COLLISION_BOUNDS CogSwitchBounds =
	{
		-512, 512,
		0, 0,
		-1536, -512,
		-ANGLE(10), ANGLE(10),
		-ANGLE(30), ANGLE(30),
		-ANGLE(10), ANGLE(10)
	};
	PHD_VECTOR CogSwitchPos(0, 0, -856);

	void CogSwitchCollision(short itemNum, ITEM_INFO* l, COLL_INFO* coll)
	{
		auto item = &g_Level.Items[itemNum];
		auto triggerIndex = GetTriggerIndex(item);

		if (!triggerIndex)
		{
			ObjectCollision(itemNum, l, coll);
			return;
		}

		short* trigger = triggerIndex;
		for (int i = *triggerIndex; (i & 0x1F) != 4; trigger++)
		{
			if (i < 0)
				break;
			i = trigger[1];
		}
		int targetItemNum = trigger[3] & 0x3FF;
		ITEM_INFO* target = &g_Level.Items[targetItemNum];
		DOOR_DATA* door = (DOOR_DATA*)target->data;

		if (item->status == ITEM_NOT_ACTIVE)
		{
			if (!(item->flags & ONESHOT)
				&& (TrInput & IN_ACTION
					&& !Lara.gunStatus
					&& !item->gravityStatus
					&& l->currentAnimState == LS_STOP
					&& l->animNumber == LA_STAND_IDLE
					|| Lara.isMoving
					&& Lara.interactedItem == itemNum))
			{
				if (TestLaraPosition(&CogSwitchBounds, item, l))
				{
					if (MoveLaraPosition(&CogSwitchPos, item, l))
					{
						Lara.isMoving = false;
						Lara.headYrot = 0;
						Lara.headXrot = 0;
						Lara.torsoYrot = 0;
						Lara.torsoXrot = 0;
						Lara.gunStatus = LG_HANDS_BUSY;
						Lara.interactedItem = targetItemNum;
						l->animNumber = LA_COGWHEEL_GRAB;
						l->goalAnimState = LS_COGWHEEL;
						l->currentAnimState = LS_COGWHEEL;
						l->frameNumber = g_Level.Anims[l->animNumber].frameBase;

						AddActiveItem(itemNum);
						item->goalAnimState = SWITCH_ON;
						item->status = ITEM_ACTIVE;

						if (door != NULL)
						{
							if (!door->opened)
							{
								AddActiveItem((target - g_Level.Items.data()));
								target->status = ITEM_ACTIVE;
							}
						}
					}
					else
					{
						Lara.interactedItem = itemNum;
					}
					return;
				}
				else if (Lara.isMoving && Lara.interactedItem == itemNum)
				{
					Lara.isMoving = false;
					Lara.gunStatus = LG_NO_ARMS;
				}
			}

			ObjectCollision(itemNum, l, coll);
		}
	}

	void CogSwitchControl(short itemNumber)
	{
		ITEM_INFO* item = &g_Level.Items[itemNumber];

		AnimateItem(item);

		if (item->currentAnimState == SWITCH_ON)
		{
			if (item->goalAnimState == SWITCH_ON && !(TrInput & IN_ACTION))
			{
				LaraItem->goalAnimState = LS_STOP;
				item->goalAnimState = SWITCH_OFF;
			}

			if (LaraItem->animNumber == LA_COGWHEEL_PULL)
			{
				if (LaraItem->frameNumber == g_Level.Anims[LaraItem->animNumber].frameBase + 10)
				{
					ITEM_INFO* doorItem = &g_Level.Items[Lara.interactedItem];
					doorItem->itemFlags[0] = COG_DOOR_TURN;
				}
			}
		}
		else
		{
			if (item->frameNumber == g_Level.Anims[item->animNumber].frameEnd)
			{
				item->currentAnimState = SWITCH_OFF;
				item->status = ITEM_NOT_ACTIVE;

				RemoveActiveItem(itemNumber);

				LaraItem->animNumber = LA_STAND_SOLID;
				LaraItem->frameNumber = g_Level.Anims[LaraItem->animNumber].frameBase;
				LaraItem->goalAnimState = LS_STOP;
				LaraItem->currentAnimState = LS_STOP;
				Lara.gunStatus = LG_NO_ARMS;
			}
		}
	}
}