ImageSharp/tests/ImageSharp.Tests/Formats/Jpg/JpegProfilingBenchmarks.cs

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

// ReSharper disable InconsistentNaming

using SixLabors.ImageSharp.PixelFormats;

namespace SixLabors.ImageSharp.Tests.Formats.Jpg
{
    using System;
    using System.IO;
    using System.Linq;
    using System.Numerics;

    using SixLabors.ImageSharp.Formats;
    using SixLabors.ImageSharp.Formats.Jpeg;
    using SixLabors.ImageSharp.Formats.Jpeg.GolangPort;
    using SixLabors.ImageSharp.Formats.Jpeg.PdfJsPort;

    using Xunit;
    using Xunit.Abstractions;

    public class JpegProfilingBenchmarks : MeasureFixture
    {
        public JpegProfilingBenchmarks(ITestOutputHelper output)
            : base(output)
        {
        }

        public static readonly TheoryData<string> DecodeJpegData = new TheoryData<string>()
        {
            TestImages.Jpeg.Baseline.Cmyk,
            TestImages.Jpeg.Baseline.Ycck,
            TestImages.Jpeg.Baseline.Calliphora,
            TestImages.Jpeg.Baseline.Jpeg400,
            TestImages.Jpeg.Baseline.Jpeg420Exif,
            TestImages.Jpeg.Baseline.Jpeg444,
        };

        //[Theory] // Benchmark, enable manually
        //[MemberData(nameof(DecodeJpegData))]
        public void DecodeJpeg_Original(string fileName)
        {
            this.DecodeJpegBenchmarkImpl(fileName, new OrigJpegDecoder());
        }

        // [Theory] // Benchmark, enable manually
        // [MemberData(nameof(DecodeJpegData))]
        public void DecodeJpeg_PdfJs(string fileName)
        {
            this.DecodeJpegBenchmarkImpl(fileName, new PdfJsJpegDecoder());
        }

        private void DecodeJpegBenchmarkImpl(string fileName, IImageDecoder decoder)
        {
            // do not run this on CI even by accident
            if (TestEnvironment.RunsOnCI)
            {
                return;
            }

            const int ExecutionCount = 30;

            if (!Vector.IsHardwareAccelerated)
            {
                throw new Exception("Vector.IsHardwareAccelerated == false! ('prefer32 bit' enabled?)");
            }

            string path = TestFile.GetInputFileFullPath(fileName);
            byte[] bytes = File.ReadAllBytes(path);

            this.Measure(
                ExecutionCount,
                () =>
                    {
                        Image<Rgba32> img = Image.Load<Rgba32>(bytes, decoder);
                    },
                // ReSharper disable once ExplicitCallerInfoArgument
                $"Decode {fileName}");
        }

        // Benchmark, enable manually!
        // [Theory]
        // [InlineData(1, 75, JpegSubsample.Ratio420)]
        // [InlineData(30, 75, JpegSubsample.Ratio420)]
        // [InlineData(30, 75, JpegSubsample.Ratio444)]
        // [InlineData(30, 100, JpegSubsample.Ratio444)]
        public void EncodeJpeg(int executionCount, int quality, JpegSubsample subsample)
        {
            // do not run this on CI even by accident
            if (TestEnvironment.RunsOnCI)
            {
                return;
            }

            string[] testFiles = TestImages.Bmp.All
                .Concat(new[] { TestImages.Jpeg.Baseline.Calliphora, TestImages.Jpeg.Baseline.Cmyk })
                .ToArray();

            Image<Rgba32>[] testImages =
                testFiles.Select(
                        tf => TestImageProvider<Rgba32>.File(tf, pixelTypeOverride: PixelTypes.Rgba32).GetImage())
                    .ToArray();

            using (MemoryStream ms = new MemoryStream())
            {
                this.Measure(executionCount,
                    () =>
                    {
                        foreach (Image<Rgba32> img in testImages)
                        {
                            JpegEncoder options = new JpegEncoder { Quality = quality, Subsample = subsample };
                            img.Save(ms, options);
                            ms.Seek(0, SeekOrigin.Begin);
                        }
                    },
                    // ReSharper disable once ExplicitCallerInfoArgument
                    $@"Encode {testFiles.Length} images"
                    );
            }
        }

    }
}