326 lines
10 KiB
C#

#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;
}
}