ShareX/ShareX.ScreenCaptureLib/Shapes/NodeManager.cs

#region License Information (GPL v3)

/*
    ShareX - A program that allows you to take screenshots and share any file type
    Copyright (c) 2007-2016 ShareX Team

    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License
    as published by the Free Software Foundation; either version 2
    of the License, or (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    Optionally you can also view the license at <http://www.gnu.org/licenses/>.
*/

#endregion License Information (GPL v3)

using ShareX.HelpersLib;
using System.Drawing;
using System.Linq;
using System.Windows.Forms;

namespace ShareX.ScreenCaptureLib
{
    internal class NodeManager
    {
        private bool visible;

        public bool Visible
        {
            get
            {
                return visible;
            }
            set
            {
                visible = value;

                if (!visible)
                {
                    foreach (NodeObject node in nodes)
                    {
                        node.Visible = visible;
                    }
                }
                else
                {
                    BaseShape shape = shapeManager.CurrentShape;

                    if (shape != null)
                    {
                        UpdateNodePositions();

                        if (shape.NodeType == NodeType.Rectangle)
                        {
                            foreach (NodeObject node in nodes)
                            {
                                node.Shape = NodeShape.Square;
                                node.Visible = visible;
                            }
                        }
                        else if (shape.NodeType == NodeType.Line)
                        {
                            nodes[(int)NodePosition.TopLeft].Shape = nodes[(int)NodePosition.BottomRight].Shape = NodeShape.Circle;
                            nodes[(int)NodePosition.TopLeft].Visible = nodes[(int)NodePosition.BottomRight].Visible = true;
                        }
                        else if (shape.NodeType == NodeType.Freehand)
                        {
                            nodes[(int)NodePosition.TopLeft].Shape = NodeShape.Circle;
                            nodes[(int)NodePosition.TopLeft].Visible = true;
                        }
                    }
                }
            }
        }

        public bool IsResizing { get; private set; }

        public bool IsCursorOnNode
        {
            get
            {
                return Visible && nodes.Any(node => node.IsCursorHover);
            }
        }

        private ShapeManager shapeManager;
        private NodeObject[] nodes;
        private Rectangle tempRect;

        public NodeManager(BaseRegionForm form, ShapeManager shapeManager)
        {
            this.shapeManager = shapeManager;

            nodes = new NodeObject[8];

            for (int i = 0; i < 8; i++)
            {
                nodes[i] = form.MakeNode();
            }

            nodes[(int)NodePosition.BottomRight].Order = 10;
        }

        public void Update()
        {
            BaseShape shape = shapeManager.CurrentShape;

            if (shape != null && Visible && nodes != null)
            {
                if (InputManager.IsMouseDown(MouseButtons.Left))
                {
                    if (shape.NodeType == NodeType.Rectangle)
                    {
                        for (int i = 0; i < 8; i++)
                        {
                            if (nodes[i].IsDragging)
                            {
                                IsResizing = true;

                                if (!InputManager.IsBeforeMouseDown(MouseButtons.Left))
                                {
                                    tempRect = shape.Rectangle;
                                }

                                NodePosition nodePosition = (NodePosition)i;

                                int x = InputManager.MouseVelocity.X;

                                switch (nodePosition)
                                {
                                    case NodePosition.TopLeft:
                                    case NodePosition.Left:
                                    case NodePosition.BottomLeft:
                                        tempRect.X += x;
                                        tempRect.Width -= x;
                                        break;
                                    case NodePosition.TopRight:
                                    case NodePosition.Right:
                                    case NodePosition.BottomRight:
                                        tempRect.Width += x;
                                        break;
                                }

                                int y = InputManager.MouseVelocity.Y;

                                switch (nodePosition)
                                {
                                    case NodePosition.TopLeft:
                                    case NodePosition.Top:
                                    case NodePosition.TopRight:
                                        tempRect.Y += y;
                                        tempRect.Height -= y;
                                        break;
                                    case NodePosition.BottomLeft:
                                    case NodePosition.Bottom:
                                    case NodePosition.BottomRight:
                                        tempRect.Height += y;
                                        break;
                                }

                                shape.Rectangle = CaptureHelpers.FixRectangle(tempRect);

                                break;
                            }
                        }
                    }
                    else if (shape.NodeType == NodeType.Line)
                    {
                        if (nodes[(int)NodePosition.TopLeft].IsDragging)
                        {
                            IsResizing = true;

                            shape.StartPosition = InputManager.MousePosition0Based;
                        }
                        else if (nodes[(int)NodePosition.BottomRight].IsDragging)
                        {
                            IsResizing = true;

                            shape.EndPosition = InputManager.MousePosition0Based;
                        }
                    }
                    else if (shape.NodeType == NodeType.Freehand)
                    {
                        if (nodes[(int)NodePosition.TopLeft].IsDragging)
                        {
                            shapeManager.IsCreating = true;

                            Visible = false;
                        }
                    }
                }
                else
                {
                    IsResizing = false;
                }

                UpdateNodePositions();
            }
        }

        private void UpdateNodePositions()
        {
            BaseShape shape = shapeManager.CurrentShape;

            if (shape != null)
            {
                if (shape.NodeType == NodeType.Rectangle)
                {
                    Rectangle rect = shape.Rectangle;

                    int xStart = rect.X;
                    int xMid = rect.X + rect.Width / 2;
                    int xEnd = rect.X + rect.Width - 1;

                    int yStart = rect.Y;
                    int yMid = rect.Y + rect.Height / 2;
                    int yEnd = rect.Y + rect.Height - 1;

                    nodes[(int)NodePosition.TopLeft].Position = new Point(xStart, yStart);
                    nodes[(int)NodePosition.Top].Position = new Point(xMid, yStart);
                    nodes[(int)NodePosition.TopRight].Position = new Point(xEnd, yStart);
                    nodes[(int)NodePosition.Right].Position = new Point(xEnd, yMid);
                    nodes[(int)NodePosition.BottomRight].Position = new Point(xEnd, yEnd);
                    nodes[(int)NodePosition.Bottom].Position = new Point(xMid, yEnd);
                    nodes[(int)NodePosition.BottomLeft].Position = new Point(xStart, yEnd);
                    nodes[(int)NodePosition.Left].Position = new Point(xStart, yMid);
                }
                else if (shape.NodeType == NodeType.Line)
                {
                    nodes[(int)NodePosition.TopLeft].Position = shape.StartPosition;
                    nodes[(int)NodePosition.BottomRight].Position = shape.EndPosition;
                }
                else if (shape.NodeType == NodeType.Freehand)
                {
                    FreehandRegionShape freehandRegionShape = (FreehandRegionShape)shape;
                    nodes[(int)NodePosition.TopLeft].Position = freehandRegionShape.LastPosition;
                }
            }
        }
    }
}