如何在GLSL计算着色器中生成自定义取值范围的伪随机数

种子来源处理 #

计算着色器无需依赖顶点着色器的st变量，直接使用内置变量gl_GlobalInvocationID作为随机数种子即可，该变量代表当前工作项的全局调用坐标，每个工作项取值唯一，完全可以满足伪随机数的种子输入要求。

自定义范围实现逻辑 #

你现有的两个rand函数返回值都是[0, 1)区间的浮点数，要映射到自定义区间[minVal, maxVal)只需要做线性变换即可：
随机值 = 基础随机值 * (maxVal - minVal) + minVal
如果需要生成整数区间的随机值，对变换后的结果取整即可。

完整实现代码 #

// 优化版基础随机数函数，返回[0,1)区间浮点数
highp float rand(vec2 seed)
{
    highp float a = 12.9898;
    highp float b = 78.233;
    highp float c = 43758.5453;
    highp float dt = dot(seed, vec2(a,b));
    highp float sn = mod(dt, 3.14159265359); // 补充π精度避免周期性问题
    return fract(sin(sn) * c);
}

// 生成[minVal, maxVal)区间的浮点数随机数
highp float randRange(vec2 seed, float minVal, float maxVal)
{
    return rand(seed) * (maxVal - minVal) + minVal;
}

// 生成[minVal, maxVal]区间的整数随机数
highp int randIntRange(vec2 seed, int minVal, int maxVal)
{
    float range = float(maxVal - minVal + 1);
    return int(floor(rand(seed) * range)) + minVal;
}

// 计算着色器使用示例
layout (local_size_x = 16, local_size_y = 16) in;
void main() {
    // 用当前工作项的全局坐标作为种子
    vec2 seed = vec2(gl_GlobalInvocationID.xy);
    // 示例：生成1~10的浮点数随机数
    float randomFloat1to10 = randRange(seed, 1.0, 10.0);
    // 示例：生成1~10的整数随机数
    int randomInt1to10 = randIntRange(seed, 1, 10);
    // 如果需要同一个坐标生成多个不同的随机数，给种子加固定偏移即可
    float anotherRandom = randRange(seed + vec2(987.654, 321.098), 0.0, 100.0);
    // 后续业务逻辑
}

补充说明 #

• 如果你的计算着色器工作项是3维的，也可以把gl_GlobalInvocationID.xyz转成vec2作为种子，比如用vec2(gl_GlobalInvocationID.x + gl_GlobalInvocationID.z * 65535.0, gl_GlobalInvocationID.y)避免重复。
• 对随机数质量要求更高的场景，可以替换为性能更好的哈希算法，上述方案已经可以满足绝大多数普通场景的需求。

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