Function __raygen__render

examples/cuda/gpu/path_tracer_gpu/src/optix.rs:102–155

()

#[cfg(feature = "optix")]
#[kernel]
pub unsafe fn __raygen__render() {
    let i = get_launch_index().xy();
    let size = PARAMS.size;

    let idx = i.x + i.y * size.x;

    let rng = PARAMS.rand_states.add(idx as usize);
    let offset = (*rng).normal_f32_2();
    let mut cur_ray = generate_ray(Vec2::from(i.to_array()), &PARAMS.viewport, offset.into());

    let mut attenuation = Vec3::one();
    let mut color = Vec3::zero();

    for _ in 0..MAX_BOUNCES {
        let mut prd = PerRayData {
            hit: false,
            attenuation,
            rand: rng,
            scattered: None,
        };

        let (mut p0, mut p1) = pack_pointer(core::ptr::addr_of_mut!(prd));

        raygen::trace(
            PARAMS.handle,
            (cur_ray.origin.into_array()).into(),
            (cur_ray.dir.into_array()).into(),
            0.001,
            1e20,
            0.0,
            255,
            RayFlags::empty(),
            0,
            1,
            0,
            [&mut p0, &mut p1],
        );

        if !prd.hit {
            let miss_color = render::color(cur_ray);
            color = attenuation * miss_color;
            break;
        }

        if let Some(scatter) = prd.scattered {
            cur_ray = scatter;
            attenuation *= prd.attenuation;
        } else {
            break;
        }
    }

    *PARAMS.image_buf.add(idx as usize) += color;
}