checkpoint

src/bike.rs

@@ -1,6 +1,7 @@
-use std::f32::consts::FRAC_PI_2;
+use avian3d::{
-
+    math::{Scalar, FRAC_PI_2},
-use avian3d::prelude::*;
+    prelude::*,
+};
 use bevy::prelude::*;
 
 use crate::physics::CatControllerState;
@@ -19,35 +20,40 @@ pub enum CyberWheel {
     Rear,
 }
 
-// marker for front suspension joint
+#[derive(Component, Clone, Copy, Debug, Reflect, Default)]
-#[derive(Component)]
+#[reflect(Component)]
-pub struct Steering;
+#[require(LinearVelocity, RigidBody(|| RigidBody::Kinematic))]
-
+pub struct WheelState {
-// marker for rear suspension joint
+    pub displacement: Scalar,
-#[derive(Component)]
+    pub sliding: bool,
-pub struct Rearing;
+    pub grounded: bool,
-
-#[derive(Debug, Component, Clone, Default)]
-pub struct PreviousDisplacement(pub f32);
-
-#[derive(Debug, Resource, Reflect)]
-#[reflect(Resource)]
-pub struct SuspensionConfig {
-    pub rest_dist: f32,
-    pub konstant: f32,
-    pub damping: f32,
-    pub front_attach: Vec3,
-    pub rear_attach: Vec3,
 }
 
-impl Default for SuspensionConfig {
+#[derive(Component, Clone, Copy, Debug, Reflect)]
-    fn default() -> Self {
+#[reflect(Component)]
-        Self {
+#[require(WheelState)]
-            rest_dist: REST_DISTANCE,
+pub struct WheelConfig {
-            konstant: 1600.0,
+    pub attach: Vec3,
-            damping: -160.0,
+    pub rest_dist: Scalar,
-            front_attach: FRONT_ATTACH,
+    pub konstant: Scalar,
-            rear_attach: REAR_ATTACH,
+    pub damping: Scalar,
+    pub friction: Scalar,
+}
+
+impl WheelConfig {
+    fn new(
+        attach: Vec3,
+        rest_dist: Scalar,
+        konstant: Scalar,
+        damping: Scalar,
+        friction: Scalar,
+    ) -> Self {
+        WheelConfig {
+            attach,
+            rest_dist,
+            konstant,
+            damping,
+            friction,
         }
     }
 }
@@ -93,43 +99,26 @@ fn spawn_bike(
         });
 }
 
-fn spawn_caster(
-    commands: &mut ChildBuilder,
-    collider: Collider,
-    xform: Transform,
-    direction: Dir3,
-    parent: Entity,
-    is_front: bool,
-) {
-    let caster = ShapeCaster::new(collider, Vec3::ZERO, Quat::IDENTITY, direction)
-        .with_query_filter(SpatialQueryFilter::from_excluded_entities([parent]));
-
-    let mut entity = commands.spawn((caster, xform, PreviousDisplacement(REST_DISTANCE)));
-    if is_front {
-        entity.insert((Steering, Name::new("front wheel")));
-    } else {
-        entity.insert((Rearing, Name::new("rear wheel")));
-    }
-}
-
 fn spawn_wheels(
     commands: &mut ChildBuilder,
     mut meshes: ResMut<Assets<Mesh>>,
     mut materials: ResMut<Assets<StandardMaterial>>,
     body: Entity,
 ) {
-    let (mesh, collider) = gen_tire();
+    let mesh: Mesh = Sphere::new(WHEEL_RADIUS).into();
+    let collider = Collider::sphere(WHEEL_RADIUS);
 
     let front_rake = Vec3::new(0.0, -1.0, -0.57).normalize(); // about 30 degrees
     let front_wheel_pos = FRONT_ATTACH + (front_rake * REST_DISTANCE);
 
-    spawn_caster(
+    wheel_caster(
         commands,
         collider.clone(),
         Transform::from_translation(FRONT_ATTACH),
         Dir3::new_unchecked(front_rake),
         body,
-        true,
+        REST_DISTANCE,
+        CyberWheel::Front,
     );
     wheel_mesh(
         commands,
@@ -137,20 +126,21 @@ fn spawn_wheels(
         &mut materials,
         front_wheel_pos,
         mesh.clone(),
+        WheelConfig::new(FRONT_ATTACH, REST_DISTANCE, 1200., -160., 0.5),
         CyberWheel::Front,
     );
 
     let rear_rake = Vec3::new(0.0, -1.0, 0.57).normalize();
-    //let rear_attach = xform.translation + xform.back() * 0.7;
     let rear_wheel_pos = REAR_ATTACH + (rear_rake * REST_DISTANCE);
 
-    spawn_caster(
+    wheel_caster(
         commands,
         collider,
         Transform::from_translation(REAR_ATTACH),
         Dir3::new_unchecked(rear_rake),
         body,
-        false,
+        REST_DISTANCE,
+        CyberWheel::Rear,
     );
     wheel_mesh(
         commands,
@@ -158,6 +148,7 @@ fn spawn_wheels(
         &mut materials,
         rear_wheel_pos,
         mesh,
+        WheelConfig::new(REAR_ATTACH, REST_DISTANCE, 1200., -160., 0.5),
         CyberWheel::Rear,
     );
 }
@@ -166,11 +157,20 @@ fn spawn_wheels(
 // helper fns for the wheels
 //-************************************************************************
 
-fn gen_tire() -> (Mesh, Collider) {
+fn wheel_caster(
-    let tire_mesh: Mesh = Sphere::new(WHEEL_RADIUS).into();
+    commands: &mut ChildBuilder,
-    let collider = Collider::sphere(WHEEL_RADIUS);
+    collider: Collider,
+    xform: Transform,
+    direction: Dir3,
+    parent: Entity,
+    rest_dist: Scalar,
+    wheel: CyberWheel,
+) {
+    let caster = ShapeCaster::new(collider, xform.translation, Quat::IDENTITY, direction)
+        .with_max_distance(rest_dist)
+        .with_query_filter(SpatialQueryFilter::from_excluded_entities([parent]));
 
-    (tire_mesh, collider)
+    commands.spawn((caster, wheel));
 }
 
 fn wheel_mesh(
@@ -179,6 +179,7 @@ fn wheel_mesh(
     materials: &mut ResMut<Assets<StandardMaterial>>,
     position: Vec3,
     tire_mesh: Mesh,
+    config: WheelConfig,
     wheel: CyberWheel,
 ) {
     let wheel_material = &StandardMaterial {
@@ -192,10 +193,10 @@ fn wheel_mesh(
 
     commands.spawn((
         Name::new("tire"),
+        config,
         Mesh3d(meshes.add(tire_mesh)),
         MeshMaterial3d(materials.add(wheel_material.clone())),
         xform,
-        PreviousDisplacement::default(),
         wheel,
     ));
 }
@@ -204,8 +205,6 @@ pub struct CyberBikePlugin;
 
 impl Plugin for CyberBikePlugin {
     fn build(&self, app: &mut App) {
-        app.init_resource::<SuspensionConfig>();
-        app.register_type::<SuspensionConfig>();
         app.add_systems(Startup, spawn_bike);
     }
 }

src/main.rs

@@ -2,7 +2,7 @@ use avian3d::prelude::{
     Collider, ColliderDensity, CollisionLayers, CollisionMargin, Friction, PhysicsGizmos, RigidBody,
 };
 use bevy::{
-    color::palettes::css::SILVER,
+    color::{palettes::css::SILVER, Alpha},
     pbr::MeshMaterial3d,
     prelude::{
         default, App, AppGizmoBuilder, Assets, BuildChildren, ButtonInput, ChildBuild, Color,
@@ -60,7 +60,7 @@ fn ground_and_light(
         .spawn((
             spatial_bundle,
             RigidBody::Static,
-            Collider::cuboid(50.0, 5., 50.0),
+            Collider::cuboid(500.0, 5., 500.0),
             CollisionMargin(0.1),
             ColliderDensity(1000.0),
             CollisionLayers::ALL,
@@ -68,8 +68,8 @@ fn ground_and_light(
         ))
         .with_children(|p| {
             let bundle = (
-                Mesh3d(meshes.add(Plane3d::default().mesh().size(50.0, 50.0))),
+                Mesh3d(meshes.add(Plane3d::default().mesh().size(500.0, 500.0))),
-                MeshMaterial3d(materials.add(Color::from(SILVER))),
+                MeshMaterial3d(materials.add(Color::from(SILVER).with_alpha(0.7))),
                 Transform::from_xyz(0.0, 2.5, 0.0),
                 Visibility::Visible,
             );

src/physics.rs

@@ -57,14 +57,12 @@ impl CatControllerState {
 mod systems {
     use std::f32::consts::{FRAC_PI_3, FRAC_PI_4};
 
-    use avian3d::prelude::*;
+    use avian3d::{math::Scalar, prelude::*};
-    use bevy::prelude::*;
+    use bevy::{color, math::VectorSpace, prelude::*};
 
     use super::{CatControllerSettings, CatControllerState, CyberLean};
     use crate::{
-        bike::{
+        bike::{CyberBikeBody, CyberWheel, WheelConfig, WheelState},
-            CyberBikeBody, CyberWheel, PreviousDisplacement, Rearing, Steering, SuspensionConfig,
-        },
         input::InputState,
     };
 
@@ -143,7 +141,7 @@ mod systems {
                 torque.apply_torque(tork);
                 gizmos.arrow(
                     xform.translation + Vec3::Y,
-                    xform.translation + tork + Vec3::Y,
+                    xform.translation + mag * *xform.right() + Vec3::Y,
                     Color::WHITE,
                 );
             }
@@ -161,76 +159,77 @@ mod systems {
 
     pub(super) fn suspension(
         mut bike_body_query: Query<(&Transform, &mut ExternalForce), With<CyberBikeBody>>,
-        mut wheel_mesh_query: Query<(&mut Transform, &CyberWheel), Without<CyberBikeBody>>,
+        mut wheel_mesh_query: Query<
-        mut front_wheel_query: Query<
+            (
-            (&ShapeCaster, &ShapeHits, &mut PreviousDisplacement),
+                &mut Transform,
-            (With<Steering>, Without<Rearing>),
+                &mut WheelState,
+                &WheelConfig,
+                &LinearVelocity,
+                &CyberWheel,
+            ),
+            Without<CyberBikeBody>,
         >,
-        mut rear_wheel_query: Query<
+        caster_query: Query<(&ShapeCaster, &ShapeHits, &CyberWheel)>,
-            (&ShapeCaster, &ShapeHits, &mut PreviousDisplacement),
-            (With<Rearing>, Without<Steering>),
-        >,
-        config: Res<SuspensionConfig>,
         time: Res<Time>,
         input: Res<InputState>,
+        mut gizmos: Gizmos,
     ) {
-        let max_thrust = 30.0;
+        let max_thrust = 100.0;
-        let max_yaw = 10.0;
+        let max_yaw = 50.0;
-        let mut thrust = (input.throttle * max_thrust);
+        let mut thrust = input.throttle * max_thrust;
         if input.brake {
             thrust *= -1.0;
         }
         let yaw = input.yaw * max_yaw;
-
         let dt = time.delta().as_secs_f32();
-        let mut front_wheel_xform = None;
+
-        let mut rear_wheel_xform = None;
+        let mut front_caster = &ShapeCaster::default();
-        for (xform, wheel) in wheel_mesh_query.iter_mut() {
+        let mut rear_caster = &ShapeCaster::default();
+
+        let mut front_hits = &ShapeHits::default();
+        let mut rear_hits = &ShapeHits::default();
+
+        for (caster, hits, wheel) in caster_query.iter() {
             match wheel {
                 CyberWheel::Front => {
-                    front_wheel_xform = Some(xform);
+                    front_caster = caster;
+                    front_hits = hits;
                 }
                 CyberWheel::Rear => {
-                    rear_wheel_xform = Some(xform);
+                    rear_caster = caster;
+                    rear_hits = hits;
                 }
             }
         }
-
         let (bike_xform, mut bike_forces) = bike_body_query.single_mut();
+        for (mut xform, mut state, config, velocity, wheel) in wheel_mesh_query.iter_mut() {
+            let (caster, hits) = match wheel {
+                CyberWheel::Front => (front_caster, front_hits),
+                CyberWheel::Rear => (rear_caster, rear_hits),
+            };
 
-        for (caster, hits, mut prev) in front_wheel_query.iter_mut() {
+            let prev = &mut state.displacement;
             for hit in hits.iter() {
-                if let Some(ref mut xform) = front_wheel_xform {
+                let force = suspension_force(caster, hit, config, prev, dt, &mut xform);
-                    let force =
+                bike_forces.apply_force_at_point(
-                        suspension_force(caster, hit, &config, prev.as_mut(), dt, xform, true);
+                    force,
-                    bike_forces.apply_force_at_point(
+                    caster.global_origin(),
-                        force,
+                    bike_xform.translation,
-                        caster.global_origin(),
+                );
-                        bike_xform.translation,
+                dbg!(velocity);
-                    );
+                let normal = hit.normal1;
-                    let normal = hit.normal1;
+                let steering = match wheel {
-                    let steering = normal.cross(*bike_xform.back()) * yaw;
+                    CyberWheel::Front => normal.cross(*bike_xform.back()) * yaw,
-                    let thrust = normal.cross(*bike_xform.right()) * thrust;
+                    _ => normal.cross(*bike_xform.forward()) * yaw, // Vec3::ZERO,
-                    let total = thrust + steering;
+                };
-                    bike_forces.apply_force_at_point(total, hit.point1, bike_xform.translation);
+                let thrust = normal.cross(*bike_xform.right()) * thrust;
-                }
+                let total = (thrust + steering) * dt;
-            }
+                bike_forces.apply_force_at_point(total, hit.point1, bike_xform.translation);
-        }
+                gizmos.arrow(
-
+                    hit.point1,
-        for (caster, hits, mut prev) in rear_wheel_query.iter_mut() {
+                    hit.point1 + total,
-            for hit in hits.iter() {
+                    Color::linear_rgb(1., 1., 0.2),
-                if let Some(ref mut xform) = rear_wheel_xform {
+                );
-                    let force =
-                        suspension_force(caster, hit, &config, prev.as_mut(), dt, xform, false);
-                    bike_forces.apply_force_at_point(
-                        force,
-                        caster.global_origin(),
-                        bike_xform.translation,
-                    );
-                    let normal = hit.normal1;
-                    let thrust = normal.cross(*bike_xform.right()) * thrust;
-                    bike_forces.apply_force_at_point(thrust, hit.point1, bike_xform.translation);
-                }
             }
         }
     }
@@ -238,33 +237,24 @@ mod systems {
     fn suspension_force(
         caster: &ShapeCaster,
         hit: &ShapeHitData,
-        config: &SuspensionConfig,
+        config: &WheelConfig,
-        previous_dispacement: &mut PreviousDisplacement,
+        previous_dispacement: &mut Scalar,
-        dt: f32,
+        dt: Scalar,
         wheel_xform: &mut Transform,
-        is_front: bool,
     ) -> Vec3 {
         let mut up_force = Vec3::ZERO;
-        let attach = if is_front {
-            config.front_attach
-        } else {
-            config.rear_attach
-        };
         let dist = hit.distance;
         let cdir = caster.direction.as_vec3();
         let dir = caster.global_direction().as_vec3();
-        let loc = if dist < config.rest_dist {
+        let loc = {
             let displacement = config.rest_dist - dist;
-            let damper_vel = (displacement - previous_dispacement.0) / dt;
+            let damper_vel = (displacement - *previous_dispacement) / dt;
             dbg!(damper_vel);
-            *previous_dispacement = PreviousDisplacement(displacement);
+            *previous_dispacement = displacement;
             let mag = config.konstant * displacement - config.damping * damper_vel;
             up_force = -dir * mag;
-            attach + (cdir * dist)
+            config.attach + (cdir * dist)
-        } else {
-            attach + (cdir * config.rest_dist)
         };
-
         wheel_xform.translation = loc;
 
         up_force
@@ -284,6 +274,9 @@ impl Plugin for CyberPhysicsPlugin {
             .register_type::<CyberLean>()
             .add_plugins((PhysicsPlugins::default(), PhysicsDebugPlugin::default()))
             .insert_resource(SubstepCount(12))
-            .add_systems(Update, (apply_lean, calculate_lean, suspension));
+            .add_systems(
+                FixedUpdate,
+                (calculate_lean, apply_lean, suspension).chain(),
+            );
     }
 }