1 files changed, 98 insertions, 0 deletions
diff --git a/src/engine/duel/ai/init.asm b/src/engine/duel/ai/init.asm
new file mode 100644
index 0000000..33132cf
--- /dev/null
+++ b/src/engine/duel/ai/init.asm
@@ -0,0 +1,98 @@
+InitAIDuelVars: ; 15636 (5:5636)
+	ld a, wAIDuelVarsEnd - wAIDuelVars
+	ld hl, wAIDuelVars
+	call ClearMemory_Bank5
+	ld a, 5
+	ld [wAIPokedexCounter], a
+	ld a, $ff
+	ld [wAIPeekedPrizes], a
+	ret
+
+; initializes some variables and sets value of wAIBarrierFlagCounter.
+; if Player uses Barrier 3 times in a row, AI checks if Player's deck
+; has only Mewtwo1 Pokemon cards (running a Mewtwo1 mill deck).
+InitAITurnVars: ; 15649 (5:5649)
+; increase Pokedex counter by 1
+	ld a, [wAIPokedexCounter]
+	inc a
+	ld [wAIPokedexCounter], a
+
+	xor a
+	ld [wPreviousAIFlags], a
+	ld [wAITriedAttack], a
+	ld [wcddc], a
+	ld [wAIRetreatedThisTurn], a
+
+; checks if the Player used an attack last turn
+; and if it was the second attack of their card.
+	ld a, [wPlayerAttackingAttackIndex]
+	cp $ff
+	jr z, .check_flag
+	or a
+	jr z, .check_flag
+	ld a, [wPlayerAttackingCardIndex]
+	cp $ff
+	jr z, .check_flag
+
+; if the card is Mewtwo1, it means the Player
+; used its second attack, Barrier.
+	call SwapTurn
+	call GetCardIDFromDeckIndex
+	call SwapTurn
+	ld a, e
+	cp MEWTWO1
+	jr nz, .check_flag
+	; Player used Barrier last turn
+
+; check if flag was already set, if so,
+; reset wAIBarrierFlagCounter to $80.
+	ld a, [wAIBarrierFlagCounter]
+	bit AI_MEWTWO_MILL_F, a
+	jr nz, .set_flag
+
+; if not, increase it by 1 and check if it exceeds 2.
+	inc a
+	ld [wAIBarrierFlagCounter], a
+	cp 3
+	jr c, .done
+
+; this means that the Player used Barrier
+; at least 3 turns in a row.
+; check if Player is running Mewtwo1-only deck,
+; if so, set wAIBarrierFlagCounter flag.
+	ld a, DUELVARS_ARENA_CARD
+	call GetNonTurnDuelistVariable
+	call SwapTurn
+	call GetCardIDFromDeckIndex
+	call SwapTurn
+	ld a, e
+	cp MEWTWO1
+	jr nz, .reset_1
+	farcall CheckIfPlayerHasPokemonOtherThanMewtwo1
+	jr nc, .set_flag
+.reset_1
+; reset wAIBarrierFlagCounter
+	xor a
+	ld [wAIBarrierFlagCounter], a
+	jr .done
+
+.set_flag
+	ld a, AI_MEWTWO_MILL
+	ld [wAIBarrierFlagCounter], a
+	jr .done
+
+.check_flag
+; increase counter by 1 if flag is set
+	ld a, [wAIBarrierFlagCounter]
+	bit AI_MEWTWO_MILL_F, a
+	jr z, .reset_2
+	inc a
+	ld [wAIBarrierFlagCounter], a
+	jr .done
+
+.reset_2
+; reset wAIBarrierFlagCounter
+	xor a
+	ld [wAIBarrierFlagCounter], a
+.done
+	ret