OpenGL ES计算着色器无法向rgba32f格式目标纹理写入数据

问题背景 #

作为OpenGL ES新手，在Android设备上编写计算着色器时，使用rgba8格式的图像可以正常填充为红色，但修改为rgba32f格式后，图像显示空白，需要实现rgba32f格式纹理的全屏红色填充。

正常运行的代码（rgba8格式） #

#version 310 es
precision highp image2D;

layout(local_size_x = 1, local_size_y = 1, local_size_z = 1) in;

layout(binding = 1, rgba8) writeonly uniform image2D dstImage;

void main()
{
    ivec2 pixelCoords = ivec2(gl_GlobalInvocationID.xy);
    vec4 pixelDst = vec4(1.0, 0.0, 0.0, 1.0);
    imageStore(dstImage, pixelCoords, pixelDst);
}

上述代码可将dstImage完全填充为红色。

出现问题的代码（rgba32f格式） #

#version 310 es
precision highp image2D;

layout(local_size_x = 1, local_size_y = 1, local_size_z = 1) in;

layout(binding = 1, rgba32f) writeonly uniform image2D dstImage;

void main()
{
    ivec2 pixelCoords = ivec2(gl_GlobalInvocationID.xy);
    vec4 pixelDst = vec4(1.0, 0.0, 0.0, 1.0);
    imageStore(dstImage, pixelCoords, pixelDst);
}

修改格式为rgba32f后，dstImage显示空白。

解决方案 #

1. 确保纹理创建参数正确 #

在Android端创建rgba32f纹理时，必须明确指定内部格式、数据格式和类型，否则会导致纹理数据存储错误：

// 生成纹理ID
int[] textureId = new int[1];
GLES31.glGenTextures(1, textureId, 0);
GLES31.glBindTexture(GLES31.GL_TEXTURE_2D, textureId[0]);

// 设置纹理采样参数
GLES31.glTexParameteri(GLES31.GL_TEXTURE_2D, GLES31.GL_TEXTURE_WRAP_S, GLES31.GL_CLAMP_TO_EDGE);
GLES31.glTexParameteri(GLES31.GL_TEXTURE_2D, GLES31.GL_TEXTURE_WRAP_T, GLES31.GL_CLAMP_TO_EDGE);
GLES31.glTexParameteri(GLES31.GL_TEXTURE_2D, GLES31.GL_TEXTURE_MIN_FILTER, GLES31.GL_NEAREST);
GLES31.glTexParameteri(GLES31.GL_TEXTURE_2D, GLES31.GL_TEXTURE_MAG_FILTER, GLES31.GL_NEAREST);

// 分配rgba32f格式的纹理内存
GLES31.glTexImage2D(
    GLES31.GL_TEXTURE_2D, 
    0, 
    GLES31.GL_RGBA32F,  // 内部格式必须为GL_RGBA32F
    width, 
    height, 
    0, 
    GLES31.GL_RGBA,     // 数据格式
    GLES31.GL_FLOAT,    // 数据类型必须为GL_FLOAT
    null
);

2. 正确绑定图像到计算着色器 #

绑定纹理到图像单元时，格式必须与着色器中声明的rgba32f一致：

// 绑定纹理到图像单元1，与着色器中的binding = 1对应
GLES31.glBindImageTexture(
    1, 
    textureId[0], 
    0, 
    false, 
    0, 
    GLES31.GL_WRITE_ONLY, 
    GLES31.GL_RGBA32F  // 必须匹配着色器中的格式
);

3. 优化计算着色器（可选但推荐） #

原工作组大小1x1效率极低，建议改为16x16或其他合适的大小，同时增加边界检查避免越界：

#version 310 es
precision highp float;
precision highp image2D;

layout(local_size_x = 16, local_size_y = 16, local_size_z = 1) in;

layout(binding = 1, rgba32f) writeonly uniform image2D dstImage;

void main()
{
    ivec2 pixelCoords = ivec2(gl_GlobalInvocationID.xy);
    ivec2 imageSize = imageSize(dstImage);
    
    // 确保坐标不超出纹理范围
    if(pixelCoords.x >= imageSize.x || pixelCoords.y >= imageSize.y)
        return;
        
    vec4 pixelDst = vec4(1.0, 0.0, 0.0, 1.0);
    imageStore(dstImage, pixelCoords, pixelDst);
}

4. 渲染浮点纹理到屏幕的片段着色器处理 #

将rgba32f纹理渲染到屏幕时，片段着色器直接采样输出即可，OpenGL会自动将浮点范围0.0-1.0映射到8位帧缓冲的0-255：

#version 310 es
precision highp float;

uniform sampler2D uTexture;
in vec2 vTexCoord;
out vec4 fragColor;

void main() {
    fragColor = texture(uTexture, vTexCoord);
}

5. 检查设备兼容性 #

部分旧设备可能不支持浮点纹理，可通过以下代码检查：

// 检查是否支持浮点纹理扩展
boolean supportsFloatTexture = GLES31.glGetString(GLES31.GL_EXTENSIONS).contains("GL_OES_texture_float");
// 检查是否支持浮点颜色缓冲（若需渲染到浮点帧缓冲）
boolean supportsFloatColorBuffer = GLES31.glGetString(GLES31.GL_EXTENSIONS).contains("GL_EXT_color_buffer_float");

内容的提问来源于stack exchange，提问作者Zappy.Mans