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

public class Baller : MonoBehaviour {
    public GameObject leftHandAnchor;
    public GameObject rightHandAnchor;
    public GameObject ballPrefab;
    public GameObject haloPrefab;
    public GameObject cylinderPrefab;
    public float threshold = 0.03f;
    public float r = 0.1f;
    // For keeping track
    private List<GameObject> BList = new List<GameObject>();
    private List<GameObject> HList = new List<GameObject>();
    private List<GameObject> CList = new List<GameObject>();
    private List<List<GameObject>> VertexList = new List<List<GameObject>>();
    private List<List<GameObject>> EdgeList = new List<List<GameObject>>();
    private List<GameObject> TempEdgeList1 = new List<GameObject>();
    private List<GameObject> TempEdgeList2 = new List<GameObject>();
    private int vertex;
    private int edge;
    private int radius;
    // For adding
    private GameObject clone;
    private GameObject halo;
    private GameObject cylinder;
    static int num;
    // For deleting
    private GameObject nearBall;
    private GameObject nearHalo;
    private Rigidbody rb;

    // Update is called once per frame
    void Update () {
        // Mask to not detect hands when deleting objects in scene
        int layerMask = 1 << 2;
        layerMask = ~layerMask;

        // When to add balls and halos
        if (OVRInput.GetDown(OVRInput.RawButton.B)) {
            if (rightHandAnchor != null) {
                AddBall(rightHandAnchor);
                UpdateRadii();
            }
        }
        if (OVRInput.GetDown(OVRInput.RawButton.Y)) {
            if (leftHandAnchor != null) {
                AddBall(leftHandAnchor);
                UpdateRadii();
            }
        }

        // When to remove balls and halos
        if (OVRInput.GetDown(OVRInput.RawButton.A)) {
            if (rightHandAnchor != null) {
                RemoveBall(rightHandAnchor);
            }
        }
        if (OVRInput.GetDown(OVRInput.RawButton.X)) {
            if (leftHandAnchor != null) {
                RemoveBall(leftHandAnchor);
            }
        }

        // When to change radius of halos
        if(OVRInput.Get(OVRInput.Button.PrimaryThumbstickUp)) {
            r += 0.02f;
            UpdateRadii();
        }

        if(OVRInput.Get(OVRInput.Button.PrimaryThumbstickDown)) {
            r -= 0.02f;
            UpdateRadii();
        }

        // When to adjust connecting cylinders
        foreach(GameObject ball in BList){
            if(ball.transform.hasChanged){
                Debug.Log("Whoa!");
                int vertex = BList.IndexOf(ball);
                foreach(GameObject cylinder in EdgeList[vertex]){
                    int edge = CList.IndexOf(cylinder);
                    cylinder.transform.position = (VertexList[edge][0].transform.position + VertexList[edge][1].transform.position)/2;
                    cylinder.transform.rotation = Quaternion.FromToRotation(Vector3.up, VertexList[edge][0].transform.position - VertexList[edge][1].transform.position);
                    cylinder.transform.localScale = new Vector3(0.015f, Vector3.Distance(VertexList[edge][0].transform.position/2, VertexList[edge][1].transform.position/2), 0.015f);
                }
                ball.transform.hasChanged = false;
            }
        }
    }

    // Adds a ball and a halo at the input anchor position
    void AddBall (GameObject anchor) {
        // Instantiate ball at anchor position
        GameObject clone = (GameObject)Instantiate(ballPrefab, anchor.transform.position, anchor.transform.rotation);
        clone.name = "clone" + num.ToString();
        TempEdgeList1.Clear();
        // Instantiate cylinder between anchor and other clones
        foreach (GameObject existing_ball in BList){
            GameObject cylinder = (GameObject)Instantiate(cylinderPrefab, (anchor.transform.position + existing_ball.transform.position)/2, Quaternion.FromToRotation(Vector3.up, anchor.transform.position - existing_ball.transform.position));
            cylinder.transform.localScale = new Vector3(0.015f, Vector3.Distance(existing_ball.transform.position/2, anchor.transform.position/2), 0.015f);
            int vertex = BList.IndexOf(existing_ball);
            cylinder.name = "cylinder-" + num.ToString() + "-" + vertex.ToString();
            // Add cylinder to lists
            CList.Add(cylinder);
            TempEdgeList1.Add(cylinder);
            //EdgeList[vertex].Add(cylinder);
            //Debug.Log("Ball " + vertex + " now has " + EdgeList[vertex].Count + " neighbors");
            // Add vertices of cylinder to list
            VertexList.Add(new List<GameObject> { clone, existing_ball });
        }
        // Add ball to list
        BList.Add(clone);
        // Add edges of ball to list
        EdgeList.Add(TempEdgeList1);
        // Instantiate halo at anchor position, as child of ball
        GameObject halo = (GameObject)Instantiate(haloPrefab, anchor.transform.position, anchor.transform.rotation);
        halo.name = "halo" + num.ToString();
        halo.transform.SetParent(clone.transform);
        HList.Add(halo);
        // Increase counter
        num++;
    }

    // Removes all balls and halos within a threshold distance of the anchor position
    void RemoveBall (GameObject anchor) {
        // Find objects within threshold of anchor
        RaycastHit[] hits = Physics.SphereCastAll(anchor.transform.position, threshold, anchor.transform.forward, 0f);
        foreach (RaycastHit ball in hits){
            Debug.Log("GONNA DESTROY: " + ball.collider.name);
            nearBall = ball.collider.gameObject;
            nearHalo = nearBall.transform.GetChild(0).gameObject;
            // Remove from lists
            BList.Remove(nearBall);
            HList.Remove(nearHalo);
            // Drop ball to bottom, make invisible
            rb = nearBall.GetComponent<Rigidbody>();
            rb.isKinematic = false;
            rb.useGravity = true;
            Destroy(nearBall.GetComponent<MeshRenderer>());
            Destroy(nearHalo.GetComponent<MeshRenderer>());
        }
    }

    void UpdateRadii () {
        foreach (GameObject halo in HList) {
            halo.transform.localScale = new Vector3(r, r, r);
        }
    }
}