mirror of
https://github.com/XCharts-Team/XCharts.git
synced 2026-05-16 13:30:10 +00:00
393 lines
14 KiB
C#
393 lines
14 KiB
C#
using System.Text;
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
using UnityEngine.EventSystems;
|
|
using UnityEngine.UI;
|
|
|
|
namespace XCharts
|
|
{
|
|
[AddComponentMenu("XCharts/PieChart", 15)]
|
|
[ExecuteInEditMode]
|
|
[RequireComponent(typeof(RectTransform))]
|
|
[DisallowMultipleComponent]
|
|
public class PieChart : BaseChart
|
|
{
|
|
private class PieTempData
|
|
{
|
|
public List<float> angleList = new List<float>();
|
|
public Vector2 center;
|
|
public float insideRadius;
|
|
public float outsideRadius;
|
|
public float dataMax;
|
|
public float dataTotal;
|
|
}
|
|
|
|
[SerializeField] private Pie m_Pie = Pie.defaultPie;
|
|
|
|
private bool isDrawPie;
|
|
private bool m_IsEnterLegendButtom;
|
|
private List<PieTempData> m_PieTempDataList = new List<PieTempData>();
|
|
|
|
public Pie pie { get { return m_Pie; } }
|
|
|
|
/// <summary>
|
|
/// Where legend is activated.
|
|
/// </summary>
|
|
/// <param name="legendName">the name of legend</param>
|
|
/// <returns></returns>
|
|
public override bool IsLegendActive(string legendName)
|
|
{
|
|
foreach (var serie in m_Series.series)
|
|
{
|
|
foreach (var serieData in serie.data)
|
|
{
|
|
if (string.IsNullOrEmpty(serieData.name))
|
|
{
|
|
if (string.IsNullOrEmpty(legendName)) return true;
|
|
}
|
|
else if (serieData.name.Equals(legendName))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
protected override void Awake()
|
|
{
|
|
base.Awake();
|
|
raycastTarget = false;
|
|
}
|
|
|
|
#if UNITY_EDITOR
|
|
protected override void Reset()
|
|
{
|
|
base.Reset();
|
|
m_Pie = Pie.defaultPie;
|
|
m_Title.text = "PieChart";
|
|
RemoveData();
|
|
AddSerie("serie1", SerieType.Pie);
|
|
AddData(0, 70, "pie1");
|
|
AddData(0, 20, "pie2");
|
|
AddData(0, 10, "pie3");
|
|
}
|
|
#endif
|
|
|
|
protected override void Update()
|
|
{
|
|
base.Update();
|
|
if (!isDrawPie) RefreshChart();
|
|
}
|
|
|
|
Dictionary<string, int> serieNameSet = new Dictionary<string, int>();
|
|
protected override void DrawChart(VertexHelper vh)
|
|
{
|
|
base.DrawChart(vh);
|
|
serieNameSet.Clear();
|
|
int serieNameCount = -1;
|
|
bool isClickOffset = false;
|
|
bool isDataHighlight = false;
|
|
for (int i = 0; i < m_Series.Count; i++)
|
|
{
|
|
var serie = m_Series.series[i];
|
|
serie.index = i;
|
|
|
|
if (!serie.show)
|
|
{
|
|
continue;
|
|
}
|
|
if (serie.clickOffset) isClickOffset = true;
|
|
PieTempData tempData;
|
|
if (i < m_PieTempDataList.Count)
|
|
{
|
|
tempData = m_PieTempDataList[i];
|
|
tempData.angleList.Clear();
|
|
}
|
|
else
|
|
{
|
|
tempData = new PieTempData();
|
|
m_PieTempDataList.Add(tempData);
|
|
}
|
|
tempData.angleList.Clear();
|
|
tempData.dataMax = serie.yMax;
|
|
tempData.dataTotal = serie.yTotal;
|
|
UpdatePieCenter(serie);
|
|
var data = serie.data;
|
|
|
|
float totalDegree = 360;
|
|
float startDegree = 0;
|
|
for (int n = 0; n < data.Count; n++)
|
|
{
|
|
var serieData = data[n];
|
|
float value = serieData.data[1];
|
|
string dataName = serieData.name;
|
|
Color color;
|
|
if (string.IsNullOrEmpty(dataName))
|
|
{
|
|
serieNameCount++;
|
|
color = m_ThemeInfo.GetColor(serieNameCount);
|
|
}
|
|
else if (!serieNameSet.ContainsKey(dataName))
|
|
{
|
|
serieNameSet.Add(dataName, serieNameCount);
|
|
serieNameCount++;
|
|
color = m_ThemeInfo.GetColor(serieNameCount);
|
|
}
|
|
else
|
|
{
|
|
color = m_ThemeInfo.GetColor(serieNameSet[dataName]);
|
|
}
|
|
if (!serieData.show)
|
|
{
|
|
tempData.angleList.Add(0);
|
|
continue;
|
|
}
|
|
float degree = totalDegree * value / tempData.dataTotal;
|
|
float toDegree = startDegree + degree;
|
|
|
|
float outSideRadius = serie.roseType > 0 ?
|
|
tempData.insideRadius + (tempData.outsideRadius - tempData.insideRadius) * value / tempData.dataMax :
|
|
tempData.outsideRadius;
|
|
if (serieData.highlighted)
|
|
{
|
|
isDataHighlight = true;
|
|
color *= 1.2f;
|
|
outSideRadius += m_Pie.tooltipExtraRadius;
|
|
}
|
|
var offset = serie.space;
|
|
if (serie.clickOffset && serieData.selected)
|
|
{
|
|
offset += m_Pie.selectedOffset;
|
|
}
|
|
if (offset > 0)
|
|
{
|
|
var halfDegree = (toDegree - startDegree) / 2;
|
|
float offsetRadius = serie.space / Mathf.Sin(halfDegree * Mathf.Deg2Rad);
|
|
var insideRadius = tempData.insideRadius - offsetRadius;
|
|
var outsideRadius = outSideRadius - offsetRadius;
|
|
if (serie.clickOffset && serieData.selected)
|
|
{
|
|
offsetRadius += m_Pie.selectedOffset;
|
|
if (insideRadius > 0) insideRadius += m_Pie.selectedOffset;
|
|
outsideRadius += m_Pie.selectedOffset;
|
|
}
|
|
float currAngle = (startDegree + halfDegree) * Mathf.Deg2Rad;
|
|
var offestCenter = new Vector3(tempData.center.x + offsetRadius * Mathf.Sin(currAngle),
|
|
tempData.center.y + offsetRadius * Mathf.Cos(currAngle));
|
|
|
|
ChartHelper.DrawDoughnut(vh, offestCenter, insideRadius, outsideRadius,
|
|
startDegree, toDegree, color);
|
|
}
|
|
else
|
|
{
|
|
ChartHelper.DrawDoughnut(vh, tempData.center, tempData.insideRadius, outSideRadius,
|
|
startDegree, toDegree, color);
|
|
}
|
|
isDrawPie = true;
|
|
tempData.angleList.Add(toDegree);
|
|
startDegree = toDegree;
|
|
}
|
|
}
|
|
raycastTarget = isClickOffset && isDataHighlight;
|
|
}
|
|
|
|
protected override void OnLegendButtonClick(int index, string legendName)
|
|
{
|
|
bool active = CheckDataShow(legendName);
|
|
var bgColor1 = active ? m_ThemeInfo.GetColor(index) : m_ThemeInfo.legendUnableColor;
|
|
m_Legend.UpdateButtonColor(legendName, bgColor1);
|
|
RefreshChart();
|
|
}
|
|
|
|
protected override void OnLegendButtonEnter(int index, string legendName)
|
|
{
|
|
m_IsEnterLegendButtom = true;
|
|
CheckDataHighlighted(legendName);
|
|
RefreshChart();
|
|
}
|
|
|
|
protected override void OnLegendButtonExit(int index, string legendName)
|
|
{
|
|
m_IsEnterLegendButtom = false;
|
|
CheckDataHighlighted(legendName);
|
|
RefreshChart();
|
|
}
|
|
|
|
private bool CheckDataShow(string legendName)
|
|
{
|
|
bool show = false;
|
|
foreach (var serie in m_Series.series)
|
|
{
|
|
foreach (var data in serie.data)
|
|
{
|
|
if (legendName.Equals(data.name))
|
|
{
|
|
data.show = !data.show;
|
|
data.highlighted = false;
|
|
if (data.show) show = true;
|
|
}
|
|
}
|
|
}
|
|
return show;
|
|
}
|
|
|
|
private bool CheckDataHighlighted(string legendName)
|
|
{
|
|
bool show = false;
|
|
foreach (var serie in m_Series.series)
|
|
{
|
|
foreach (var data in serie.data)
|
|
{
|
|
if (legendName.Equals(data.name))
|
|
{
|
|
data.highlighted = !data.highlighted;
|
|
if (data.highlighted) show = true;
|
|
}
|
|
}
|
|
}
|
|
return show;
|
|
}
|
|
|
|
private void UpdatePieCenter(Serie serie)
|
|
{
|
|
if (serie.center.Length < 2) return;
|
|
var tempData = m_PieTempDataList[serie.index];
|
|
var centerX = serie.center[0] <= 1 ? chartWidth * serie.center[0] : serie.center[0];
|
|
var centerY = serie.center[1] <= 1 ? chartHeight * serie.center[1] : serie.center[1];
|
|
tempData.center = new Vector2(centerX, centerY);
|
|
var minWidth = Mathf.Min(chartWidth, chartHeight);
|
|
tempData.insideRadius = serie.radius[0] <= 1 ? minWidth * serie.radius[0] : serie.radius[0];
|
|
tempData.outsideRadius = serie.radius[1] <= 1 ? minWidth * serie.radius[1] : serie.radius[1];
|
|
}
|
|
|
|
protected override void CheckTootipArea(Vector2 local)
|
|
{
|
|
if (m_IsEnterLegendButtom) return;
|
|
m_Tooltip.dataIndex.Clear();
|
|
bool selected = false;
|
|
for (int i = 0; i < m_PieTempDataList.Count; i++)
|
|
{
|
|
var serie = m_Series.GetSerie(i);
|
|
var tempData = m_PieTempDataList[i];
|
|
int index = GetPosPieIndex(tempData, local);
|
|
m_Tooltip.dataIndex.Add(index);
|
|
|
|
bool refresh = false;
|
|
for (int j = 0; j < serie.data.Count; j++)
|
|
{
|
|
var serieData = serie.data[j];
|
|
if (serieData.highlighted != (j == index)) refresh = true;
|
|
serieData.highlighted = j == index;
|
|
}
|
|
|
|
if (index >= 0) selected = true;
|
|
if (refresh) RefreshChart();
|
|
}
|
|
if (selected)
|
|
{
|
|
m_Tooltip.UpdateContentPos(new Vector2(local.x + 18, local.y - 25));
|
|
RefreshTooltip();
|
|
}
|
|
else
|
|
{
|
|
m_Tooltip.SetActive(false);
|
|
}
|
|
}
|
|
|
|
private int GetPosPieIndex(PieTempData tempData, Vector2 local)
|
|
{
|
|
var dist = Vector2.Distance(local, tempData.center);
|
|
if (dist < tempData.insideRadius || dist > tempData.outsideRadius) return -1;
|
|
Vector2 dir = local - tempData.center;
|
|
float angle = VectorAngle(Vector2.up, dir);
|
|
var angleList = tempData.angleList;
|
|
for (int i = angleList.Count - 1; i >= 0; i--)
|
|
{
|
|
if (i == 0)
|
|
{
|
|
if (angle <= angleList[i]) return 0;
|
|
}
|
|
else if (angle <= angleList[i] && angle > angleList[i - 1])
|
|
{
|
|
return i;
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
float VectorAngle(Vector2 from, Vector2 to)
|
|
{
|
|
float angle;
|
|
|
|
Vector3 cross = Vector3.Cross(from, to);
|
|
angle = Vector2.Angle(from, to);
|
|
angle = cross.z > 0 ? -angle : angle;
|
|
angle = (angle + 360) % 360;
|
|
return angle;
|
|
}
|
|
|
|
StringBuilder sb = new StringBuilder();
|
|
protected override void RefreshTooltip()
|
|
{
|
|
base.RefreshTooltip();
|
|
bool showTooltip = false;
|
|
for (int i = 0; i < m_PieTempDataList.Count; i++)
|
|
{
|
|
int index = m_Tooltip.dataIndex[i];
|
|
if (index < 0) continue;
|
|
showTooltip = true;
|
|
var serie = m_Series.GetSerie(i);
|
|
string key = serie.data[index].name;
|
|
if (string.IsNullOrEmpty(key)) key = m_Legend.GetData(index);
|
|
float value = serie.data[index].data[1];
|
|
sb.Length = 0;
|
|
if (!string.IsNullOrEmpty(serie.name))
|
|
{
|
|
sb.Append(serie.name).Append("\n");
|
|
}
|
|
sb.Append("<color=#").Append(m_ThemeInfo.GetColorStr(index)).Append(">● </color>")
|
|
.Append(key).Append(": ").Append(ChartCached.FloatToStr(value));
|
|
m_Tooltip.UpdateContentText(sb.ToString());
|
|
|
|
var pos = m_Tooltip.GetContentPos();
|
|
if (pos.x + m_Tooltip.width > chartWidth)
|
|
{
|
|
pos.x = chartWidth - m_Tooltip.width;
|
|
}
|
|
if (pos.y - m_Tooltip.height < 0)
|
|
{
|
|
pos.y = m_Tooltip.height;
|
|
}
|
|
m_Tooltip.UpdateContentPos(pos);
|
|
}
|
|
m_Tooltip.SetActive(showTooltip);
|
|
}
|
|
|
|
public override void OnPointerDown(PointerEventData eventData)
|
|
{
|
|
Vector2 local;
|
|
if (!RectTransformUtility.ScreenPointToLocalPointInRectangle(rectTransform,
|
|
eventData.position, canvas.worldCamera, out local))
|
|
{
|
|
return;
|
|
}
|
|
for (int i = 0; i < m_PieTempDataList.Count; i++)
|
|
{
|
|
var tempData = m_PieTempDataList[i];
|
|
int index = GetPosPieIndex(tempData, local);
|
|
if (index >= 0)
|
|
{
|
|
var serie = m_Series.GetSerie(i);
|
|
for (int j = 0; j < serie.data.Count; j++)
|
|
{
|
|
if (j == index) serie.data[j].selected = !serie.data[j].selected;
|
|
else serie.data[j].selected = false;
|
|
}
|
|
}
|
|
}
|
|
RefreshChart();
|
|
}
|
|
}
|
|
}
|