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: 71498f1c3a90058f0aef9a34eecb88de56690f29 (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
130
131
132
133
134
135
/*
 *                     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 Emgu.CV;
using Emgu.CV.CvEnum;
using System;
using System.Threading.Tasks;
using UVtools.Core.EmguCV;
using UVtools.Core.Extensions;
using UVtools.Core.FileFormats;

namespace UVtools.Core.Operations;


public sealed class OperationSolidify : Operation
{
    #region Enums
    public enum AreaCheckTypes
    {
        More,
        Less
    }
    #endregion

    #region Members
    private uint _minimumArea = 1;
    private AreaCheckTypes _areaCheckType = AreaCheckTypes.More;
    #endregion

    #region Overrides

    public override string IconClass => "fa-solid fa-diamond";
    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 Constructor

    public OperationSolidify() { }

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

    #endregion

    #region Methods

    protected override bool ExecuteInternally(OperationProgress progress)
    {
        Parallel.For(LayerIndexStart, LayerIndexEnd + 1, CoreSettings.GetParallelOptions(progress), layerIndex =>
        {
            using var mat = SlicerFile[layerIndex].LayerMat;
            Execute(mat);
            SlicerFile[layerIndex].LayerMat = mat;
            progress.LockAndIncrement();
        });

        return !progress.Token.IsCancellationRequested;
    }

    public override bool Execute(Mat mat, params object[]? arguments)
    {
        using Mat filteredMat = new();
        using var original = mat.Clone();
        var target = GetRoiOrDefault(mat);

        CvInvoke.Threshold(target, filteredMat, 127, 255, ThresholdType.Binary); // Clean AA
        using var contours = filteredMat.FindContours(out var hierarchy, RetrType.Ccomp);
        for (int i = 0; i < contours.Size; i++)
        {
            if (hierarchy[i, EmguContour.HierarchyFirstChild] != -1 || hierarchy[i, EmguContour.HierarchyParent] == -1) continue;
            if (MinimumArea >= 1)
            {
                var area = CvInvoke.ContourArea(contours[i]);
                if (AreaCheckType == AreaCheckTypes.More)
                {
                    if (area < MinimumArea) continue;
                }
                else
                {
                    if (area > MinimumArea) continue;
                }

            }

            CvInvoke.DrawContours(target, contours, i, EmguExtensions.WhiteColor, -1);
        }

        ApplyMask(original, target);

        return true;
    }

    #endregion
}