feat: gaussian camera filter

Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "bevy_gaussian_splatting"
 description = "bevy gaussian splatting render pipeline plugin"
-version = "2.3.0"
+version = "2.4.0"
 edition = "2021"
 authors = ["mosure <[email protected]>"]
 license = "MIT"
@@ -134,9 +134,9 @@ webgpu = ["bevy/webgpu"]
 
 [dependencies]
 bevy_args = "1.6"
-bevy-inspector-egui = { version = "0.25", optional = true }
+bevy-inspector-egui = { version = "0.26", optional = true }
 bevy_mod_picking = { version = "0.20", optional = true }
-bevy_panorbit_camera = { version = "0.19", optional = true }
+bevy_panorbit_camera = { version = "0.19", optional = true, features = ["bevy_egui"] }
 bevy_transform_gizmo = { version = "0.12", optional = true }
 bincode2 = { version = "2.0", optional = true }
 byte-unit = { version = "5.0", optional = true }

examples/headless.rs

@@ -12,6 +12,7 @@ use bevy::{
 use bevy_args::BevyArgsPlugin;
 
 use bevy_gaussian_splatting::{
+    GaussianCamera,
     GaussianCloud,
     GaussianSplattingBundle,
     GaussianSplattingPlugin,
@@ -407,6 +408,7 @@ fn setup_gaussian_cloud(
             },
             ..default()
         },
+        GaussianCamera,
     ));
 }
 

src/lib.rs

@@ -11,6 +11,7 @@ pub use material::spherical_harmonics::SphericalHarmonicCoefficients;
 
 use io::loader::GaussianCloudLoader;
 
+pub use render::GaussianCamera;
 use render::RenderPipelinePlugin;
 
 pub mod gaussian;

src/morph/particle.rs

@@ -79,6 +79,7 @@ use serde::{
 };
 
 use crate::render::{
+    GaussianCamera,
     GaussianCloudBindGroup,
     GaussianCloudPipeline,
     GaussianCloudPipelineKey,
@@ -313,6 +314,7 @@ pub struct ParticleBehaviorNode {
     )>,
     initialized: bool,
     view_bind_group: QueryState<(
+        &'static GaussianCamera,
         &'static GaussianViewBindGroup,
         &'static ViewUniformOffset,
     )>,
@@ -367,6 +369,7 @@ impl Node for ParticleBehaviorNode {
         let command_encoder = render_context.command_encoder();
 
         for (
+            _gaussian_camera,
             view_bind_group,
             view_uniform_offset,
         ) in self.view_bind_group.iter_manual(world) {

src/render/mod.rs

@@ -1,6 +1,7 @@
 use std::hash::Hash;
 
 use bevy::{
+    prelude::*,
     asset::{
         load_internal_asset,
         LoadState,
@@ -13,12 +14,13 @@ use bevy::{
             SystemParamItem,
         }
     },
-    prelude::*,
     render::{
         Extract,
         extract_component::{
             ComponentUniforms,
             DynamicUniformIndex,
+            ExtractComponent,
+            ExtractComponentPlugin,
             UniformComponentPlugin,
         },
         globals::{
@@ -144,6 +146,7 @@ impl Plugin for RenderPipelinePlugin {
             Shader::from_wgsl
         );
 
+        app.add_plugins(ExtractComponentPlugin::<GaussianCamera>::default());
         app.add_plugins(RenderAssetPlugin::<GpuGaussianCloud>::default());
         app.add_plugins(UniformComponentPlugin::<GaussianCloudUniform>::default());
 
@@ -273,7 +276,13 @@ fn queue_gaussians(
     gaussian_clouds: Res<RenderAssets<GpuGaussianCloud>>,
     sorted_entries: Res<RenderAssets<GpuSortedEntry>>,
     mut transparent_render_phases: ResMut<ViewSortedRenderPhases<Transparent3d>>,
-    mut views: Query<(Entity, &ExtractedView)>,
+    mut views: Query<
+        (
+            Entity,
+            &ExtractedView
+        ),
+        With<GaussianCamera>,
+    >,
     msaa: Res<Msaa>,
     gaussian_splatting_bundles: Query<GpuGaussianBundleQuery>,
 ) {
@@ -845,6 +854,16 @@ fn queue_gaussian_bind_group(
     }
 }
 
+#[derive(
+    Clone,
+    Component,
+    Debug,
+    Default,
+    ExtractComponent,
+    Reflect,
+)]
+pub struct GaussianCamera;
+
 #[derive(Component)]
 pub struct GaussianViewBindGroup {
     pub value: BindGroup,
@@ -855,10 +874,13 @@ pub fn queue_gaussian_view_bind_groups(
     render_device: Res<RenderDevice>,
     gaussian_cloud_pipeline: Res<GaussianCloudPipeline>,
     view_uniforms: Res<ViewUniforms>,
-    views: Query<(
-        Entity,
-        &ExtractedView,
-    )>,
+    views: Query<
+        (
+            Entity,
+            &ExtractedView,
+        ),
+        With<GaussianCamera>,
+    >,
     globals_buffer: Res<GlobalsBuffer>,
 ) {
     if let (
@@ -902,15 +924,20 @@ pub struct SetGaussianViewBindGroup<const I: usize>;
 impl<P: PhaseItem, const I: usize> RenderCommand<P> for SetGaussianViewBindGroup<I> {
     type Param = ();
     type ViewQuery = (
-        Read<ViewUniformOffset>,
+        Read<GaussianCamera>,
         Read<GaussianViewBindGroup>,
+        Read<ViewUniformOffset>,
     );
     type ItemQuery = ();
 
     #[inline]
     fn render<'w>(
         _item: &P,
-        (view_uniform, gaussian_view_bind_group): ROQueryItem<
+        (
+            _gaussian_camera,
+            gaussian_view_bind_group,
+            view_uniform,
+        ): ROQueryItem<
             'w,
             Self::ViewQuery,
         >,

src/sort/radix.rs

@@ -500,6 +500,7 @@ pub struct RadixSortNode {
     )>,
     initialized: bool,
     view_bind_group: QueryState<(
+        &'static GaussianCamera,
         &'static GaussianViewBindGroup,
         &'static ViewUniformOffset,
     )>,

tests/gpu/gaussian.rs

@@ -13,6 +13,7 @@ use bevy::{
 };
 
 use bevy_gaussian_splatting::{
+    GaussianCamera,
     GaussianCloud,
     GaussianSplattingBundle,
     random_gaussians,
@@ -59,6 +60,7 @@ fn setup(
             tonemapping: Tonemapping::None,
             ..default()
         },
+        GaussianCamera,
     ));
 }
 

tests/gpu/radix.rs

@@ -36,6 +36,7 @@ use bevy::{
 };
 
 use bevy_gaussian_splatting::{
+    GaussianCamera,
     GaussianCloud,
     GaussianSplattingBundle,
     random_gaussians,
@@ -105,6 +106,7 @@ fn setup(
             tonemapping: Tonemapping::None,
             ..default()
         },
+        GaussianCamera,
     ));
 }
 

tools/compare_aabb_obb.rs

@@ -15,6 +15,7 @@ use bevy_panorbit_camera::{
 
 use bevy_gaussian_splatting::{
     Gaussian,
+    GaussianCamera,
     GaussianCloud,
     GaussianCloudSettings,
     GaussianSplattingBundle,
@@ -97,6 +98,7 @@ pub fn setup_aabb_obb_compare(
             allow_upside_down: true,
             ..default()
         },
+        GaussianCamera,
     ));
 }
 

viewer/viewer.rs

@@ -25,6 +25,7 @@ use bevy_panorbit_camera::{
 };
 
 use bevy_gaussian_splatting::{
+    GaussianCamera,
     GaussianCloud,
     GaussianSplattingBundle,
     GaussianSplattingPlugin,
@@ -93,6 +94,7 @@ fn setup_gaussian_cloud(
             zoom_smoothness: 0.0,
             ..default()
         },
+        GaussianCamera,
     ));
 }