using System;
using NUnit.Framework;
using System.IO;
using UniGLTF;
using Unity.Collections;
using Unity.PerformanceTesting;
using UnityEngine;
using UnityEngine.Profiling;
using UniVRM10;

public class GltfParsePerformanceTests
{
    static string AliciaPath => Path.GetFullPath(
        Path.Combine(Application.dataPath, "../Tests/Models/Alicia_vrm-0.51/AliciaSolid_vrm-0.51.vrm"));

    static Vrm10Instance CreateVrm10Instance(NativeArray<byte> bytes)
    {
        return Vrm10.LoadBytesAsync(
            bytes,
            canLoadVrm0X: true,
            awaitCaller: new ImmediateCaller(),
            materialGenerator: new BuiltInVrm10MaterialDescriptorGenerator()
        ).Result;
    }

    [Test, Performance]
    public void ParseGlb()
    {
        using var bytes = NativeFile.ReadAllBytes(AliciaPath);

        Measure.Method(() =>
            {
                using var data = GlbLowLevelParser.Parse(AliciaPath, bytes);
            })
            .WarmupCount(3)
            .MeasurementCount(20)
            .GC()
            .Run();
    }

    [Test, Performance]
    public void LoadVrm10Instance()
    {
        using var bytes = NativeFile.ReadAllBytes(AliciaPath);

        Measure.Method(() =>
            {
                var instance = CreateVrm10Instance(bytes);
                UnityEngine.Object.DestroyImmediate(instance.gameObject);
            })
            .WarmupCount(1)
            .MeasurementCount(10)
            .GC()
            .Run();
    }

    [Test, Performance]
    public void NativeAllocatedBytes()
    {
        using var bytes = NativeFile.ReadAllBytes(AliciaPath);

        var parseNative = new SampleGroup("Parse.Native.AllocatedBytes", SampleUnit.Kilobyte);
        var loadNative = new SampleGroup("Load.Native.AllocatedBytes", SampleUnit.Kilobyte);

        // warmup
        using (GlbLowLevelParser.Parse(AliciaPath, bytes)) { }
        UnityEngine.Object.DestroyImmediate(CreateVrm10Instance(bytes).gameObject);

        for (var i = 0; i < 5; ++i)
        {
            GC.Collect();
            GC.WaitForPendingFinalizers();

            var before = Profiler.GetTotalAllocatedMemoryLong();
            using (GlbLowLevelParser.Parse(AliciaPath, bytes))
            {
                var after = Profiler.GetTotalAllocatedMemoryLong();
                Measure.Custom(parseNative, (after - before) / 1024.0);
            }

            before = Profiler.GetTotalAllocatedMemoryLong();
            var instance = CreateVrm10Instance(bytes);
            var afterLoad = Profiler.GetTotalAllocatedMemoryLong();
            UnityEngine.Object.DestroyImmediate(instance.gameObject);
            Measure.Custom(loadNative, (afterLoad - before) / 1024.0);
        }
    }
}