修复进行处理之后的图片没有保存

Author: lindexi

src/MediaConverters/MediaConverters.Tests/Framework/TestFileProvider.cs

@@ -17,6 +17,11 @@ public static Image<Rgba32> GetDefaultTestImage()
 
     public static string GetTestFilePath(string fileName)
     {
+        if (File.Exists(fileName))
+        {
+            return Path.GetFullPath(fileName);
+        }
+
         return System.IO.Path.Join(AppContext.BaseDirectory, "Assets", "TestFiles", fileName);
     }
 

src/MediaConverters/MediaConverters.Tests/Tool/MediaConverterTests.cs

@@ -1,6 +1,15 @@
-using System.Text.Json;
+using DotNetCampus.MediaConverters.Contexts;
+using DotNetCampus.MediaConverters.Imaging.Effects;
+using DotNetCampus.MediaConverters.Imaging.Effects.Colors;
+
+using SixLabors.ImageSharp;
+using SixLabors.ImageSharp.Formats.Png;
+using SixLabors.ImageSharp.PixelFormats;
+
+using System.Text.Json;
 using System.Text.Json.Serialization;
-using DotNetCampus.MediaConverters.Contexts;
+
+using TextVisionComparer;
 
 namespace DotNetCampus.MediaConverters.Tests.Tool;
 
@@ -27,6 +36,103 @@ public async Task ReplaceColorTask1()
         TestHelper.OpenFileInExplorer(new FileInfo(options.OutputFile));
     }
 
+    [TestMethod]
+    public async Task ReplaceColorTask2()
+    {
+        using Image<Rgba32> image = TestFileProvider.GetDefaultTestImage();
+        var list = image.GetColorCountList();
+        list = list.OrderByDescending(t => t.Count).ToList();
+
+        var replaceColorInfoList = new List<ReplaceColorInfo>();
+
+        for (var i = 0; i < list.Count && i < 10; i++)
+        {
+            var color = list[i].Color;
+            replaceColorInfoList.Add(new ReplaceColorInfo($"#{color.A:X2}{color.R:X2}{color.G:X2}{color.B:X2}", "#00FFFFFF"));
+        }
+
+        var imageConvertContext = new ImageConvertContext()
+        {
+            ImageConvertTaskList =
+            [
+                new ReplaceColorTask()
+                {
+                    ReplaceColorInfoList = replaceColorInfoList
+                },
+            ]
+        };
+
+        var options = ToOptions(TestFileProvider.DefaultTestImageName, imageConvertContext);
+
+        var result = await Program.RunAsync(options);
+        Assert.AreEqual(ErrorCode.Success, result);
+        AssertReplaceColor(options);
+        TestHelper.OpenFileInExplorer(new FileInfo(options.OutputFile));
+    }
+
+    [TestMethod]
+    public async Task ReplaceColorTask3()
+    {
+        var name = "file_example_TIFF_1MB.tiff";
+        Image<Rgba32> image = await Image.LoadAsync<Rgba32>(TestFileProvider.GetTestFilePath(name));
+        var list = image.GetColorCountList();
+        list = list.OrderByDescending(t => t.Count).ToList();
+
+        var replaceColorInfoList = new List<ReplaceColorInfo>();
+
+        for (var i = 0; i < list.Count && i < 10; i++)
+        {
+            var color = list[i].Color;
+            replaceColorInfoList.Add(new ReplaceColorInfo($"#{color.A:X2}{color.R:X2}{color.G:X2}{color.B:X2}", "#00FFFFFF"));
+        }
+
+        var imageConvertContext = new ImageConvertContext()
+        {
+            ImageConvertTaskList =
+            [
+                new ReplaceColorTask()
+                {
+                    ReplaceColorInfoList = replaceColorInfoList
+                },
+            ]
+        };
+
+        var options = ToOptions(name, imageConvertContext);
+
+        var result = await Program.RunAsync(options);
+        Assert.AreEqual(ErrorCode.Success, result);
+        AssertReplaceColor(options);
+        TestHelper.OpenFileInExplorer(new FileInfo(options.OutputFile));
+    }
+
+    private void AssertReplaceColor(Options options)
+    {
+        var inputFile = options.InputFile;
+        using var image = Image.Load<Rgba32>(inputFile);
+
+        var list = image.GetColorCountList();
+        list = list.OrderByDescending(t => t.Count).ToList();
+        var replaceList = new List<(ColorMetadata SourceMetadata, ColorMetadata TargetMetadata)>();
+        for (var i = 0; i < list.Count && i < 10; i++)
+        {
+            var color = list[i].Color;
+            var targetMetadata = new ColorMetadata(new Rgba32(0xFF, 0xFF, 0xFF, 0x00));
+            replaceList.Add((new ColorMetadata(color), targetMetadata));
+        }
+
+        image.ReplaceColor(replaceList);
+
+        var tempFile = Path.Join(options.WorkingFolder, $"Assert_{Path.GetRandomFileName()}.png");
+        image.SaveAsPng(tempFile, new PngEncoder()
+        {
+            ColorType = PngColorType.RgbWithAlpha
+        });
+
+        var visionComparer = new VisionComparer();
+        var visionCompareResult = visionComparer.Compare(new FileInfo(tempFile), new FileInfo(options.OutputFile));
+        Assert.IsTrue(visionCompareResult.IsSimilar());
+    }
+
     [TestMethod]
     public async Task OptimizeImageFile1()
     {

src/MediaConverters/MediaConverters.Tool/Program.cs

@@ -9,7 +9,7 @@
 
 using System.Diagnostics;
 using System.Text.Json;
-
+using SixLabors.ImageSharp.Formats.Png;
 using SourceGenerationContext = DotNetCampus.MediaConverters.Contexts.SourceGenerationContext;
 
 namespace DotNetCampus.MediaConverters;
@@ -75,9 +75,18 @@ internal static async Task<ErrorCode> RunAsync(Options options)
             {
                 workerProvider.Run(image, imageConvertTask);
             }
+
+            await image.SaveAsPngAsync(options.OutputFile, new PngEncoder()
+            {
+                ColorType = PngColorType.RgbWithAlpha,
+                BitDepth = PngBitDepth.Bit8,
+            });
         }
 
-        optimizedImageFile.CopyTo(options.OutputFile, overwrite: true);
+        else
+        {
+            optimizedImageFile.CopyTo(options.OutputFile, overwrite: true);
+        }
 
         return ErrorCode.Success;
     }