//
// Duel
//

#include "stagesetup.h"

// Characters
#define CHR_GUARD    0x00
#define CHR_JONATHAN 0x01
#define CHR_TRENT    0x02

// Stage flags
#define STAGEFLAG_GUARD_DEAD      0x00000100
#define STAGEFLAG_JONATHAN_DEAD   0x00000200
#define STAGEFLAG_TRENT_DEAD      0x00000400
#define STAGEFLAG_SA_WALK_TRIGGER 0x00000800
#define STAGEFLAG_PA_WALK_TRIGGER 0x00001000

// AI Lists
#define AILIST_MOVE_TO_PAD  0x0401
#define AILIST_GUARD_COMBAT 0x0402
#define AILIST_JON_COMBAT   0x0403
#define AILIST_TRENT_COMBAT 0x0404
#define AILIST_GUARD_INIT   0x0405
#define AILIST_JON_INIT     0x0406
#define AILIST_TRENT_INIT   0x0407

extern s32 intro[];
extern u32 props[];
extern struct path paths[];
extern struct ailist ailists[];

struct stagesetup setup = {
	NULL,
	NULL,
	NULL,
	intro,
	props,
	paths,
	ailists,
	NULL,
};

u32 props[] = {
	briefing(BRIEFINGTYPE_LOCATION, L_ATE_001)
	briefing(BRIEFINGTYPE_TEXT_PA,  L_ATE_000)
	briefing(BRIEFINGTYPE_TEXT_SA,  L_ATE_002)
	briefing(BRIEFINGTYPE_TEXT_A,   L_ATE_003)

	beginobjective(0, L_ATE_004, (DIFFBIT_A | DIFFBIT_SA | DIFFBIT_PA | DIFFBIT_PD)) // "Defeat dataDyne guard"
		complete_flags(STAGEFLAG_GUARD_DEAD)
	endobjective

	beginobjective(1, L_ATE_005, (DIFFBIT_SA | DIFFBIT_PA | DIFFBIT_PD)) // "Defeat Jonathan Dark"
		complete_flags(STAGEFLAG_JONATHAN_DEAD)
	endobjective

	beginobjective(2, L_ATE_006, (DIFFBIT_PA | DIFFBIT_PD)) // "Defeat Trent Easton"
		complete_flags(STAGEFLAG_TRENT_DEAD)
	endobjective

	chr(SPAWNFLAG_00000200, 0x00, PAD_ATE_0279, BODY_DDSHOCK, HEAD_RANDOM, AILIST_GUARD_INIT, -1, -1, 100, 100, CHRFLAG0_CAN_THROW_GRENADES | CHRFLAG0_SQUADALERTANYDIST | CHRFLAG0_HOLD_POSITION | CHRFLAG0_COVER_TYPE2 | CHRFLAG0_NOHEAR, CHRFLAG1_STOPTRACKINGIFLOOKEDAT, TEAM_ENEMY, SQUADRON_04, -1, 0, 0, 0, 0, 0)
	weapon(0x0100, MODEL_CHRFALCON2, CHR_GUARD, OBJFLAG_ASSIGNEDTOCHR, 0, 0, WEAPON_FALCON2_SCOPE)
	chr(SPAWNFLAG_00000200, 0x01, PAD_ATE_0023, BODY_CISOLDIER, HEAD_JONATHAN, AILIST_JON_INIT, -1, -1, 100, 100, CHRFLAG0_CAN_THROW_GRENADES | CHRFLAG0_SQUADALERTANYDIST | CHRFLAG0_HOLD_POSITION | CHRFLAG0_COVER_TYPE2 | CHRFLAG0_NOHEAR, CHRFLAG1_STOPTRACKINGIFLOOKEDAT, TEAM_ENEMY, SQUADRON_04, -1, 0, 0, 0, 0, 0)
	weapon(0x0100, MODEL_CHRDY357, CHR_JONATHAN, OBJFLAG_ASSIGNEDTOCHR, 0, 0, WEAPON_DY357MAGNUM)
	chr(SPAWNFLAG_00000200, 0x02, PAD_ATE_0024, BODY_TRENT, HEAD_TRENT, AILIST_TRENT_INIT, -1, -1, 100, 100, CHRFLAG0_CAN_THROW_GRENADES | CHRFLAG0_SQUADALERTANYDIST | CHRFLAG0_HOLD_POSITION | CHRFLAG0_COVER_TYPE2 | CHRFLAG0_NOHEAR, CHRFLAG1_STOPTRACKINGIFLOOKEDAT, TEAM_ENEMY, SQUADRON_04, -1, 0, 0, 0, 0, 0)
	weapon(0x0100, MODEL_CHRDY357TRENT, CHR_TRENT, OBJFLAG_ASSIGNEDTOCHR, 0, 0, WEAPON_DY357LX)
	door(0x0100, MODEL_DD_OFFICEDOOR, PAD_ATE_0158, OBJFLAG_IGNOREFLOORCOLOUR, OBJFLAG2_AICANNOTUSE, 0, 1000, 0x0000f333, 0x00010000, 0x00007fff, 0x00014000, 0x00000666, DOORFLAG_0004, DOORTYPE_SLIDING, 0x00000040, 0x00000384, 0x00000000, 0x00000001, 0x00000800)
	door(0x0100, MODEL_DD_OFFICEDOOR, PAD_ATE_0159, OBJFLAG_IGNOREFLOORCOLOUR, OBJFLAG2_AICANNOTUSE, 0, 1000, 0x0000f333, 0x00010000, 0x00007fff, 0x00014000, 0x00000666, DOORFLAG_0004, DOORTYPE_SLIDING, 0x00000040, 0x00000384, 0x00000000, 0xffffffff, 0x00000000)
	endprops
};

s32 intro[] = {
	outfit(OUTFIT_DEFAULT)
	spawn(PAD_ATE_027A)
	intro_weapon(WEAPON_FALCON2_SCOPE, -1)
	ammo(AMMOTYPE_PISTOL, 8)
	endintro
};

struct path paths[] = {
	{ NULL, 0, 0 },
};

u8 func1000_counterop_setup[] = {
	yield
	set_chr_team(CHR_ANTI, TEAM_ENEMY)
	set_ailist(CHR_SELF, GAILIST_REBUILD_GROUPS)
	endlist
};

u8 func1001_objectives_failed[] = {
	yield
	set_ailist(CHR_SELF, GAILIST_SHOW_OBJ_FAILED_MSG)
	endlist
};

u8 func1002_first_walk[] = {
	set_chr_chrflag(CHR_COOP, CHRCFLAG_HIDDEN)
	set_chr_chrflag(CHR_ANTI, CHRCFLAG_HIDDEN)
	camera_movement(ANIM_CUT_ATE_INTRO_CAM)
	unset_chr_chrflag(CHR_BOND, CHRCFLAG_HIDDEN)
	set_chr_chrflag(CHR_BOND, CHRCFLAG_UNPLAYABLE)
	set_chr_hiddenflag(CHR_BOND, CHRHFLAG_00020000)
	show_hudmsg_middle(0x00, COLOR_02_WHITE, L_ATE_007) // "THE DUEL"
	chr_do_animation(ANIM_RELOAD, -1, -1, CHRANIMFLAG_MOVEWHENINVIS | CHRANIMFLAG_PAUSEATEND, 0, CHR_GUARD, 2)
	label(0x2d)
	show_hudmsg(CHR_BOND, L_ATE_012) // "Opponent skill level: AGENT"

	// Wait for camera to stop animating
	restart_timer

	beginloop(0x08)
		if_camera_animating(/*goto*/ 0x2d)
		goto_next(0x06)
		label(0x2d)
	endloop(0x08)

	// Start walking
	label(0x06)
	remove_hudmsgs
	revoke_control(CHR_BOND, 0)
	enter_firstperson
	force_walk(CHR_BOND, 0x0275, 40, 5, 0, 100)
	chr_move_to_pad(CHR_GUARD, PAD_ATE_0274, 0x01, /*goto*/ 0x0b)

	// Wait for walk to finish
	beginloop(0x0b)
		if_force_walk_finished(CHR_BOND, /*goto*/ 0x0c)
	endloop(0x0b)

	label(0x0c)
	grant_control(CHR_BOND)
	set_ailist(CHR_GUARD, AILIST_GUARD_COMBAT)
	terminate
	endlist
};

u8 func1007_second_walk[] = {
	// Wait until SA walk trigger
	beginloop(0x58)
		if_chr_dead(CHR_BOND, /*goto*/ 0x2d)
		if_chr_death_animation_finished(CHR_BOND, /*goto*/ 0x2d)
		if_stage_flag_eq(STAGEFLAG_SA_WALK_TRIGGER, TRUE, /*goto*/ 0x06)
	endloop(0x58)

	// Begin cutscene
	label(0x2d)
	terminate
	label(0x06)
	revoke_control(CHR_BOND, 0)
	camera_movement(ANIM_CUT_ATE_INTRO_CAM)

	// Place chrs
	label(0x8f)
	set_ailist(CHR_P1P2, AILIST_MOVE_TO_PAD)
	chr_move_to_pad(CHR_JONATHAN, PAD_ATE_0279, 0x01, /*goto*/ 0x2d)
	yield
	goto_first(0x8f)

	label(0x2d)
	unset_chr_chrflag(CHR_JONATHAN, CHRCFLAG_HIDDEN)
	unset_chr_chrflag(CHR_JONATHAN, CHRCFLAG_INVINCIBLE)
	unset_chr_chrflag(CHR_JONATHAN, CHRCFLAG_UNEXPLODABLE)
	yield
	unset_chr_chrflag(CHR_BOND, CHRCFLAG_HIDDEN)
	set_chr_chrflag(CHR_BOND, CHRCFLAG_UNPLAYABLE)
	set_chr_hiddenflag(CHR_BOND, CHRHFLAG_00020000)
	chr_do_animation(ANIM_GRAB_CROTCH, -1, -1, CHRANIMFLAG_MOVEWHENINVIS | CHRANIMFLAG_PAUSEATEND, 0, CHR_JONATHAN, 2)

	label(0x2d)
	show_hudmsg(CHR_BOND, L_ATE_013) // "Opponent skill level: SPECIAL AGENT"
	restart_timer

	beginloop(0x08)
		if_camera_animating(/*goto*/ 0x2d)
		goto_next(0x06)
		label(0x2d)
	endloop(0x08)

	// Start force walk
	label(0x06)
	label(0x2d)
	remove_hudmsgs
	revoke_control(CHR_BOND, 0)
	enter_firstperson
	yield
	force_walk(CHR_BOND, 0x0275, 40, 5, 0, 100)

	// Place Jon
	call_rng
	if_rand_lt(128, /*goto*/ 0x8f)
	chr_move_to_pad(CHR_JONATHAN, PAD_ATE_0271, 0x01, /*goto*/ 0x0b)
	label(0x8f)
	chr_move_to_pad(CHR_JONATHAN, PAD_ATE_0270, 0x01, /*goto*/ 0x0b)

	// Wait for force walk to finish
	beginloop(0x0b)
		if_force_walk_finished(CHR_BOND, /*goto*/ 0x0c)
	endloop(0x0b)

	// Give control back to Jo
	label(0x0c)
	grant_control(CHR_BOND)
	set_ailist(CHR_JONATHAN, AILIST_JON_COMBAT)
	terminate
	endlist
};

u8 func0401_move_to_pad[] = {
	chr_move_to_pad(CHR_SELF, PAD_ATE_027A, 0x01, /*goto*/ 0x06)
	label(0x06)
	yield
	stop_chr
	terminate
	endlist
};

u8 func1008_third_walk[] = {
	// Wait for flag or Jo dying
	beginloop(0x58)
		if_chr_dead(CHR_BOND, /*goto*/ 0x2d)
		if_chr_death_animation_finished(CHR_BOND, /*goto*/ 0x2d)
		if_stage_flag_eq(STAGEFLAG_PA_WALK_TRIGGER, TRUE, /*goto*/ 0x06)
	endloop(0x58)

	// Jo dead
	label(0x2d)
	terminate

	// Do cutscene
	label(0x06)
	revoke_control(CHR_BOND, 0)
	camera_movement(ANIM_CUT_ATE_INTRO_CAM)

	label(0x8f)
	set_ailist(CHR_P1P2, AILIST_MOVE_TO_PAD)
	chr_move_to_pad(CHR_TRENT, PAD_ATE_0279, 0x01, /*goto*/ 0x2d)
	yield
	goto_first(0x8f)

	label(0x2d)
	unset_chr_chrflag(CHR_TRENT, CHRCFLAG_HIDDEN)
	unset_chr_chrflag(CHR_TRENT, CHRCFLAG_INVINCIBLE)
	unset_chr_chrflag(CHR_TRENT, CHRCFLAG_UNEXPLODABLE)
	yield
	unset_chr_chrflag(CHR_BOND, CHRCFLAG_HIDDEN)
	set_chr_chrflag(CHR_BOND, CHRCFLAG_UNPLAYABLE)
	set_chr_hiddenflag(CHR_BOND, CHRHFLAG_00020000)
	chr_do_animation(ANIM_HEAD_ROLL, -1, -1, CHRANIMFLAG_MOVEWHENINVIS | CHRANIMFLAG_PAUSEATEND, 0, CHR_TRENT, 2)
	label(0x2d)
	show_hudmsg(CHR_BOND, L_ATE_014) // "Opponent skill level: PERFECT AGENT"

	// Wait for cutscene to finish
	restart_timer

	beginloop(0x08)
		if_camera_animating(/*goto*/ 0x2d)
		goto_next(0x06)
		label(0x2d)
	endloop(0x08)

	// Begin force walk
	label(0x06)
	remove_hudmsgs
	revoke_control(CHR_BOND, 0)
	enter_firstperson
	force_walk(CHR_BOND, 0x0275, 40, 5, 0, 100)
	chr_move_to_pad(CHR_TRENT, PAD_ATE_0272, 0x01, /*goto*/ 0x0b)

	beginloop(0x0b)
		if_force_walk_finished(CHR_BOND, /*goto*/ 0x0c)
	endloop(0x0b)

	label(0x0c)
	set_ailist(CHR_TRENT, AILIST_TRENT_COMBAT)
	grant_control(CHR_BOND)
	terminate
	endlist
};

u8 func1004_revoke_control[] = {
	yield
	revoke_control(CHR_BOND, 0)
	terminate
	endlist
};

u8 func1005_main[] = {
	// Wait until guard dead
	beginloop(0x04)
		if_chr_deadish(CHR_GUARD, /*goto*/ 0x2d)
	endloop(0x04)

	label(0x2d)
	show_hudmsg(CHR_BOND, L_ATE_009) // "Well done! You were too quick for him!"
	set_stage_flag(STAGEFLAG_GUARD_DEAD)
	restart_timer

	beginloop(0x08)
		if_timer_gt(120, /*goto*/ 0x2d)
	endloop(0x08)

	// If Agent, return
	label(0x2d)
	if_difficulty_lt(DIFF_SA, /*goto*/ 0x2d)
	goto_next(0x06)

	label(0x2d)
	terminate

	label(0x06)
	set_stage_flag(STAGEFLAG_SA_WALK_TRIGGER)

	// Wait until Jonathan dead
	beginloop(0x09)
		if_chr_deadish(CHR_JONATHAN, /*goto*/ 0x2d)
	endloop(0x09)

	label(0x2d)
	show_hudmsg(CHR_BOND, L_ATE_010) // "Well done! You were too quick for him!"
	set_stage_flag(STAGEFLAG_JONATHAN_DEAD)
	restart_timer

	beginloop(0x0b)
		if_timer_gt(120, /*goto*/ 0x2d)
	endloop(0x0b)

	// If Special Agent, return
	label(0x2d)
	if_difficulty_lt(DIFF_PA, /*goto*/ 0x2d)
	goto_next(0x06)

	label(0x2d)
	terminate

	label(0x06)
	set_stage_flag(STAGEFLAG_PA_WALK_TRIGGER)

	// Wait until Trent dead
	beginloop(0x0c)
		if_chr_deadish(CHR_TRENT, /*goto*/ 0x2d)
	endloop(0x0c)

	label(0x2d)
	show_hudmsg(CHR_BOND, L_ATE_011) // "Well done! You were too quick for him!"
	set_stage_flag(STAGEFLAG_TRENT_DEAD)
	restart_timer

	beginloop(0x0d)
		if_timer_gt(120, /*goto*/ 0x2d)
	endloop(0x0d)

	label(0x2d)
	terminate
	endlist
};

u8 func1003_init_audio[] = {
	stop_cutscene_track
	stop_ambient_track
	terminate
	endlist
};

u8 func1006_check_objectives_complete[] = {
	// Wait for all objectives complete
	beginloop(0x04)
		if_all_objectives_complete(/*goto*/ 0x2d)
	endloop(0x04)

	// Wait 3 seconds
	label(0x2d)
	restart_timer

	beginloop(0x08)
		if_timer_gt(180, /*goto*/ 0x2d)
	endloop(0x08)

	// End level
	label(0x2d)
	end_level
	terminate
	endlist
};

u8 func0402_guard_combat[] = {
	set_shotlist(AILIST_GUARD_COMBAT)
	if_just_injured(CHR_SELF, /*goto*/ 0x03)
	set_chr_hiddenflag(CHR_SELF, CHRHFLAG_PERFECTACCURACY)

	// Wait 2 seconds or for guard to come into view
	restart_timer

	beginloop(0x04)
		if_timer_gt(120, /*goto*/ 0x06)
		if_chr_in_view(/*goto*/ 0x2d)
	endloop(0x04)

	// Guard has come into view. Jump sideways.
	label(0x2d)
	try_sidestep(/*goto*/ 0x08)

	beginloop(0x08)
		if_distance_to_target_lt(250, /*goto*/ 0x0e)
		if_chr_stopped(/*goto*/ 0x03)
	endloop(0x08)

	// Guard has been shot, or finished jump
	beginloop(0x03)
		if_distance_to_target_lt(250, /*goto*/ 0x0e)
		if_can_see_target(/*goto*/ 0x06)
		reloop(0x03)

		// 2 seconds have passed without seeing guard, or guard has been shot
		label(0x06)
		if_chr_dead(CHR_TARGET, /*goto*/ 0x2e)
		try_attack_stand(ATTACKFLAG_AIMATTARGET | ATTACKFLAG_AIMONLY, 0, /*goto*/ 0x2d)
		label(0x2d)
		try_modify_attack(ATTACKFLAG_AIMATTARGET, 0, /*goto*/ 0x0c)

		beginloop(0x0c)
			if_chr_stopped(/*goto*/ 0x06)
		endloop(0x0c)

		// Finished shooting
		label(0x06)
	endloop(0x03)

	// Dying
	label(0x2e)
	terminate

	// Hand combat
	label(0x0e)
	set_returnlist(CHR_SELF, AILIST_TRENT_COMBAT)
	set_ailist(CHR_SELF, GAILIST_HAND_COMBAT)
	endlist
};

u8 func0403_jon_combat[] = {
	set_shotlist(AILIST_JON_COMBAT)
	if_just_injured(CHR_SELF, /*goto*/ 0x58)
	set_chr_hiddenflag(CHR_SELF, CHRHFLAG_PERFECTACCURACY)
	restart_timer

	beginloop(0x04)
		if_timer_gt(240, /*goto*/ 0x58)
		if_nearly_in_targets_sight(30, /*goto*/ 0x58)
	endloop(0x04)

	label(0x58)
	try_run_to_target(/*goto*/ 0x08)

	beginloop(0x08)
		if_distance_to_target_lt(250, /*goto*/ 0x0e)
		if_can_see_attack_target(/*goto*/ 0x06)
		reloop(0x08)

		label(0x06)
		try_attack_kneel(ATTACKFLAG_AIMATTARGET | ATTACKFLAG_AIMONLY, 0, /*goto*/ 0x2d)
		label(0x2d)
		try_attack_roll(/*goto*/ 0x0c)
		try_attack_kneel(ATTACKFLAG_AIMATTARGET, 0, /*goto*/ 0x0c)

		beginloop(0x0c)
			if_chr_stopped(/*goto*/ 0x06)
		endloop(0x0c)

		label(0x06)
	endloop(0x58)

	label(0x0e)
	set_returnlist(CHR_SELF, AILIST_JON_COMBAT)
	set_ailist(CHR_SELF, GAILIST_HAND_COMBAT)
	terminate
	endlist
};

u8 func0404_trent_combat[] = {
	set_shotlist(AILIST_TRENT_COMBAT)
	if_just_injured(CHR_SELF, /*goto*/ 0x03)
	set_chr_hiddenflag(CHR_SELF, CHRHFLAG_PERFECTACCURACY)
	restart_timer

	beginloop(0x04)
		if_timer_gt(120, /*goto*/ 0x03)
		if_chr_in_view(/*goto*/ 0x03)
	endloop(0x04)

	// Wait until player in sight or in close range
	beginloop(0x03)
		if_distance_to_target_lt(250, /*goto*/ 0x0e)
		if_can_see_target(/*goto*/ 0x06)
		reloop(0x03)

		// Attack
		label(0x06)
		if_chr_dead(CHR_TARGET, /*goto*/ 0x2e)
		try_attack_stand(ATTACKFLAG_AIMATTARGET | ATTACKFLAG_AIMONLY, 0, /*goto*/ 0x2d)
		label(0x2d)
		try_modify_attack(ATTACKFLAG_AIMATTARGET, 0, /*goto*/ 0x0c)

		beginloop(0x0c)
			if_chr_stopped(/*goto*/ 0x06)
		endloop(0x0c)

		label(0x06)
	endloop(0x03)

	// Hand combat
	label(0x0e)
	set_returnlist(CHR_SELF, AILIST_TRENT_COMBAT)
	set_ailist(CHR_SELF, GAILIST_HAND_COMBAT)
	label(0x2e)
	terminate
	endlist
};

u8 func0405_guard_init[] = {
	set_self_chrflag(CHRCFLAG_NOAUTOAIM)
	// @bug: The value for set_accuracy is interpreted as a signed byte,
	// so 200 is interpreted as -56 which reduces the chr's accuracy.
	// However this isn't an issue because the chr has CHRHFLAG_PERFECTACCURACY.
	// This bug is repeated for Jon and Trent below.
	set_accuracy(200)
	set_reaction_speed(50)
	set_chr_maxdamage(CHR_SELF, 20)
	add_health_or_armor(0)
	set_recovery_speed(0)
	set_shield(0)
	terminate
	endlist
};

u8 func0406_jon_init[] = {
	set_self_chrflag(CHRCFLAG_NOAUTOAIM)
	set_accuracy(200)
	set_reaction_speed(75)
	set_chr_maxdamage(CHR_SELF, 40)
	add_health_or_armor(0)
	set_recovery_speed(0)
	set_shield(0)
	set_self_chrflag(CHRCFLAG_HIDDEN)
	set_self_chrflag(CHRCFLAG_INVINCIBLE)
	set_self_chrflag(CHRCFLAG_UNEXPLODABLE)
	terminate
	endlist
};

u8 func0407_trent_init[] = {
	set_self_chrflag(CHRCFLAG_NOAUTOAIM)
	set_accuracy(200)
	set_reaction_speed(75)
	set_chr_maxdamage(CHR_SELF, 40)
	add_health_or_armor(0)
	set_recovery_speed(0)
	set_shield(0)
	set_self_chrflag(CHRCFLAG_HIDDEN)
	set_self_chrflag(CHRCFLAG_INVINCIBLE)
	set_self_chrflag(CHRCFLAG_UNEXPLODABLE)
	terminate
	endlist
};

struct ailist ailists[] = {
	{ func1000_counterop_setup,           0x1000 },
	{ func1001_objectives_failed,         0x1001 },
	{ func1002_first_walk,                0x1002 },
	{ func1003_init_audio,                0x1003 },
	{ func1004_revoke_control,            0x1004 },
	{ func1005_main,                      0x1005 },
	{ func1006_check_objectives_complete, 0x1006 },
	{ func1007_second_walk,               0x1007 },
	{ func1008_third_walk,                0x1008 },
	{ func0401_move_to_pad,               0x0401 },
	{ func0402_guard_combat,              0x0402 },
	{ func0403_jon_combat,                0x0403 },
	{ func0404_trent_combat,              0x0404 },
	{ func0405_guard_init,                0x0405 },
	{ func0406_jon_init,                  0x0406 },
	{ func0407_trent_init,                0x0407 },
	{ NULL, 0 },
};