public void SetupBattle(GameObject playerInBattle, GameObject enemyInBattle) { player = playerInBattle; enemy = enemyInBattle; playerUnit = player.GetComponent <Unit>(); enemyUnit = enemy.GetComponent <Unit>(); StartCoroutine(infoBarManager.UpdateText("You have entered a battle against " + enemyUnit.unitName)); playerUnit.SetHUD(); enemyUnit.SetHUD(); agent = enemy.GetComponent <EnemyAgent>(); if (!agent.trainingMode) { agent.UnfreezeAgent(); DialoguePanel.SetActive(true); } else { enemyUnit.currentHP = enemyUnit.maxHP; enemyUnit.SetHP(); playerUnit.currentHP = playerUnit.maxHP; playerUnit.SetHP(); } //agent.BattleSystemSc = this; SwitchTurn(); }