UVtools/UVtools.Core/Operations/OperationChangeResolution.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 UVtools.Core.FileFormats;
using UVtools.Core.Objects;

namespace UVtools.Core.Operations
{
    [Serializable]
    public sealed class OperationChangeResolution : Operation
    {
        #region Members
        private uint _newResolutionX;
        private uint _newResolutionY;
        #endregion

        #region Subclasses
        public class Resolution
        {
            public uint ResolutionX { get; }
            public uint ResolutionY { get; }
            public string Name { get; }
            public bool IsEmpty => ResolutionX == 0 && ResolutionY == 0;

            public Resolution(uint resolutionX, uint resolutionY, string name = null)
            {
                ResolutionX = resolutionX;
                ResolutionY = resolutionY;
                Name = name;
            }

            public override string ToString()
            {
                if(IsEmpty) return string.Empty;
                var str = $"{ResolutionX} x {ResolutionY}";
                if (!string.IsNullOrEmpty(Name))
                {
                    str += $" ({Name})";
                }
                return str;
            }
        }
        #endregion

        #region Overrides

        public override Enumerations.LayerRangeSelection StartLayerRangeSelection => Enumerations.LayerRangeSelection.None;
        public override bool CanROI => false;
        public override string Title => "Change print resolution";
        public override string Description =>
            "Crops or resizes all layer images to fit an alternate print resolution\n" +
            "Useful to make files printable on a different printer than they were originally sliced for without the need to re-slice.\n\n" +
            "NOTE: Please ensure that the actual model will fit within the new print resolution. The operation will be aborted if it will result in any of the actual model being clipped.\n" +
            "Only use this tool if both source and target printer have the same pixel pitch spec, otherwise the model size will be invalidated and result in a different size than the originally sliced for. " +
            "As alternative is possible to resize the model to match the new pixel pitch.";

        public override string ConfirmationText => 
            "change print resolution " +
            $"from {SlicerFile.ResolutionX}x{SlicerFile.ResolutionY} " +
            $"to {NewResolutionX}x{NewResolutionY}?";

        public override string ProgressTitle =>
            $"Changing print resolution from ({SlicerFile.ResolutionX}x{SlicerFile.ResolutionY}) to ({NewResolutionX}x{NewResolutionY})";

        public override string ProgressAction => "Changed layers";

        public override string ValidateInternally()
        {
            var sb = new StringBuilder();
            if (SlicerFile.ResolutionX == NewResolutionX && SlicerFile.ResolutionY == NewResolutionY)
            {
                sb.AppendLine($"The new resolution must be different from current resolution ({SlicerFile.ResolutionX} x {SlicerFile.ResolutionY}).");
            }

            if (NewResolutionX < SlicerFile.BoundingRectangle.Width || NewResolutionY < SlicerFile.BoundingRectangle.Height)
            {
                sb.AppendLine($"The new resolution ({NewResolutionX} x {NewResolutionY}) is not large enough to hold the model volume ({SlicerFile.BoundingRectangle.Width} x {SlicerFile.BoundingRectangle.Height}), continuing operation would clip the model");
                sb.AppendLine("To fix this, try to rotate the object and/or resize to fit on this new resolution.");
            }

            return sb.ToString();
        }

        public override string ToString()
        {
            var result = $"{_newResolutionX} x {_newResolutionY}";
            if (!string.IsNullOrEmpty(ProfileName)) result = $"{ProfileName}: {result}";
            return result;
        }

        #endregion

        #region Properties
        public uint NewResolutionX
        {
            get => _newResolutionX;
            set => RaiseAndSetIfChanged(ref _newResolutionX, value);
        }

        public uint NewResolutionY
        {
            get => _newResolutionY;
            set => RaiseAndSetIfChanged(ref _newResolutionY, value);
        }

        public Size VolumeBondsSize => SlicerFile.BoundingRectangle.Size;

        #endregion

        #region Constructor

        public OperationChangeResolution() { }

        public OperationChangeResolution(FileFormat slicerFile) : base(slicerFile)
        { }

        public override void InitWithSlicerFile()
        {
            base.InitWithSlicerFile();
            NewResolutionX = SlicerFile.ResolutionX;
            NewResolutionY = SlicerFile.ResolutionY;
        }

        #endregion

        #region Methods
        public static Resolution[] GetResolutions()
        {
            return new [] {
                //new Resolution(0, 0, string.Empty),
                new Resolution(854, 480, "FWVGA"),
                new Resolution(960, 1708),
                new Resolution(1080, 1920, "FHD"),
                new Resolution(1440, 2560, "QHD"),
                new Resolution(1600, 2560, "WQXGA"),
                new Resolution(1620, 2560, "WQXGA"),
                new Resolution(1920, 1080, "FHD"),
                new Resolution(2160, 3840, "4K UHD"),
                new Resolution(2531, 1410, "QHD"),
                new Resolution(2560, 1440, "QHD"),
                new Resolution(2560, 1600, "WQXGA"),
                new Resolution(2560, 1620, "WQXGA"),
                new Resolution(3840, 2160, "4K UHD"),
                new Resolution(3840, 2400, "WQUXGA"),
            };
        }

        public static Resolution[] Presets => GetResolutions();


        protected override bool ExecuteInternally(OperationProgress progress)
        {
            progress.ItemCount = SlicerFile.LayerCount;
            var boundingRectangle = SlicerFile.BoundingRectangle;
            var newSize = new Size((int) NewResolutionX, (int) NewResolutionY);
            Parallel.For(0, SlicerFile.LayerCount, layerIndex =>
            {
                if (progress.Token.IsCancellationRequested) return;

                using var mat = SlicerFile[layerIndex].LayerMat;

                if (mat.Size != newSize)
                {
                    using var matRoi = new Mat(mat, boundingRectangle);
                    using var matDst = new Mat(newSize, mat.Depth, mat.NumberOfChannels);
                    using var matDstRoi = new Mat(matDst,
                        new Rectangle((int) (NewResolutionX / 2 - boundingRectangle.Width / 2),
                            (int) NewResolutionY / 2 - boundingRectangle.Height / 2,
                            boundingRectangle.Width, boundingRectangle.Height));
                    matRoi.CopyTo(matDstRoi);
                    //Execute(mat);
                    SlicerFile[layerIndex].LayerMat = matDst;
                }

                progress.LockAndIncrement();
            });

            progress.Token.ThrowIfCancellationRequested();

            SlicerFile.ResolutionX = NewResolutionX;
            SlicerFile.ResolutionY = NewResolutionY;

            return !progress.Token.IsCancellationRequested;
        }

        /*public override bool Execute(Mat mat, params object[] arguments)
        {
            //mat.Transform(1, 1, newResolutionX-mat.Width+roi.Width, newResolutionY-mat.Height - roi.Height/2, new Size((int) newResolutionX, (int) newResolutionY));
            using var matRoi = new Mat(mat, VolumeBonds);
            using var matDst = new Mat(new Size((int)NewResolutionX, (int)NewResolutionY), mat.Depth, mat.NumberOfChannels);
            using var matDstRoi = new Mat(matDst, 
                    new Rectangle((int)(NewResolutionX / 2 - VolumeBonds.Width / 2),
                    (int)NewResolutionY / 2 - VolumeBonds.Height / 2,
                    VolumeBonds.Width, VolumeBonds.Height));
            matRoi.CopyTo(matDstRoi);

            return true;
        }*/

        #endregion

        #region Equality

        private bool Equals(OperationChangeResolution other)
        {
            return _newResolutionX == other._newResolutionX && _newResolutionY == other._newResolutionY;
        }

        public override bool Equals(object obj)
        {
            return ReferenceEquals(this, obj) || obj is OperationChangeResolution other && Equals(other);
        }

        public override int GetHashCode()
        {
            unchecked
            {
                return ((int) _newResolutionX * 397) ^ (int) _newResolutionY;
            }
        }

        #endregion

    }
}