path: root/UVtools.Core/Operations/OperationMove.cs

/*
 *                     GNU AFFERO GENERAL PUBLIC LICENSE
 *                       Version 3, 19 November 2007
 *  Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
 *  Everyone is permitted to copy and distribute verbatim copies
 *  of this license document, but changing it is not allowed.
 */
using System;
using System.Drawing;
using System.Text;
using System.Threading.Tasks;
using Emgu.CV;
using Emgu.CV.Structure;
using UVtools.Core.FileFormats;
using UVtools.Core.Objects;

namespace UVtools.Core.Operations
{
    [Serializable]
    public class OperationMove : Operation
    {
        #region Overrides

        public override bool CanMask => false;

        public override string Title => "Move";
        public override string Description =>
            "Change or copy the position of the model on the build plate.\n" +
            "Note: Before perform this operation, un-rotate the layer preview to see the real orientation.";

        public override string ConfirmationText =>
            (IsCutMove ? "move" : "copy") + $" model layers {LayerIndexStart} through {LayerIndexEnd} from " +
            $"location {{X={ROI.X},Y={ROI.Y}}} to " +
            $"location {{X={DstRoi.X},Y={DstRoi.Y}}}?";

        public override string ProgressTitle =>
            (IsCutMove ? "Moving" : "Copying") +$" model to {{X={DstRoi.X},Y={DstRoi.Y}}}";

        public override string ProgressAction => (IsCutMove ? "Moved" : "Copied")+" layers";

        public override bool CanHaveProfiles => false;

        public override StringTag Validate(params object[] parameters)
        {
            var sb = new StringBuilder();

            if (!ValidateBounds())
            {
                sb.AppendLine("Your parameters will put the model outside of build plate. Please adjust the location and margins.");
            }

            return new StringTag(sb.ToString());
        }

        public override string ToString()
        {
            var result = $"[{ROI} -> {DstRoi}] [Cut: {IsCutMove}]" + LayerRangeString;
            if (!string.IsNullOrEmpty(ProfileName)) result = $"{ProfileName}: {result}";
            return result;
        }
        #endregion

        #region Members
        private Rectangle _dstRoi = Rectangle.Empty;
        private uint _imageWidth;
        private uint _imageHeight;
        private Enumerations.Anchor _anchor = Enumerations.Anchor.MiddleCenter;
        private int _marginLeft;
        private int _marginTop;
        private int _marginRight;
        private int _marginBottom;
        private bool _isCutMove = true;
        private bool _isWithinBoundary;
        #endregion

        #region Properties

        public Rectangle DstRoi
        {
            get
            {
                if(!_dstRoi.IsEmpty) return _dstRoi;
                CalculateDstRoi();
                return _dstRoi;
            }
        }

        public uint ImageWidth
        {
            get => _imageWidth;
            set => RaiseAndSetIfChanged(ref _imageWidth, value);
        }

        public uint ImageHeight
        {
            get => _imageHeight;
            set => RaiseAndSetIfChanged(ref _imageHeight, value);
        }

        public Enumerations.Anchor Anchor
        {
            get => _anchor;
            set
            {
                RaiseAndSetIfChanged(ref _anchor, value);
                CalculateDstRoi();
            }
        }

        public int MarginLeft
        {
            get => _marginLeft;
            set
            {
                RaiseAndSetIfChanged(ref _marginLeft, value);
                CalculateDstRoi();
            }
        }

        public int MarginTop
        {
            get => _marginTop;
            set
            {
                RaiseAndSetIfChanged(ref _marginTop, value);
                CalculateDstRoi();
            }
        }

        public int MarginRight
        {
            get => _marginRight;
            set
            {
                RaiseAndSetIfChanged(ref _marginRight, value);
                CalculateDstRoi();
            }
        }

        public int MarginBottom
        {
            get => _marginBottom;
            set
            {
                RaiseAndSetIfChanged(ref _marginBottom, value);
                CalculateDstRoi();
            }
        }

        public bool IsCutMove
        {
            get => _isCutMove;
            set => RaiseAndSetIfChanged(ref _isCutMove, value);
        }

        public string LocationXStr => $"X: {DstRoi.X} / {ImageWidth - ROI.Width}";
        public string LocationYStr => $"Y: {DstRoi.Y} / {ImageHeight - ROI.Height}";

        public string LocationWidthStr => $"Width: {ROI.Width} / {ImageWidth}";
        public string LocationHeightStr => $"Height: {ROI.Height} / {ImageHeight}";

        public bool IsWithinBoundary
        {
            get => _isWithinBoundary;
            set
            {
                if (!RaiseAndSetIfChanged(ref _isWithinBoundary, value)) return;
                RaisePropertyChanged(nameof(IsWithinBoundaryStr));
            }
        }

        public string IsWithinBoundaryStr => "Model within boundary: " + (_isWithinBoundary ? "Yes" : "No");

        #endregion

        #region Constructor

        public OperationMove() { }

        public OperationMove(FileFormat slicerFile, Enumerations.Anchor anchor = Enumerations.Anchor.MiddleCenter) : base(slicerFile)
        {
            ROI = slicerFile.LayerManager.BoundingRectangle;
            _imageWidth = slicerFile.ResolutionX;
            _imageHeight = slicerFile.ResolutionY;
            _anchor = anchor;
        }

        public OperationMove(FileFormat slicerFile, Rectangle srcRoi, Enumerations.Anchor anchor = Enumerations.Anchor.MiddleCenter) : this(slicerFile, anchor)
        {
            if(!srcRoi.IsEmpty) ROI = srcRoi;
        }

        /*public OperationMove(FileFormat slicerFile, Rectangle srcRoi, Mat mat, Enumerations.Anchor anchor = Enumerations.Anchor.MiddleCenter) : this(slicerFile, srcRoi, anchor)
        {
            ImageWidth = (uint) mat.Width;
            ImageHeight = (uint) mat.Height;
        }*/

        #endregion

        #region Methods
        public void CalculateDstRoi()
        {
            _dstRoi.Size = ROI.Size;

            switch (Anchor)
            {
                case Enumerations.Anchor.TopLeft:
                    _dstRoi.Location = new Point(0, 0);
                    break;
                case Enumerations.Anchor.TopCenter:
                    _dstRoi.Location = new Point((int)(ImageWidth / 2 - ROI.Width / 2), 0);
                    break;
                case Enumerations.Anchor.TopRight:
                    _dstRoi.Location = new Point((int)(ImageWidth - ROI.Width), 0);
                    break;
                case Enumerations.Anchor.MiddleLeft:
                    _dstRoi.Location = new Point(0, (int)(ImageHeight / 2 - ROI.Height / 2));
                    break;
                case Enumerations.Anchor.MiddleCenter:
                //case Anchor.None:
                    _dstRoi.Location = new Point((int)(ImageWidth / 2 - ROI.Width / 2), (int)(ImageHeight / 2 - ROI.Height / 2));
                    break;
                case Enumerations.Anchor.MiddleRight:
                    _dstRoi.Location = new Point((int)(ImageWidth - ROI.Width), (int)(ImageHeight / 2 - ROI.Height / 2));
                    break;
                case Enumerations.Anchor.BottomLeft:
                    _dstRoi.Location = new Point(0, (int)(ImageHeight - ROI.Height));
                    break;
                case Enumerations.Anchor.BottomCenter:
                    _dstRoi.Location = new Point((int)(ImageWidth / 2 - ROI.Width / 2), (int)(ImageHeight - ROI.Height));
                    break;
                case Enumerations.Anchor.BottomRight:
                    _dstRoi.Location = new Point((int)(ImageWidth - ROI.Width), (int)(ImageHeight - ROI.Height));
                    break;
                default:
                    throw new ArgumentOutOfRangeException();
            }

            _dstRoi.X += MarginLeft;
            _dstRoi.X -= MarginRight;
            _dstRoi.Y += MarginTop;
            _dstRoi.Y -= MarginBottom;

            IsWithinBoundary = !(_dstRoi.IsEmpty || _dstRoi.X < 0 || _dstRoi.Y < 0 ||
                                 _dstRoi.Width == 0 || _dstRoi.Right > ImageWidth ||
                                 _dstRoi.Height == 0 || _dstRoi.Bottom > ImageHeight);

            RaisePropertyChanged(nameof(DstRoi));
            RaisePropertyChanged(nameof(LocationXStr));
            RaisePropertyChanged(nameof(LocationYStr));
        }


        public void Reset()
        {
            MarginLeft = MarginTop = MarginRight = MarginBottom = 0;
            Anchor = Enumerations.Anchor.MiddleCenter;
            IsCutMove = true;
        }


        public void SetAnchor(byte value)
        {
            Anchor = (Enumerations.Anchor)value;
        }


        public bool ValidateBounds()
        {
            CalculateDstRoi();
            return IsWithinBoundary;
        }

        protected override bool ExecuteInternally(OperationProgress progress)
        {
            if (ROI.IsEmpty) ROI = SlicerFile.LayerManager.GetBoundingRectangle(progress);
            CalculateDstRoi();

            Parallel.For(LayerIndexStart, LayerIndexEnd + 1, layerIndex =>
            {
                if (progress.Token.IsCancellationRequested) return;

                using (var mat = SlicerFile[layerIndex].LayerMat)
                {
                    Execute(mat);
                    SlicerFile[layerIndex].LayerMat = mat;
                }

                progress.LockAndIncrement();
            });

            SlicerFile.LayerManager.BoundingRectangle = Rectangle.Empty;

            return !progress.Token.IsCancellationRequested;
        }

        public override bool Execute(Mat mat, params object[] arguments)
        {
            if (_imageWidth == 0) _imageWidth = (uint) mat.Width;
            if (_imageHeight == 0) _imageHeight = (uint) mat.Height;

            using var srcRoi = new Mat(mat, ROI);
            using var dstRoi = new Mat(mat, DstRoi);
            if (IsCutMove)
            {
                using var targetRoi = srcRoi.Clone();
                srcRoi.SetTo(new MCvScalar(0));
                targetRoi.CopyTo(dstRoi);
            }
            else
            {
                srcRoi.CopyTo(dstRoi);
            }

            return true;
        }

        #endregion
    }
}