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
|
/*
* 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.Collections.Generic;
using System.Text;
using System.Threading.Tasks;
using UVtools.Core.Scripting;
using Emgu.CV;
using Emgu.CV.Structure;
using UVtools.Core.Extensions;
namespace UVtools.ScriptSample
{
/// <summary>
/// Performs a black inset around objects
/// </summary>
public class ScriptLightBleedCompensationSample : ScriptGlobals
{
ScriptTextBoxInput BrightnessesInput = new()
{
Label = "Brightnesses",
ToolTip = "Brightness to reduce each subsequent repeated pixels",
Unit = "1-255",
Value = "25,20,15,10,5"
};
public byte[] Levels
{
get
{
List<byte> levels = new();
var split = BrightnessesInput.Value.Split(',', StringSplitOptions.TrimEntries);
foreach (var str in split)
{
if(!byte.TryParse(str, out var brightness)) continue;
if(brightness is byte.MinValue or byte.MaxValue) continue;
levels.Add(brightness);
}
return levels.ToArray();
}
}
/// <summary>
/// Set configurations here, this function trigger just after load a script
/// </summary>
public void ScriptInit()
{
Script.Name = "Light bleed compensation";
Script.Description = "Dim sequential pixels";
Script.Author = "Tiago Conceição";
Script.Version = new Version(0, 1);
Script.UserInputs.Add(BrightnessesInput);
}
/// <summary>
/// Validate user inputs here, this function trigger when user click on execute
/// </summary>
/// <returns>A error message, empty or null if validation passes.</returns>
public string ScriptValidate()
{
StringBuilder sb = new();
if (Levels.Length == 0)
{
sb.AppendLine($"No brightness levels are set");
}
return sb.ToString();
}
/// <summary>
/// Execute the script, this function trigger when when user click on execute and validation passes
/// </summary>
/// <returns>True if executes successfully to the end, otherwise false.</returns>
public bool ScriptExecute()
{
Progress.Reset("Bleed compensation", Operation.LayerRangeCount); // Sets the progress name and number of items to process
var brightnesses = Levels;
// Loop user selected layers in parallel, this will put each core of CPU working here on parallel
Parallel.For(Operation.LayerIndexStart, Operation.LayerIndexEnd+1, layerIndex =>
{
if (Progress.Token.IsCancellationRequested) return; // Abort operation, user requested cancellation
var layer = SlicerFile[layerIndex]; // Unpack and expose layer variable for easier use
using var mat = layer.LayerMat; // Gets this layer mat/image
var original = mat.Clone(); // Keep a original mat copy
var target = Operation.GetRoiOrDefault(mat); // Get ROI from mat if user selected an region
for (byte i = 0; i < brightnesses.Length; i++)
{
uint layerIndexNext = (uint) (layerIndex + i + 1);
if (layerIndexNext > Operation.LayerIndexEnd) break;
using var subtractMat = EmguExtensions.InitMat(target.Size, new MCvScalar(brightnesses[i]));
using var nextMat = SlicerFile[layerIndexNext].LayerMat;
var nextMatRoi = Operation.GetRoiOrDefault(nextMat);
CvInvoke.Subtract(target, subtractMat, target, nextMatRoi);
}
// Apply the results only to the selected masked area, if user selected one
Operation.ApplyMask(original, target);
// Set current layer image with the modified mat we just manipulated
layer.LayerMat = mat;
// Increment progress bar by 1
Progress.LockAndIncrement();
});
// return true if not cancelled by user
return !Progress.Token.IsCancellationRequested;
}
}
}
|
