Welcome to mirror list, hosted at ThFree Co, Russian Federation.

OperationSolidify.cs « Operations « UVtools.Core - github.com/sn4k3/UVtools.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: ca6577fa718df4dcbffd02e172947f9d78c29597 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/*
 *                     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.Threading.Tasks;
using Emgu.CV;
using Emgu.CV.CvEnum;
using Emgu.CV.Structure;
using Emgu.CV.Util;
using UVtools.Core.Extensions;
using UVtools.Core.FileFormats;

namespace UVtools.Core.Operations
{
    [Serializable]
    public sealed class OperationSolidify : Operation
    {
        public enum AreaCheckTypes
        {
            More,
            Less
        }
        private uint _minimumArea = 1;
        private AreaCheckTypes _areaCheckType = AreaCheckTypes.More;

        #region Overrides

        public override string Title => "Solidify";

        public override string Description =>
            "Solidifies the selected layers, closing all interior holes.\n\n" +
            "NOTE: All open areas of the layer that are completely surrounded by pixels will be filled. Please ensure that none of the holes in the layer are required before proceeding.";

        public override string ConfirmationText =>
            $"solidify layers {LayerIndexStart} through {LayerIndexEnd}?";

        public override string ProgressTitle =>
            $"Solidifying layers {LayerIndexStart} through {LayerIndexEnd}";

        public override string ProgressAction => "Solidified layers";

        /// <summary>
        /// Gets the minimum required area to solidify it
        /// </summary>
        public uint MinimumArea
        {
            get => _minimumArea;
            set => RaiseAndSetIfChanged(ref _minimumArea, Math.Max(1, value));
        }

        public AreaCheckTypes AreaCheckType
        {
            get => _areaCheckType;
            set => RaiseAndSetIfChanged(ref _areaCheckType, value);
        }

        public static Array AreaCheckTypeItems => Enum.GetValues(typeof(AreaCheckTypes));

        public override string ToString()
        {
            var result = $"[Area: ={_areaCheckType} than {_minimumArea}px²]" + LayerRangeString;
            if (!string.IsNullOrEmpty(ProfileName)) result = $"{ProfileName}: {result}";
            return result;
        }

        #endregion

        #region Methods

        public override bool Execute(FileFormat slicerFile, OperationProgress progress = null)
        {
            progress ??= new OperationProgress();
            progress.Reset(ProgressAction, LayerRangeCount);
            Parallel.For(LayerIndexStart, LayerIndexEnd + 1, layerIndex =>
            {
                if (progress.Token.IsCancellationRequested) return;
                using var mat = slicerFile[layerIndex].LayerMat;
                Execute(mat);
                slicerFile[layerIndex].LayerMat = mat;
                lock (progress.Mutex)
                {
                    progress++;
                }
            });
            progress.Token.ThrowIfCancellationRequested();
            return true;
        }

        public override bool Execute(Mat mat, params object[] arguments)
        {
            using Mat filteredMat = new Mat();
            using VectorOfVectorOfPoint contours = new VectorOfVectorOfPoint();
            using Mat hierarchy = new Mat();
            Mat target = GetRoiOrDefault(mat);

            CvInvoke.Threshold(target, filteredMat, 127, 255, ThresholdType.Binary); // Clean AA
            CvInvoke.FindContours(filteredMat, contours, hierarchy, RetrType.Ccomp, ChainApproxMethod.ChainApproxSimple);
            var arr = hierarchy.GetData();
            for (int i = 0; i < contours.Size; i++)
            {
                if ((int)arr.GetValue(0, i, 2) != -1 || (int)arr.GetValue(0, i, 3) == -1) continue;
                if (MinimumArea >= 1)
                {
                    var rectangle = CvInvoke.BoundingRectangle(contours[i]);
                    if (AreaCheckType == AreaCheckTypes.More)
                    {
                        if (rectangle.GetArea() < MinimumArea) continue;
                    }
                    else
                    {
                        if (rectangle.GetArea() > MinimumArea) continue;
                    }

                }

                CvInvoke.DrawContours(target, contours, i, new MCvScalar(255), -1);
            }

            return true;
        }

        #endregion
    }
}