skeleton-wars/Assets/UnitCreator.cs

using UnityEngine;
using System.Collections;
using System.Collections.Generic;

public class UnitCreator : MonoBehaviour {

	public GameObject lichNone;
	public GameObject lichCyan;
	public GameObject lichPurple;
	public GameObject lichPuke;
	public GameObject lichRed;
	public GameObject skeletonNone;
	public GameObject skeletonCyan;
	public GameObject skeletonPurple;
	public GameObject skeletonPuke;
	public GameObject skeletonRed;
	public GameObject archerNone;
	public GameObject archerCyan;
	public GameObject archerPurple;
	public GameObject archerPuke;
	public GameObject archerRed;
	public GameObject impNone;
	public GameObject impCyan;
	public GameObject impPurple;
	public GameObject impPuke;
	public GameObject impRed;
	public GameObject zombieNone;
	public GameObject zombieCyan;
	public GameObject zombiePurple;
	public GameObject zombiePuke;
	public GameObject zombieRed;
	public GameObject[,] unitsByMap;
	public Vector2 mapSize;
	public List<List<GameObject>> unitsByTeam;

	private UnitBehaviour playerLichScript;
	private List<GameObject> teamNone;
	private List<GameObject> teamPlayer;
	private List<GameObject> teamEnemy1;

	public enum UnitType {
		Lich = 0,
		Skeleton,
		Archer,
		Imp,
		Zombie,
		Size
	}

	// Use this for initialization
	void Start () {
		playerLichScript = null;
	}
	
	// Update is called once per frame
	void Update () {
	
	}

	void createUnit(int posX, int posZ, UnitBehaviour.Team team, UnitType type = UnitType.Skeleton) {
		GameObject unit = null;

		if (type == UnitType.Lich) {
			if (team == UnitBehaviour.Team.Player)
				unit = Instantiate (lichCyan, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			else if (team == UnitBehaviour.Team.Enemy1)
				unit = Instantiate (lichRed, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			else
				unit = Instantiate (lichNone, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;

			unit.GetComponent<UnitBehaviour> ().SetupStats (posX, posZ, team, 10, 100, 0, 0, 0, 3, 3);
			unit.GetComponent<UnitBehaviour> ().SetupBaseAttack (4, 3, 2);
			unit.transform.localScale = new Vector3 (unit.transform.localScale.x*0.3f, unit.transform.localScale.y*0.3f, unit.transform.localScale.z*0.3f);

			if (team == UnitBehaviour.Team.Player && playerLichScript == null )
				playerLichScript = unit.GetComponent<UnitBehaviour> ();
			
		} else if (type == UnitType.Skeleton) {
			if (team == UnitBehaviour.Team.Player)
				unit = Instantiate (skeletonCyan, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			else if (team == UnitBehaviour.Team.Enemy1)
				unit = Instantiate (skeletonRed, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			else
				unit = Instantiate (skeletonNone, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;

			unit.GetComponent<UnitBehaviour> ().SetupStats (posX, posZ, team);
			unit.transform.localScale = new Vector3 (unit.transform.localScale.x*0.3f, unit.transform.localScale.y*0.3f, unit.transform.localScale.z*0.3f);

		} else if (type == UnitType.Archer) {
			if (team == UnitBehaviour.Team.Player)
				unit = Instantiate (archerCyan, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			else if (team == UnitBehaviour.Team.Enemy1)
				unit = Instantiate (archerRed, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			else
				unit = Instantiate (archerNone, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			
			unit.GetComponent<UnitBehaviour> ().SetupStats (posX, posZ, team, 2, 80, 10, 15, 5, 7, 1);
			unit.GetComponent<UnitBehaviour> ().SetupBaseAttack (5, 2);

		} else if (type == UnitType.Imp) {
			if (team == UnitBehaviour.Team.Player)
				unit = Instantiate (impCyan, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			else if (team == UnitBehaviour.Team.Enemy1)
				unit = Instantiate (impRed, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			else
				unit = Instantiate (impNone, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;

			unit.GetComponent<UnitBehaviour> ().SetupStats (posX, posZ, team, 2, 30, 10, 30, 15, 10, 2);

		} else if (type == UnitType.Zombie) {
			if (team == UnitBehaviour.Team.Player)
				unit = Instantiate (zombieCyan, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			else if (team == UnitBehaviour.Team.Enemy1)
				unit = Instantiate (zombieRed, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;
			else
				unit = Instantiate (zombieNone, new Vector3(posX, 0f, posZ), Quaternion.identity) as GameObject;

			unit.GetComponent<UnitBehaviour> ().SetupStats (posX, posZ, team);

		}

		unitsByMap[posZ, posX] = unit;
		Color color = new Color (Random.value, Random.value, Random.value);

		if (team == UnitBehaviour.Team.Player) {
			teamPlayer.Add (unit);
		} else if (team == UnitBehaviour.Team.Enemy1) {
			unit.transform.Rotate (new Vector3 (0, 180, 0));
			teamEnemy1.Add (unit);
		}
	}

	void setMap(int x, int y) {
		unitsByMap = new GameObject[y, x];
		for (int i = 0; i < y; i++) {
			for (int j = 0; j < x; j++) {
				unitsByMap[i, j] = null;
			}
		}

		unitsByTeam = new List<List<GameObject>> ();

		teamNone = new List<GameObject> ();
		teamPlayer = new List<GameObject> ();
		teamEnemy1 = new List<GameObject> ();
	}
		

	public void tutorialUnits() {
		setMap (20, 40);

		createUnit (9, 5, UnitBehaviour.Team.Player, UnitType.Lich);
		
		createUnit (8, 7, UnitBehaviour.Team.Player);
		createUnit (10, 7, UnitBehaviour.Team.Player);
		createUnit (6, 6, UnitBehaviour.Team.Player, UnitType.Archer);
		createUnit (12, 6, UnitBehaviour.Team.Player, UnitType.Archer);

		createUnit (8, 12, UnitBehaviour.Team.Enemy1);
		createUnit (10, 12, UnitBehaviour.Team.Enemy1);
		createUnit (9, 17, UnitBehaviour.Team.Enemy1, UnitType.Imp);
		createUnit (5, 17, UnitBehaviour.Team.Enemy1);
		createUnit (4, 18, UnitBehaviour.Team.Enemy1);
		createUnit (5, 19, UnitBehaviour.Team.Enemy1);
		createUnit (2, 22, UnitBehaviour.Team.Enemy1);
		createUnit (15, 23, UnitBehaviour.Team.Enemy1);
		createUnit (13, 24, UnitBehaviour.Team.Enemy1);
		createUnit (16, 24, UnitBehaviour.Team.Enemy1);
		createUnit (17, 26, UnitBehaviour.Team.Enemy1, UnitType.Archer);

		unitsByTeam.Add (teamPlayer);
		unitsByTeam.Add (teamEnemy1);
	}

	public List<List<GameObject>> getUnitLists() {
		return unitsByTeam;
	}

	public GameObject[,] getUnitMap() {
		return unitsByMap;
	}
}