478 lines
17 KiB
C#
478 lines
17 KiB
C#
using Graphing.Graphables;
|
|
using Graphing.Parts;
|
|
using System.Drawing.Drawing2D;
|
|
|
|
namespace Graphing.Forms;
|
|
|
|
public partial class GraphForm : Form
|
|
{
|
|
public static readonly Color MainAxisColor = Color.Black;
|
|
public static readonly Color SemiAxisColor = Color.FromArgb(unchecked((int)0xFF_999999));
|
|
public static readonly Color QuarterAxisColor = Color.FromArgb(unchecked((int)0xFF_E0E0E0));
|
|
public static readonly Color UnitsTextColor = Color.Black;
|
|
|
|
public Float2 ScreenCenter { get; private set; }
|
|
public Float2 Dpi { get; private set; }
|
|
|
|
public float DpiFloat { get; private set; }
|
|
|
|
public double ZoomLevel
|
|
{
|
|
get => _zoomLevel;
|
|
set
|
|
{
|
|
double oldZoom = ZoomLevel;
|
|
|
|
_zoomLevel = Math.Clamp(value, 1e-5, 1e3);
|
|
|
|
int totalSegments = 0;
|
|
foreach (Graphable able in ables) totalSegments += able.GetItemsToRender(this).Count();
|
|
|
|
if (totalSegments > 10_000)
|
|
{
|
|
_zoomLevel = oldZoom;
|
|
return; // Too many segments, stop.
|
|
}
|
|
}
|
|
}
|
|
private double _zoomLevel;
|
|
|
|
private readonly Point initialWindowPos;
|
|
private readonly Size initialWindowSize;
|
|
|
|
public Graphable[] Graphables => ables.ToArray();
|
|
|
|
public Float2 MinVisibleGraph => ScreenSpaceToGraphSpace(new Int2(0, ClientRectangle.Height));
|
|
public Float2 MaxVisibleGraph => ScreenSpaceToGraphSpace(new Int2(ClientRectangle.Width, 0));
|
|
|
|
private readonly List<Graphable> ables;
|
|
|
|
public GraphForm(string title)
|
|
{
|
|
SetStyle(ControlStyles.OptimizedDoubleBuffer, true);
|
|
SetStyle(ControlStyles.AllPaintingInWmPaint, true);
|
|
SetStyle(ControlStyles.UserPaint, true);
|
|
|
|
InitializeComponent();
|
|
Text = title;
|
|
|
|
Graphics tempG = CreateGraphics();
|
|
Dpi = new(tempG.DpiX, tempG.DpiY);
|
|
tempG.Dispose();
|
|
|
|
DpiFloat = (float)((Dpi.x + Dpi.y) / 2);
|
|
|
|
ables = [];
|
|
ZoomLevel = 1;
|
|
initialWindowPos = Location;
|
|
initialWindowSize = Size;
|
|
}
|
|
|
|
public Int2 GraphSpaceToScreenSpace(Float2 graphPoint)
|
|
{
|
|
graphPoint.y = -graphPoint.y;
|
|
|
|
graphPoint.x -= ScreenCenter.x;
|
|
graphPoint.y -= ScreenCenter.y;
|
|
|
|
graphPoint.x *= Dpi.x / ZoomLevel;
|
|
graphPoint.y *= Dpi.y / ZoomLevel;
|
|
|
|
graphPoint.x += ClientRectangle.Width / 2.0;
|
|
graphPoint.y += ClientRectangle.Height / 2.0;
|
|
|
|
return new((int)graphPoint.x, (int)graphPoint.y);
|
|
}
|
|
public Float2 ScreenSpaceToGraphSpace(Int2 screenPoint)
|
|
{
|
|
Float2 result = new(screenPoint.x, screenPoint.y);
|
|
|
|
result.x -= ClientRectangle.Width / 2.0;
|
|
result.y -= ClientRectangle.Height / 2.0;
|
|
|
|
result.x /= Dpi.x / ZoomLevel;
|
|
result.y /= Dpi.y / ZoomLevel;
|
|
|
|
result.x += ScreenCenter.x;
|
|
result.y += ScreenCenter.y;
|
|
|
|
result.y = -result.y;
|
|
|
|
return result;
|
|
}
|
|
|
|
protected virtual void PaintGrid(Graphics g)
|
|
{
|
|
double axisScale = Math.Pow(2, Math.Round(Math.Log2(ZoomLevel)));
|
|
|
|
// Draw horizontal/vertical quarter-axis.
|
|
Brush quarterBrush = new SolidBrush(QuarterAxisColor);
|
|
Pen quarterPen = new(quarterBrush, DpiFloat * 2 / 192);
|
|
|
|
for (double x = Math.Ceiling(MinVisibleGraph.x * 4 / axisScale) * axisScale / 4; x <= Math.Floor(MaxVisibleGraph.x * 4 / axisScale) * axisScale / 4; x += axisScale / 4)
|
|
{
|
|
Int2 startPos = GraphSpaceToScreenSpace(new Float2(x, MinVisibleGraph.y)),
|
|
endPos = GraphSpaceToScreenSpace(new Float2(x, MaxVisibleGraph.y));
|
|
g.DrawLine(quarterPen, startPos, endPos);
|
|
}
|
|
for (double y = Math.Ceiling(MinVisibleGraph.y * 4 / axisScale) * axisScale / 4; y <= Math.Floor(MaxVisibleGraph.y * 4 / axisScale) * axisScale / 4; y += axisScale / 4)
|
|
{
|
|
Int2 startPos = GraphSpaceToScreenSpace(new Float2(MinVisibleGraph.x, y)),
|
|
endPos = GraphSpaceToScreenSpace(new Float2(MaxVisibleGraph.x, y));
|
|
g.DrawLine(quarterPen, startPos, endPos);
|
|
}
|
|
|
|
// Draw horizontal/vertical semi-axis.
|
|
Brush semiBrush = new SolidBrush(SemiAxisColor);
|
|
Pen semiPen = new(semiBrush, DpiFloat * 2 / 192);
|
|
|
|
for (double x = Math.Ceiling(MinVisibleGraph.x / axisScale) * axisScale; x <= Math.Floor(MaxVisibleGraph.x / axisScale) * axisScale; x += axisScale)
|
|
{
|
|
Int2 startPos = GraphSpaceToScreenSpace(new Float2(x, MinVisibleGraph.y)),
|
|
endPos = GraphSpaceToScreenSpace(new Float2(x, MaxVisibleGraph.y));
|
|
g.DrawLine(semiPen, startPos, endPos);
|
|
}
|
|
for (double y = Math.Ceiling(MinVisibleGraph.y / axisScale) * axisScale; y <= Math.Floor(MaxVisibleGraph.y / axisScale) * axisScale; y += axisScale)
|
|
{
|
|
Int2 startPos = GraphSpaceToScreenSpace(new Float2(MinVisibleGraph.x, y)),
|
|
endPos = GraphSpaceToScreenSpace(new Float2(MaxVisibleGraph.x, y));
|
|
g.DrawLine(semiPen, startPos, endPos);
|
|
}
|
|
|
|
Brush mainLineBrush = new SolidBrush(MainAxisColor);
|
|
Pen mainLinePen = new(mainLineBrush, DpiFloat * 3 / 192);
|
|
|
|
// Draw the main axis (on top of the semi axis).
|
|
Int2 startCenterY = GraphSpaceToScreenSpace(new Float2(0, MinVisibleGraph.y)),
|
|
endCenterY = GraphSpaceToScreenSpace(new Float2(0, MaxVisibleGraph.y)),
|
|
startCenterX = GraphSpaceToScreenSpace(new Float2(MinVisibleGraph.x, 0)),
|
|
endCenterX = GraphSpaceToScreenSpace(new Float2(MaxVisibleGraph.x, 0));
|
|
|
|
g.DrawLine(mainLinePen, startCenterX, endCenterX);
|
|
g.DrawLine(mainLinePen, startCenterY, endCenterY);
|
|
}
|
|
protected virtual void PaintUnits(Graphics g)
|
|
{
|
|
double axisScale = Math.Pow(2, Math.Round(Math.Log(ZoomLevel, 2)));
|
|
Brush textBrush = new SolidBrush(UnitsTextColor);
|
|
Font textFont = new(Font.Name, 9, FontStyle.Regular);
|
|
|
|
// X-axis
|
|
int minX = (int)(DpiFloat * 50 / 192),
|
|
maxX = ClientRectangle.Height - (int)(DpiFloat * 40 / 192);
|
|
for (double x = Math.Ceiling(MinVisibleGraph.x / axisScale) * axisScale; x <= MaxVisibleGraph.x; x += axisScale)
|
|
{
|
|
if (x == 0) x = 0; // Fixes -0
|
|
|
|
Int2 screenPos = GraphSpaceToScreenSpace(new Float2(x, 0));
|
|
|
|
if (screenPos.y < minX) screenPos.y = minX;
|
|
else if (screenPos.y > maxX) screenPos.y = maxX;
|
|
|
|
g.DrawString($"{x}", textFont, textBrush, screenPos.x, screenPos.y);
|
|
}
|
|
|
|
// Y-axis
|
|
int minY = (int)(DpiFloat * 10 / 192);
|
|
for (double y = Math.Ceiling(MinVisibleGraph.y / axisScale) * axisScale; y <= MaxVisibleGraph.y; y += axisScale)
|
|
{
|
|
if (y == 0) continue;
|
|
|
|
Int2 screenPos = GraphSpaceToScreenSpace(new Float2(0, y));
|
|
|
|
string result = y.ToString();
|
|
int maxY = ClientRectangle.Width - (int)(DpiFloat * (textFont.Height * result.Length * 0.40 + 15) / 192);
|
|
|
|
if (screenPos.x < minY) screenPos.x = minY;
|
|
else if (screenPos.x > maxY) screenPos.x = maxY;
|
|
|
|
g.DrawString($"{y}", textFont, textBrush, screenPos.x, screenPos.y);
|
|
}
|
|
}
|
|
|
|
protected override void OnPaint(PaintEventArgs e)
|
|
{
|
|
Graphics g = e.Graphics;
|
|
g.SmoothingMode = SmoothingMode.HighQuality;
|
|
|
|
Brush background = new SolidBrush(Color.White);
|
|
g.FillRectangle(background, e.ClipRectangle);
|
|
|
|
PaintGrid(g);
|
|
PaintUnits(g);
|
|
|
|
Point clientMousePos = PointToClient(Cursor.Position);
|
|
Float2 graphMousePos = ScreenSpaceToGraphSpace(new(clientMousePos.X,
|
|
clientMousePos.Y));
|
|
|
|
// Draw the actual graphs.
|
|
for (int i = 0; i < ables.Count; i++)
|
|
{
|
|
IEnumerable<IGraphPart> lines = ables[i].GetItemsToRender(this);
|
|
Brush graphBrush = new SolidBrush(ables[i].Color);
|
|
Pen graphPen = new(graphBrush, DpiFloat * 3 / 192);
|
|
foreach (IGraphPart gp in lines) gp.Render(this, g, graphPen);
|
|
|
|
// Equation selection detection.
|
|
// This system lets you select multiple graphs, and that's cool by me.
|
|
if (ableDrag)
|
|
{
|
|
if (ables[i].ShouldSelectGraphable(this, graphMousePos, 2.5))
|
|
{
|
|
Float2 selectedPoint = ables[i].GetSelectedPoint(this, graphMousePos);
|
|
GraphUiCircle select = new(selectedPoint, 8);
|
|
select.Render(this, g, graphPen);
|
|
}
|
|
}
|
|
}
|
|
|
|
base.OnPaint(e);
|
|
}
|
|
protected override void OnResize(EventArgs e)
|
|
{
|
|
base.OnResize(e);
|
|
Invalidate(false);
|
|
}
|
|
|
|
public void Graph(params Graphable[] able)
|
|
{
|
|
ables.AddRange(able);
|
|
RegenerateMenuItems();
|
|
Invalidate(false);
|
|
}
|
|
|
|
private bool mouseDrag = false;
|
|
private Int2 initialMouseLocation;
|
|
private Float2 initialScreenCenter;
|
|
|
|
private bool ableDrag = false;
|
|
protected override void OnMouseDown(MouseEventArgs e)
|
|
{
|
|
if (!mouseDrag)
|
|
{
|
|
Point clientMousePos = PointToClient(Cursor.Position);
|
|
Float2 graphMousePos = ScreenSpaceToGraphSpace(new(clientMousePos.X,
|
|
clientMousePos.Y));
|
|
foreach (Graphable able in Graphables)
|
|
{
|
|
if (able.ShouldSelectGraphable(this, graphMousePos, 1)) ableDrag = true;
|
|
}
|
|
if (ableDrag) Invalidate(false);
|
|
}
|
|
|
|
if (!ableDrag)
|
|
{
|
|
mouseDrag = true;
|
|
initialMouseLocation = new Int2(Cursor.Position.X, Cursor.Position.Y);
|
|
initialScreenCenter = ScreenCenter;
|
|
}
|
|
}
|
|
protected override void OnMouseUp(MouseEventArgs e)
|
|
{
|
|
if (mouseDrag)
|
|
{
|
|
Int2 pixelDiff = new(initialMouseLocation.x - Cursor.Position.X,
|
|
initialMouseLocation.y - Cursor.Position.Y);
|
|
Float2 graphDiff = new(pixelDiff.x * ZoomLevel / Dpi.x, pixelDiff.y * ZoomLevel / Dpi.y);
|
|
ScreenCenter = new(initialScreenCenter.x + graphDiff.x,
|
|
initialScreenCenter.y + graphDiff.y);
|
|
}
|
|
mouseDrag = false;
|
|
ableDrag = false;
|
|
Invalidate(false);
|
|
}
|
|
protected override void OnMouseMove(MouseEventArgs e)
|
|
{
|
|
if (mouseDrag)
|
|
{
|
|
Int2 pixelDiff = new(initialMouseLocation.x - Cursor.Position.X,
|
|
initialMouseLocation.y - Cursor.Position.Y);
|
|
Float2 graphDiff = new(pixelDiff.x * ZoomLevel / Dpi.x, pixelDiff.y * ZoomLevel / Dpi.y);
|
|
ScreenCenter = new(initialScreenCenter.x + graphDiff.x,
|
|
initialScreenCenter.y + graphDiff.y);
|
|
Invalidate(false);
|
|
}
|
|
else if (ableDrag) Invalidate(false);
|
|
}
|
|
protected override void OnMouseWheel(MouseEventArgs e)
|
|
{
|
|
ZoomLevel *= 1 - e.Delta * 0.00075; // Zoom factor.
|
|
Invalidate(false);
|
|
}
|
|
|
|
private void ResetViewportButton_Click(object? sender, EventArgs e)
|
|
{
|
|
ScreenCenter = new Float2(0, 0);
|
|
ZoomLevel = 1;
|
|
Invalidate(false);
|
|
}
|
|
private void GraphColorPickerButton_Click(Graphable able)
|
|
{
|
|
GraphColorPickerForm picker = new(this, able)
|
|
{
|
|
StartPosition = FormStartPosition.Manual
|
|
};
|
|
picker.Location = new Point(Location.X + ClientRectangle.Width + 10,
|
|
Location.Y + (ClientRectangle.Height - picker.ClientRectangle.Height) / 2);
|
|
|
|
if (picker.Location.X + picker.Width > Screen.FromControl(this).WorkingArea.Width)
|
|
{
|
|
picker.StartPosition = FormStartPosition.WindowsDefaultLocation;
|
|
}
|
|
|
|
picker.TopMost = true;
|
|
picker.ShowDialog();
|
|
RegenerateMenuItems();
|
|
}
|
|
|
|
private void RegenerateMenuItems()
|
|
{
|
|
MenuColors.DropDownItems.Clear();
|
|
MenuEquationsDerivative.DropDownItems.Clear();
|
|
MenuEquationsIntegral.DropDownItems.Clear();
|
|
|
|
foreach (Graphable able in ables)
|
|
{
|
|
ToolStripMenuItem colorItem = new()
|
|
{
|
|
ForeColor = able.Color,
|
|
Text = able.Name
|
|
};
|
|
colorItem.Click += (o, e) => GraphColorPickerButton_Click(able);
|
|
MenuColors.DropDownItems.Add(colorItem);
|
|
|
|
if (able is Equation equ)
|
|
{
|
|
ToolStripMenuItem derivativeItem = new()
|
|
{
|
|
ForeColor = able.Color,
|
|
Text = able.Name
|
|
};
|
|
derivativeItem.Click += (o, e) => EquationComputeDerivative_Click(equ);
|
|
MenuEquationsDerivative.DropDownItems.Add(derivativeItem);
|
|
|
|
ToolStripMenuItem integralItem = new()
|
|
{
|
|
ForeColor = able.Color,
|
|
Text = able.Name
|
|
};
|
|
integralItem.Click += (o, e) => EquationComputeIntegral_Click(equ);
|
|
MenuEquationsIntegral.DropDownItems.Add(integralItem);
|
|
}
|
|
}
|
|
}
|
|
|
|
private void ButtonViewportSetZoom_Click(object? sender, EventArgs e)
|
|
{
|
|
SetZoomForm picker = new(this)
|
|
{
|
|
StartPosition = FormStartPosition.Manual,
|
|
};
|
|
picker.Location = new Point(Location.X + ClientRectangle.Width + 10,
|
|
Location.Y + (ClientRectangle.Height - picker.ClientRectangle.Height) / 2);
|
|
picker.ShowDialog();
|
|
}
|
|
private void ButtonViewportSetCenter_Click(object? sender, EventArgs e)
|
|
{
|
|
MessageBox.Show("TODO", "Set Center Position", MessageBoxButtons.OK, MessageBoxIcon.Error);
|
|
}
|
|
private void ButtonViewportReset_Click(object? sender, EventArgs e)
|
|
{
|
|
ScreenCenter = new Float2(0, 0);
|
|
ZoomLevel = 1;
|
|
Invalidate(false);
|
|
}
|
|
private void ButtonViewportResetWindow_Click(object? sender, EventArgs e)
|
|
{
|
|
Location = initialWindowPos;
|
|
Size = initialWindowSize;
|
|
WindowState = FormWindowState.Normal;
|
|
}
|
|
private void EquationComputeDerivative_Click(Equation equation)
|
|
{
|
|
EquationDelegate equ = equation.GetDelegate();
|
|
string oldName = equation.Name, newName;
|
|
if (oldName.StartsWith("Derivative of ")) newName = "Second Derivative of " + oldName[14..];
|
|
else if (oldName.StartsWith("Second Derivative of ")) newName = "Third Derivative of " + oldName[21..];
|
|
else newName = "Derivative of " + oldName;
|
|
// TODO: anti-integrate (maybe).
|
|
|
|
Graph(new Equation(DerivativeAtPoint(equ))
|
|
{
|
|
Name = newName
|
|
});
|
|
|
|
static EquationDelegate DerivativeAtPoint(EquationDelegate e)
|
|
{
|
|
const double step = 1e-3;
|
|
return x => (e(x + step) - e(x)) / step;
|
|
}
|
|
}
|
|
private void EquationComputeIntegral_Click(Equation equation)
|
|
{
|
|
EquationDelegate equ = equation.GetDelegate();
|
|
string oldName = equation.Name, newName;
|
|
if (oldName.StartsWith("Integral of ")) newName = "Second Integral of " + oldName[12..];
|
|
else if (oldName.StartsWith("Second Integral of ")) newName = "Third Integral of " + oldName[19..];
|
|
else newName = "Integral of " + oldName;
|
|
// TODO: anti-derive (maybe)
|
|
|
|
Graph(new Equation(x => Integrate(equ, 0, x))
|
|
{
|
|
Name = newName
|
|
});
|
|
|
|
static double Integrate(EquationDelegate e, double lower, double upper)
|
|
{
|
|
// TODO: a better rendering method could make this much faster.
|
|
const double step = 1e-2;
|
|
|
|
double factor = 1;
|
|
if (upper < lower)
|
|
{
|
|
factor = -1;
|
|
(lower, upper) = (upper, lower);
|
|
}
|
|
|
|
double sum = 0;
|
|
for (double x = lower; x <= upper; x += step)
|
|
{
|
|
sum += e(x) * step;
|
|
}
|
|
|
|
return sum * factor;
|
|
}
|
|
}
|
|
|
|
private void MenuMiscCaches_Click(object? sender, EventArgs e)
|
|
{
|
|
ViewCacheForm cacheForm = new(this)
|
|
{
|
|
StartPosition = FormStartPosition.Manual
|
|
};
|
|
|
|
cacheForm.Location = new Point(Location.X + ClientRectangle.Width + 10,
|
|
Location.Y + (ClientRectangle.Height - cacheForm.ClientRectangle.Height) / 2);
|
|
|
|
if (cacheForm.Location.X + cacheForm.Width > Screen.FromControl(this).WorkingArea.Width)
|
|
{
|
|
cacheForm.StartPosition = FormStartPosition.WindowsDefaultLocation;
|
|
}
|
|
cacheForm.TopMost = true;
|
|
cacheForm.Show();
|
|
}
|
|
private void MiscMenuPreload_Click(object sender, EventArgs e)
|
|
{
|
|
Float2 min = MinVisibleGraph, max = MaxVisibleGraph;
|
|
Float2 add = new(max.x - min.x, max.y - min.y);
|
|
add.x *= 0.75; // Expansion
|
|
add.y *= 0.75; //
|
|
|
|
Float2 xRange = new(min.x - add.x, max.x + add.x),
|
|
yRange = new(min.y - add.y, max.y + add.y);
|
|
|
|
foreach (Graphable able in Graphables) able.Preload(xRange, yRange);
|
|
Invalidate(false);
|
|
}
|
|
}
|