// 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.Alpha1, 1000, 307, 32)]
        [InlineData(Lossy.Alpha2, 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.NoFilter01, PixelTypes.Rgba32)]
        [WithFile(Lossy.NoFilter02, PixelTypes.Rgba32)]
        [WithFile(Lossy.NoFilter03, PixelTypes.Rgba32)]
        [WithFile(Lossy.NoFilter04, PixelTypes.Rgba32)]
        [WithFile(Lossy.NoFilter05, PixelTypes.Rgba32)]
        [WithFile(Lossy.SegmentationNoFilter01, PixelTypes.Rgba32)]
        [WithFile(Lossy.SegmentationNoFilter02, PixelTypes.Rgba32)]
        [WithFile(Lossy.SegmentationNoFilter03, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossy_WithoutFilter<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.SimpleFilter01, PixelTypes.Rgba32)]
        [WithFile(Lossy.SimpleFilter02, PixelTypes.Rgba32)]
        [WithFile(Lossy.SimpleFilter03, PixelTypes.Rgba32)]
        [WithFile(Lossy.SimpleFilter04, PixelTypes.Rgba32)]
        [WithFile(Lossy.SimpleFilter05, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossy_WithSimpleFilter<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.IccpComplexFilter, PixelTypes.Rgba32)]
        [WithFile(Lossy.VeryShort, PixelTypes.Rgba32)]
        [WithFile(Lossy.BikeComplexFilter, PixelTypes.Rgba32)]
        [WithFile(Lossy.ComplexFilter01, PixelTypes.Rgba32)]
        [WithFile(Lossy.ComplexFilter02, PixelTypes.Rgba32)]
        [WithFile(Lossy.ComplexFilter03, PixelTypes.Rgba32)]
        [WithFile(Lossy.ComplexFilter04, PixelTypes.Rgba32)]
        [WithFile(Lossy.ComplexFilter05, PixelTypes.Rgba32)]
        [WithFile(Lossy.ComplexFilter06, PixelTypes.Rgba32)]
        [WithFile(Lossy.ComplexFilter07, PixelTypes.Rgba32)]
        [WithFile(Lossy.ComplexFilter08, PixelTypes.Rgba32)]
        [WithFile(Lossy.ComplexFilter09, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossy_WithComplexFilter<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.Small01, PixelTypes.Rgba32)]
        [WithFile(Lossy.Small02, PixelTypes.Rgba32)]
        [WithFile(Lossy.Small03, PixelTypes.Rgba32)]
        [WithFile(Lossy.Small04, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossy_VerySmall<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.SegmentationNoFilter04, PixelTypes.Rgba32)]
        [WithFile(Lossy.SegmentationNoFilter05, PixelTypes.Rgba32)]
        [WithFile(Lossy.SegmentationNoFilter06, PixelTypes.Rgba32)]
        [WithFile(Lossy.SegmentationComplexFilter01, PixelTypes.Rgba32)]
        [WithFile(Lossy.SegmentationComplexFilter02, PixelTypes.Rgba32)]
        [WithFile(Lossy.SegmentationComplexFilter03, PixelTypes.Rgba32)]
        [WithFile(Lossy.SegmentationComplexFilter04, PixelTypes.Rgba32)]
        [WithFile(Lossy.SegmentationComplexFilter05, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossy_WithPartitions<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.Partitions01, PixelTypes.Rgba32)]
        [WithFile(Lossy.Partitions02, PixelTypes.Rgba32)]
        [WithFile(Lossy.Partitions03, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossy_WithSegmentation<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.Sharpness01, PixelTypes.Rgba32)]
        [WithFile(Lossy.Sharpness02, PixelTypes.Rgba32)]
        [WithFile(Lossy.Sharpness03, PixelTypes.Rgba32)]
        [WithFile(Lossy.Sharpness04, PixelTypes.Rgba32)]
        [WithFile(Lossy.Sharpness05, PixelTypes.Rgba32)]
        [WithFile(Lossy.Sharpness06, PixelTypes.Rgba32)]
        public void WebpDecoder_CanDecode_Lossy_WithSharpnessLevel<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.Alpha1, PixelTypes.Rgba32)]
        [WithFile(Lossy.Alpha2, PixelTypes.Rgba32)]
        [WithFile(Lossy.Alpha3, PixelTypes.Rgba32)]
        [WithFile(Lossy.Alpha4, PixelTypes.Rgba32)]
        [WithFile(Lossy.AlphaNoCompression, 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.BikeThreeTransforms, 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)) { } });
        }
    }
}