ImageSharp/tests/ImageSharp.Tests/Drawing/RecolorImageTests.cs

// Copyright (c) Six Labors and contributors.
// Licensed under the Apache License, Version 2.0.

using System;

using SixLabors.ImageSharp.PixelFormats;
using SixLabors.ImageSharp.Processing;
using SixLabors.Primitives;

using Xunit;

namespace SixLabors.ImageSharp.Tests.Drawing
{
    [GroupOutput("Drawing")]
    public class RecolorImageTests
    {
        [Theory]
        [WithFile(TestImages.Png.CalliphoraPartial, PixelTypes.Rgba32, "Yellow", "Pink", 0.2f)]
        [WithFile(TestImages.Png.CalliphoraPartial, PixelTypes.Bgra32, "Yellow", "Pink", 0.5f)]
        [WithTestPatternImages(100, 100, PixelTypes.Rgba32, "Red", "Blue", 0.2f)]
        [WithTestPatternImages(100, 100, PixelTypes.Rgba32, "Red", "Blue", 0.6f)]
        public void Recolor<TPixel>(TestImageProvider<TPixel> provider, string sourceColorName, string targetColorName, float threshold)
            where TPixel : struct, IPixel<TPixel>
        {
            Color sourceColor = TestUtils.GetColorByName(sourceColorName);
            Color targetColor = TestUtils.GetColorByName(targetColorName);
            var brush = new RecolorBrush<TPixel>(sourceColor.ToPixel<TPixel>(), targetColor.ToPixel<TPixel>(), threshold);

            FormattableString testInfo = $"{sourceColorName}-{targetColorName}-{threshold}";
            provider.RunValidatingProcessorTest(x => x.Fill(brush), testInfo);
        }
        
        [Theory]
        [WithFile(TestImages.Png.CalliphoraPartial, PixelTypes.Bgra32, "Yellow", "Pink", 0.5f)]
        [WithTestPatternImages(100, 100, PixelTypes.Rgba32, "Red", "Blue", 0.2f)]
        public void Recolor_InBox<TPixel>(TestImageProvider<TPixel> provider, string sourceColorName, string targetColorName, float threshold)
            where TPixel : struct, IPixel<TPixel>
        {
            Color sourceColor = TestUtils.GetColorByName(sourceColorName);
            Color targetColor = TestUtils.GetColorByName(targetColorName);
            var brush = new RecolorBrush<TPixel>(sourceColor.ToPixel<TPixel>(), targetColor.ToPixel<TPixel>(), threshold);

            FormattableString testInfo = $"{sourceColorName}-{targetColorName}-{threshold}";
            provider.RunValidatingProcessorTest(x =>
                {
                    Size size = x.GetCurrentSize();
                    var rectangle = new Rectangle(0, size.Height / 2 - size.Height / 4, size.Width, size.Height / 2);
                    x.Fill(brush, rectangle);
                }, testInfo);
        }
    }
}