加上更多单元测试

Author: lindexi

src/MediaConverters/MediaConverters.Lib/Imaging/Effect/BitmapEffectExtension.cs

@@ -3,6 +3,7 @@
 
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using System.Threading.Tasks;
 using DotNetCampus.MediaConverters.Imaging.Effect.Colors;
 using DotNetCampus.MediaConverters.Imaging.Effect.Extensions;
@@ -178,4 +179,36 @@ public static void SetSoftEdgeEffect(this Image<Rgba32> bitmap, float radius)
 
         SoftEdgeHelper.SetSoftEdgeMask(bitmap, radius);
     }
-}
+
+    /// <summary>
+    /// 获取一张图中颜色数量最多的颜色
+    /// </summary>
+    /// <param name="image"></param>
+    public static ColorCount GetMaxCountColor(this Image<Rgba32> image)
+    {
+        var dictionary = new Dictionary<Rgba32, int>();
+        image.ProcessPixelRows(accessor =>
+        {
+            for (int row = 0; row < accessor.Height; row++)
+            {
+                var pixelRow = accessor.GetRowSpan(row);
+                foreach (var pixel in pixelRow)
+                {
+                    if (dictionary.TryGetValue(pixel, out int count))
+                    {
+                        dictionary[pixel] = count + 1;
+                    }
+                    else
+                    {
+                        dictionary[pixel] = 1;
+                    }
+                }
+            }
+        });
+
+        var (color, count) = dictionary.MaxBy(pair => pair.Value);
+        return new ColorCount(color, count);
+    }
+}
+
+public readonly record struct ColorCount(Rgba32 Color, int Count);

src/MediaConverters/MediaConverters.Tests/Assets/TestFiles/ReplaceColorTest1.png

@@ -2,6 +2,7 @@
 using SixLabors.ImageSharp;
 using SixLabors.ImageSharp.Formats.Png;
 using SixLabors.ImageSharp.PixelFormats;
+using TextVisionComparer;
 
 namespace DotNetCampus.MediaConverters.Tests;
 
@@ -16,31 +17,51 @@ public static void AssemblyInit(TestContext context)
 
     private static DirectoryInfo WorkingDirectory { get; set; } = null!;
 
-    public static string SaveAsTestImageFile(this Image<Rgba32> image)
+    public static FileInfo SaveAsTestImageFile(this Image<Rgba32> image)
     {
         var file = Path.Join(WorkingDirectory.FullName, Path.GetRandomFileName() + ".png");
         using var fileStream = File.OpenWrite(file);
         image.SaveAsPng(fileStream, new PngEncoder()
         {
             ColorType = PngColorType.RgbWithAlpha
         });
-        return file;
+        return new FileInfo(file);
     }
 
-    public static void OpenFileInExplorer(string filePath)
+    public static FileInfo SaveAndCompareTestFile(this Image<Rgba32> image, string? testFileName = null)
     {
-        if (File.Exists(filePath))
+        FileInfo testFile = image.SaveAsTestImageFile();
+
+        if (testFileName != null)
+        {
+            var file = TestFileProvider.GetTestFile(testFileName);
+            CompareImageFile(testFile, file);
+        }
+
+        return testFile;
+    }
+
+    public static void CompareImageFile(FileInfo file1, FileInfo file2)
+    {
+        var visionComparer = new VisionComparer();
+        var visionCompareResult = visionComparer.Compare(file1,file2);
+        Assert.IsTrue(visionCompareResult.IsSimilar());
+    }
+
+    public static void OpenFileInExplorer(FileInfo file)
+    {
+        if (File.Exists(file.FullName))
         {
 #if DEBUG
             if (Debugger.IsAttached)
             {
-                Process.Start(new ProcessStartInfo("explorer", $"\"{filePath}\"") { UseShellExecute = true });
+                Process.Start(new ProcessStartInfo("explorer", $"\"{file.FullName}\"") { UseShellExecute = true });
             }
 #endif
         }
         else
         {
-            throw new FileNotFoundException($"The file '{filePath}' does not exist.");
+            throw new FileNotFoundException($"The file '{file}' does not exist.");
         }
     }
 }

src/MediaConverters/MediaConverters.Tests/Assets/TestFiles/SetSoftEdgeMaskResult1.png

@@ -0,0 +1,34 @@
+using DotNetCampus.MediaConverters.Imaging.Effect;
+
+using SixLabors.ImageSharp;
+using SixLabors.ImageSharp.PixelFormats;
+
+namespace DotNetCampus.MediaConverters.Tests.Imaging.Effect;
+
+[TestClass()]
+public class BitmapEffectExtensionTests
+{
+    [TestMethod()]
+    public void TestGetMaxCountColor()
+    {
+        Image<Rgba32> image = TestFileProvider.GetDefaultTestImage();
+        var (rgba32, count) = image.GetMaxCountColor();
+
+        var expected = new Rgba32(0xF1, 0xD7, 0xA6, 0xFF);
+        Assert.AreEqual(expected, rgba32);
+        Assert.AreEqual(1215, count);
+    }
+
+    [TestMethod()]
+    public void ReplaceColorTest1()
+    {
+        Image<Rgba32> image = TestFileProvider.GetDefaultTestImage();
+        var (rgba32, count) = image.GetMaxCountColor();
+        _ = count;
+        var targetColor = new Rgba32(0xFF, 0xFF, 0xFF, 0x00);
+        image.ReplaceColor(rgba32, targetColor);
+
+        var file = image.SaveAndCompareTestFile("ReplaceColorTest1.png");
+        TestHelper.OpenFileInExplorer(file);
+    }
+}

src/MediaConverters/MediaConverters.Tests/Framework/TestHelper.cs

@@ -18,9 +18,8 @@ public void SetSoftEdgeMaskTest()
 
         SoftEdgeHelper.SetSoftEdgeMask(image, 50.0f);
 
-        var file = image.SaveAsTestImageFile();
+        var file = image.SaveAndCompareTestFile("SetSoftEdgeMaskResult1.png");
 
-        Assert.IsTrue(File.Exists(file));
         TestHelper.OpenFileInExplorer(file);
     }
 

src/MediaConverters/MediaConverters.Tests/Imaging/Effect/BitmapEffectExtensionTests.cs

@@ -16,6 +16,7 @@
 
   <ItemGroup>
     <ProjectReference Include="..\MediaConverters.Lib\MediaConverters.Lib.csproj" />
+    <ProjectReference Include="..\TextVisionComparer\TextVisionComparer.csproj" />
   </ItemGroup>
 
   <ItemGroup>