测试替换颜色

Author: lindexi

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

@@ -1,13 +1,16 @@
-using SixLabors.ImageSharp;
+using DotNetCampus.MediaConverters.Imaging.Effect.Colors;
+using DotNetCampus.MediaConverters.Imaging.Effect.Extensions;
+
+using SixLabors.ImageSharp;
+using SixLabors.ImageSharp.Advanced;
 using SixLabors.ImageSharp.PixelFormats;
 
 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;
-using SixLabors.ImageSharp.Advanced;
+
+using static System.Net.Mime.MediaTypeNames;
 
 namespace DotNetCampus.MediaConverters.Imaging.Effect;
 
@@ -185,6 +188,18 @@ public static void SetSoftEdgeEffect(this Image<Rgba32> bitmap, float radius)
     /// </summary>
     /// <param name="image"></param>
     public static ColorCount GetMaxCountColor(this Image<Rgba32> image)
+    {
+        var list = image.GetColorCount();
+
+        return list.MaxBy(t => t.Count);
+    }
+
+    /// <summary>
+    /// 获取无序的颜色数量列表
+    /// </summary>
+    /// <param name="image"></param>
+    /// <returns></returns>
+    public static IReadOnlyList<ColorCount> GetColorCount(this Image<Rgba32> image)
     {
         var dictionary = new Dictionary<Rgba32, int>();
         image.ProcessPixelRows(accessor =>
@@ -206,8 +221,7 @@ public static ColorCount GetMaxCountColor(this Image<Rgba32> image)
             }
         });
 
-        var (color, count) = dictionary.MaxBy(pair => pair.Value);
-        return new ColorCount(color, count);
+        return [..dictionary.Select(t => new ColorCount(t.Key, t.Value))];
     }
 }
 

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

@@ -46,4 +46,25 @@ public void ReplaceColorTest2()
         var file = image.SaveAndCompareTestFile("ReplaceColorTest1.png");
         TestHelper.OpenFileInExplorer(file);
     }
+
+    [TestMethod()]
+    public void ReplaceColorTest3()
+    {
+        Image<Rgba32> image = TestFileProvider.GetDefaultTestImage();
+        var list = image.GetColorCount();
+        list = list.OrderByDescending(t => t.Count).ToList();
+
+        var targetColor = new Rgba32(0xFF, 0xFF, 0xFF, 0x00);
+        var targetMetadata = new ColorMetadata(targetColor);
+
+        Dictionary<ColorMetadata, ColorMetadata> colorInfos = [];
+        for (var i = 0; i < list.Count && i < 10; i++)
+        {
+            colorInfos[new ColorMetadata(list[i].Color)] = targetMetadata;
+        }
+
+        image.ReplaceColor(colorInfos);
+        var file = image.SaveAndCompareTestFile("ReplaceColorTest3.png");
+        TestHelper.OpenFileInExplorer(file);
+    }
 }