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

use super::SegmentView;
use crate::math::{Real, Rot, Vect};
use rapier::parry::shape::Capsule;

/// Read-only access to the properties of a capsule.
#[derive(Copy, Clone)]
pub struct CapsuleView<'a> {
    /// The raw shape from Rapier.
    pub raw: &'a Capsule,
}

macro_rules! impl_ref_methods(
    ($View: ident) => {
        impl<'a> $View<'a> {
            /// The axis and endpoint of the capsule.
            pub fn segment(&self) -> SegmentView {
                SegmentView {
                    raw: &self.raw.segment,
                }
            }

            /// The radius of the capsule.
            pub fn radius(&self) -> Real {
                self.raw.radius
            }

            /// The height of this capsule.
            pub fn height(&self) -> Real {
                self.raw.height()
            }

            /// The half-height of this capsule.
            pub fn half_height(&self) -> Real {
                self.raw.half_height()
            }

            /// The center of this capsule.
            pub fn center(&self) -> Vect {
                self.raw.center().into()
            }

            /// The transformation such that `t * Y` is collinear with `b - a` and `t * origin` equals
            /// the capsule's center.
            pub fn canonical_transform(&self) -> (Vect, Rot) {
                self.raw.canonical_transform().into()
            }

            /// The rotation `r` such that `r * Y` is collinear with `b - a`.
            #[cfg(feature = "dim2")]
            pub fn rotation_wrt_y(&self) -> Rot {
                self.raw.rotation_wrt_y().angle()
            }

            /// The rotation `r` such that `r * Y` is collinear with `b - a`.
            #[cfg(feature = "dim3")]
            pub fn rotation_wrt_y(&self) -> Rot {
                self.raw.rotation_wrt_y().into()
            }

            /// The transform `t` such that `t * Y` is collinear with `b - a` and such that `t * origin = (b + a) / 2.0`.
            pub fn transform_wrt_y(&self) -> (Vect, Rot) {
                self.raw.transform_wrt_y().into()
            }
        }
    }
);

impl_ref_methods!(CapsuleView);

/// Read-write access to the properties of a capsule.
pub struct CapsuleViewMut<'a> {
    /// The raw shape from Rapier.
    pub raw: &'a mut Capsule,
}

impl_ref_methods!(CapsuleViewMut);

impl<'a> CapsuleViewMut<'a> {
    /// Set the segment of this capsule.
    pub fn set_segment(&mut self, start: Vect, end: Vect) {
        self.raw.segment.a = start.into();
        self.raw.segment.b = end.into();
    }

    /// Set the radius of this capsule.
    pub fn set_radius(&mut self, radius: Real) {
        self.raw.radius = radius;
    }
}