avian_pickup 0.1.6

A plugin for implementing picking up dynamic rigid bodies in Avian physics for the Bevy engine.Modeled after Half Life 2's gravity gun.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

use avian3d::{prelude::*, sync::ancestor_marker::AncestorMarker};
use bevy::prelude::*;

pub(crate) fn get_rigid_body_colliders(
    rigid_body: Entity,
    q_collider_ancestor: &Query<&Children, With<AncestorMarker<ColliderMarker>>>,
    q_collider: &Query<&Collider>,
) -> Option<Vec<Entity>> {
    let mut colliders = Vec::new();
    get_rigid_body_colliders_recursive(rigid_body, q_collider_ancestor, q_collider, &mut colliders);
    if colliders.is_empty() {
        None
    } else {
        Some(colliders)
    }
}

fn get_rigid_body_colliders_recursive(
    entity: Entity,
    q_collider_ancestor: &Query<&Children, With<AncestorMarker<ColliderMarker>>>,
    q_collider: &Query<&Collider>,
    colliders: &mut Vec<Entity>,
) {
    if q_collider.contains(entity) {
        colliders.push(entity);
    }
    if let Ok(children) = q_collider_ancestor.get(entity) {
        for child in children.iter() {
            get_rigid_body_colliders_recursive(*child, q_collider_ancestor, q_collider, colliders);
        }
    }
}