blob: ccec76f79e31d31157569bda05214024bfb67497 [file] [log] [blame]
Neslisah Torosdagli50a72282020-03-20 18:23:27 -04001#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
Neslisah Torosdagli915f1192020-03-26 18:51:18 -040013layout(binding = 0, set = 0) uniform accelerationStructureEXT rtas;
Neslisah Torosdagli50a72282020-03-20 18:23:27 -040014layout(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{
Neslisah Torosdagli7f689be2020-03-23 23:09:44 -040023 rayQueryInitializeEXT(rayQuery, rtas, gl_RayFlagsNoneEXT, gl_RayFlagsCullBackFacingTrianglesEXT, ray.pos, ray.tmin, ray.dir, ray.tmax);
Neslisah Torosdagli50a72282020-03-20 18:23:27 -040024}
25
26void main()
27{
28 uint index = launchIndex();
29 Ray ray = rays[index];
30 rayQueryEXT rayQuery1;
31 rayQueryEXT rayQuery2;
32 doInitialize(rayQuery1, ray);
Neslisah Torosdagli7f689be2020-03-23 23:09:44 -040033 rayQueryInitializeEXT(rayQuery1, rtas, gl_RayFlagsOpaqueEXT, gl_RayFlagsCullFrontFacingTrianglesEXT, ray.pos, ray.tmin, ray.dir, ray.tmax);
Neslisah Torosdagli50a72282020-03-20 18:23:27 -040034 doInitialize(rayQuery2, ray);
35}