mattercontrol/MatterControlLib/DesignTools/EditorTools/RotateControls/PathControl.cs

/*
Copyright (c) 2019, Lars Brubaker, John Lewin
All rights reserved.

Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:

1. Redistributions of source code must retain the above copyright notice, this
   list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright notice,
   this list of conditions and the following disclaimer in the documentation
   and/or other materials provided with the distribution.

THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

The views and conclusions contained in the software and documentation are those
of the authors and should not be interpreted as representing official policies,
either expressed or implied, of the FreeBSD Project.
*/

using System;
using System.Collections.Generic;
using System.Linq;
using MatterHackers.Agg;
using MatterHackers.Agg.UI;
using MatterHackers.Agg.VertexSource;
using MatterHackers.DataConverters3D;
using MatterHackers.MatterControl;
using MatterHackers.MatterControl.DesignTools;
using MatterHackers.MatterControl.PartPreviewWindow;
using MatterHackers.MeshVisualizer;
using MatterHackers.RayTracer;
using MatterHackers.RenderOpenGl;
using MatterHackers.RenderOpenGl.OpenGl;
using MatterHackers.VectorMath;

namespace MatterHackers.Plugins.EditorTools
{
	public class PathControl : IObject3DControl, IObject3DControlsProvider
	{
		private readonly IObject3DControlContext context;

		private readonly ThemeConfig theme;

		private readonly WorldView world;

		private IObject3D lastItem;

		private readonly List<VertexPointWidget> targets = new List<VertexPointWidget>();

		private IEnumerable<VertexData> activePoints;
		private FlattenCurves flattened;
		private bool visible;
		private PointWidget _activeItem;

		public PathControl(IObject3DControlContext context)
		{
			this.context = context;
			theme = MatterControl.AppContext.Theme;

			world = context.World;
		}

		public string Name => "Path Control";

		public bool Visible
		{
			get => visible;
			set
			{
				if (visible != value)
				{
					visible = value;

					foreach (var widget in targets)
					{
						widget.Visible = visible && !(widget is CurveControlPoint);
					}
				}
			}
		}

		public PointWidget ActiveItem
		{
			get => _activeItem;
			set
			{
				if (_activeItem != null)
				{
					_activeItem.Selected = false;
				}

				_activeItem = value;
				_activeItem.Selected = true;

				foreach (var item in targets)
				{
					if (item is VertexPointWidget vertexPoint
						&& vertexPoint.ControlPoint != null)
					{
						vertexPoint.ControlPoint.Visible = item == _activeItem || vertexPoint.LinkedPoint == _activeItem;
					}
				}
			}
		}

		public bool DrawOnTop => false;

		public string UiHint => "";

		public void CancelOperation()
		{
		}

		public void Draw(DrawGlContentEventArgs e)
		{
			if (flattened != null
				&& e.Graphics2D is Graphics2DOpenGL glGraphics)
			{
				var pixelWidth = world.GetWorldUnitsPerScreenPixelAtPosition(new Vector3(activePoints.First().Position));

				world.RenderPath(
					new Stroke(flattened, pixelWidth * .02),
					theme.PrimaryAccentColor,
					false);

				GL.Begin(BeginMode.Lines);
				{
					GL.Color4(Color.Red);

					// Render control point connection lines
					GL.Color4(theme.PrimaryAccentColor);
					foreach (var widget in targets.Where(t => t.ControlPoint != null))
					{
						if (widget == this.ActiveItem || widget.LinkedPoint == this.ActiveItem)
						{
							widget.ControlPoint.Visible = true;
							widget.LinkedPoint.ControlPoint.Visible = true;

							var linkedPoint = widget.LinkedPoint;

							GL.Vertex3(new Vector3(widget.Point));
							GL.Vertex3(new Vector3(widget.ControlPoint.Point));

							GL.Vertex3(new Vector3(linkedPoint.Point));
							GL.Vertex3(new Vector3(linkedPoint.ControlPoint.Point));

							break;
						}
					}
				}

				GL.End();
			}
		}

		AxisAlignedBoundingBox IObject3DControl.GetWorldspaceAABB()
		{
			// TODO: Untested.
			if (flattened != null)
				return new AxisAlignedBoundingBox(targets.Select(t => t.Point).ToArray());
			else
				return AxisAlignedBoundingBox.Empty();
		}

		private void Reset()
		{
			// Clear and close selection targets
			foreach (var widget in targets)
			{
				widget.Close();
			}

			targets.Clear();

			lastItem = null;
		}

		public void SetPosition(IObject3D selectedItem, MeshSelectInfo selectInfo)
		{
			if (selectedItem != lastItem)
			{
				this.Reset();

				lastItem = selectedItem;

				if (selectedItem is PathContainerObject3D pathObject)
				{
					var vertexStorage = pathObject.VertexStorage;

					activePoints = vertexStorage.Vertices();

					flattened = new FlattenCurves(vertexStorage)
					{
						ResolutionScale = 6
					};

					VertexPointWidget widget = null;

					for (var i = 0; i < vertexStorage.Count; i++)
					{
						var command = vertexStorage.Vertex(i, out double x, out double y);

						if (ShapePath.IsVertex(command))
						{
							if (command == FlagsAndCommand.Curve4)
							{
								// vertexDataManager.AddVertex(x_ctrl1, y_ctrl1, FlagsAndCommand.Curve4);
								// vertexDataManager.AddVertex(x_ctrl2, y_ctrl2, FlagsAndCommand.Curve4);
								// vertexDataManager.AddVertex(x_to, y_to, FlagsAndCommand.Curve4);

								var lastItem = targets.LastOrDefault();

								var controlPoint1 = new CurveControlPoint(context, this, vertexStorage, new Vector3(x, y, 0), command, i);
								context.GuiSurface.AddChild(controlPoint1);
								targets.Add(controlPoint1);

								command = vertexStorage.Vertex(i + 1, out x, out y);
								var controlPoint2 = new CurveControlPoint(context, this, vertexStorage, new Vector3(x, y, 0), command, i + 1);
								context.GuiSurface.AddChild(controlPoint2);
								targets.Add(controlPoint2);

								command = vertexStorage.Vertex(i + 2, out x, out y);
								var curveWidget = new Curve4AnchorWidget(context, this, vertexStorage, new Vector3(x, y, 0), command, i + 2)
								{
									ControlPoint = controlPoint2,
								};

								if (lastItem is VertexPointWidget vertexPointWidget)
								{
									vertexPointWidget.ControlPoint = controlPoint1;
									vertexPointWidget.LinkedPoint = curveWidget;
									curveWidget.LinkedPoint = vertexPointWidget;
								}

								// controlPoint1.ParentPoint = curveWidget;
								// controlPoint2.ParentPoint = curveWidget;

								widget = curveWidget;

								// Advance to account for 3 commands in Curve4
								i += 2;
							}
							else
							{
								widget = new VertexPointWidget(context, this, vertexStorage, new Vector3(x, y, 0), command, i);
							}

							// widget.Click += (s, e) =>

							targets.Add(widget);

							context.GuiSurface.AddChild(widget);
						}
					}

					// Highlight last
					if (widget != null)
					{
						widget.PointColor = Color.Red;
					}
				}
			}

			foreach (var item in targets)
			{
				item.UpdatePosition();
			}
		}

		public ITraceable GetTraceable()
		{
			throw new NotImplementedException();
		}

		public void OnMouseDown(Mouse3DEventArgs mouseEvent3D)
		{
			throw new NotImplementedException();
		}

		public void OnMouseMove(Mouse3DEventArgs mouseEvent3D, bool mouseIsOver)
		{
			throw new NotImplementedException();
		}

		public void OnMouseUp(Mouse3DEventArgs mouseEvent3D)
		{
			throw new NotImplementedException();
		}

		public IEnumerable<IObject3DControl> CreateObject3DControls(IObject3DControlContext context)
		{
			return new List<IObject3DControl> { new PathControl(context) };
		}

		public void Dispose()
		{
			Reset();
		}

		public void AddObject3DControls(Object3DControlsLayer object3DControlsLayer)
		{
			throw new NotImplementedException();
		}

		private class CurveControlPoint : VertexPointWidget
		{
			public CurveControlPoint(IObject3DControlContext context, PathControl pathControl, VertexStorage vertexStorage, Vector3 point, FlagsAndCommand flagsandCommand, int index)
				: base(context, pathControl, vertexStorage, point, flagsandCommand, index)
			{
				this.ClaimSelection = false;
				this.Visible = false;
				this.HandleStyle = HandleStyle.Circle;
			}

			// public Curve4PointWidget ParentPoint { get; internal set; }
		}

		private class Curve4AnchorWidget : VertexPointWidget
		{
			private readonly bool _focused;

			public Curve4AnchorWidget(IObject3DControlContext context, PathControl pathControl, VertexStorage vertexStorage, Vector3 point, FlagsAndCommand flagsandCommand, int index)
				: base(context, pathControl, vertexStorage, point, flagsandCommand, index)
			{
			}

			public override void OnFocusChanged(EventArgs e)
			{
				if (this.Focused)
				{
					this.ControlPoint.Visible = true;
				}
				else
				{
					UiThread.RunOnIdle(() =>
					{
						this.ControlPoint.Visible = this.PointActive || this.LinkedPoint?.PointActive == true;
					}, .1);
				}

				base.OnFocusChanged(e);
			}

			// public VertexPointWidget ControlPoint1 { get; set; }

			// public VertexPointWidget ControlPoint2 { get; set; }
		}

		private class VertexPointWidget : PointWidget
		{
			public PathControl PathControl { get; }

			private readonly FlagsAndCommand command;

			private readonly VertexStorage vertexStorage;
			private Vector3 _point = Vector3.NegativeInfinity;
			private Vector3 controlPointDelta;
			private readonly int index;

			public VertexPointWidget(IObject3DControlContext context, PathControl pathControl, VertexStorage vertexStorage, Vector3 point, FlagsAndCommand flagsandCommand, int index)
				: base(context, point)
			{
				this.PathControl = pathControl;
				this.command = flagsandCommand;
				this.index = index;
				this.vertexStorage = vertexStorage;
			}

			public VertexPointWidget ControlPoint { get; set; }

			public VertexPointWidget LinkedPoint { get; set; }

			public bool ClaimSelection { get; protected set; } = true;

			public bool PointActive => this.Focused || this.ControlPoint?.Focused == true;

			public override Vector3 Point
			{
				get => _point;
				set
				{
					if (_point != value)
					{
						_point = value;

						if (vertexStorage?.Count >= index)
						{
							vertexStorage.ModifyVertex(index, _point.X, _point.Y);
						}

						this.Invalidate();
					}
				}
			}

			public override void OnMouseDown(MouseEventArgs mouseEvent)
			{
				if (ControlPoint != null)
				{
					controlPointDelta = ControlPoint.Point - this.Point;
				}

				if (mouseEvent.Button == MouseButtons.Left
					&& this.ClaimSelection
					&& this.PositionWithinLocalBounds(mouseEvent.Position))
				{
					this.Selected = true;
					this.PathControl.ActiveItem = this;
				}

				base.OnMouseDown(mouseEvent);
			}

			protected override void OnDragTo(IntersectInfo info)
			{
				if (ControlPoint != null)
				{
					ControlPoint.Point = info.HitPosition + controlPointDelta;
				}

				this.Point = info.HitPosition;
				this.Invalidate();
				base.OnDragTo(info);
			}
		}

		public class PointWidget : GuiWidget
		{
			private readonly WorldView world;
			private readonly GuiWidget guiSurface;
			private bool mouseInBounds;
			private GuiWidget systemWindow;
			private bool mouseDownOnWidget;
			private readonly IObject3DControlContext object3DControlContext;
			private static readonly PlaneShape BedPlane = new PlaneShape(Vector3.UnitZ, 0, null);
			private readonly ThemeConfig theme;

			public PointWidget(IObject3DControlContext object3DControlContext, Vector3 point)
			{
				this.theme = MatterControl.AppContext.Theme;
				this.object3DControlContext = object3DControlContext;
				this.HAnchor = HAnchor.Absolute;
				this.VAnchor = VAnchor.Absolute;
				this.Width = 8;
				this.Height = 8;
				this.Point = point;
				this.PointColor = theme.PrimaryAccentColor;

				world = object3DControlContext.World;
				guiSurface = object3DControlContext.GuiSurface;
			}

			public virtual Vector3 Point { get; set; }

			public override void OnLoad(EventArgs args)
			{
				// Register listeners
				systemWindow = this.Parents<SystemWindow>().First();
				systemWindow.AfterDraw += this.Parent_AfterDraw;

				base.OnLoad(args);
			}

			public override void OnClosed(EventArgs e)
			{
				// Unregister listeners
				if (systemWindow != null)
				{
					systemWindow.AfterDraw -= this.Parent_AfterDraw;
				}

				base.OnClosed(e);
			}

			public override void OnMouseDown(MouseEventArgs mouseEvent)
			{
				mouseDownOnWidget = mouseEvent.Button == MouseButtons.Left && this.PositionWithinLocalBounds(mouseEvent.Position);
				base.OnMouseDown(mouseEvent);
			}

			public override void OnMouseUp(MouseEventArgs mouseEvent)
			{
				mouseDownOnWidget = false;
				base.OnMouseUp(mouseEvent);
			}

			public override void OnMouseMove(MouseEventArgs mouseEvent)
			{
				// Drag item
				if (mouseDownOnWidget)
				{
					var localMousePosition = mouseEvent.Position;

					Vector2 meshViewerWidgetScreenPosition = this.TransformToParentSpace(guiSurface, localMousePosition);
					Ray ray = world.GetRayForLocalBounds(meshViewerWidgetScreenPosition);

					if (BedPlane.GetClosestIntersection(ray) is IntersectInfo info)
					{
						this.OnDragTo(info);
					}
				}

				base.OnMouseMove(mouseEvent);
			}

			protected virtual void OnDragTo(IntersectInfo info)
			{
			}

			public override void OnMouseEnterBounds(MouseEventArgs mouseEvent)
			{
				mouseInBounds = true;
				base.OnMouseEnterBounds(mouseEvent);
				this.Invalidate();
			}

			public override void OnMouseLeaveBounds(MouseEventArgs mouseEvent)
			{
				mouseInBounds = false;
				base.OnMouseLeaveBounds(mouseEvent);
				this.Invalidate();
			}

			private void Parent_AfterDraw(object sender, DrawEventArgs e)
			{
				// AfterDraw listener registered on parent to draw outside of bounds
				if (mouseInBounds)
				{
					var position = this.TransformToScreenSpace(LocalBounds.Center);
					e.Graphics2D.Circle(position, 9, Color.Blue.WithAlpha(80));
				}
			}

			public Color PointColor { get; set; }

			protected HandleStyle HandleStyle { get; set; } = HandleStyle.Square;

			public bool Selected { get; set; }

			public override void OnDraw(Graphics2D graphics2D)
			{
				if (this.HandleStyle == HandleStyle.Square)
				{
					if (this.Selected)
					{
						graphics2D.FillRectangle(0, 0, this.Width, this.Height, theme.PrimaryAccentColor);
						graphics2D.Rectangle(0, 0, this.Width, this.Height, Color.White);
					}
					else
					{
						graphics2D.FillRectangle(0, 0, this.Width, this.Height, Color.White);
						graphics2D.Rectangle(0, 0, this.Width, this.Height, theme.PrimaryAccentColor);
					}
				}
				else
				{
					graphics2D.Circle(this.LocalBounds.Center, 3.5, this.PointColor);
				}

				base.OnDraw(graphics2D);
			}

			public void UpdatePosition()
			{
				this.Position = world.GetScreenPosition(Point) - new Vector2(this.LocalBounds.Width / 2, this.LocalBounds.Height / 2);
			}
		}

		public enum HandleStyle
		{
			Square,
			Circle
		}
	}
}