mattercontrol/DesignTools/Operations/Object3DExtensions.cs

/*
Copyright (c) 2018, 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.Diagnostics;
using System.Linq;
using ClipperLib;
using MatterHackers.Agg;
using MatterHackers.Agg.VertexSource;
using MatterHackers.DataConverters2D;
using MatterHackers.DataConverters3D;
using MatterHackers.DataConverters3D.UndoCommands;
using MatterHackers.MatterControl.PartPreviewWindow.View3D;
using MatterHackers.VectorMath;

namespace MatterHackers.MatterControl.DesignTools.Operations
{
	public static class Object3DExtensions
	{
		public static bool IsRoot(this IObject3D object3D)
		{
			return object3D.Parent == null;
		}

		public static void CopyProperties(this IObject3D copyTo, IObject3D copyFrom, Object3DPropertyFlags flags)
		{
			if (flags.HasFlag(Object3DPropertyFlags.Matrix))
			{
				copyTo.Matrix = copyFrom.Matrix;
			}
			if (flags.HasFlag(Object3DPropertyFlags.Color))
			{
				copyTo.Color = copyFrom.Color;
			}
			if (flags.HasFlag(Object3DPropertyFlags.MaterialIndex))
			{
				copyTo.MaterialIndex = copyFrom.MaterialIndex;
			}
			if (flags.HasFlag(Object3DPropertyFlags.Name))
			{
				copyTo.Name = copyFrom.Name;
			}
			if (flags.HasFlag(Object3DPropertyFlags.OutputType))
			{
				copyTo.OutputType = copyFrom.OutputType;
			}
			if (flags.HasFlag(Object3DPropertyFlags.Visible))
			{
				copyTo.Visible = copyFrom.Visible;
			}
		}

		public static IObject3D Translate(this IObject3D objectToTranslate, double x = 0, double y = 0, double z = 0, string name = "")
		{
			return objectToTranslate.Translate(new Vector3(x, y, z), name);
		}

		public static IObject3D Translate(this IObject3D objectToTranslate, Vector3 translation, string name = "")
		{
			objectToTranslate.Matrix *= Matrix4X4.CreateTranslation(translation);
			return objectToTranslate;
		}


		public static IObject3D Scale(this IObject3D objectToTranslate, double x = 0, double y = 0, double z = 0, string name = "")
		{
			return objectToTranslate.Scale(new Vector3(x, y, z), name);
		}

		public static IObject3D Scale(this IObject3D objectToTranslate, Vector3 translation, string name = "")
		{
			objectToTranslate.Matrix *= Matrix4X4.CreateScale(translation);
			return objectToTranslate;
		}

		public static IObject3D Minus(this IObject3D a, IObject3D b)
		{
			var resultsA = a.Clone();
			SubtractObject3D.Subtract(resultsA.VisibleMeshes().Select((i) => i).ToList(), b.VisibleMeshes().Select((i) => i).ToList());
			return resultsA;
		}

		private static VertexStorage CombinePaths(IVertexSource a, IVertexSource b, ClipType clipType)
		{
			List<List<IntPoint>> aPolys = a.CreatePolygons();
			List<List<IntPoint>> bPolys = b.CreatePolygons();

			Clipper clipper = new Clipper();

			clipper.AddPaths(aPolys, PolyType.ptSubject, true);
			clipper.AddPaths(bPolys, PolyType.ptClip, true);

			List<List<IntPoint>> intersectedPolys = new List<List<IntPoint>>();
			clipper.Execute(clipType, intersectedPolys);

			Clipper.CleanPolygons(intersectedPolys);

			VertexStorage output = intersectedPolys.CreateVertexStorage();

			output.Add(0, 0, ShapePath.FlagsAndCommand.Stop);

			return output;
		}

		public static VertexStorage Offset(this IVertexSource a, double distance)
		{
			List<List<IntPoint>> aPolys = a.CreatePolygons();

			ClipperOffset offseter = new ClipperOffset();
			offseter.AddPaths(aPolys, JoinType.jtMiter, EndType.etClosedPolygon);
			var solution = new List<List<IntPoint>>();
			offseter.Execute(ref solution, distance * 1000);

			Clipper.CleanPolygons(solution);

			VertexStorage output = solution.CreateVertexStorage();

			output.Add(0, 0, ShapePath.FlagsAndCommand.Stop);

			return output;
		}

		public static IObject3D Plus(this IObject3D a, IObject3D b)
		{
			var results = new Object3D();

			results.Children.Add(a.Clone());
			results.Children.Add(b.Clone());

			return results;
		}

		public static IVertexSource Minus(this IVertexSource a, IVertexSource b)
		{
			return CombinePaths(a, b, ClipType.ctDifference);
		}

		public static IVertexSource Plus(this IVertexSource a, IVertexSource b)
		{
			return CombinePaths(a, b, ClipType.ctUnion);
		}

		public static double XSize(this IObject3D item)
		{
			return item.GetAxisAlignedBoundingBox().XSize;
		}

		public static double YSize(this IObject3D item)
		{
			return item.GetAxisAlignedBoundingBox().YSize;
		}

		public static double ZSize(this IObject3D item)
		{
			return item.GetAxisAlignedBoundingBox().ZSize;
		}

		public static void AddSelectionAsChildren(this InteractiveScene scene, IObject3D newParent)
		{
			if (scene.HasSelection)
			{
				List<IObject3D> itemsToReplace;

				if (scene.SelectedItem is SelectionGroup)
				{
					itemsToReplace = scene.SelectedItem.Children.ToList();
					foreach (var child in itemsToReplace)
					{
						newParent.Children.Add(child.Clone());
					}
				}
				else
				{
					itemsToReplace = new List<IObject3D> { scene.SelectedItem };
					newParent.Children.Add(scene.SelectedItem.Clone());
				}

				scene.SelectedItem = null;

				newParent.MakeNameNonColliding();

				scene.UndoBuffer.AddAndDo(
					new ReplaceCommand(
						itemsToReplace,
						new List<IObject3D> { newParent }));

				scene.SelectedItem = newParent;
			}
		}

		public static void WrapWith(this IObject3D originalItem, IObject3D wrapper, InteractiveScene scene)
		{
			originalItem.SuspendRebuild();
			originalItem.Parent.Children.Modify(list =>
			{
				list.Remove(originalItem);

				wrapper.Matrix = originalItem.Matrix;

				originalItem.Matrix = Matrix4X4.Identity;
				wrapper.Children.Add(originalItem);

				list.Add(wrapper);
			});

			scene.SelectedItem = wrapper;
			originalItem.ResumeRebuild();
		}

		public static Matrix4X4 ApplyAtBoundsCenter(this IObject3D objectWithBounds, Matrix4X4 transformToApply)
		{
			return ApplyAtCenter(objectWithBounds.Matrix, objectWithBounds.GetAxisAlignedBoundingBox(Matrix4X4.Identity), transformToApply);
		}

		public static Matrix4X4 ApplyAtCenter(this Matrix4X4 currentTransform, AxisAlignedBoundingBox boundsToApplyTo, Matrix4X4 transformToApply)
		{
			return ApplyAtPosition(currentTransform, transformToApply, boundsToApplyTo.Center);
		}

		public static Matrix4X4 ApplyAtPosition(this Matrix4X4 currentTransform, Matrix4X4 transformToApply, Vector3 positionToApplyAt)
		{
			currentTransform *= Matrix4X4.CreateTranslation(-positionToApplyAt);
			currentTransform *= transformToApply;
			currentTransform *= Matrix4X4.CreateTranslation(positionToApplyAt);

			return currentTransform;
		}
		 
		public static Vector3 GetCenter(this IObject3D item)
		{
			return item.GetAxisAlignedBoundingBox(Matrix4X4.Identity).Center;
		}

		public static IObject3D SetChildren(this IObject3D parent, IEnumerable<IObject3D> newChildren)
		{
			parent.Children.Modify((list) =>
			{
				list.Clear();
				list.AddRange(newChildren);
			});

			return parent;
		}

		public static void SetChildren(this IObject3D parent, IObject3D newChild)
		{
			parent.Children.Modify((list) =>
			{
				list.Clear();
				list.Add(newChild);
			});
		}
	}
}