Rust光线追踪器并行化：如何实现可种子化RNG？

问题背景 #

单线程光线追踪原型使用带固定种子（123）的StdRng包装采样器，渲染时每个像素通过采样器生成多个样本射线。改用par_into_iter实现多线程渲染时，编译器报错：

cannot borrow sampler as mutable, as it is a captured variable in a Fn closure

问题根源 #

Rust的并行迭代要求闭包实现Fn（而非FnMut），且必须满足Send+Sync。单个可变采样器无法被多个线程同时可变借用——这既违反内存安全规则，也会导致随机数序列混乱，破坏渲染结果的可重复性。

解决方案：为每个像素创建独立的确定性采样器 #

核心思路是避免共享采样器实例，为每个像素生成一个独立的采样器，且通过像素坐标与全局种子计算唯一采样器种子，保证结果可重复。

步骤1：修改采样器创建函数 #

扩展采样器创建逻辑，支持传入自定义种子参数：

// 示例：新增带种子的采样器创建函数
fn create_sampler_with_seed(sampler_config: &SamplerValue, seed: u64) -> impl Sampler {
    match sampler_config {
        SamplerValue::StdRng => StdSampler::new(StdRng::seed_from_u64(seed)),
        // 后续新增的采样器类型按此模式扩展
    }
}

步骤2：重构并行渲染循环 #

将像素迭代改为并行模式，为每个像素生成独立采样器：

let (width, height) = (image.size_x, image.size_y);
// 提前获取全局采样次数，避免重复创建采样器
let sample_count = create_sampler(&self.sampler_value).sample_count();

println!("Rendering ...");
let progress_bar = get_progress_bar(image.size());

// 并行遍历所有像素（需确保image支持并行可变迭代，比如依赖rayon库）
image.enumerate_pixels_mut().par_iter_mut().for_each(|(x, y, pixel)| {
    // 基于全局种子+像素坐标生成唯一种子，保证渲染结果可重复
    let seed = 123 ^ ((x as u64) << 32) ^ (y as u64);
    let mut sampler = create_sampler_with_seed(&self.sampler_value, seed);
    
    *pixel = (0..sample_count)
        .into_iter()
        .map(|_| {
            let pixel_pos = Vec2::new(x as f32, y as f32) + sampler.next2f();
            let ray = self.camera.generate_ray(&pixel_pos);
            self.integrator.li(self, &mut sampler, &ray)
        })
        .sum::<Vec3>()
        / (sample_count as f32);
    
    progress_bar.inc(1);
});

方案优势 #

• 符合Rust并发安全规则：每个线程操作独立的采样器实例，无共享可变状态
• 保证结果可重复性：通过确定性种子计算，每次渲染的随机数序列完全一致
• 扩展性好：后续新增其他采样器类型时，只需在create_sampler_with_seed中添加对应分支即可

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