Handle naga's SpecialTypes

[JMS55]

Ray tracing APIs do not currently work with naga_oil. DerivedModule throws out special_type info.

[robtfm]

a minimal example would be really useful - just the wgsl.

[JMS55]

Try calling this function from a different module.

@group(0) @binding(0) var tlas: acceleration_structure;

fn trace_ray(ray_origin: vec3<f32>, ray_direction: vec3<f32>, ray_t_min: f32, ray_t_max: f32) -> RayIntersection {
    let ray_flags = RAY_FLAG_NONE;
    let ray_cull_mask = 0xFFu;
    let ray = RayDesc(ray_flags, ray_cull_mask, ray_t_min, ray_t_max, ray_origin, ray_direction);

    var rq: ray_query;
    rayQueryInitialize(&rq, tlas, ray);
    rayQueryProceed(&rq);
    return rayQueryGetCommittedIntersection(&rq);
}

[MiniaczQ]

If I understood what's going on correctly when experimenting, naga can interpret acceleration_structure type, but it cannot write it again in WGSL.

[JMS55]

@robtfm I modified naga to support writing out acceleration_structure. However we still need to populate the RT SpecialTypes. I worked around it by hardcoding them: JMS55@9bcdd3b

[JMS55]

The fix for being unable to write the acceleration structure binding is gfx-rs/wgpu#5261, but using RayIntersection across modules and stuff is still and issue and needs to be fixed by populating the special types.