aboutsummaryrefslogtreecommitdiff
path: root/benchmarks2d/joint_prismatic2.rs
blob: 0cbf859fdda424eb484ee491e8820252c7828c8c (plain)
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
use rapier2d::prelude::*;
use rapier_testbed2d::Testbed;

pub fn init_world(testbed: &mut Testbed) {
    /*
     * World
     */
    let mut bodies = RigidBodySet::new();
    let mut colliders = ColliderSet::new();
    let mut joints = JointSet::new();

    /*
     * Create the balls
     */
    // Build the rigid body.
    let rad = 0.4;
    let num = 10;
    let shift = 1.0;

    for l in 0..25 {
        let y = l as f32 * shift * (num as f32 + 2.0) * 2.0;

        for j in 0..50 {
            let x = j as f32 * shift * 4.0;

            let ground = RigidBodyBuilder::new_static()
                .translation(vector![x, y])
                .build();
            let mut curr_parent = bodies.insert(ground);
            let collider = ColliderBuilder::cuboid(rad, rad).build();
            colliders.insert_with_parent(collider, curr_parent, &mut bodies);

            for i in 0..num {
                let y = y - (i + 1) as f32 * shift;
                let density = 1.0;
                let rigid_body = RigidBodyBuilder::new_dynamic()
                    .translation(vector![x, y])
                    .build();
                let curr_child = bodies.insert(rigid_body);
                let collider = ColliderBuilder::cuboid(rad, rad).density(density).build();
                colliders.insert_with_parent(collider, curr_child, &mut bodies);

                let axis = if i % 2 == 0 {
                    UnitVector::new_normalize(vector![1.0, 1.0])
                } else {
                    UnitVector::new_normalize(vector![-1.0, 1.0])
                };

                let mut prism =
                    PrismaticJoint::new(Point::origin(), axis, point![0.0, shift], axis);
                prism.limits_enabled = true;
                prism.limits[0] = -1.5;
                prism.limits[1] = 1.5;
                joints.insert(&mut bodies, curr_parent, curr_child, prism);

                curr_parent = curr_child;
            }
        }
    }

    /*
     * Set up the testbed.
     */
    testbed.set_world(bodies, colliders, joints);
    testbed.look_at(point![80.0, 80.0], 15.0);
}