using AcDream.App.Rendering.Wb;
using AcDream.Core.Meshing;

namespace AcDream.Core.Tests.Rendering.Wb;

public sealed class AcSurfaceMetadataTableTests
{
    [Fact]
    public void Add_ThenLookup_RoundTripsSameMetadata()
    {
        var table = new AcSurfaceMetadataTable();
        var meta = new AcSurfaceMetadata(
            Translucency: TranslucencyKind.AlphaBlend,
            Luminosity: 0.5f,
            Diffuse: 0.8f,
            SurfOpacity: 0.7f,
            NeedsUvRepeat: true,
            DisableFog: false);

        table.Add(gfxObjId: 0x01000123ul, surfaceIdx: 2, meta);

        Assert.True(table.TryLookup(0x01000123ul, 2, out var got));
        Assert.Equal(meta, got);
    }

    [Fact]
    public void Lookup_MissingKey_ReturnsFalse()
    {
        var table = new AcSurfaceMetadataTable();
        Assert.False(table.TryLookup(0xDEADBEEFul, 0, out _));
    }

    [Fact]
    public void Add_OverwritesPreviousMetadata()
    {
        var table = new AcSurfaceMetadataTable();
        var first = new AcSurfaceMetadata(TranslucencyKind.Opaque, 0f, 1f, 1f, false, false);
        var second = new AcSurfaceMetadata(TranslucencyKind.Additive, 1f, 1f, 1f, false, true);

        table.Add(0xAAAA, 0, first);
        table.Add(0xAAAA, 0, second);

        Assert.True(table.TryLookup(0xAAAA, 0, out var got));
        Assert.Equal(second, got);
    }

    [Fact]
    public void Add_FromMultipleThreads_IsThreadSafe()
    {
        var table = new AcSurfaceMetadataTable();
        var threads = new System.Threading.Tasks.Task[8];
        for (int t = 0; t < 8; t++)
        {
            int threadIdx = t;
            threads[t] = System.Threading.Tasks.Task.Run(() =>
            {
                for (int i = 0; i < 1000; i++)
                {
                    ulong key = (ulong)(threadIdx * 1000 + i);
                    table.Add(key, 0, new AcSurfaceMetadata(
                        TranslucencyKind.Opaque, 0f, 1f, 1f, false, false));
                }
            });
        }
        System.Threading.Tasks.Task.WaitAll(threads);

        // 8000 entries should be present.
        for (int t = 0; t < 8; t++)
        for (int i = 0; i < 1000; i++)
            Assert.True(table.TryLookup((ulong)(t * 1000 + i), 0, out _));
    }
}