diff options
Diffstat (limited to 'shaders/compute.comp')
| -rw-r--r-- | shaders/compute.comp | 22 |
1 files changed, 18 insertions, 4 deletions
diff --git a/shaders/compute.comp b/shaders/compute.comp index 6bf4205..7763538 100644 --- a/shaders/compute.comp +++ b/shaders/compute.comp @@ -5,6 +5,9 @@ struct ray { struct material { vec4 albedo; + float fuzz; + uint type; + uint[2] padding; }; struct hit_info { @@ -36,7 +39,7 @@ layout(set = 0, binding = 0) buffer Inputs { vec4[1000] random_in_sphere; - sphere[2] spheres; + sphere[4] spheres; uint max_bounce; float [3] padding; @@ -97,7 +100,7 @@ bool hit_spheres(ray r, float min_t, float max_t, out hit_info info) { float closest_t = max_t; hit_info temp_info; bool hit = false; - for (uint sphere_index = 0; sphere_index < 2; sphere_index++) { + for (uint sphere_index = 0; sphere_index < 4; sphere_index++) { if (hit_sphere(spheres[sphere_index], r, min_t, closest_t, temp_info)) { info = temp_info; closest_t = temp_info.t; @@ -108,7 +111,7 @@ bool hit_spheres(ray r, float min_t, float max_t, out hit_info info) { return hit; } -bool lambertian_scatter(material mat, ray incident, vec2 seed, hit_info info, out vec4 attenuation, out ray scattered) { +bool lambertian_scattering(material mat, ray incident, vec2 seed, hit_info info, out vec4 attenuation, out ray scattered) { vec3 unit_hemisphere_vec = to_hemisphere_vec(random_in_sphere[uint(rand(seed) * 999.0)].xyz, info.normal); attenuation = mat.albedo; @@ -117,6 +120,16 @@ bool lambertian_scatter(material mat, ray incident, vec2 seed, hit_info info, ou return true; } +bool metal_scattering(material mat, ray incident, vec2 seed, hit_info info, out vec4 attenuation, out ray scattered) { + vec3 reflected = reflect(normalize(incident.direction), info.normal); + vec3 unit_sphere_vec = random_in_sphere[uint(rand(seed) * 999.0)].xyz; + + attenuation = mat.albedo; + scattered = ray(info.point, reflected + mat.fuzz * unit_sphere_vec); + + return (dot(scattered.direction, info.normal) > 0.0); +} + vec4 ray_color(ray r, uint sample_index) { hit_info info; uint bounces = 0; @@ -127,7 +140,8 @@ vec4 ray_color(ray r, uint sample_index) { vec2 seed = vec2(info.point.x * sample_index, info.point.y * bounces) * gl_GlobalInvocationID.xy; ray scattered; vec4 attenuation; - if (lambertian_scatter(info.mat, current_ray, seed, info, attenuation, scattered)) { + if (info.mat.type == 0 && lambertian_scattering(info.mat, current_ray, seed, info, attenuation, scattered) || + info.mat.type == 1 && metal_scattering(info.mat, current_ray, seed, info, attenuation, scattered)) { final_color *= attenuation; current_ray = scattered; } else { |