#define Graph_And_Chart_PRO
using UnityEngine;
using ChartAndGraph;
using System.Collections.Generic;
using System;
using System.IO;

public partial class LargeDataFeed : MonoBehaviour, IComparer<DoubleVector2>
{
    public string Category = "Player 1";
    public int DownSampleToPoints = 100;
    List<DoubleVector2> mData = new List<DoubleVector2>();  // the data held by the chart
    double pageSize = 2f;
    double currentPagePosition = 0.0;
    double currentZoom = 0f;
    GraphChartBase graph;
    double mCurrentPageSizeFactor = double.NegativeInfinity;
    public bool LoadExample = true;
    public bool ControlViewPortion = false;
    double? mSlideEndThreshold = null;
    int mStart = 0;
    [HideInInspector]
    public GraphChartBase AlternativeGraph = null;
    void Start()
    {
        graph = GetComponent<GraphChartBase>();
        if (AlternativeGraph != null)
            graph = AlternativeGraph;
        if (LoadExample)
            SetInitialData();
        else
            SetData(mData);
    }

    public DoubleVector2 GetLastPoint()
    {
        if (mData.Count == 0)
            return new DoubleVector2();
        return mData[mData.Count - 1];
    }
    /// <summary>
    /// called with Start(). These will be used to load the data into the large data feed. You should replace this with your own loading logic.
    /// </summary> 
    void SetInitialData()
    {
        List<DoubleVector2> data = new List<DoubleVector2>(250000);
        double x = 0f;
        double y = 200f;
        for (int i = 0; i < 25000; i++)    // initialize with random data
        {
            data.Add(new DoubleVector2(x, y));
            y += UnityEngine.Random.value * 10f - 5f;
            x += UnityEngine.Random.value;
        }
        SetData(data);
    }

    public void SaveToFile(string path)
    {
        using (StreamWriter file = new StreamWriter(path))
        {
            file.WriteLine(mData.Count);
            for (int i = 0; i < mData.Count; i++)
            {
                DoubleVector2 item = mData[i];
                file.WriteLine(item.x);
                file.WriteLine(item.y);
            }
        }
    }
    public void LoadFromFile(string path)
    {
        try
        {
            List<DoubleVector2> data = new List<DoubleVector2>();
            using (StreamReader file = new StreamReader(path))
            {
                int count = int.Parse(file.ReadLine());
                for (int i = 0; i < count; i++)
                {
                    double x = double.Parse(file.ReadLine());
                    double y = double.Parse(file.ReadLine());
                    data.Add(new DoubleVector2(x, y));
                }
            }
            SetData(data);
        }
        catch (Exception)
        {
            throw new Exception("Invalid file format");
        }
    }

    /// <summary>
    /// vertify's that the graph data is sorted so it can be searched using a binary search.
    /// </summary>
    /// <returns></returns>
    bool VerifySorted(List<DoubleVector2> data)
    {
        if (data == null)
            return true;
        for (int i = 1; i < data.Count; i++)
        {
            if (data[i].x < data[i - 1].x)
                return false;
        }
        return true;
    }

    partial void OnDataLoaded();
    /// <summary>
    /// set the data of the large data graph
    /// </summary>
    /// <param name="data"></param>
    public void SetData(List<DoubleVector2> data)
    {
        if (data == null)
            data = new List<DoubleVector2>(); // set up an empty list instead of null
        if (VerifySorted(data) == false)
        {
            Debug.LogWarning("The data used with large data feed must be sorted acoording to the x value, aborting operation");
            return;
        }
        mData = data;
        OnDataLoaded();
        LoadPage(currentPagePosition); // load the page at position 0
    }

    int FindClosestIndex(double position) // if you want to know what is index is currently displayed . use binary search to find it
    {
        //NOTE :: this method assumes your data is sorted !!! 
        int res = mData.BinarySearch(new DoubleVector2(position, 0.0), this);
        if (res >= 0)
            return res;
        return ~res;
    }

    double PageSizeFactor
    {
        get
        {
            return pageSize * graph.DataSource.HorizontalViewSize;
        }
    }
    void AdjustVerticalView()
    {
        int start, end;
        findAdjustPosition(graph.HorizontalScrolling,graph.DataSource.HorizontalViewSize,out start,out end);
        double minY = double.MaxValue, maxY = double.MinValue;

        bool show = graph.AutoScrollHorizontally;
        if (mData.Count == 0)
            show = true;
        else
        {
            double viewX = mData[mData.Count - 1].x;
            double pageStartThreshold = currentPagePosition - mCurrentPageSizeFactor;
            double pageEndThreshold = currentPagePosition + mCurrentPageSizeFactor - graph.DataSource.HorizontalViewSize;
            if (viewX >= pageStartThreshold && viewX <= pageEndThreshold)
                show = true;
        }
        if (show)
            --end;
        for (int i=start; i<=end; i++)
        {
            double y = mData[i].y;
            minY = Math.Min(y, minY);
            maxY = Math.Max(y, maxY);
        }

        if(show)
        {
            DoubleVector3 p;
            if (graph.DataSource.GetLastPoint(Category, out p))
            {
                minY = Math.Min(p.y, minY);
                maxY = Math.Max(p.y, maxY);
            }
        }
        graph.VerticalScrolling = minY;
        graph.DataSource.VerticalViewSize = maxY - minY;
    }
    void findAdjustPosition(double position,double size,out int start,out int end)
    {
        int index = FindClosestIndex(position); // use binary search to find the closest position to the current scroll point

        double endPosition = position + size;

        start = index;

        for (end = index; end < mData.Count; end++)
        {
            if (mData[end].x > endPosition) // take the first point that is out of the page
                break;
        }

    }
    void findPointsForPage(double position, out int start, out int end) // given a page position , find the right most and left most indices in the data for that page. 
    {
        int index = FindClosestIndex(position); // use binary search to find the closest position to the current scroll point

        double endPosition = position + PageSizeFactor;
        double startPosition = position - PageSizeFactor;

        //starting from the current index , we find the page boundries
        for (start = index; start > 0; start--)
        {
            if (mData[start].x < startPosition) // take the first point that is out of the page. so the graph doesn't break at the edge
                break;
        }

        for (end = index; end < mData.Count; end++)
        {
            if (mData[end].x > endPosition) // take the first point that is out of the page
                break;
        }

    }

    public void Update()
    {
        if (graph != null)
        {

            //check the scrolling position of the graph. if we are past the view size , load a new page
            double pageStartThreshold = currentPagePosition - mCurrentPageSizeFactor;
            double pageEnd = currentPagePosition + mCurrentPageSizeFactor;
            if (mSlideEndThreshold.HasValue)
                pageEnd = Math.Max(mSlideEndThreshold.Value, pageEnd);
            double pageEndThreshold = pageEnd - graph.DataSource.HorizontalViewSize*1.0001;
            if (graph.HorizontalScrolling < pageStartThreshold || graph.HorizontalScrolling > pageEndThreshold || currentZoom >= graph.DataSource.HorizontalViewSize * 2f)
            {
                currentZoom = graph.DataSource.HorizontalViewSize;
                mCurrentPageSizeFactor = PageSizeFactor * 0.9f;
                LoadPage(graph.HorizontalScrolling);
            }
            if (ControlViewPortion)
                AdjustVerticalView();
        }
    }

    void LoadWithoutDownSampling(int start, int end)
    {
        for (int i = start; i < end; i++) // load the data
        {
            graph.DataSource.AddPointToCategory(Category, mData[i].x, mData[i].y);
        }
    }

    void LoadWithDownSampling(int start, int end)
    {
        int total = end - start;

        if (DownSampleToPoints >= total)
        {
            LoadWithoutDownSampling(start, end);
            return;
        }

        double sampleCount = ((double)total) / (double)DownSampleToPoints;
        // graph.DataSource.AddPointToCategory(Category, mData[start].x, mData[start].y);
        for (int i = 0; i < DownSampleToPoints; i++)
        {
            int fractionStart = start + (int)(i * sampleCount); // the first point with a fraction
            int fractionEnd = start + (int)((i + 1) * sampleCount); // the first point with a fraction
            fractionEnd = Math.Min(fractionEnd, mData.Count - 1);
            double x = 0, y = 0;
            double divide = 0.0;
            for (int j = fractionStart; j < fractionEnd; j++)  // avarge the points
            {
                x += mData[j].x;
                y += mData[j].y;
                divide++;
            }
            if (divide > 0.0)
            {
                x /= divide;
                y /= divide;
                graph.DataSource.AddPointToCategory(Category, x, y);
            }
            else
                Debug.Log("error");
        }
        //   graph.DataSource.AddPointToCategory(Category, mData[last].x, mData[last].y);
    }

    public int GetIndex(int inGraphIndex)
    {
        return mStart + inGraphIndex;
    }

    void LoadPage(double pagePosition)
    {
        mSlideEndThreshold = null;
        if (graph != null)
        {

            Debug.Log("Loading page :" + pagePosition);
            graph.DataSource.StartBatch(); // call start batch 
            graph.DataSource.HorizontalViewOrigin = 0;
            int start, end;
            findPointsForPage(pagePosition, out start, out end); // get the page edges
            graph.DataSource.ClearCategory(Category); // clear the cateogry
            mStart = start;
            if (DownSampleToPoints <= 0)
                LoadWithoutDownSampling(start, end);
            else
                LoadWithDownSampling(start, end);
            graph.DataSource.EndBatch();
            graph.HorizontalScrolling = pagePosition;
        }
        currentPagePosition = pagePosition;
    }


    public int Compare(DoubleVector2 x, DoubleVector2 y)
    {
        if (x.x < y.x)
            return -1;
        if (x.x > y.x)
            return 1;
        return 0;
    }
}