//DISABLE_TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj

/* A test around use of an array like container. 

Here the attempt is to make the equality test separate from the type. 
This isn't a great answer because it relies on the right types set
for T and E to work together.

This doesn't work because...
.slang(25): error 30019: expected an expression of type 'Type', got 'T'
            if (E::isEqual(v, elements[i]))

 */

//TEST_INPUT:ubuffer(data=[0 0 0 0], stride=4):out,name outputBuffer
RWStructuredBuffer<float> outputBuffer;

interface IEquality
{
    associatedtype Type;
    static bool isEqual(Type a, Type b);
};

struct IntEquality : IEquality
{
    typedef int Type;
    static bool isEqual(Type a, Type b) { return a == b; }
};

struct FixedArray<T, E : IEquality> 
{
    static const int SIZE = 4;
    
    [mutating] void setAt(int i, T value) { elements[i] = value; }
    T getAt(int i) { return elements[i]; }
    int indexOf(T v)
    {
        for (int i = 0; i < SIZE; ++i)
        {
            if (E::isEqual(v, elements[i]))
            {
                return i;
            }
            return -1;
        }
    }
    T elements[SIZE];
};


[numthreads(4, 1, 1)]
void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
{
    int index = dispatchThreadID.x;

    FixedArray<int, IntEquality> arr;
    arr.setAt(0, index);
    
	outputBuffer[index] = 1 + arr.getAt(0);
}