ImageSharp/tests/ImageSharp.Tests/Formats/WebP/WebPDecoderTests.cs

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

using System.IO;

using SixLabors.ImageSharp.Formats.WebP;
using SixLabors.ImageSharp.PixelFormats;
using SixLabors.ImageSharp.Tests.TestUtilities.ReferenceCodecs;

using Xunit;

// ReSharper disable InconsistentNaming

namespace SixLabors.ImageSharp.Tests.Formats.WebP
{
    using static SixLabors.ImageSharp.Tests.TestImages.WebP;

    public class WebPDecoderTests
    {
        private static WebPDecoder WebpDecoder => new WebPDecoder();
        private static MagickReferenceDecoder ReferenceDecoder => new MagickReferenceDecoder();

        [Theory]
        [InlineData(Lossless.Lossless1, 1000, 307, 24)]
        [InlineData(Lossless.Lossless2, 1000, 307, 24)]
        [InlineData(Lossy.Alpha.LossyAlpha1, 1000, 307, 32)]
        [InlineData(Lossy.Alpha.LossyAlpha2, 1000, 307, 32)]
        public void Identify_DetectsCorrectDimensions(
            string imagePath,
            int expectedWidth,
            int expectedHeight,
            int expectedBitsPerPixel)
        {
            var testFile = TestFile.Create(imagePath);
            using (var stream = new MemoryStream(testFile.Bytes, false))
            {
                IImageInfo imageInfo = Image.Identify(stream);
                Assert.NotNull(imageInfo);
                Assert.Equal(expectedWidth, imageInfo.Width);
                Assert.Equal(expectedHeight, imageInfo.Height);
                Assert.Equal(expectedBitsPerPixel, imageInfo.PixelType.BitsPerPixel);
            }
        }

        [Theory]
        [WithFile(Lossy.Bike, PixelTypes.Rgba32)]
        [WithFile(Lossy.LenaIccp, PixelTypes.Rgba32)]
        [WithFile(Lossy.Lossy01, PixelTypes.Rgba32)]
        [WithFile(Lossy.Lossy02, PixelTypes.Rgba32)]
        [WithFile(Lossy.Lossy03, PixelTypes.Rgba32)]
        [WithFile(Lossy.Lossy04, PixelTypes.Rgba32)]
        [WithFile(Lossy.Lossy05, PixelTypes.Rgba32)]
        [WithFile(Lossy.Lossy06, PixelTypes.Rgba32)]
        [WithFile(Lossy.Lossy07, PixelTypes.Rgba32)]
        [WithFile(Lossy.Lossy08, PixelTypes.Rgba32)]
        [WithFile(Lossy.Lossy09, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossy<TPixel>(TestImageProvider<TPixel> provider)
            where TPixel : struct, IPixel<TPixel>
        {
            using (Image<TPixel> image = provider.GetImage(WebpDecoder))
            {
                image.DebugSave(provider);
                image.CompareToOriginal(provider, ReferenceDecoder);
            }
        }

        [Theory]
        [WithFile(Lossy.Alpha.LossyAlpha1, PixelTypes.Rgba32)]
        [WithFile(Lossy.Alpha.LossyAlpha2, PixelTypes.Rgba32)]
        [WithFile(Lossy.Alpha.LossyAlpha3, PixelTypes.Rgba32)]
        [WithFile(Lossy.Alpha.LossyAlpha4, PixelTypes.Rgba32)]
        [WithFile(Lossy.Alpha.LossyAlphaNoCompression, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossy_WithAlpha<TPixel>(TestImageProvider<TPixel> provider)
            where TPixel : struct, IPixel<TPixel>
        {
            using (Image<TPixel> image = provider.GetImage(WebpDecoder))
            {
                image.DebugSave(provider);
                image.CompareToOriginal(provider, ReferenceDecoder);
            }
        }

        [Theory]
        [WithFile(Lossless.NoTransform1, PixelTypes.Rgba32)]
        [WithFile(Lossless.NoTransform2, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossless_WithoutTransforms<TPixel>(TestImageProvider<TPixel> provider)
            where TPixel : struct, IPixel<TPixel>
        {
            using (Image<TPixel> image = provider.GetImage(WebpDecoder))
            {
                image.DebugSave(provider);
                image.CompareToOriginal(provider, ReferenceDecoder);
            }
        }

        [Theory]
        [WithFile(Lossless.GreenTransform1, PixelTypes.Rgba32)]
        [WithFile(Lossless.GreenTransform2, PixelTypes.Rgba32)]
        [WithFile(Lossless.GreenTransform3, PixelTypes.Rgba32)]
        [WithFile(Lossless.GreenTransform4, PixelTypes.Rgba32)]
        // TODO: Reference decoder throws here MagickCorruptImageErrorException, webpinfo also indicates an error here, but decoding the image seems to work.
        // [WithFile(Lossless.GreenTransform5, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossless_WithSubstractGreenTransform<TPixel>(
            TestImageProvider<TPixel> provider)
            where TPixel : struct, IPixel<TPixel>
        {
            using (Image<TPixel> image = provider.GetImage(WebpDecoder))
            {
                image.DebugSave(provider);
                image.CompareToOriginal(provider, ReferenceDecoder);
            }
        }

        [Theory]
        [WithFile(Lossless.ColorIndexTransform1, PixelTypes.Rgba32)]
        [WithFile(Lossless.ColorIndexTransform2, PixelTypes.Rgba32)]
        [WithFile(Lossless.ColorIndexTransform3, PixelTypes.Rgba32)]
        [WithFile(Lossless.ColorIndexTransform4, PixelTypes.Rgba32)]
        [WithFile(Lossless.ColorIndexTransform5, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossless_WithColorIndexTransform<TPixel>(TestImageProvider<TPixel> provider)
            where TPixel : struct, IPixel<TPixel>
        {
            using (Image<TPixel> image = provider.GetImage(WebpDecoder))
            {
                image.DebugSave(provider);
                image.CompareToOriginal(provider, ReferenceDecoder);
            }
        }

        [Theory]
        [WithFile(Lossless.PredictorTransform1, PixelTypes.Rgba32)]
        [WithFile(Lossless.PredictorTransform2, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossless_WithPredictorTransform<TPixel>(TestImageProvider<TPixel> provider)
            where TPixel : struct, IPixel<TPixel>
        {
            using (Image<TPixel> image = provider.GetImage(WebpDecoder))
            {
                image.DebugSave(provider);
                image.CompareToOriginal(provider, ReferenceDecoder);
            }
        }

        [Theory]
        [WithFile(Lossless.CrossColorTransform1, PixelTypes.Rgba32)]
        [WithFile(Lossless.CrossColorTransform2, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossless_WithCrossColorTransform<TPixel>(TestImageProvider<TPixel> provider)
            where TPixel : struct, IPixel<TPixel>
        {
            using (Image<TPixel> image = provider.GetImage(WebpDecoder))
            {
                image.DebugSave(provider);
                image.CompareToOriginal(provider, ReferenceDecoder);
            }
        }

        [Theory]
        [WithFile(Lossless.TwoTransforms1, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms2, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms3, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms4, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms5, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms6, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms7, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms8, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms9, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms10, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms11, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms12, PixelTypes.Rgba32)]
        [WithFile(Lossless.TwoTransforms13, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossless_WithTwoTransforms<TPixel>(TestImageProvider<TPixel> provider)
            where TPixel : struct, IPixel<TPixel>
        {
            using (Image<TPixel> image = provider.GetImage(WebpDecoder))
            {
                image.DebugSave(provider);
                image.CompareToOriginal(provider, ReferenceDecoder);
            }
        }

        [Theory]
        [WithFile(Lossless.ThreeTransforms1, PixelTypes.Rgba32)]
        [WithFile(Lossless.ThreeTransforms2, PixelTypes.Rgba32)]
        [WithFile(Lossless.ThreeTransforms3, PixelTypes.Rgba32)]
        [WithFile(Lossless.ThreeTransforms4, PixelTypes.Rgba32)]
        [WithFile(Lossless.ThreeTransforms5, PixelTypes.Rgba32)]
        [WithFile(Lossless.ThreeTransforms6, PixelTypes.Rgba32)]
        [WithFile(Lossless.ThreeTransforms7, PixelTypes.Rgba32)]
        [WithFile(Lossless.ThreeTransforms8, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossless_WithThreeTransforms<TPixel>(TestImageProvider<TPixel> provider)
            where TPixel : struct, IPixel<TPixel>
        {
            using (Image<TPixel> image = provider.GetImage(WebpDecoder))
            {
                image.DebugSave(provider);
                image.CompareToOriginal(provider, ReferenceDecoder);
            }
        }

        [Theory]
        [WithFile(Lossless.LossLessCorruptImage1, PixelTypes.Rgba32)]
        [WithFile(Lossless.LossLessCorruptImage2, PixelTypes.Rgba32)]
        [WithFile(Lossless.LossLessCorruptImage3, PixelTypes.Rgba32)]
        [WithFile(Lossless.LossLessCorruptImage4, PixelTypes.Rgba32)]
        public void WebpDecoder_ThrowImageFormatException_OnInvalidImages<TPixel>(TestImageProvider<TPixel> provider)
            where TPixel : struct, IPixel<TPixel>
        {
            Assert.Throws<ImageFormatException>(() => { using (provider.GetImage(WebpDecoder)) { } });
        }
    }
}