// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using BenchmarkDotNet.Attributes;
using NUnit.Framework;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Framework.Graphics.Shapes;
using osuTK;
using osuTK.Graphics;

namespace osu.Framework.Benchmarks
{
    [MemoryDiagnoser]
    public class BenchmarkDrawableLoad : GameBenchmark
    {
        private TestGame game;

        [Test]
        [Benchmark]
        public void NonRecursive()
        {
            game.Schedule(() =>
            {
                Container container = new Container();

                for (int i = 0; i < 1000; i++)
                {
                    var box = new Box
                    {
                        Size = new Vector2(100),
                        Colour = Color4.Black
                    };

                    container.Add(box);
                }

                game.Clear();
                game.Add(container);
            });

            RunSingleFrame();
        }

        [Test]
        [Benchmark]
        public void SlightlyNested()
        {
            game.Schedule(() =>
            {
                Container container = new Container();

                for (int i = 0; i < 1000; i++)
                {
                    container.Add(new Container
                    {
                        Size = new Vector2(100),
                        Colour = Color4.Black,
                        Children = new Drawable[]
                        {
                            new Box
                            {
                                Colour = Color4.Black,
                                RelativeSizeAxes = Axes.Both,
                            },
                        }
                    });
                }

                game.Clear();
                game.Add(container);
            });

            RunSingleFrame();
        }

        [Test]
        [Benchmark]
        public void VeryNested()
        {
            game.Schedule(() =>
            {
                Container container = new Container();
                Container target = container;

                for (int i = 0; i < 1000; i++)
                {
                    var newContainer = new Container { Size = new Vector2(100), Colour = Color4.Black };

                    target.Add(newContainer);
                    target = newContainer;
                }

                game.Clear();
                game.Add(container);
            });

            RunSingleFrame();
        }

        protected override Game CreateGame() => game = new TestGame();

        private class TestGame : Game
        {
        }
    }
}