Solari radiance cache

crates/bevy_pbr/src/render/utils.wgsl

@@ -115,6 +115,10 @@ fn sample_uniform_hemisphere(normal: vec3<f32>, rng: ptr<function, u32>) -> vec3
     return orthonormalize(normal) * vec3(x, y, z);
 }
 
+fn uniform_hemisphere_inverse_pdf() -> f32 {
+    return PI_2;
+}
+
 // https://www.realtimerendering.com/raytracinggems/unofficial_RayTracingGems_v1.9.pdf#0004286901.INDD%3ASec19%3A294
 fn sample_disk(disk_radius: f32, rng: ptr<function, u32>) -> vec2<f32> {
     let ab = 2.0 * rand_vec2f(rng) - 1.0;
@@ -141,7 +145,7 @@ fn sample_disk(disk_radius: f32, rng: ptr<function, u32>) -> vec2<f32> {
 fn sample_cube_dir(uv: vec2f, face: u32) -> vec3f {
     // Convert from [0,1] to [-1,1]
     let uvc = 2.0 * uv - 1.0;
-    
+
     // Generate direction based on the cube face
     var dir: vec3f;
     switch(face) {
@@ -165,7 +169,7 @@ fn dir_to_cube_uv(dir: vec3f) -> CubeUV {
     let abs_dir = abs(dir);
     var face: u32 = 0u;
     var uv: vec2f = vec2f(0.0);
-    
+
     // Find the dominant axis to determine face
     if (abs_dir.x >= abs_dir.y && abs_dir.x >= abs_dir.z) {
         // X axis is dominant
@@ -195,7 +199,7 @@ fn dir_to_cube_uv(dir: vec3f) -> CubeUV {
             uv = vec2f(-dir.x, -dir.y) / abs_dir.z;
         }
     }
-    
+
     // Convert from [-1,1] to [0,1]
     return CubeUV(uv * 0.5 + 0.5, face);
 }

crates/bevy_solari/src/realtime/mod.rs

@@ -13,6 +13,7 @@ use bevy_ecs::{component::Component, reflect::ReflectComponent, schedule::IntoSc
 use bevy_pbr::DefaultOpaqueRendererMethod;
 use bevy_reflect::{std_traits::ReflectDefault, Reflect};
 use bevy_render::{
+    load_shader_library,
     render_graph::{RenderGraphExt, ViewNodeRunner},
     renderer::RenderDevice,
     view::Hdr,
@@ -31,9 +32,12 @@ pub struct SolariLightingPlugin;
 
 impl Plugin for SolariLightingPlugin {
     fn build(&self, app: &mut App) {
-        embedded_asset!(app, "presample_light_tiles.wgsl");
+        load_shader_library!(app, "presample_light_tiles.wgsl");
         embedded_asset!(app, "restir_di.wgsl");
         embedded_asset!(app, "restir_gi.wgsl");
+        load_shader_library!(app, "world_cache_query.wgsl");
+        embedded_asset!(app, "world_cache_compact.wgsl");
+        embedded_asset!(app, "world_cache_update.wgsl");
 
         app.insert_resource(DefaultOpaqueRendererMethod::deferred());
     }