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

use crate::math::{Point, Real};
use crate::query::{PointProjection, PointQuery};
use crate::shape::{Cylinder, FeatureId};
use na;

impl PointQuery for Cylinder {
    #[inline]
    fn project_local_point(&self, pt: &Point<Real>, solid: bool) -> PointProjection {
        // Project on the basis.
        let mut dir_from_basis_center = pt.coords.xz();
        let planar_dist_from_basis_center = dir_from_basis_center.normalize_mut();

        if planar_dist_from_basis_center <= crate::math::DEFAULT_EPSILON {
            dir_from_basis_center = na::Vector2::x();
        }

        let proj2d = dir_from_basis_center * self.radius;

        if pt.y >= -self.half_height
            && pt.y <= self.half_height
            && planar_dist_from_basis_center <= self.radius
        {
            // The point is inside of the cylinder.
            if solid {
                PointProjection::new(true, *pt)
            } else {
                let dist_to_top = self.half_height - pt.coords.y;
                let dist_to_bottom = pt.coords.y - (-self.half_height);
                let dist_to_side = self.radius - planar_dist_from_basis_center;

                if dist_to_top < dist_to_bottom && dist_to_top < dist_to_side {
                    let projection_on_top = Point::new(pt.coords.x, self.half_height, pt.coords.z);
                    PointProjection::new(true, projection_on_top)
                } else if dist_to_bottom < dist_to_top && dist_to_bottom < dist_to_side {
                    let projection_on_bottom =
                        Point::new(pt.coords.x, -self.half_height, pt.coords.z);
                    PointProjection::new(true, projection_on_bottom)
                } else {
                    let projection_on_side = Point::new(proj2d[0], pt.y, proj2d[1]);
                    PointProjection::new(true, projection_on_side)
                }
            }
        } else {
            // The point is outside of the cylinder.
            if pt.y > self.half_height {
                if planar_dist_from_basis_center <= self.radius {
                    let projection_on_top = Point::new(pt.coords.x, self.half_height, pt.coords.z);
                    return PointProjection::new(false, projection_on_top);
                } else {
                    let projection_on_top_circle =
                        Point::new(proj2d[0], self.half_height, proj2d[1]);
                    return PointProjection::new(false, projection_on_top_circle);
                }
            } else if pt.y < -self.half_height {
                // Project on the bottom plane or the bottom circle.
                if planar_dist_from_basis_center <= self.radius {
                    let projection_on_bottom =
                        Point::new(pt.coords.x, -self.half_height, pt.coords.z);
                    return PointProjection::new(false, projection_on_bottom);
                } else {
                    let projection_on_bottom_circle =
                        Point::new(proj2d[0], -self.half_height, proj2d[1]);
                    return PointProjection::new(false, projection_on_bottom_circle);
                }
            } else {
                // Project on the side.
                let projection_on_side = Point::new(proj2d[0], pt.y, proj2d[1]);
                return PointProjection::new(false, projection_on_side);
            }
        }
    }

    #[inline]
    fn project_local_point_and_get_feature(
        &self,
        pt: &Point<Real>,
    ) -> (PointProjection, FeatureId) {
        // TODO: get the actual feature.
        (self.project_local_point(pt, false), FeatureId::Unknown)
    }
}