#include <ultra64.h>
#include "constants.h"
#include "game/botmgr.h"
#include "game/chr.h"
#include "game/body.h"
#include "game/prop.h"
#include "game/botcmd.h"
#include "game/botinvinit.h"
#include "game/mplayer/mplayer.h"
#include "bss.h"
#include "lib/memp.h"
#include "lib/model.h"
#include "lib/rng.h"
#include "data.h"
#include "types.h"

void botmgrAllocateBot(s32 chrnum, s32 botcfgnum, s32 botnum)
{
	s16 rooms[1];
	struct prop *prop;
	struct chrdata *chr;
	struct aibot *aibot;
	struct model *model;
	s32 i;
	s32 bodynum;
	s32 headnum;

	rooms[0] = -1;

	headnum = mpGetHeadId(g_BotConfigsArray[botcfgnum].base.mpheadnum);
	bodynum = mpGetBodyId(g_BotConfigsArray[botcfgnum].base.mpbodynum);

	model = bodyAllocateModel(bodynum, headnum, 0);

	if (model != NULL) {
		struct coord pos = {0.0f, 0.0f, 0.0f};
		u32 stack;

		prop = chrAllocate(model, &pos, rooms, 0.0f);

		if (prop != NULL) {
			propActivate(prop);
			propEnable(prop);

			chr = prop->chr;

			g_MpChrs[chrnum].chr = chr;

			chrSetChrnum(chr, chrnum);

			chr->chrnum = chrnum;
			chr->padpreset1 = -1;
			chr->chrpreset1 = -1;
			chr->hearingscale = 1;
			chr->visionrange = 80;
			chr->headnum = headnum;
			chr->bodynum = bodynum;
			chr->race = bodyGetRace(chr->bodynum);
			chr->flags = CHRFLAG0_CAN_EXAMINE_BODY; // reused flag?
			chr->flags2 = 0;
			chr->team = g_BotConfigsArray[botcfgnum].base.team;
			chr->squadron = 0;

			aibot = mempAlloc(sizeof(struct aibot), MEMPOOL_STAGE);
			chr->aibot = aibot;

			if (aibot == NULL) {
				return;
			}

			aibot->chrdistances = mempAlloc(g_MpNumChrs * sizeof(f32), MEMPOOL_STAGE);
			aibot->chrsinsight = mempAlloc(g_MpNumChrs * sizeof(u8), MEMPOOL_STAGE);
			aibot->chrslastseen60 = mempAlloc(g_MpNumChrs * sizeof(s32), MEMPOOL_STAGE);
			aibot->chrrooms = mempAlloc(g_MpNumChrs * sizeof(s8), MEMPOOL_STAGE);

			if (!aibot->chrdistances || !aibot->chrsinsight || !aibot->chrslastseen60 || !aibot->chrrooms) {
				return;
			}

			chr->tude = 0;
			chr->voicebox = random() % 3;

			if (g_HeadsAndBodies[chr->bodynum].ismale == false) {
				chr->voicebox = VOICEBOX_FEMALE;
			}

			chr->target = -1;
			chr->naturalanim = 0;
			chr->myspecial = -1;
			chr->yvisang = 0;
			chr->teamscandist = 0;
			chr->convtalk = 0;
			chr->myaction = MA_AIBOTMAINLOOP;

			aibot->ammoheld = mempAlloc(36 * sizeof(s32), MEMPOOL_STAGE);

			for (i = 0; i < 33; i++) {
				aibot->ammoheld[i] = 0;
			}

			aibot->config = &g_BotConfigsArray[botcfgnum];

			switch (g_BotConfigsArray[botcfgnum].difficulty) {
			case BOTDIFF_MEAT:
				aibot->followchance = 0;
				break;
			case BOTDIFF_EASY:
				aibot->followchance = 10;
				break;
			default:
			case BOTDIFF_NORMAL:
				aibot->followchance = 20;
				break;
			case BOTDIFF_HARD:
				aibot->followchance = 40;
				break;
			case BOTDIFF_PERFECT:
				aibot->followchance = 60;
				break;
			case BOTDIFF_DARK:
				aibot->followchance = 0;
				break;
			}

			aibot->aibotnum = botnum;
			aibot->unk064 = 0;
			aibot->gotoprop = NULL;
			aibot->timeuntilreload60[0] = 0;
			aibot->timeuntilreload60[1] = 0;
			aibot->throwtimer60 = 0;
			aibot->unk040 = 0.0f;
			aibot->loadedammo[0] = 0;
			aibot->loadedammo[1] = 0;
			aibot->unk058 = 0;
			aibot->unk059 = 0;
			aibot->nextbullettimer60[0] = 0;
			aibot->nextbullettimer60[1] = 0;
			aibot->unk030 = 301;
			aibot->unk038 = 0;
			aibot->punchtimer60[HAND_LEFT] = -1;
			aibot->punchtimer60[HAND_RIGHT] = 0;
			aibot->attackingplayernum = -1;
			aibot->followingplayernum = -1;
			aibot->dangerouspropnum = -1;
			aibot->unk068 = 0;
			aibot->unk06c = 0.0f;
			aibot->unk070 = 0.0f;
			aibot->distmode = -1;
			aibot->lastkilledbyplayernum = -1;
			aibot->feudplayernum = -1;
			aibot->command = AIBOTCMD_NORMAL;
			aibot->unk098 = 0.0f;

			aibot->defendholdpos.x = pos.x;
			aibot->defendholdpos.y = pos.y;
			aibot->defendholdpos.z = pos.z;

			aibot->weaponnum = WEAPON_UNARMED;

			aibot->burstsdone[0] = 0;
			aibot->burstsdone[1] = 0;
			aibot->skrocket = NULL;
			aibot->unk0a0 = 0;

			aibot->gunfunc = FUNC_PRIMARY;
			aibot->iscloserangeweapon = true;
			aibot->teamisonlyai = false;
			aibot->hasbriefcase = false;
			aibot->hascase = false;
			aibot->cloakdeviceenabled = false;
			aibot->rcp120cloakenabled = false;
			aibot->unk04c_04 = false;
			aibot->unk04c_03 = false;
			aibot->hasuplink = false;
			aibot->unk04c_00 = false;

			aibot->hillpadnum = -1;
			aibot->hillcovernum = -1;

			aibot->lastknownhill = -1;
			aibot->cyclonedischarging[HAND_LEFT] = false;
			aibot->cyclonedischarging[HAND_RIGHT] = false;
			aibot->changeguntimer60 = 0;
			aibot->distmodettl60 = 0;
			aibot->forcemainloop = false;
			aibot->returntodefendtimer60 = 0;
			aibot->reaperspeed[HAND_LEFT] = 0;
			aibot->reaperspeed[HAND_RIGHT] = 0;
			aibot->maulercharge[HAND_LEFT] = 0.0f;
			aibot->maulercharge[HAND_RIGHT] = 0.0f;
			aibot->unk0a4 = model0001ae44(chr->model);
			aibot->angleoffset = 0.0f;
			aibot->speedtheta = 0.0f;
			aibot->unk0b0 = model0001ae44(chr->model);

			aibot->unk0b4 = 0.0f;
			aibot->unk0b8 = 0.0f;
			aibot->shotspeed.x = 0.0f;
			aibot->shotspeed.y = 0.0f;
			aibot->shotspeed.z = 0.0f;
			aibot->commandtimer60 = 0;
			aibot->shootdelaytimer60 = 0;
			aibot->targetlastseen60 = -1;
			aibot->lastseenanytarget60 = -1;
			aibot->targetinsight = false;
			aibot->queryplayernum = 0;

			for (i = 0; i < ARRAYCOUNT(aibot->chrnumsbydistanceasc); i++) {
				aibot->chrnumsbydistanceasc[i] = -1;
			}

			for (i = 0; i < g_MpNumChrs; i++) {
				aibot->chrdistances[i] = U32_MAX;
				aibot->chrsinsight[i] = 0;
				aibot->chrslastseen60[i] = -1;
				aibot->chrrooms[i] = -1;
			}

			aibot->unk1c0 = 0.0f;
			aibot->unk1c4 = 0.0f;
			aibot->unk1c8 = 0.0f;
			aibot->unk1cc = -1;
			aibot->unk1d0 = 0;
			aibot->unk1d4 = 0.0f;
			aibot->unk1e4 = -1;
			aibot->waypoints[0] = NULL;
			aibot->unk208 = 0;
			aibot->random1 = random();
			aibot->random1ttl60 = 0;

			for (i = 0; i < 6; i++) {
				aibot->killsbygunfunc[i][0] = 0.0f;
				aibot->killsbygunfunc[i][1] = 0.0f;

				aibot->suicidesbygunfunc[i][0] = 0.0f;
				aibot->suicidesbygunfunc[i][1] = 0.0f;

				aibot->equipdurations60[i][0] = 0;
				aibot->equipdurations60[i][1] = 0;

				aibot->unk2a8[i] = 0;
			}

			aibot->unk2a4 = 0;
			aibot->dampensuicidesttl60 = 0;
			aibot->unk2c4 = 0.0f;
			aibot->targetcloaktimer60 = 0;
			aibot->canseecloaked = false;

			aibot->random2ttl60 = 0;
			aibot->random2 = random();
			aibot->randomfrac = RANDOMFRAC();
			aibot->cheap = false;
			aibot->unk078 = 0;
			aibot->unk050 = 0;
			aibot->unk09d = 0;

			botinvInit(chr, 10);
		}
	}
}