fix: prioritize storage buffers and disable rayon sort camera debounce

mosure · Jan 2, 2024 · b8f2b30 · b8f2b30
1 parent 6f57b78
commit b8f2b30
Show file tree

Hide file tree

Showing 8 changed files with 11 additions and 68 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -40,16 +40,16 @@ default = [
   # "packed",
   "planar",
 
-  # "buffer_storage",
-  "buffer_texture",
+  "buffer_storage",
+  # "buffer_texture",
 
   # "f32",
   "f16",
 
   "query_select",
   "query_sparse",
 
-  # "sort_radix",
+  "sort_radix",
   "sort_rayon",
   "sort_std",
 

diff --git a/src/render/mod.rs b/src/render/mod.rs
@@ -41,43 +41,7 @@ use bevy::{
             SetItemPipeline,
             TrackedRenderPass,
         },
-        render_resource::{
-            BindGroup,
-            BindGroupEntry,
-            BindGroupLayout,
-            BindGroupLayoutDescriptor,
-            BindGroupLayoutEntry,
-            BindingResource,
-            BindingType,
-            BlendState,
-            Buffer,
-            BufferBinding,
-            BufferBindingType,
-            BufferInitDescriptor,
-            BufferUsages,
-            ColorTargetState,
-            ColorWrites,
-            CompareFunction,
-            DepthBiasState,
-            DepthStencilState,
-            FragmentState,
-            FrontFace,
-            MultisampleState,
-            PipelineCache,
-            PolygonMode,
-            PrimitiveState,
-            PrimitiveTopology,
-            RenderPipelineDescriptor,
-            ShaderDefVal,
-            ShaderStages,
-            ShaderType,
-            SpecializedRenderPipeline,
-            SpecializedRenderPipelines,
-            StencilFaceState,
-            StencilState,
-            TextureFormat,
-            VertexState,
-        },
+        render_resource::*,
         renderer::RenderDevice,
         Render,
         RenderApp,

diff --git a/src/render/packed.wgsl b/src/render/packed.wgsl
@@ -18,7 +18,7 @@ fn get_color(
     ray_direction: vec3<f32>,
 ) -> vec3<f32> {
     let sh = get_spherical_harmonics(index);
-    let color = spherical_harmonics_lookup(sh, ray_direction);
+    let color = spherical_harmonics_lookup(ray_direction, sh);
     return srgb_to_linear(color);
 }
 

diff --git a/src/render/planar.rs b/src/render/planar.rs
@@ -1,13 +1,6 @@
+#[allow(unused_imports)]
 use bevy::render::{
-    render_resource::{
-        Buffer,
-        BufferInitDescriptor,
-        BufferUsages,
-        Extent3d,
-        ShaderType,
-        TextureDimension,
-        TextureFormat,
-    },
+    render_resource::*,
     renderer::RenderDevice,
 };
 

diff --git a/src/render/planar.wgsl b/src/render/planar.wgsl
@@ -24,7 +24,7 @@ fn get_color(
     ray_direction: vec3<f32>,
 ) -> vec3<f32> {
     let sh = get_spherical_harmonics(index);
-    let color = spherical_harmonics_lookup(sh, ray_direction);
+    let color = spherical_harmonics_lookup(ray_direction, sh);
     return srgb_to_linear(color);
 }
 

diff --git a/src/render/texture.wgsl b/src/render/texture.wgsl
@@ -255,7 +255,7 @@ fn get_color(
     ray_direction: vec3<f32>,
 ) -> vec3<f32> {
     let sh = get_spherical_harmonics(index);
-    let color = spherical_harmonics_lookup(sh, ray_direction);
+    let color = spherical_harmonics_lookup(ray_direction, sh);
     return srgb_to_linear(color);
 }
 #endif

diff --git a/src/sort/mod.rs b/src/sort/mod.rs
@@ -7,15 +7,7 @@ use bevy::{
     },
     reflect::TypeUuid,
     render::{
-        render_resource::{
-            Buffer,
-            BufferInitDescriptor,
-            BufferUsages,
-            Extent3d,
-            ShaderType,
-            TextureDimension,
-            TextureFormat,
-        },
+        render_resource::*,
         render_asset::{
             RenderAsset,
             RenderAssetPlugin,

diff --git a/src/sort/rayon.rs b/src/sort/rayon.rs
@@ -41,7 +41,6 @@ pub fn rayon_sort(
     mut last_camera_position: Local<Vec3>,
     mut last_sort_time: Local<Option<Instant>>,
     mut period: Local<std::time::Duration>,
-    mut camera_debounce: Local<bool>,
     mut sort_done: Local<bool>,
 ) {
     if last_sort_time.is_none() {
@@ -68,18 +67,13 @@ pub fn rayon_sort(
 
         if camera_movement {
             *sort_done = false;
-            *camera_debounce = true;
         } else {
             if *sort_done {
                 return;
             }
         }
 
-        if *camera_debounce {
-            *last_camera_position = camera_position;
-            *camera_debounce = false;
-            return;
-        }
+        *last_camera_position = camera_position;
 
         for (
             gaussian_cloud_handle,