#include "lara2gun.h"
#include "..\Global\global.h"
#include "larafire.h"
#include "lara.h"
#include "effect2.h"

PISTOL_DEF PistolsTable[4] =
{
	{ ID_LARA, 0, 0, 0, 0 },
	{ ID_PISTOLS_ANIM, 4, 5, 0xD, 0x18 },
	{ ID_REVOLVER_ANIM , 7, 8, 0xF, 0x1D },
	{ ID_UZI_ANIM, 4, 5, 0xD, 0x18 }
};

int UziLeft;
int UziRight;

void AnimatePistols(int weaponType)
{
	int soundPlayed = false;

	PHD_VECTOR pos;

	if (LaraItem->meshBits)
	{
		if (SmokeCountL)
		{
			switch (SmokeWeapon)
			{
			case WEAPON_PISTOLS:
				pos.x = 4;
				pos.y = 128;
				pos.z = 40;
				break;
			case WEAPON_REVOLVER:
				pos.x = 16;
				pos.y = 160;
				pos.z = 56;
				break;
			case WEAPON_UZI:
				pos.x = 8;
				pos.y = 140;
				pos.z = 48;
				break;
			}
			GetLaraJointPosition(&pos, LJ_LHAND);
			TriggerGunSmoke(pos.x, pos.y, pos.z, 0, 0, 0, 0, SmokeWeapon, SmokeCountL);
		}

		if (SmokeCountR)
		{
			switch (SmokeWeapon)
			{
			case WEAPON_PISTOLS:
				pos.x = -16;
				pos.y = 128;
				pos.z = 40;
				break;
			case WEAPON_REVOLVER:
				pos.x = -32;
				pos.y = 160;
				pos.z = 56;
				break;
			case WEAPON_UZI:
				pos.x = -16;
				pos.y = 140;
				pos.z = 48;
				break;
			}
			GetLaraJointPosition(&pos, LJ_RHAND);
			TriggerGunSmoke(pos.x, pos.y, pos.z, 0, 0, 0, 0, SmokeWeapon, SmokeCountR);
		}
	}

	PISTOL_DEF* p = &PistolsTable[Lara.gunType];
	WEAPON_INFO* weapon = &Weapons[weaponType];

	short frameRight = Lara.rightArm.frameNumber;
	short angles[2];

	if (Lara.rightArm.lock || TrInput & IN_ACTION && !Lara.target)
	{
		if (frameRight < 0 || frameRight >= p->draw1Anim2)
		{
			if (frameRight == p->draw1Anim2)
			{
				if (TrInput & IN_ACTION)
				{
					if (weaponType != WEAPON_REVOLVER)
					{
						angles[0] = Lara.rightArm.yRot + LaraItem->pos.yRot;
						angles[1] = Lara.rightArm.xRot;

						if (FireWeapon(weaponType, Lara.target, LaraItem, angles))
						{
							SmokeCountR = 28;
							SmokeWeapon = weaponType;
							/*RIGHT_HAND*/		TriggerGunShell(1, ID_GUNSHELL, weaponType);

							Lara.rightArm.flash_gun = weapon->flashTime;

							SoundEffect(SFX_EXPLOSION1, &LaraItem->pos, 33554436);
							SoundEffect(weapon->sampleNum, &LaraItem->pos, 0);
							soundPlayed = true;

							if (weaponType == WEAPON_UZI)
								UziRight = true;

							Savegame.Game.AmmoUsed++;
						}
					}

					frameRight = p->recoilAnim;
				}

				else if (UziRight)
				{
					SoundEffect(weapon->sampleNum + 1, &LaraItem->pos, 0);
					UziRight = false;
				}
			}
			else
			{
				if (Lara.rightArm.frameNumber >= p->recoilAnim)
				{
					if (weaponType == WEAPON_UZI)
					{
						SoundEffect(weapon->sampleNum, &LaraItem->pos, 0);
						UziRight = true;
					}

					frameRight++;
					if (frameRight == p->recoilAnim + weapon->recoilFrame)
						frameRight = p->recoilAnim + weapon->recoilFrame;
				}
			}
		}
		else
		{
			frameRight++;
		}
	}
	else
	{
		if (frameRight >= p->recoilAnim)
			frameRight = p->draw1Anim2;
		else if (frameRight > 0 && frameRight <= p->draw1Anim2)
			frameRight--;

		if (UziRight)
		{
			SoundEffect(weapon->sampleNum + 1, &LaraItem->pos, 0);
			UziRight = false;
		}
	}

	set_arm_info(&Lara.rightArm, frameRight);

	int frameLeft = Lara.leftArm.frameNumber;
	if (!Lara.leftArm.lock && (!(TrInput & IN_ACTION) || Lara.target))
	{
		if (frameLeft >= 0 && frameLeft < p->draw1Anim2)
		{
			frameLeft++;
		}
		else if (frameLeft == p->draw1Anim2)
		{
			if (TrInput & IN_ACTION)
			{
				angles[0] = Lara.leftArm.yRot + LaraItem->pos.yRot;
				angles[1] = Lara.leftArm.xRot;

				if (FireWeapon(weaponType, Lara.target, LaraItem, angles))
				{
					if (weaponType == WEAPON_REVOLVER)
					{
						SmokeCountR = 28;
						SmokeWeapon = WEAPON_REVOLVER;
						Lara.rightArm.flash_gun = weapon->flashTime;
					}
					else
					{
						SmokeCountL = 28;
						SmokeWeapon = weaponType;
						/*LEFT_HAND*/	TriggerGunShell(0, ID_GUNSHELL, weaponType);
						Lara.leftArm.flash_gun = weapon->flashTime;
					}

					if (!soundPlayed)
					{
						SoundEffect(SFX_EXPLOSION1, &LaraItem->pos, PITCH_SHIFT | 0x2000000);
						SoundEffect(weapon->sampleNum, &LaraItem->pos, 0);
					}

					if (weaponType == WEAPON_UZI)
						UziLeft = true;

					Savegame.Game.AmmoUsed++;
				}
			}
			else if (UziLeft)
			{
				SoundEffect(weapon->sampleNum + 1, &LaraItem->pos, 0);
				UziLeft = false;
			}
		}
	}
	else
	{
		if (frameLeft >= p->recoilAnim)
			frameLeft = p->recoilAnim;
		else if (frameLeft > 0 && frameLeft <= p->draw1Anim2)
			frameLeft--;
		
		if (UziLeft)
		{
			SoundEffect(weapon->sampleNum + 1, &LaraItem->pos, 0);
			UziLeft = false;
		}
	}

	set_arm_info(&Lara.leftArm, frameLeft);
}

void PistolHandler(int weaponType)
{
	WEAPON_INFO* weapon = &Weapons[weaponType];

	LaraGetNewTarget(weapon);
	if (TrInput & IN_ACTION)
		LaraTargetInfo(weapon);

	AimWeapon(weapon, &Lara.leftArm);
	AimWeapon(weapon, &Lara.rightArm);

	if (Lara.leftArm.lock)
	{
		if (Lara.rightArm.lock)
		{
			Lara.torsoYrot = (Lara.leftArm.yRot + Lara.rightArm.yRot) / 4;
			Lara.torsoXrot = (Lara.leftArm.xRot + Lara.rightArm.xRot) / 4;
			if (Camera.oldType != LOOK_CAMERA)
			{
				Lara.headYrot = Lara.torsoYrot;
				Lara.headXrot = Lara.torsoXrot;
			}
		}
		else
		{
			Lara.torsoYrot = Lara.leftArm.yRot / 2;
			Lara.torsoXrot = Lara.leftArm.xRot / 2;
			if (Camera.oldType != LOOK_CAMERA)
			{
				Lara.headYrot = Lara.torsoYrot;
				Lara.headXrot = Lara.torsoXrot;
			}
		}
	}
	else if (Lara.rightArm.lock)
	{
		Lara.torsoYrot = Lara.rightArm.yRot / 2;
		Lara.torsoXrot = Lara.rightArm.xRot / 2;
		if (Camera.oldType != LOOK_CAMERA)
		{
			Lara.headYrot = Lara.torsoYrot;
			Lara.headXrot = Lara.torsoXrot;
		}
	}

	AnimatePistols(weaponType);
	
	if (Lara.leftArm.flash_gun || Lara.rightArm.flash_gun)
	{
		PHD_VECTOR pos;

		pos.x = GetRandomControl() - 128;
		pos.y = (GetRandomControl() & 0x7F) - 63;
		pos.z = GetRandomControl() - 128;

		GetLaraJointPosition(&pos, Lara.leftArm.flash_gun != 0 ? LJ_LHAND : LJ_RHAND);
		/*if (gfLevelFlags & 0x2000 && LaraItem->room_number == gfMirrorRoom)
		{
			v8 = GetRandomControl() & 0x3F;
			v9 = (GetRandomControl() & 0x1F) + 128;
			v10 = GetRandomControl();
			result = sub_4015A5(v14, v15, v16, 10, (v10 & 0x3F) + 192, v9, v8);
		}
		else
		{*/
			TriggerDynamicLight(pos.x, pos.y, pos.z, 10, (GetRandomControl() & 0x3F) + 192, (GetRandomControl() & 0x1F) + 128, GetRandomControl() & 0x3F);
		//}
	}
}

void undraw_pistol_mesh_right(int weaponType)
{
	short objectNumber = WeaponObject(weaponType);
	
	//Lara.meshPtrs[THIGH_R] = Meshes[Objects[objectNumber].meshIndex + THIGH_R];
	Lara.meshPtrs[HAND_R] = Meshes[Objects[ID_LARA].meshIndex + HAND_R];
	
	switch (weaponType)
	{
	case WEAPON_PISTOLS:
		Lara.holster = 14;
		break;
	case WEAPON_UZI:
		Lara.holster = 15;
		break;
	case WEAPON_REVOLVER:
		Lara.holster = 16;
		break;
	}

}

void undraw_pistol_mesh_left(int weaponType)
{
	int result; // eax

	if (weaponType != WEAPON_REVOLVER)
	{
		WeaponObject(weaponType);

		Lara.meshPtrs[HAND_L] = Meshes[Objects[ID_LARA].meshIndex + HAND_L];
		if (weaponType == WEAPON_PISTOLS)
		{
			Lara.holster = 14;
		}
		else if (weaponType == WEAPON_UZI)
		{
			Lara.holster = 15;
		}
	}
}

void draw_pistol_meshes(int weaponType)
{
	Lara.holster = 13;
	Lara.meshPtrs[HAND_L] = Meshes[Objects[WeaponObjectMesh(weaponType)].meshIndex + HAND_R];
	
	if (weaponType != WEAPON_REVOLVER)
	{
		Lara.meshPtrs[HAND_L] = Meshes[Objects[WeaponObjectMesh(weaponType)].meshIndex + HAND_R];
	}
}

void ready_pistols(int weaponType)
{
	Lara.gunStatus = LG_READY;
	Lara.leftArm.zRot = 0;
	Lara.leftArm.yRot = 0;
	Lara.leftArm.xRot = 0;
	Lara.rightArm.zRot = 0;
	Lara.rightArm.yRot = 0;
	Lara.rightArm.xRot = 0;
	Lara.rightArm.frameNumber = 0;
	Lara.leftArm.frameNumber = 0;
	Lara.target = NULL;
	Lara.rightArm.lock = 0;
	Lara.leftArm.lock = 0;
	Lara.rightArm.frameBase = Objects[WeaponObject(weaponType)].frameBase;
	Lara.leftArm.frameBase = Objects[WeaponObject(weaponType)].frameBase;
}

void undraw_pistols(int weaponType)
{
	PISTOL_DEF* p = &PistolsTable[Lara.gunType];

	int frameLeft = Lara.leftArm.frameNumber;
	if (frameLeft >= p->recoilAnim)
	{
		frameLeft = p->recoilAnim;
	}
	else if (frameLeft > 0 && frameLeft < p->draw1Anim)
	{
		Lara.leftArm.xRot -= Lara.leftArm.xRot / frameLeft;
		Lara.leftArm.yRot -= Lara.leftArm.yRot / frameLeft;
		frameLeft--;
	}
	else if (frameLeft == 0)
	{
		Lara.leftArm.zRot = 0;
		Lara.leftArm.yRot = 0;
		Lara.leftArm.xRot = 0;
		frameLeft = p->recoilAnim - 1;
	}
	else if (frameLeft > p->draw1Anim)
	{
		frameLeft--;

		if (frameLeft == p->draw2Anim - 1)
		{
			undraw_pistol_mesh_left(weaponType);
			SoundEffect(SFX_LARA_HOLSTER_AWAY, &LaraItem->pos, 0);
		}
	}

	set_arm_info(&Lara.leftArm, frameLeft);

	int frameRight = Lara.rightArm.frameNumber;
	if (frameRight < p->recoilAnim)
	{
		if (frameRight > 0 && frameRight < p->draw1Anim)
		{
			Lara.rightArm.xRot -= Lara.rightArm.xRot / frameRight;
			Lara.rightArm.yRot -= Lara.rightArm.yRot / frameRight;
			frameRight--;
		}
		else if (frameRight == 0)
		{
			Lara.rightArm.zRot = 0;
			Lara.rightArm.yRot = 0;
			Lara.rightArm.xRot = 0;
			frameRight = p->recoilAnim - 1;
		}
		else if (Lara.rightArm.frameNumber > p->draw1Anim)
		{
			frameRight--;

			if (frameRight == p->draw2Anim - 1)
			{
				undraw_pistol_mesh_right(weaponType);
				SoundEffect(SFX_LARA_HOLSTER_AWAY, &LaraItem->pos, 0);
			}
		}
	}
	else
	{
		frameRight = p->draw1Anim2;
	}

	set_arm_info(&Lara.rightArm, frameRight);

	if (frameLeft == p->draw1Anim && frameRight == p->draw1Anim)
	{
		Lara.gunStatus = LG_NO_ARMS;
		Lara.leftArm.frameNumber = 0;
		Lara.rightArm.frameNumber = 0;
		Lara.target = NULL;
		Lara.rightArm.lock = 0;
		Lara.leftArm.lock = 0;
	}

	if (!(TrInput & IN_LOOK))
	{
		Lara.headYrot = (Lara.leftArm.yRot + Lara.rightArm.yRot) / 4;
		Lara.torsoYrot = (Lara.leftArm.yRot + Lara.rightArm.yRot) / 4;
		Lara.headXrot = (Lara.leftArm.xRot + Lara.rightArm.xRot) / 4;
		Lara.torsoXrot = (Lara.leftArm.xRot + Lara.rightArm.xRot) / 4;
	}
}

void set_arm_info(LARA_ARM* arm, int frame)
{
	PISTOL_DEF* p = &PistolsTable[Lara.gunType];
	short animBase = Objects[p->objectNum].animIndex;
	
	if (frame < p->draw1Anim)
		arm->animNumber = animBase;
	else if (frame < p->draw2Anim)
		arm->animNumber = animBase + 1;
	else if (frame < p->recoilAnim)
		arm->animNumber = animBase + 2;
	else
		arm->animNumber = animBase + 3;

	arm->frameNumber = frame;
	arm->frameBase = Anims[animBase].framePtr;
}

void draw_pistols(int weaponType)
{
	short frame = Lara.leftArm.frameNumber + 1;
	PISTOL_DEF* p = &PistolsTable[Lara.gunType];

	if (frame < p->draw1Anim|| frame > p->recoilAnim - 1)
	{
		frame = p->draw1Anim;
	}
	else if (frame == p->draw2Anim)
	{
		draw_pistol_meshes(weaponType);
		SoundEffect(SFX_LARA_HOLSTER_DRAW, &LaraItem->pos, 0);
	}
	else if (frame == p->recoilAnim - 1)
	{
		ready_pistols(weaponType);
		frame = 0;
	}

	set_arm_info(&Lara.rightArm, frame);
	set_arm_info(&Lara.leftArm, frame);
}