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
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

use super::{IntoEntity, IntoHandle};
use bevy::prelude::*;
use rapier::data::{ComponentSet, ComponentSetMut, ComponentSetOption, Index};
use rapier::prelude::*;

impl IntoHandle<ColliderHandle> for Entity {
    #[inline]
    fn handle(self) -> ColliderHandle {
        ColliderHandle::from_raw_parts(self.id(), self.generation())
    }
}

impl IntoEntity for ColliderHandle {
    #[inline]
    fn entity(self) -> Entity {
        self.0.entity()
    }
}

pub type QueryPipelineColliderComponentsQuery<'a, 'b> = Query<
    'a,
    (
        Entity,
        &'b ColliderPosition,
        &'b ColliderShape,
        &'b ColliderFlags,
    ),
>;

pub struct QueryPipelineColliderComponentsSet<'a, 'b, 'c>(
    pub &'c QueryPipelineColliderComponentsQuery<'a, 'b>,
);

impl_component_set_wo_query_set!(
    QueryPipelineColliderComponentsSet,
    ColliderPosition,
    |data| data.1
);
impl_component_set_wo_query_set!(QueryPipelineColliderComponentsSet, ColliderShape, |data| {
    data.2
});
impl_component_set_wo_query_set!(QueryPipelineColliderComponentsSet, ColliderFlags, |data| {
    data.3
});

pub struct ColliderComponentsSet<'a, 'b, 'c>(pub ColliderComponentsQuery<'a, 'b, 'c>);

pub type ColliderComponentsQuery<'a, 'b, 'c> = QuerySet<(
    Query<
        'a,
        (
            Entity,
            &'b ColliderChanges,
            &'b ColliderPosition,
            &'b ColliderBroadPhaseData,
            &'b ColliderShape,
            &'b ColliderType,
            &'b ColliderMaterial,
            &'b ColliderFlags,
            Option<&'b ColliderParent>,
        ),
    >,
    Query<
        'a,
        (
            Entity,
            &'c mut ColliderChanges,
            &'c mut ColliderPosition,
            &'c mut ColliderBroadPhaseData,
        ),
    >,
    Query<
        'a,
        (
            Entity,
            &'c mut ColliderChanges,
            Or<(Changed<ColliderPosition>, Added<ColliderPosition>)>,
            Or<(Changed<ColliderFlags>, Added<ColliderFlags>)>,
            Or<(Changed<ColliderShape>, Added<ColliderShape>)>,
            Or<(Changed<ColliderType>, Added<ColliderType>)>,
            Option<Or<(Changed<ColliderParent>, Added<ColliderParent>)>>,
        ),
        Or<(
            Changed<ColliderPosition>,
            Added<ColliderPosition>,
            Changed<ColliderFlags>,
            Added<ColliderFlags>,
            Changed<ColliderShape>,
            Added<ColliderShape>,
            Changed<ColliderType>,
            Added<ColliderType>,
            Changed<ColliderParent>,
            Added<ColliderParent>,
        )>,
    >,
)>;

impl_component_set_mut!(ColliderComponentsSet, ColliderChanges, |data| data.1);
impl_component_set_mut!(ColliderComponentsSet, ColliderPosition, |data| data.2);
impl_component_set_mut!(ColliderComponentsSet, ColliderBroadPhaseData, |d| d.3);

impl_component_set!(ColliderComponentsSet, ColliderShape, |data| data.4);
impl_component_set!(ColliderComponentsSet, ColliderType, |data| data.5);
impl_component_set!(ColliderComponentsSet, ColliderMaterial, |data| data.6);
impl_component_set!(ColliderComponentsSet, ColliderFlags, |data| data.7);

impl_component_set_option!(ColliderComponentsSet, ColliderParent);

#[derive(Bundle)]
pub struct ColliderBundle {
    pub collider_type: ColliderType,
    pub shape: ColliderShape,
    pub position: ColliderPosition,
    pub material: ColliderMaterial,
    pub flags: ColliderFlags,
    pub mass_properties: ColliderMassProps,
    pub changes: ColliderChanges,
    pub bf_data: ColliderBroadPhaseData,
}

impl Default for ColliderBundle {
    fn default() -> Self {
        Self {
            collider_type: ColliderType::Solid,
            shape: ColliderShape::ball(0.5),
            position: ColliderPosition::default(),
            material: ColliderMaterial::default(),
            flags: ColliderFlags::default(),
            mass_properties: ColliderMassProps::default(),
            changes: ColliderChanges::default(),
            bf_data: ColliderBroadPhaseData::default(),
        }
    }
}