1#version 460
2#extension GL_NV_ray_tracing : enable
3#extension GL_EXT_ray_query : enable
4
5struct Ray
6{
7    vec3 pos;
8    float tmin;
9    vec3 dir;
10    float tmax;
11};
12
13layout(binding = 0, set = 0) uniform accelerationStructureEXT rtas;
14layout(std430, set = 0, binding = 2) buffer Rays { Ray rays[]; };
15
16uint launchIndex()
17{
18    return gl_LaunchIDNV.z*gl_LaunchSizeNV.x*gl_LaunchSizeNV.y + gl_LaunchIDNV.y*gl_LaunchSizeNV.x + gl_LaunchIDNV.x;
19}
20
21void doInitialize(rayQueryEXT rayQuery, Ray ray)
22{
23    rayQueryInitializeEXT(rayQuery, rtas, gl_RayFlagsNoneEXT, gl_RayFlagsCullBackFacingTrianglesEXT, ray.pos, ray.tmin, ray.dir, ray.tmax);
24}
25
26void main()
27{
28    uint index = launchIndex();
29    Ray ray = rays[index];
30    rayQueryEXT rayQuery1;
31    rayQueryEXT rayQuery2;
32    doInitialize(rayQuery1, ray);
33    rayQueryInitializeEXT(rayQuery1, rtas, gl_RayFlagsOpaqueEXT, gl_RayFlagsCullFrontFacingTrianglesEXT, ray.pos, ray.tmin, ray.dir, ray.tmax);
34    doInitialize(rayQuery2, ray);
35}
36