import FluidGridData;

struct Params
{
    FluidGridData<float> velocityX;
    FluidGridData<float> velocityY;
    FluidGridData<float> velocityZ;
    float3 force;
    float dt;
};
ParameterBlock<Params> params;

[shader("compute")]
[numthreads(32, 8, 1)]
void applyForces(uint3 dispatchThreadID : SV_DispatchThreadID)
{
    FluidGridData<float> velocityX = params.velocityX;
    FluidGridData<float> velocityY = params.velocityY;
    FluidGridData<float> velocityZ = params.velocityZ;
    float3 force = params.force;
    float dt = params.dt;

    int x = dispatchThreadID.x + 1;
    int y = dispatchThreadID.y + 1;
    int z = dispatchThreadID.z + 1;
    if(x >= gridParams.gridSize.x - 1 || y >= gridParams.gridSize.y - 1 || z >= gridParams.gridSize.z - 1) return;

    velocityX[x, y, z] += force.x * dt;
    velocityY[x, y, z] += force.y * dt;
    velocityZ[x, y, z] += force.z * dt;
}