使用ModernGL执行GLSL计算着色器无预期效果，求排查

你的计算着色器输出全黑的核心原因是纹理格式不匹配，以及缺少必要的纹理使用权限设置。以下是具体问题和修复步骤：

关键问题点 #

1. 纹理格式不匹配：着色器声明的是rgba8ui（无符号整数格式），但ModernGL默认创建的是RGBA8（归一化浮点格式）。这种类型不匹配会导致着色器读取/写入数据时出现错误，最终输出黑色。
2. 缺少存储权限：用于图像加载/存储的纹理需要启用TEXTURE_USAGE_STORAGE_BIT权限，否则OpenGL不允许计算着色器访问。
3. 不必要的读写权限：源纹理只需读权限，目标纹理只需写权限，过度授权虽不直接导致错误，但不符合最佳实践。

修复后的完整代码 #

import moderngl
import numpy as np
import cv2

compute_shader_source = """
#version 450

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

layout (binding = 0, rgba8ui) uniform uimage2D src_texture;
layout (binding = 1, rgba8ui) uniform uimage2D dest_texture;

// 旋转通道顺序
void main() {
    ivec2 texel_pos = ivec2(gl_GlobalInvocationID.xy);
    uvec4 color = imageLoad(src_texture, texel_pos);
    uvec4 flipped_color = color.gbra; // G, B, R, A
    imageStore(dest_texture, texel_pos, flipped_color);
}
"""

dims = (256, 128)

// 确保上下文支持OpenGL 4.5及以上版本
ctx = moderngl.create_standalone_context(require=450)
compute_shader = ctx.compute_shader(compute_shader_source)
compute_shader['src_texture'] = 0
compute_shader['dest_texture'] = 1

// 输入图像（RGBA格式：R=0, G=0, B=100, A=255 → 蓝色）
in_color = np.array([0, 0, 100, 255] * dims[0] * dims[1], dtype=np.uint8)
in_texture = ctx.texture(
    dims,
    4,
    data=in_color,
    internalformat='rgba8ui',
    usage=moderngl.TEXTURE_USAGE_STORAGE_BIT | moderngl.TEXTURE_USAGE_SAMPLED_BIT
)

// 输出图像（初始化为RGBA格式：R=100, G=0, B=0, A=255 → 红色）
out_color = np.array([100, 0, 0, 255] * dims[0] * dims[1], dtype=np.uint8)
out_texture = ctx.texture(
    dims,
    4,
    data=out_color,
    internalformat='rgba8ui',
    usage=moderngl.TEXTURE_USAGE_STORAGE_BIT | moderngl.TEXTURE_USAGE_SAMPLED_BIT
)

// 绑定纹理并设置正确的读写权限
in_texture.bind_to_image(0, read=True, write=False)
out_texture.bind_to_image(1, read=False, write=True)

// 显示输入图像（cv2使用BGR格式，会显示为蓝色）
a = np.frombuffer(in_texture.read(), dtype=np.uint8).reshape(in_texture.height, in_texture.width, -1)
cv2.imshow('input', a)

// 显示着色器执行前的目标图像（会显示为红色）
a = np.frombuffer(out_texture.read(), dtype=np.uint8).reshape(out_texture.height, out_texture.width, -1)
cv2.imshow('destination before', a)

// 执行计算着色器（16x8个工作组覆盖256x128像素）
compute_shader.run(dims[0] // 16, dims[1] // 16, 1)

// 显示着色器执行后的目标图像（会显示为绿色）
a = np.frombuffer(out_texture.read(), dtype=np.uint8).reshape(out_texture.height, out_texture.width, -1)
cv2.imshow('destination after', a)
cv2.waitKey()
cv2.destroyAllWindows()

修复细节说明 #

1. 上下文版本强制：添加require=450确保OpenGL上下文支持计算着色器和图像加载/存储特性。
2. 纹理格式匹配：使用internalformat='rgba8ui'创建纹理，与着色器中的rgba8ui声明完全一致，保证数据类型正确。
3. 存储权限启用：通过usage=moderngl.TEXTURE_USAGE_STORAGE_BIT允许纹理被计算着色器作为图像访问。
4. 权限优化：为源纹理设置只读权限，目标纹理设置只写权限，遵循最小权限原则。

内容的提问来源于stack exchange，提问作者fst