MapGen.cs 3.06 KB
Newer Older
Zac Olsen's avatar
Pathing  
Zac Olsen committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137
using UnityEngine;
using System.Collections;
using System.Collections.Generic;

public class MapGen : MonoBehaviour {

	public int rows = 6;
	public int cols = 6;

	public GameObject tile;
	public GameObject enemy;
	public static Tile[,] map;

	public Tile startPos;
	public Tile endPos;

	public List<Tile> open;
	public List<Tile> closed;

	void Start () {

		map = new Tile[rows, cols];

		float startx = -(float)rows / 2f * tile.transform.lossyScale.x;
		float starty = (float)cols / 2f * tile.transform.lossyScale.y;

		for (int y = 0; y < cols; y++) {
			for (int x = 0; x < rows; x++) {

				map[x, y] = ((GameObject) Instantiate (tile, new Vector2(startx + x * tile.GetComponent<BoxCollider2D>().size.x,
					starty - y * tile.GetComponent<BoxCollider2D>().size.x), Quaternion.identity)).GetComponent<Tile> ();

				map [x, y].x = x;
				map [x, y].y = y;
			}
		}

		open = new List<Tile> ();
		closed = new List<Tile> ();

		enemy = (GameObject)(Instantiate (enemy, map[0, 0].transform.position, Quaternion.identity));
	}

	void FixedUpdate () {

		if (map[rows - 1, cols - 1].prev != null) {

			Tile current = map [rows - 1, cols - 1];
			while (current.prev != null) {
				Debug.DrawLine (current.transform.position, current.prev.transform.position, Color.red);
				current = current.prev;
			}
		}
	}

	public void BeginPath () {
		Path (map[0, 0], map[rows - 1, cols - 1]);
	}

	void Path (Tile start, Tile finish) {

		if (start == finish) {
			return;
		}

		open.Add (start);
		start.goal = 0;
		start.fitness = EstimateHeuristic (start, finish);

		while (open.Count != 0) {

			Tile current = null;
			int index = 0;
			float lowestFitness = open[0].fitness;
			for (int i = 0; i < open.Count; i++) {
				if (open[i].fitness < lowestFitness) {
					lowestFitness = open [i].fitness;
					index = i;
				}
			}

			current = open[index];

			open.RemoveAt (index);
			closed.Add (current);

			List<Tile> neighbors = current.GetNeighbors();
			for (int i = 0; i < neighbors.Count; i++) {
				if (closed.Contains (neighbors[i])) {
					continue;
				}

				float tempGoal = current.goal + DistBetween(current, neighbors[i]);
				if (!open.Contains (neighbors[i])) {
					open.Add (neighbors [i]);
				}else if (tempGoal >= neighbors[i].goal) {
					continue;
				}

				neighbors [i].prev = current;
				neighbors [i].goal = tempGoal;
				neighbors [i].fitness = neighbors[i].goal + EstimateHeuristic(neighbors[i], finish);
			}
		}

//		PrintPathRetrace (finish);
		enemy.GetComponent<TraversePath> ().BeginTraversingPath (CreatePath (finish));
	}

	float EstimateHeuristic (Tile current, Tile finish) {
		return Mathf.Abs (current.x - finish.x) + Mathf.Abs (current.y - finish.y);
	}

	float DistBetween (Tile current, Tile neighbor) {
		return Mathf.Pow(current.x - neighbor.x, 2) + Mathf.Pow(current.y - neighbor.y, 2);
	}

	void PrintPathRetrace (Tile end) {

		while (end != null) {
			Debug.Log (end.x + ", " + end.y);
			end = end.prev;
		}
	}

	Path CreatePath (Tile finish) {

		Path path = new Path ();

		while (finish != null) {
			path.Push (finish);
			finish = finish.prev;
		}

		return path;
	}
}