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
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
/*
* 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 System;
using System.CommandLine;
using System.IO;
using Emgu.CV.CvEnum;
namespace UVtools.Cmd.Symbols;
internal static class SetThumbnailCommand
{
private const string HeatmapArg = ":heatmap";
private const string RandomLayerArg = ":random-layer";
internal static Command CreateCommand()
{
var sourceArgument = new Argument<string>($"file path|layer index|{RandomLayerArg}|{HeatmapArg}", () => HeatmapArg, "Choose from a file, layer index, random layer or generate a heatmap");
var thumbnailIndexesOption = new Option<byte[]>(new []{"-i", "--indexes" }, "Prints only the matching thumbnail(s) index(es)")
{
AllowMultipleArgumentsPerToken = true
};
var command = new Command("set-thumbnail", "Sets and replace thumbnail(s) in the file")
{
GlobalArguments.InputFileArgument,
sourceArgument,
thumbnailIndexesOption,
};
command.AddAlias("set-preview");
command.SetHandler(async (inputFile, source, thumbnailIndexes) =>
{
if (string.IsNullOrWhiteSpace(source))
{
Program.WriteLineError("Invalid empty source argument.");
}
bool result = false;
if (string.Equals(source, HeatmapArg, StringComparison.OrdinalIgnoreCase))
{
var slicerFile = Program.OpenInputFile(inputFile);
using var mat = await Program.ProgressBarWork($"Generating a heatmap from layers 0 through {slicerFile.LastLayerIndex}",
() => slicerFile.GenerateHeatmapAsync(slicerFile.GetBoundingRectangle(50, 100, Program.Progress), Program.Progress));
CvInvoke.CvtColor(mat, mat, ColorConversion.Gray2Bgr);
if (thumbnailIndexes.Length > 0)
{
foreach (var thumbnailIndex in thumbnailIndexes)
{
result = slicerFile.SetThumbnail(thumbnailIndex, mat);
}
}
else
{
result = slicerFile.SetThumbnails(mat) > 0;
}
if(result) Program.SaveFile(slicerFile);
return;
}
if (string.Equals(source, RandomLayerArg, StringComparison.OrdinalIgnoreCase))
{
var slicerFile = Program.OpenInputFile(inputFile);
if(slicerFile.LayerCount == 0) Program.WriteLineError("The file have no valid layers");
using var matRoi = slicerFile[Random.Shared.Next((int) slicerFile.LayerCount)].GetLayerMatBoundingRectangle(50, 100);
CvInvoke.CvtColor(matRoi.RoiMat, matRoi.RoiMat, ColorConversion.Gray2Bgr);
if (thumbnailIndexes.Length > 0)
{
foreach (var thumbnailIndex in thumbnailIndexes)
{
result = slicerFile.SetThumbnail(thumbnailIndex, matRoi.RoiMat);
}
}
else
{
result = slicerFile.SetThumbnails(matRoi.RoiMat) > 0;
}
if (result) Program.SaveFile(slicerFile);
return;
}
if (uint.TryParse(source, out var layerIndex))
{
var slicerFile = Program.OpenInputFile(inputFile);
if (slicerFile.LayerCount == 0) Program.WriteLineError("The file have no valid layers");
slicerFile.SanitizeLayerIndex(ref layerIndex);
using var matRoi = slicerFile[layerIndex].GetLayerMatBoundingRectangle(50, 100);
CvInvoke.CvtColor(matRoi.RoiMat, matRoi.RoiMat, ColorConversion.Gray2Bgr);
if (thumbnailIndexes.Length > 0)
{
foreach (var thumbnailIndex in thumbnailIndexes)
{
result = slicerFile.SetThumbnail(thumbnailIndex, matRoi.RoiMat);
}
}
else
{
result = slicerFile.SetThumbnails(matRoi.RoiMat) > 0;
}
if (result) Program.SaveFile(slicerFile);
return;
}
if (File.Exists(source))
{
var slicerFile = Program.OpenInputFile(inputFile);
if (thumbnailIndexes.Length > 0)
{
foreach (var thumbnailIndex in thumbnailIndexes)
{
result = slicerFile.SetThumbnail(thumbnailIndex, source);
}
}
else
{
result = slicerFile.SetThumbnails(source) > 0;
}
if (result) Program.SaveFile(slicerFile);
return;
}
Program.WriteLineError($"'{source}' is not a file nor layer index nor {RandomLayerArg} nor {HeatmapArg}");
}, GlobalArguments.InputFileArgument, sourceArgument, thumbnailIndexesOption);
return command;
}
}
|
