feat: rasterize position

Author: mosure

src/gaussian/settings.rs

@@ -88,8 +88,9 @@ pub enum RasterizeMode {
     #[default]
     Color,
     Depth,
-    OpticalFlow,
     Normal,
+    OpticalFlow,
+    Position,
     Velocity,
 }
 

src/material/mod.rs

@@ -3,6 +3,7 @@ use bevy::prelude::*;
 pub mod classification;
 pub mod depth;
 pub mod optical_flow;
+pub mod position;
 pub mod spherical_harmonics;
 pub mod spherindrical_harmonics;
 
@@ -23,6 +24,7 @@ impl Plugin for MaterialPlugin {
             classification::ClassificationMaterialPlugin,
             depth::DepthMaterialPlugin,
             optical_flow::OpticalFlowMaterialPlugin,
+            position::PositionMaterialPlugin,
             spherical_harmonics::SphericalHarmonicCoefficientsPlugin,
             spherindrical_harmonics::SpherindricalHarmonicCoefficientsPlugin,
         ));

src/material/position.rs

@@ -0,0 +1,21 @@
+use bevy::{
+    prelude::*,
+    asset::load_internal_asset,
+};
+
+
+const POSITION_SHADER_HANDLE: Handle<Shader> = Handle::weak_from_u128(62346645534);
+
+
+pub struct PositionMaterialPlugin;
+
+impl Plugin for PositionMaterialPlugin {
+    fn build(&self, app: &mut App) {
+        load_internal_asset!(
+            app,
+            POSITION_SHADER_HANDLE,
+            "position.wgsl",
+            Shader::from_wgsl
+        );
+    }
+}

src/material/position.wgsl

@@ -19,6 +19,8 @@ struct GaussianUniforms {
     time_start: f32,
     time_stop: f32,
     num_classes: u32,
+    min: vec4<f32>,
+    max: vec4<f32>,
 };
 @group(1) @binding(0) var<uniform> gaussian_uniforms: GaussianUniforms;
 

src/render/bindings.wgsl

@@ -359,6 +359,8 @@ fn vs_points(
     );
 
     rgb = optical_flow_to_rgb(motion_vector);
+#else ifdef RASTERIZE_POSITION
+    rgb = (transformed_position - gaussian_uniforms.min.xyz) / (gaussian_uniforms.max.xyz - gaussian_uniforms.min.xyz);
 #else ifdef RASTERIZE_VELOCITY
     let time_delta = 1e-3;
     let future_gaussian_4d = conditional_cov3d(

src/render/gaussian.wgsl

@@ -687,8 +687,9 @@ pub fn shader_defs(
         RasterizeMode::Classification => shader_defs.push("RASTERIZE_CLASSIFICATION".into()),
         RasterizeMode::Color => shader_defs.push("RASTERIZE_COLOR".into()),
         RasterizeMode::Depth => shader_defs.push("RASTERIZE_DEPTH".into()),
-        RasterizeMode::OpticalFlow => shader_defs.push("RASTERIZE_OPTICAL_FLOW".into()),
         RasterizeMode::Normal => shader_defs.push("RASTERIZE_NORMAL".into()),
+        RasterizeMode::OpticalFlow => shader_defs.push("RASTERIZE_OPTICAL_FLOW".into()),
+        RasterizeMode::Position => shader_defs.push("RASTERIZE_POSITION".into()),
         RasterizeMode::Velocity => shader_defs.push("RASTERIZE_VELOCITY".into()),
     }
 
@@ -807,6 +808,8 @@ pub struct CloudUniform {
     pub time_start: f32,
     pub time_stop: f32,
     pub num_classes: u32,
+    pub min: Vec4,
+    pub max: Vec4,
 }
 
 #[allow(clippy::type_complexity)]
@@ -870,6 +873,8 @@ pub fn extract_gaussians<R: PlanarStorage>(
             time_start: settings.time_start,
             time_stop: settings.time_stop,
             num_classes: settings.num_classes as u32,
+            min: aabb.min().extend(1.0),
+            max: aabb.max().extend(1.0),
         };
 
         commands_list.push((