using BenchmarkDotNet.Attributes;
using BenchmarkDotNet.Jobs;
using BenchmarkDotNet.Running;
using ColorMine.ColorSpaces;
using static System.Net.Mime.MediaTypeNames;
using System.Drawing;

namespace com.clusterrr.Famicom.NesTiler.Benchmarks
{
    [SimpleJob(RuntimeMoniker.Net60, warmupCount: 2, targetCount: 5)]
    [MemoryDiagnoser]
    public class Benchmarks
    {
        const string ImagesPath = "Images";

        static void Main()
        {
            var summary = BenchmarkRunner.Run<Benchmarks>();
        }

        [Benchmark]
        public void BelayaAkula()
        {
            var imagePath = Path.Combine(ImagesPath, "belaya_akula.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Buhanka()
        {
            var imagePath = Path.Combine(ImagesPath, "buhanka.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Chernobyl()
        {
            var imagePath = Path.Combine(ImagesPath, "chernobyl.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Dira()
        {
            var imagePath = Path.Combine(ImagesPath, "dira.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Glaza()
        {
            var imagePath = Path.Combine(ImagesPath, "glaza.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Gorgona()
        {
            var imagePath = Path.Combine(ImagesPath, "gorgona.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Myatejl()
        {
            var imagePath = Path.Combine(ImagesPath, "myatej.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Pagoda()
        {
            var imagePath = Path.Combine(ImagesPath, "pagoda.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Rayon4()
        {
            var imagePath = Path.Combine(ImagesPath, "rayon4.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Shkola()
        {
            var imagePath = Path.Combine(ImagesPath, "shkola.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Sindikat()
        {
            var imagePath = Path.Combine(ImagesPath, "sindikat.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Sputnik()
        {
            var imagePath = Path.Combine(ImagesPath, "sputnik.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Sworm()
        {
            var imagePath = Path.Combine(ImagesPath, "sworm.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void TrailerPark()
        {
            var imagePath = Path.Combine(ImagesPath, "trailer-park.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void WarfaceLogo()
        {
            var imagePath = Path.Combine(ImagesPath, "warface_logo.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Zapravka()
        {
            var imagePath = Path.Combine(ImagesPath, "zapravka.gif");
            DoBenchmarkSplit4(imagePath);
        }

        [Benchmark]
        public void Jurassic()
        {
            var imagePath = Path.Combine(ImagesPath, "jurassic.png");
            DoBenchmarkSplit2(imagePath);
        }

        [Benchmark]
        public void Jurassic2()
        {
            var imagePath = Path.Combine(ImagesPath, "jurassic2.png");
            DoBenchmarkSplit2(imagePath);
        }

        [Benchmark]
        public void BlasterMasterLeft()
        {
            var imagePath = Path.Combine(ImagesPath, "blaster_master_left.png");
            DoBenchmarkNoSplit(imagePath);
        }

        [Benchmark]
        public void BlasterMasterRight()
        {
            var imagePath = Path.Combine(ImagesPath, "blaster_master_right.png");
            DoBenchmarkNoSplit(imagePath);
        }

        [Benchmark]
        public void BlasterMasterSharedPattern()
        {
            var imagePath1 = Path.Combine(ImagesPath, "blaster_master_left.png");
            var imagePath2 = Path.Combine(ImagesPath, "blaster_master_right.png");
            DoBenchmarkSharedPattern(imagePath1, imagePath2);
        }

        [Benchmark]
        public void MeLossy()
        {
            var imagePath = Path.Combine(ImagesPath, "me.png");
            DoBenchmarkSplit2Lossy(imagePath);
        }

        [Benchmark]
        public void Sprites8x8()
        {
            var imagePath = Path.Combine(ImagesPath, "sprites1.png");
            DoBenchmarkSprites8x8(imagePath);
        }

        [Benchmark]
        public void Sprites8x16()
        {
            var imagePath = Path.Combine(ImagesPath, "sprites2.png");
            DoBenchmarkSprites8x16(imagePath);
        }

        private string PatternTablePath(string prefix, int number) => $"{prefix}_pattern_{number}.bin";
        private string NameTablePath(string prefix, int number) => $"{prefix}_name_table_{number}.bin";
        private string AttrTablePath(string prefix, int number) => $"{prefix}_attr_table_{number}.bin";
        private string PalettePath(string prefix, int number) => $"{prefix}_palette_{number}.bin";
        private string TilesCsvPath(string prefix) => $"{prefix}_tiles.csv";
        private string PalettesCsvPath(string prefix) => $"{prefix}_palettes.csv";
        private string SpritesCsvPath(string prefix) => $"{prefix}_sprites.csv";

        public void DoBenchmarkNoSplit(string imagePath)
        {
            var prefix = Path.GetFileNameWithoutExtension(imagePath);
            var args = new string[] {
                "--enable-palettes", "0,1,2,3",
                "--in-0", $"{imagePath}",
                "--out-pattern-table-0", PatternTablePath(prefix, 0),
                "--out-name-table-0", NameTablePath(prefix, 0),
                "--out-attribute-table-0", AttrTablePath(prefix, 0),
                "--out-palette-0", PalettePath(prefix, 0),
                "--out-palette-1", PalettePath(prefix, 1),
                "--out-palette-2", PalettePath(prefix, 2),
                "--out-palette-3", PalettePath(prefix, 3),
                "--out-tiles-csv", TilesCsvPath(prefix),
                "--out-palettes-csv", PalettesCsvPath(prefix),
                "--quiet",
            };
            var r = Program.Main(args);
            if (r != 0) throw new InvalidOperationException($"Return code: {r}");
        }

        public void DoBenchmarkSharedPattern(string imagePath1, string imagePath2)
        {
            var prefix = Path.GetFileNameWithoutExtension(imagePath1) + "_" + Path.GetFileNameWithoutExtension(imagePath2);
            var args = new string[] {
                "--enable-palettes", "0,1,2,3",
                "--in-0", $"{imagePath1}",
                "--in-1", $"{imagePath2}",
                "--out-pattern-table", PatternTablePath(prefix, 0),
                "--out-name-table-0", NameTablePath(prefix, 0),
                "--out-name-table-1", NameTablePath(prefix, 1),
                "--out-attribute-table-0", AttrTablePath(prefix, 0),
                "--out-attribute-table-1", AttrTablePath(prefix, 1),
                "--out-palette-0", PalettePath(prefix, 0),
                "--out-palette-1", PalettePath(prefix, 1),
                "--out-palette-2", PalettePath(prefix, 2),
                "--out-palette-3", PalettePath(prefix, 3),
                "--share-pattern-table",
                "--out-tiles-csv", TilesCsvPath(prefix),
                "--out-palettes-csv", PalettesCsvPath(prefix),
                "--quiet",
            };
            var r = Program.Main(args);
            if (r != 0) throw new InvalidOperationException($"Return code: {r}");
        }

        public void DoBenchmarkSplit2(string imagePath)
        {
            var prefix = Path.GetFileNameWithoutExtension(imagePath);
            var args = new string[] {
                "--enable-palettes", "0,1,2,3",
                "--in-0", $"{imagePath}:0:128",
                "--in-1", $"{imagePath}:128:112",
                "--out-pattern-table-0", PatternTablePath(prefix, 0),
                "--out-pattern-table-1", PatternTablePath(prefix, 1),
                "--out-name-table-0", NameTablePath(prefix, 0),
                "--out-name-table-1", NameTablePath(prefix, 1),
                "--out-attribute-table-0", AttrTablePath(prefix, 0),
                "--out-attribute-table-1", AttrTablePath(prefix, 1),
                "--out-palette-0", PalettePath(prefix, 0),
                "--out-palette-1", PalettePath(prefix, 1),
                "--out-palette-2", PalettePath(prefix, 2),
                "--out-palette-3", PalettePath(prefix, 3),
                "--out-tiles-csv", TilesCsvPath(prefix),
                "--out-palettes-csv", PalettesCsvPath(prefix),
                "--quiet",
            };
            var r = Program.Main(args);
            if (r != 0) throw new InvalidOperationException($"Return code: {r}");
        }

        public void DoBenchmarkSplit2Lossy(string imagePath)
        {
            var prefix = Path.GetFileNameWithoutExtension(imagePath);
            var args = new string[] {
                "--enable-palettes", "0,1,2,3",
                "--in-0", $"{imagePath}:0:128",
                "--in-1", $"{imagePath}:128:112",
                "--out-pattern-table-0", PatternTablePath(prefix, 0),
                "--out-pattern-table-1", PatternTablePath(prefix, 1),
                "--out-name-table-0", NameTablePath(prefix, 0),
                "--out-name-table-1", NameTablePath(prefix, 1),
                "--out-attribute-table-0", AttrTablePath(prefix, 0),
                "--out-attribute-table-1", AttrTablePath(prefix, 1),
                "--out-palette-0", PalettePath(prefix, 0),
                "--out-palette-1", PalettePath(prefix, 1),
                "--out-palette-2", PalettePath(prefix, 2),
                "--out-palette-3", PalettePath(prefix, 3),
                "--lossy", "3",
                "--out-tiles-csv", TilesCsvPath(prefix),
                "--out-palettes-csv", PalettesCsvPath(prefix),
                "--quiet",
            };
            var r = Program.Main(args);
            if (r != 0) throw new InvalidOperationException($"Return code: {r}");
        }

        public void DoBenchmarkSplit4(string imagePath)
        {
            var prefix = Path.GetFileNameWithoutExtension(imagePath);
            var args = new string[] {
                "--enable-palettes", "0,1,2,3",
                "--in-0", $"{imagePath}:0:64",
                "--in-1", $"{imagePath}:64:64",
                "--in-2", $"{imagePath}:128:64",
                "--in-3", $"{imagePath}:192:48",
                "--out-pattern-table-0", PatternTablePath(prefix, 0),
                "--out-pattern-table-1", PatternTablePath(prefix, 1),
                "--out-pattern-table-2", PatternTablePath(prefix, 2),
                "--out-pattern-table-3", PatternTablePath(prefix, 3),
                "--out-name-table-0", NameTablePath(prefix, 0),
                "--out-name-table-1", NameTablePath(prefix, 1),
                "--out-name-table-2", NameTablePath(prefix, 2),
                "--out-name-table-3", NameTablePath(prefix, 3),
                "--out-attribute-table-0", AttrTablePath(prefix, 0),
                "--out-attribute-table-1", AttrTablePath(prefix, 1),
                "--out-attribute-table-2", AttrTablePath(prefix, 2),
                "--out-attribute-table-3", AttrTablePath(prefix, 3),
                "--out-palette-0", PalettePath(prefix, 0),
                "--out-palette-1", PalettePath(prefix, 1),
                "--out-palette-2", PalettePath(prefix, 2),
                "--out-palette-3", PalettePath(prefix, 3),
                "--out-tiles-csv", TilesCsvPath(prefix),
                "--out-palettes-csv", PalettesCsvPath(prefix),
                "--quiet",
            };
            var r = Program.Main(args);
            if (r != 0) throw new InvalidOperationException($"Return code: {r}");
        }

        public void DoBenchmarkSprites8x8(string imagePath)
        {
            var prefix = Path.GetFileNameWithoutExtension(imagePath);
            var args = new string[] {
                "--mode", "sprites8x8",
                "--enable-palettes", "0",
                "--in-0", imagePath,
                "--out-pattern-table-0", PatternTablePath(prefix, 0),
                "--out-palette-0", PalettePath(prefix, 0),
                "--out-tiles-csv", SpritesCsvPath(prefix),
                "--bg-color", "#000000",
                "--quiet",
            };
            var r = Program.Main(args);
            if (r != 0) throw new InvalidOperationException($"Return code: {r}");
        }

        public void DoBenchmarkSprites8x16(string imagePath)
        {
            var prefix = Path.GetFileNameWithoutExtension(imagePath);
            var args = new string[] {
                "--mode", "sprites8x16",
                "--enable-palettes", "0",
                "--in-0", imagePath,
                "--out-pattern-table-0", PatternTablePath(prefix, 0),
                "--out-palette-0", PalettePath(prefix, 0),
                "--out-tiles-csv", SpritesCsvPath(prefix),
                "--bg-color", "#000000",
                "--quiet",
            };
            var r = Program.Main(args);
            if (r != 0) throw new InvalidOperationException($"Return code: {r}");
        }
    }
}