simplified cube example

Author: Vrixyz

examples/cube.rs

@@ -11,7 +11,6 @@ use bevy_wgsparkl::resources::{AppState, PhysicsContext};
 use nalgebra::{Vector3, vector};
 use wgrapier3d::dynamics::body::{BodyCoupling, BodyCouplingEntry};
 use wgsparkl3d::models::DruckerPrager;
-use wgsparkl3d::solver::ParticlePhase;
 use wgsparkl3d::{
     models::ElasticCoefficients,
     pipeline::MpmData,
@@ -77,15 +76,6 @@ pub fn setup_mpm_particles(
     let grid_size_z = 25;
     let num_particles = grid_size_x * grid_size_y * grid_size_z;
 
-    let particle_positions = (0..num_particles)
-        .map(|i| {
-            let x = i % grid_size_x;
-            let y = (i / grid_size_x) % grid_size_y;
-            let z = (i / (grid_size_x * grid_size_y)) % grid_size_z;
-            Vector3::new(x as f32, y as f32 + 1f32, z as f32)
-        })
-        .collect::<Vec<_>>();
-
     app_state.particles_initialized = true;
 
     let coupling: Vec<_> = coupling
@@ -121,46 +111,34 @@ pub fn setup_mpm_particles(
     let cell_width = 1.0;
     let mut particles = vec![];
 
-    for x in -1..2 {
-        for z in -1..2 {
-            let x = x as f32;
-            let z = z as f32;
-            let offset = vector![x * grid_size_x as f32, 0f32, z * grid_size_z as f32] * 2f32;
-            for particle in &particle_positions {
-                let position = vector![particle.x, particle.y, particle.z];
-
-                let particle_size = vector![1.0, 1.0, 1.0];
-                let volume = particle_size.x * particle_size.y * particle_size.z;
-                let density = 1700.0;
-                particles.push(Particle {
-                    position: nalgebra::Rotation::from_axis_angle(
-                        &Vector3::z_axis(),
-                        5f32.to_radians(),
-                    ) * vector![position.x, position.y, position.z]
-                        + offset,
-                    velocity: Vector3::zeros(),
-                    volume: ParticleMassProps::new(density * volume, volume.cbrt() / 2.0),
-                    model: ElasticCoefficients::from_young_modulus(
-                        100_000_000.0 * 10f32.powf(z - 1f32),
-                        0.2 + x * 0.05f32,
-                    ),
-                    plasticity: Some(DruckerPrager {
-                        h0: (45.0f32 + x * 20f32).to_radians(),
-                        h1: (70.0f32 + x * 20f32).to_radians() + z,
-                        h2: 1.6 + z * 0.5f32,
-                        h3: 25.0f32.to_radians() + x,
-                        ..DruckerPrager::new(
-                            100_000_000.0 * 10f32.powf(z - 1f32),
-                            0.2 + x * 0.05f32,
-                        )
-                    }),
-                    phase: Some(ParticlePhase {
-                        phase: 0.5 + 0.5 * x,
-                        max_stretch: (0.0 + z).clamp(0.0, 1.0),
-                    }),
-                });
-            }
-        }
+    let density = 2700.0;
+    let rock_size = vector![1.0, 1.0, 1.0];
+    let volume = rock_size.x * rock_size.y * rock_size.z;
+    let mass_props = ParticleMassProps::new(density * volume, volume.cbrt() / 2.0);
+    let modulus = 10_000_000.0;
+    let poisson = 0.2;
+    let model = ElasticCoefficients::from_young_modulus(modulus, poisson);
+    let plasticity = Some(DruckerPrager {
+        h0: 45.0f32.to_radians(),
+        h1: 50.0f32.to_radians(),
+        h2: 0.4,
+        h3: 15.0f32.to_radians(),
+        ..DruckerPrager::new(modulus, poisson)
+    });
+
+    for i in 0..num_particles {
+        let x = i % grid_size_x;
+        let y = (i / grid_size_x) % grid_size_y;
+        let z = (i / (grid_size_x * grid_size_y)) % grid_size_z;
+        let position = vector![x, y, z];
+        particles.push(Particle {
+            position: vector![position.x as f32, position.y as f32, position.z as f32],
+            velocity: Vector3::zeros(),
+            volume: mass_props,
+            model,
+            plasticity,
+            phase: None,
+        });
     }
 
     println!("Number of simulated particles: {}", particles.len());