Trials of the Archmage

using UnityEngine;
using UnityEngine.UI;
using UnityEngine.EventSystems;

using System;
using System.Collections;
using System.Collections.Generic;
using System.IO;

using PDollarGestureRecognizer;

public class GestureController : MonoBehaviour {
	public Transform gestureOnScreenPrefab;

	private List trainingSet = new List();

	private List points = new List();
	private int strokeId = -1;

	private Vector3 virtualKeyPosition = Vector2.zero;

	private RuntimePlatform platform;
	private int vertexCount = 0;
	public GameObject gestureEffect;
    private List gestureLinesRenderer = new List();
	private LineRenderer currentGestureLineRenderer;
	private SpellCastingController spellCastingController;
    private bool isDrawingGesture = false;
    private bool isGestureButtonDown = false;
	private bool isWaitingForGesture = false;

	//GUI
	[HideInInspector]
	public string message;
	[HideInInspector]
	public float score;
	private bool recognized;
	private string newGestureName = "";
	private GameObject currentGestureEffect = null;

    // A buffer for continuous drawings
    public float gestureBuffer = 0.1f;

	void Start () {
		spellCastingController = GameObject.Find("GameManagement/Controllers/SpellCastingController").GetComponent();
		platform = Application.platform;

		//Load pre-made gestures
		TextAsset[] gesturesXml = Resources.LoadAll("GestureSet/10-stylus-MEDIUM/");
		foreach (TextAsset gestureXml in gesturesXml)
			trainingSet.Add(GestureIO.ReadGestureFromXML(gestureXml.text));

		//Load user custom gestures
		//Debug.Log(Application.persistentDataPath);
		string[] filePaths = Directory.GetFiles(Application.persistentDataPath, "*.xml");
		foreach (string filePath in filePaths)
			trainingSet.Add(GestureIO.ReadGestureFromFile(filePath));
	}

	void Update () {
        if (platform == RuntimePlatform.Android || platform == RuntimePlatform.IPhonePlayer) {
			if (Input.touchCount > 0) {
				virtualKeyPosition = new Vector3(Input.GetTouch(0).position.x, Input.GetTouch(0).position.y);
			}
		} else {
			if (Input.GetMouseButton(0)) {
				virtualKeyPosition = new Vector3(Input.mousePosition.x, Input.mousePosition.y);
			}
		}

		if (currentGestureEffect != null) {
			//Debug.Log(currentGestureEffect.transform.position);
			currentGestureEffect.transform.position = Camera.main.ScreenToWorldPoint(new Vector3(virtualKeyPosition.x, virtualKeyPosition.y, 10));
		}

        if (isDrawingGesture && !isWaitingForGesture) {
            points.Add(new Point(virtualKeyPosition.x, -virtualKeyPosition.y, strokeId));
			if (currentGestureLineRenderer != null) {
				currentGestureLineRenderer.SetVertexCount(++vertexCount);
				currentGestureLineRenderer.SetPosition(vertexCount - 1, Camera.main.ScreenToWorldPoint(new Vector3(virtualKeyPosition.x, virtualKeyPosition.y, 10)));
			}
        }
	}

    public void StartGestureDrawing(){
        isGestureButtonDown = true;
        // New Gesture!
        if (!isDrawingGesture) {
            Transform tmpGesture = Instantiate(gestureOnScreenPrefab, transform.position, transform.rotation) as Transform;
		    currentGestureLineRenderer = tmpGesture.GetComponent();
            currentGestureLineRenderer.SetVertexCount(0);
            vertexCount = 0;

			currentGestureEffect = Instantiate(gestureEffect, transform.position, Quaternion.identity);
        }
        isDrawingGesture = true;
    }

    public void EndGestureDrawing(){
        isGestureButtonDown = false;
        StartCoroutine(LateEndDrawing());
    }

    IEnumerator LateEndDrawing() {
		isWaitingForGesture = true;
        yield return new WaitForSecondsRealtime(gestureBuffer);
		isWaitingForGesture = false;
        if (!isGestureButtonDown) {
            if (isDrawingGesture) {
                // It means we get a gesture now!
                Recognize();
				spellCastingController.EndDrawingGesture(message, score);
                ClearStrokes();
            }
            isDrawingGesture = false;
        }
    }

	public void ClearStrokes(){
		strokeId = -1;

		points.Clear();
		StartCoroutine(LateDestroy(currentGestureEffect.gameObject, 2f));
		currentGestureEffect = null;
        currentGestureLineRenderer.SetVertexCount(0);
		Destroy(currentGestureLineRenderer.gameObject);

	}

	IEnumerator LateDestroy(GameObject go, float waitTime)
	{
		yield return new WaitForSeconds(waitTime);
		Destroy(go);
	}

	List CleanUpNearPoints(List points) {
		List newPoints = new List();
		newPoints.Add(points[0]);
		for (int i = 1; i < points.Count - 1; i++) {
			if (Point.GetDistance(points[i], points[i - 1]) > 5)
				newPoints.Add(points[i]);
		}
		return newPoints;
	}

	void Recognize(){
		recognized = true;
		List pointsForRecognize = CleanUpNearPoints(points);
		Gesture candidate = new Gesture(CleanUpNearPoints(points).ToArray());

		// A random touch on screen
		if (pointsForRecognize.Count == 1) {
			message = "NULL";
			score = 0;
			return;
		}

		Result gestureResult = PointCloudRecognizer.Classify(candidate, trainingSet.ToArray());

		message = gestureResult.GestureClass;
		score = gestureResult.Score;
		//Debug.Log(gestureResult.GestureClass + " " + gestureResult.Score);
	}

}