如何在 Rust 的线程之间共享不可变数据?

Cal*_*rey 3 multithreading raytracing rust

我正在使用 Rust 开发光线追踪器作为学习该语言的一种方式,并且单线程版本工作得很好。我想通过多线程来加速它,并且在 C/C++ 中对光线跟踪器进行多线程处理相对容易,因为大多数共享数据是只读的(唯一的问题发生在写入像素数据时)。但我在 Rust 及其所有权规则方面遇到了更多麻烦。

我对世界上可能受到打击的不同类型的事物(球体、网格体)有一个特征Hittable: Send + Sync,我将实现留空,Send因为Sync我实际上不需要它们中的任何一个。然后我有一个类型为世界对象的向量Vec<Box<dyn Hittable>>。对于实际的多线程,我正在尝试这样的事情:

    let pixels_mutex: Arc<Mutex<Vec<Vec<(f64, f64, f64, u32)>>>> = Arc::new(Mutex::new(pixels));
    let vec_arc: Arc<Vec<Box<dyn Hittable>>> = Arc::new(vec);

    let mut thread_vec: Vec<thread::JoinHandle<()>> = Vec::new();

    for _ in 0..NUM_THREADS {
        let camera_clone = camera.clone();
        thread_vec.push(thread::spawn(move || {
            for r in 0..RAYS_PER_THREAD {
                if r % THREAD_UPDATE == 0 {
                    println!("Thread drawing ray {} of {} ({:.2}%)", r, RAYS_PER_THREAD, (r as f64 * 100.) / (RAYS_PER_THREAD as f64));
                }
                let u: f64 = util::rand();
                let v: f64 = util::rand();
                let ray = camera_clone.get_ray(u, v);
                let res = geometry::thread_safe_cast_ray(&ray, &vec_arc, MAX_DEPTH);
                let i = (u * IMAGE_WIDTH as f64).floor() as usize;
                let j = (v * IMAGE_HEIGHT as f64).floor() as usize;
                util::thread_safe_increment_color(&pixels_mutex, j, i, &res);
            }
        }));
    }

    for handle in thread_vec {
        handle.join().unwrap();
    }
Run Code Online (Sandbox Code Playgroud)

我已经thread_safe_increment_color实现了,看起来不错,但我会推迟执行,thread_safe_cast_ray直到我让这个循环正常工作。我在这段代码中遇到的问题是每个线程都试图进入vec_arc其闭包,这违反了所有权规则。vec_arc我尝试像使用 那样进行克隆camera,但编译器不允许我这样做,我认为这是因为我的Hittable特征不需要CopyorClone特征。我的结构体Hittable不能简单地实现derive(Copy, Clone),因为它们包含一个Box<dyn Material>,其中Material是代表对象材料的另一个特征。

我认为这会更容易,因为我知道大多数数据(除 之外pixels_mutex)都是只读的。我如何在我创建的线程之间共享vec_arc(就此而言)?pixels_mutex

use*_*342 5

您应该vec_arc像使用相机一样进行克隆。由于它是一个Arc,因此克隆它很便宜,并且不需要Vec和它的元素是可克隆的。您只需要确保克隆智能指针而不是向量,并使用关联的函数:

let vec_arc = Arc::clone(&vec_arc);
Run Code Online (Sandbox Code Playgroud)

另一种选择是使用作用域线程Arc,在这种情况下您根本不需要使用,您可以简单地将闭包设置为非move并直接访问局部变量。您的代码将如下所示:

crossbeam::scope(|s| {
    for _ in 0..NUM_THREADS {
        let camera_clone = camera.clone();
        thread_vec.push(s.spawn(|_| {
            // your code here, using `vec` directly instead of `vec_arc`
            ...
        }).unwrap());
    }
}).unwrap();  // here threads are joined automatically
Run Code Online (Sandbox Code Playgroud)