#include "framework.h"
#include "hair.h"

#include "draw.h"
#include "laramisc.h"
#include "lara.h"
#include "control.h"
#include "GameFlowScript.h"
#include "setup.h"
#include "sphere.h"
#include "level.h"

int FirstHair[HAIR_MAX];
HAIR_STRUCT Hairs[HAIR_MAX][HAIR_SEGMENTS];
int WindAngle;
int DWindAngle;
int Wind;

extern GameFlow* g_GameFlow;

void InitialiseHair()
{
	for (int h = 0; h < HAIR_MAX; h++)
	{
		FirstHair[h] = 1;

		int* bone = &Bones[Objects[ID_LARA_HAIR].boneIndex];

		Hairs[h][0].pos.yRot = 0;
		Hairs[h][0].pos.xRot = -0x4000;

		for (int i = 1; i < HAIR_SEGMENTS; i++, bone += 4)
		{
			Hairs[h][i].pos.xPos = *(bone + 1);
			Hairs[h][i].pos.yPos = *(bone + 2);
			Hairs[h][i].pos.zPos = *(bone + 3);
			Hairs[h][i].pos.xRot = -0x4000;
			Hairs[h][i].pos.yRot = Hairs[h][i].pos.zRot = 0;

			Hairs[h][i].hvel.x = Hairs[h][i].hvel.y = Hairs[h][i].hvel.z = 0;
		}
	}
}


void HairControl(int cutscene, int ponytail, short* framePtr)
{
	SPHERE sphere[HAIR_SPHERE];
	ObjectInfo* object = &Objects[ID_LARA];
	short* frame;
	int spaz;
	bool youngLara = g_GameFlow->GetLevel(CurrentLevel)->LaraType == LARA_YOUNG;

	if (framePtr == NULL)
	{
		if (Lara.hitDirection >= 0)
		{
			switch (Lara.hitDirection)
			{
			case NORTH:
				if (Lara.isDucked)
					spaz = 294;
				else
					spaz = 125;
				break;

			case SOUTH:
				if (Lara.isDucked)
					spaz = 293;
				else
					spaz = 126;
				break;

			case EAST:
				if (Lara.isDucked)
					spaz = 295;
				else
					spaz = 127;
				break;

			default:
				if (Lara.isDucked)
					spaz = 296;
				else
					spaz = 128;
				break;
			}

			frame = Anims[spaz].framePtr;
			int size = Anims[spaz].interpolation >> 8;

			frame += (int)(Lara.hitFrame * size);
		}
		else
			frame = GetBestFrame(LaraItem);
	}
	else
	{
		frame = framePtr;
	}

	// Get Lara's spheres in absolute coords, for head, torso, hips and upper arms
	short* objptr = Lara.meshPtrs[LM_HIPS];
	PHD_VECTOR pos = { objptr[0], objptr[1], objptr[2] };
	GetLaraJointPosition(&pos, LM_HIPS);
	sphere[0].x = pos.x;
	sphere[0].y = pos.y;
	sphere[0].z = pos.z;
	sphere[0].r = (int) * (objptr + 3);

	objptr = Lara.meshPtrs[LM_TORSO];
	pos = { objptr[0], objptr[1], objptr[2] };
	GetLaraJointPosition(&pos, LM_TORSO);
	sphere[1].x = pos.x;
	sphere[1].y = pos.y;
	sphere[1].z = pos.z;
	sphere[1].r = (int) * (objptr + 3);
	if (youngLara)
		sphere[1].r = sphere[1].r - ((sphere[1].r >> 2) + (sphere[1].r >> 3));

	objptr = Lara.meshPtrs[LM_HEAD];
	pos = { objptr[0], objptr[1], objptr[2] };
	GetLaraJointPosition(&pos, LM_HEAD);
	sphere[2].x = pos.x;
	sphere[2].y = pos.y;
	sphere[2].z = pos.z;
	sphere[2].r = (int) * (objptr + 3);

	objptr = Lara.meshPtrs[LM_RINARM];
	pos = { objptr[0], objptr[1], objptr[2] };
	GetLaraJointPosition(&pos, LM_RINARM);
	sphere[3].x = pos.x;
	sphere[3].y = pos.y;
	sphere[3].z = pos.z;
	sphere[3].r = (int) * (objptr + 3) * 3 / 2;

	objptr = Lara.meshPtrs[LM_LINARM];
	pos = { objptr[0], objptr[1], objptr[2] };
	GetLaraJointPosition(&pos, LM_LINARM);
	sphere[4].x = pos.x;
	sphere[4].y = pos.y;
	sphere[4].z = pos.z;
	sphere[4].r = (int) * (objptr + 3) * 3 / 2;

	if (youngLara)
	{
		sphere[1].x = (sphere[1].x + sphere[2].x) / 2;
		sphere[1].y = (sphere[1].y + sphere[2].y) / 2;
		sphere[1].z = (sphere[1].z + sphere[2].z) / 2;
	}
	
	Matrix world;
	g_Renderer->GetBoneMatrix(Lara.itemNumber, LM_HEAD, &world);

	if (ponytail)
	{
		world = Matrix::CreateTranslation(44, -48, -50) * world;
	}
	else if (youngLara)
	{
		world = Matrix::CreateTranslation(-52, -48, -50) * world;
	}
	else
	{
		world = Matrix::CreateTranslation(-4, -48, -48) * world;
	}

	pos.x = world.Translation().x; 
	pos.y = world.Translation().y; 
	pos.z = world.Translation().z;

	int* bone = &Bones[Objects[ID_LARA_HAIR].boneIndex];

	if (FirstHair[ponytail])
	{
		FirstHair[ponytail] = 0;

		Hairs[ponytail][0].pos.xPos = pos.x;
		Hairs[ponytail][0].pos.yPos = pos.y;
		Hairs[ponytail][0].pos.zPos = pos.z;

		for (int i = 0; i < HAIR_SEGMENTS - 1; i++, bone += 4)
		{
			world = Matrix::CreateTranslation(Hairs[ponytail][i].pos.xPos, Hairs[ponytail][i].pos.yPos, Hairs[ponytail][i].pos.zPos);		
			world = Matrix::CreateFromYawPitchRoll(TO_RAD(Hairs[ponytail][i].pos.yRot), TO_RAD(Hairs[ponytail][i].pos.xRot), 0) * world;			
			world = Matrix::CreateTranslation(*(bone + 1), *(bone + 2), *(bone + 3)) * world;

			Hairs[ponytail][i + 1].pos.xPos = world.Translation().x;
			Hairs[ponytail][i + 1].pos.yPos = world.Translation().y;
			Hairs[ponytail][i + 1].pos.zPos = world.Translation().z;
		}

		Wind = SmokeWindX = SmokeWindZ = 0;
		WindAngle = DWindAngle = 2048;
	}
	else
	{
		Hairs[ponytail][0].pos.xPos = pos.x;
		Hairs[ponytail][0].pos.yPos = pos.y;
		Hairs[ponytail][0].pos.zPos = pos.z;

		short roomNumber = LaraItem->roomNumber;
		int wh;

		if (cutscene)
		{
			wh = NO_HEIGHT;
		}
		else
		{
			int x = LaraItem->pos.xPos + (frame[0] + frame[1]) / 2;
			int y = LaraItem->pos.yPos + (frame[2] + frame[3]) / 2;
			int z = LaraItem->pos.zPos + (frame[4] + frame[5]) / 2;
			wh = GetWaterHeight(x, y, z, roomNumber);
		}

		Wind += (GetRandomControl() & 7) - 3;
		if (Wind <= -2)
			Wind++;
		else if (Wind >= 9)
			Wind--;

		DWindAngle = (DWindAngle + 2 * (GetRandomControl() & 63) - 64) & 0x1FFE;

		if (DWindAngle < 1024)
			DWindAngle = 2048 - DWindAngle;
		else if (DWindAngle > 3072)
			DWindAngle += 6144 - 2 * DWindAngle;

		WindAngle = (WindAngle + ((DWindAngle - WindAngle) >> 3)) & 0x1FFE;

		SmokeWindX = Wind * rcossin_tbl[WindAngle] >> 12;
		SmokeWindZ = Wind * rcossin_tbl[WindAngle + 1] >> 12;

		SmokeWindX = (((rcossin_tbl[WindAngle]) * Wind) >> 12);
		SmokeWindZ = (((rcossin_tbl[WindAngle + 1]) * Wind) >> 12);

		for (int i = 1; i < HAIR_SEGMENTS; i++, bone += 4)
		{
			Hairs[ponytail][0].hvel.x = Hairs[ponytail][i].pos.xPos;
			Hairs[ponytail][0].hvel.y = Hairs[ponytail][i].pos.yPos;
			Hairs[ponytail][0].hvel.z = Hairs[ponytail][i].pos.zPos;

			int height;

			if (!cutscene)
			{
				FLOOR_INFO* floor = GetFloor(Hairs[ponytail][i].pos.xPos, Hairs[ponytail][i].pos.yPos, Hairs[ponytail][i].pos.zPos, &roomNumber);
				height = GetFloorHeight(floor, Hairs[ponytail][i].pos.xPos, Hairs[ponytail][i].pos.yPos, Hairs[ponytail][i].pos.zPos);
			}
			else
			{
				height = 32767;
			}

			Hairs[ponytail][i].pos.xPos += Hairs[ponytail][i].hvel.x * 3 / 4;
			Hairs[ponytail][i].pos.yPos += Hairs[ponytail][i].hvel.y * 3 / 4;
			Hairs[ponytail][i].pos.zPos += Hairs[ponytail][i].hvel.z * 3 / 4;

			if (Lara.waterStatus == LW_ABOVE_WATER && Rooms[roomNumber].flags & ENV_FLAG_WIND)
			{
				Hairs[ponytail][i].pos.xPos += SmokeWindX;
				Hairs[ponytail][i].pos.zPos += SmokeWindZ;
			}

			switch (Lara.waterStatus)
			{
			case LW_ABOVE_WATER:
				Hairs[ponytail][i].pos.yPos += 10;
				if (wh != NO_HEIGHT && Hairs[ponytail][i].pos.yPos > wh)
				{
					Hairs[ponytail][i].pos.yPos = wh;
				}
				else if (Hairs[ponytail][i].pos.yPos > height)
				{
					Hairs[ponytail][i].pos.xPos = Hairs[ponytail][0].hvel.x;
					Hairs[ponytail][i].pos.zPos = Hairs[ponytail][0].hvel.z;
				}
				break;

			case LW_UNDERWATER:
			case LW_SURFACE:
			case LW_WADE:
				if (Hairs[ponytail][i].pos.yPos < wh)
					Hairs[ponytail][i].pos.yPos = wh;
				else if (Hairs[ponytail][i].pos.yPos > height)
					Hairs[ponytail][i].pos.yPos = height;
				break;
			}

			for (int j = 0; j < HAIR_SPHERE; j++)
			{
				int x = Hairs[ponytail][i].pos.xPos - sphere[j].x;
				int y = Hairs[ponytail][i].pos.yPos - sphere[j].y;
				int z = Hairs[ponytail][i].pos.zPos - sphere[j].z;

				int distance = SQUARE(x) + SQUARE(y) + SQUARE(z);

				if (distance < SQUARE(sphere[j].r))
				{
					distance = sqrt(distance);

					if (distance == 0)
						distance = 1;

					Hairs[ponytail][i].pos.xPos = sphere[j].x + x * sphere[j].r / distance;
					Hairs[ponytail][i].pos.yPos = sphere[j].y + y * sphere[j].r / distance;
					Hairs[ponytail][i].pos.zPos = sphere[j].z + z * sphere[j].r / distance;
				}
			}

			int distance = sqrt(SQUARE(Hairs[ponytail][i].pos.zPos - Hairs[ponytail][i - 1].pos.zPos) + SQUARE(Hairs[ponytail][i].pos.xPos - Hairs[ponytail][i - 1].pos.xPos));
			Hairs[ponytail][i - 1].pos.yRot = phd_atan((Hairs[ponytail][i].pos.zPos - Hairs[ponytail][i - 1].pos.zPos), (Hairs[ponytail][i].pos.xPos - Hairs[ponytail][i - 1].pos.xPos));
			Hairs[ponytail][i - 1].pos.xRot = -phd_atan(distance, Hairs[ponytail][i].pos.yPos - Hairs[ponytail][i - 1].pos.yPos);

			world = Matrix::CreateTranslation(Hairs[ponytail][i - 1].pos.xPos, Hairs[ponytail][i - 1].pos.yPos, Hairs[ponytail][i - 1].pos.zPos);
			world = Matrix::CreateFromYawPitchRoll(TO_RAD(Hairs[ponytail][i - 1].pos.yRot), TO_RAD(Hairs[ponytail][i - 1].pos.xRot), 0) * world;

			if (i == HAIR_SEGMENTS - 1)
				world = Matrix::CreateTranslation(*(bone - 3), *(bone - 2), *(bone - 1)) * world;
			else
				world = Matrix::CreateTranslation(*(bone + 1), *(bone + 2), *(bone + 3)) * world;

			Hairs[ponytail][i].pos.xPos = world.Translation().x;
			Hairs[ponytail][i].pos.yPos = world.Translation().y;
			Hairs[ponytail][i].pos.zPos = world.Translation().z;

			Hairs[ponytail][i].hvel.x = Hairs[ponytail][i].pos.xPos - Hairs[ponytail][0].hvel.x;
			Hairs[ponytail][i].hvel.y = Hairs[ponytail][i].pos.yPos - Hairs[ponytail][0].hvel.y;
			Hairs[ponytail][i].hvel.z = Hairs[ponytail][i].pos.zPos - Hairs[ponytail][0].hvel.z;
		}
	}
}