File: DillCreator.cpp

package info (click to toggle)
bullet 2.87%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 14,272 kB
  • sloc: cpp: 204,241; ansic: 12,100; lisp: 12,017; python: 593; makefile: 136; sh: 8
file content (124 lines) | stat: -rw-r--r-- 4,064 bytes parent folder | download
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
#include "DillCreator.hpp"
#include <cmath>
namespace btInverseDynamics {

DillCreator::DillCreator(int level)
    : m_level(level),
      m_num_bodies(BT_ID_POW(2, level))
     {
		  m_parent.resize(m_num_bodies);
		m_parent_r_parent_body_ref.resize(m_num_bodies);
		m_body_T_parent_ref.resize(m_num_bodies);
		m_body_axis_of_motion.resize(m_num_bodies);
		m_mass.resize(m_num_bodies);
		m_body_r_body_com.resize(m_num_bodies);
		m_body_I_body.resize(m_num_bodies);

    // generate names (for debugging)
    for (int i = 0; i < m_num_bodies; i++) {
        m_parent[i] = i - 1;

        // all z-axis (DH convention)
        m_body_axis_of_motion[i](0) = 0.0;
        m_body_axis_of_motion[i](1) = 0.0;
        m_body_axis_of_motion[i](2) = 1.0;
    }

    // recursively build data structures
    m_current_body = 0;
    const int parent = -1;
    const idScalar d_DH = 0.0;
    const idScalar a_DH = 0.0;
    const idScalar alpha_DH = 0.0;

    if (-1 == recurseDill(m_level, parent, d_DH, a_DH, alpha_DH)) {
        error_message("recurseDill failed\n");
        abort();
    }
}

DillCreator::~DillCreator() {}

int DillCreator::getNumBodies(int* num_bodies) const {
    *num_bodies = m_num_bodies;
    return 0;
}

int DillCreator::getBody(const int body_index, int* parent_index, JointType* joint_type,
                         vec3* parent_r_parent_body_ref, mat33* body_T_parent_ref,
                         vec3* body_axis_of_motion, idScalar* mass, vec3* body_r_body_com,
                         mat33* body_I_body, int* user_int, void** user_ptr) const {
    if (body_index < 0 || body_index >= m_num_bodies) {
        error_message("invalid body index %d\n", body_index);
        return -1;
    }
    *parent_index = m_parent[body_index];
    *joint_type = REVOLUTE;
    *parent_r_parent_body_ref = m_parent_r_parent_body_ref[body_index];
    *body_T_parent_ref = m_body_T_parent_ref[body_index];
    *body_axis_of_motion = m_body_axis_of_motion[body_index];
    *mass = m_mass[body_index];
    *body_r_body_com = m_body_r_body_com[body_index];
    *body_I_body = m_body_I_body[body_index];

    *user_int = 0;
    *user_ptr = 0;
    return 0;
}

int DillCreator::recurseDill(const int level, const int parent, const idScalar d_DH_in,
                             const idScalar a_DH_in, const idScalar alpha_DH_in) {
    if (level < 0) {
        error_message("invalid level parameter (%d)\n", level);
        return -1;
    }

    if (m_current_body >= m_num_bodies || m_current_body < 0) {
        error_message("invalid body parameter (%d, num_bodies: %d)\n", m_current_body,
                      m_num_bodies);
        return -1;
    }

    idScalar size = BT_ID_MAX(level, 1);
    const int body = m_current_body;
    //  length = 0.1 * size;
    //  with = 2 * 0.01 * size;

    /// these parameters are from the paper ...
    /// TODO: add proper citation
    m_parent[body] = parent;
    m_mass[body] = 0.1 * BT_ID_POW(size, 3);
    m_body_r_body_com[body](0) = 0.05 * size;
    m_body_r_body_com[body](1) = 0;
    m_body_r_body_com[body](2) = 0;
    // initialization
    for (int i = 0; i < 3; i++) {
        m_parent_r_parent_body_ref[body](i) = 0;
        for (int j = 0; j < 3; j++) {
            m_body_I_body[body](i, j) = 0.0;
            m_body_T_parent_ref[body](i, j) = 0.0;
        }
    }
    const idScalar size_5 = pow(size, 5);
    m_body_I_body[body](0, 0) = size_5 / 0.2e6;
    m_body_I_body[body](1, 1) = size_5 * 403 / 1.2e6;
    m_body_I_body[body](2, 2) = m_body_I_body[body](1, 1);

    getVecMatFromDH(0, 0, a_DH_in, alpha_DH_in, &m_parent_r_parent_body_ref[body],
                    &m_body_T_parent_ref[body]);

    // attach "level" Dill systems of levels 1...level
    for (int i = 1; i <= level; i++) {
        idScalar d_DH = 0.01 * size;
        if (i == level) {
            d_DH = 0.0;
        }
        const idScalar a_DH = i * 0.1;
        const idScalar alpha_DH = i * BT_ID_PI / 3.0;
        m_current_body++;
        recurseDill(i - 1, body, d_DH, a_DH, alpha_DH);
    }

    return 0;  // ok!
}
}