acdream/tests/AcDream.Core.Tests/Rendering/Wb/WbDrawDispatcherIndirectBuilderTests.cs

using System.Numerics;
using AcDream.App.Rendering.Wb;
using AcDream.Core.Meshing;
using DatReaderWriter.Enums;
using Xunit;

namespace AcDream.Core.Tests.Rendering.Wb;

/// <summary>
/// Pure CPU test of <see cref="WbDrawDispatcher.BuildIndirectArrays"/>.
/// Verifies that a synthetic group set lays out into the indirect buffer
/// + parallel batch data with opaque section first, transparent second,
/// per-group fields propagated correctly.
/// </summary>
public sealed class WbDrawDispatcherIndirectBuilderTests
{
    [Fact]
    public void TwoOpaqueGroupsAndOneTransparent_LaysOutContiguouslyOpaqueFirst()
    {
        // Arrange — three groups: 2 opaque (12+1 instances) + 1 transparent (12 instances)
        var groups = new List<WbDrawDispatcher.IndirectGroupInput>
        {
            new(IndexCount: 100, FirstIndex: 0,   BaseVertex: 0,   InstanceCount: 12, FirstInstance: 0,  TextureIndex: 0xAA, TextureLayer: 0, Translucency: TranslucencyKind.Opaque),
            new(IndexCount: 200, FirstIndex: 100, BaseVertex: 0,   InstanceCount: 12, FirstInstance: 12, TextureIndex: 0xBB, TextureLayer: 0, Translucency: TranslucencyKind.AlphaBlend),
            new(IndexCount: 50,  FirstIndex: 300, BaseVertex: 100, InstanceCount: 1,  FirstInstance: 24, TextureIndex: 0xCC, TextureLayer: 0, Translucency: TranslucencyKind.Opaque),
        };

        var indirect = new DrawElementsIndirectCommand[16];
        var batch = new WbDrawDispatcher.BatchDataPublic[16];
        var cull = new CullMode[16];

        // Act
        var result = WbDrawDispatcher.BuildIndirectArrays(groups, indirect, batch, cull);

        // Assert layout
        Assert.Equal(2, result.OpaqueCount);
        Assert.Equal(1, result.TransparentCount);
        Assert.Equal(2 * 20, result.TransparentByteOffset);  // sizeof(DEIC) = 20

        // Opaque section, in input order (Task 10 callers sort)
        Assert.Equal(100u, indirect[0].Count);
        Assert.Equal(0u,   indirect[0].FirstIndex);
        Assert.Equal(0,    indirect[0].BaseVertex);
        Assert.Equal(12u,  indirect[0].InstanceCount);
        Assert.Equal(0u,   indirect[0].BaseInstance);

        Assert.Equal(50u,  indirect[1].Count);
        Assert.Equal(300u, indirect[1].FirstIndex);
        Assert.Equal(100,  indirect[1].BaseVertex);
        Assert.Equal(1u,   indirect[1].InstanceCount);
        Assert.Equal(24u,  indirect[1].BaseInstance);

        // Transparent section
        Assert.Equal(200u, indirect[2].Count);
        Assert.Equal(100u, indirect[2].FirstIndex);
        Assert.Equal(12u,  indirect[2].InstanceCount);
        Assert.Equal(12u,  indirect[2].BaseInstance);

        // BatchData parallel — same indices as indirect
        Assert.Equal(0xAAu, batch[0].TextureIndex);
        Assert.Equal(0xCCu, batch[1].TextureIndex);
        Assert.Equal(0xBBu, batch[2].TextureIndex);
        Assert.Equal(CullMode.CounterClockwise, cull[0]);
        Assert.Equal(CullMode.CounterClockwise, cull[1]);
        Assert.Equal(CullMode.CounterClockwise, cull[2]);
    }

    [Fact]
    public void CullModes_FollowOpaqueTransparentLayout()
    {
        var groups = new List<WbDrawDispatcher.IndirectGroupInput>
        {
            new(IndexCount: 10, FirstIndex: 0, BaseVertex: 0, InstanceCount: 1, FirstInstance: 0,
                TextureIndex: 0x1, TextureLayer: 0, Translucency: TranslucencyKind.Opaque,
                CullMode: CullMode.Clockwise),
            new(IndexCount: 20, FirstIndex: 10, BaseVertex: 0, InstanceCount: 1, FirstInstance: 1,
                TextureIndex: 0x2, TextureLayer: 0, Translucency: TranslucencyKind.AlphaBlend,
                CullMode: CullMode.None),
            new(IndexCount: 30, FirstIndex: 30, BaseVertex: 0, InstanceCount: 1, FirstInstance: 2,
                TextureIndex: 0x3, TextureLayer: 0, Translucency: TranslucencyKind.ClipMap,
                CullMode: CullMode.Landblock),
        };
        var indirect = new DrawElementsIndirectCommand[4];
        var batch = new WbDrawDispatcher.BatchDataPublic[4];
        var cull = new CullMode[4];

        var result = WbDrawDispatcher.BuildIndirectArrays(groups, indirect, batch, cull);

        Assert.Equal(2, result.OpaqueCount);
        Assert.Equal(CullMode.Clockwise, cull[0]);
        Assert.Equal(CullMode.Landblock, cull[1]);
        Assert.Equal(CullMode.None, cull[2]);
    }

    [Fact]
    public void EmptyGroupList_ProducesZeroCounts()
    {
        var groups = new List<WbDrawDispatcher.IndirectGroupInput>();
        var indirect = new DrawElementsIndirectCommand[0];
        var batch = new WbDrawDispatcher.BatchDataPublic[0];

        var result = WbDrawDispatcher.BuildIndirectArrays(groups, indirect, batch);

        Assert.Equal(0, result.OpaqueCount);
        Assert.Equal(0, result.TransparentCount);
        Assert.Equal(0, result.TransparentByteOffset);
    }

    [Fact]
    public void ClipMapTreatedAsOpaque()
    {
        // ClipMap surfaces (alpha-cutout) belong with the opaque pass
        // because the discard handles transparency, not blending.
        var groups = new List<WbDrawDispatcher.IndirectGroupInput>
        {
            new(IndexCount: 10, FirstIndex: 0, BaseVertex: 0, InstanceCount: 1, FirstInstance: 0, TextureIndex: 0x1, TextureLayer: 0, Translucency: TranslucencyKind.ClipMap),
        };
        var indirect = new DrawElementsIndirectCommand[4];
        var batch = new WbDrawDispatcher.BatchDataPublic[4];

        var result = WbDrawDispatcher.BuildIndirectArrays(groups, indirect, batch);

        Assert.Equal(1, result.OpaqueCount);
        Assert.Equal(0, result.TransparentCount);
    }

    [Fact]
    public void BatchDataPublic_LayoutMatchesPrivateBatchData()
    {
        // Task 10 will use MemoryMarshal.Cast<BatchData, BatchDataPublic> to
        // expose the dispatcher's per-frame BatchData[] scratch to BuildIndirectArrays
        // without copying. The cast is only safe if the structs have identical
        // layout (size, field offsets).
        //
        // Campaign V slice V2 (2026-07-27): TextureHandle (ulong, an
        // ARB_bindless_texture handle) became TextureIndex (uint) plus an
        // explicit Reserved pad word — a slot into the binding=9 handle table
        // (GpuBindingModel.StorageTextureTable) instead of the raw handle.
        // The struct stays 16 bytes and TextureLayer/Flags keep their offsets
        // (8/12), matching GpuBindingModel.GpuBatchDataStrideBytes and every
        // existing CPU writer, so both structs only need 4-byte packing now.
        Assert.Equal(16, System.Runtime.CompilerServices.Unsafe.SizeOf<WbDrawDispatcher.BatchDataPublic>());
        Assert.Equal(0,  (int)System.Runtime.InteropServices.Marshal.OffsetOf<WbDrawDispatcher.BatchDataPublic>(nameof(WbDrawDispatcher.BatchDataPublic.TextureIndex)));
        Assert.Equal(4,  (int)System.Runtime.InteropServices.Marshal.OffsetOf<WbDrawDispatcher.BatchDataPublic>(nameof(WbDrawDispatcher.BatchDataPublic.Reserved)));
        Assert.Equal(8,  (int)System.Runtime.InteropServices.Marshal.OffsetOf<WbDrawDispatcher.BatchDataPublic>(nameof(WbDrawDispatcher.BatchDataPublic.TextureLayer)));
        Assert.Equal(12, (int)System.Runtime.InteropServices.Marshal.OffsetOf<WbDrawDispatcher.BatchDataPublic>(nameof(WbDrawDispatcher.BatchDataPublic.Flags)));
    }

    [Fact]
    public void DrawCommandStride_MatchesStructSize()
    {
        Assert.Equal(WbDrawDispatcher.DrawCommandStride, System.Runtime.CompilerServices.Unsafe.SizeOf<DrawElementsIndirectCommand>());
    }
}