哪些方法可用于使用 GLSL 处理二维数组?

What approaches can be used to handle 2d arrays with GLSL?

我的具体情况是计算着色器,用于在 n by m matrixn legnth vector 之间进行矩阵乘法(其中 nm 在编译时未知).

要么使用二维图像(参见 glsl - 8.12. Image Functions):

layout(r32f) uniform image2D matrixImage;

void main()
{
    // [...]

    ivec2 size = ivec2(imageSize(matrixImage));
    int n = size.x;
    int m = size.y;
    for (int j = 0; j < m; j ++)
    {
        for (int i = 0; i < n; i ++)
        {
            float val = imageLoad(matrixImage, ivec2(i, j)).x;
 
            // [...]
        }
    }

    // [...]
}

或者将数据写入着色器存储缓冲区对象中的一维开放大小数组:

layout(std430) buffer TMatrix
{
  int n;
  int m;
  float data[];
} matrix;

layout(r32f) uniform image2D matrixImage;

void main()
{
    // [...]

    for (int j = 0; j < matrix.m; j ++)
    {
        for (int i = 0; i < matrix.n; i ++)
        {
            int index = j * matrix.n + i;  
            float val = matrix.data[index];
            
            // [...]
        }
    }

    // [...]
}

如果你在编译时知道大小,就没有必要使用开放大小的数组(see GLSL - 4.1.9. Arrays):

const int n = 10;
const int m = 10;

layout(std430) buffer TMatrix
{
  float data[n][m];
} matrix;

void main()
{
    // [...]

    for (int j = 0; j < matrix.m; j ++)
    {
        for (int i = 0; i < matrix.n; i ++)
        {
            float val = matrix.data[i][j];

            // [...]
        }
    }

    // [...]
}