GCC Code Coverage Report


Directory: ./
File: src/tools/robot-utils.cpp
Date: 2024-11-13 12:35:17
Exec Total Coverage
Lines: 237 330 71.8%
Branches: 207 556 37.2%

Line Branch Exec Source
1 /*
2 * Copyright 2017, 2019
3 * LAAS-CNRS
4 * A. Del Prete, T. Flayols, O. Stasse, F. Bailly
5 *
6 */
7
8 /** pinocchio is forcing the BOOST_MPL_LIMIT_VECTOR_SIZE to a specific value.
9 This happen to be not working when including the boost property_tree
10 library. For this reason if defined, the current value of
11 BOOST_MPL_LIMIT_VECTOR_SIZE is saved in the preprocessor stack and unset.
12 Once the property_tree included the pinocchio value of this variable is
13 restored.
14 */
15 #ifdef BOOST_MPL_LIMIT_VECTOR_SIZE
16 #pragma push_macro("BOOST_MPL_LIMIT_VECTOR_SIZE")
17 #undef BOOST_MPL_LIMIT_VECTOR_SIZE
18 #include <boost/property_tree/ptree.hpp>
19 #include <boost/property_tree/xml_parser.hpp>
20 #pragma pop_macro("BOOST_MPL_LIMIT_VECTOR_SIZE")
21 #else
22 #include <boost/property_tree/ptree.hpp>
23 #include <boost/property_tree/xml_parser.hpp>
24 #endif
25
26 #include <dynamic-graph/factory.h>
27
28 #include <iostream>
29 #include <sot/core/debug.hh>
30 #include <sot/core/robot-utils.hh>
31 #include <sstream>
32
33 namespace dynamicgraph {
34 namespace sot {
35 namespace dg = ::dynamicgraph;
36 namespace pt = boost::property_tree;
37 using namespace dg;
38 using namespace dg::command;
39
40 ForceLimits VoidForceLimits;
41 JointLimits VoidJointLimits;
42 Index VoidIndex(-1);
43
44 1 RobotUtilShrPtr RefVoidRobotUtil() {
45 1 return std::shared_ptr<RobotUtil>(nullptr);
46 }
47
48 ExtractJointMimics::ExtractJointMimics(std::string &robot_model) {
49 // Parsing the model from a string.
50 std::istringstream iss(robot_model);
51 /// Read the XML file in the property tree.
52 boost::property_tree::read_xml(iss, tree_);
53 /// Start the recursive parsing.
54 go_through_full();
55 }
56
57 const std::vector<std::string> &ExtractJointMimics::get_mimic_joints() {
58 return mimic_joints_;
59 }
60
61 void ExtractJointMimics::go_through_full() {
62 /// Root of the recursive parsing.
63 current_joint_name_ = "";
64 go_through(tree_, 0, 0);
65 }
66
67 void ExtractJointMimics::go_through(pt::ptree &pt, int level, int stage) {
68 /// If pt is empty (i.e. this is a leaf)
69 if (pt.empty()) {
70 /// and this is a name of a joint (stage == 3) update the
71 /// curret_joint_name_ variable.
72 if (stage == 3) current_joint_name_ = pt.data();
73 } else {
74 /// This is not a leaf
75 for (auto pos : pt) {
76 int new_stage = stage;
77
78 /// But this is joint
79 if (pos.first == "joint")
80 /// the continue the exploration.
81 new_stage = 1;
82 else if (pos.first == "<xmlattr>") {
83 /// we are exploring the xml attributes of a joint
84 /// -> continue the exploration
85 if (stage == 1) new_stage = 2;
86 }
87 /// The xml attribute of the joint is the name
88 /// next leaf is the name we are possibly looking for
89 else if (pos.first == "name") {
90 if (stage == 2) new_stage = 3;
91 }
92 /// The exploration of the tree tracback on the joint
93 /// and find that this is a mimic joint.
94 else if (pos.first == "mimic") {
95 if (stage == 1)
96 /// Save the current name of the joint
97 /// in mimic_joints.
98 mimic_joints_.push_back(current_joint_name_);
99 } else
100 new_stage = 0;
101
102 /// Explore the subtree of the XML robot description.
103 go_through(pos.second, level + 1, new_stage);
104 }
105 }
106 }
107
108 1 void ForceLimits::display(std::ostream &os) const {
109 1 os << "Lower limits:" << std::endl;
110 1 os << lower << std::endl;
111 1 os << "Upper Limits:" << std::endl;
112 1 os << upper << std::endl;
113 1 }
114
115 /******************** FootUtil ***************************/
116
117 2 void FootUtil::display(std::ostream &os) const {
118 2 os << "Right Foot Sole XYZ " << std::endl;
119 2 os << m_Right_Foot_Sole_XYZ << std::endl;
120 2 os << "Left Foot Frame Name:" << m_Left_Foot_Frame_Name << std::endl;
121 2 os << "Right Foot Frame Name:" << m_Right_Foot_Frame_Name << std::endl;
122 2 }
123
124 /******************** HandUtil ***************************/
125
126 1 void HandUtil::display(std::ostream &os) const {
127 1 os << "Left Hand Frame Name:" << m_Left_Hand_Frame_Name << std::endl;
128 1 os << "Right Hand Frame Name:" << m_Right_Hand_Frame_Name << std::endl;
129 1 }
130
131 /******************** ForceUtil ***************************/
132
133 4 void ForceUtil::set_name_to_force_id(const std::string &name,
134 const Index &force_id) {
135 4 m_name_to_force_id[name] = (Index)force_id;
136 4 create_force_id_to_name_map();
137
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 3 times.
4 if (name == "rf")
138 1 set_force_id_right_foot(m_name_to_force_id[name]);
139
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 2 times.
3 else if (name == "lf")
140 1 set_force_id_left_foot(m_name_to_force_id[name]);
141
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 1 times.
2 else if (name == "lh")
142 1 set_force_id_left_hand(m_name_to_force_id[name]);
143
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 else if (name == "rh")
144 1 set_force_id_right_hand(m_name_to_force_id[name]);
145 4 }
146
147 1 void ForceUtil::set_force_id_to_limits(const Index &force_id,
148 const dg::Vector &lf,
149 const dg::Vector &uf) {
150
2/4
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
1 m_force_id_to_limits[(Index)force_id].lower = lf;
151
2/4
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
1 m_force_id_to_limits[(Index)force_id].upper = uf;
152 1 }
153
154 4 Index ForceUtil::get_id_from_name(const std::string &name) {
155 4 std::map<std::string, Index>::const_iterator it;
156
1/2
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
4 it = m_name_to_force_id.find(name);
157
1/2
✓ Branch 3 taken 4 times.
✗ Branch 4 not taken.
4 if (it != m_name_to_force_id.end()) return it->second;
158 return -1;
159 }
160
161 std::string force_default_rtn("Force name not found");
162 std::string joint_default_rtn("Joint name not found");
163
164 4 const std::string &ForceUtil::get_name_from_id(Index idx) {
165 4 std::map<Index, std::string>::const_iterator it;
166
1/2
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
4 it = m_force_id_to_name.find(idx);
167
1/2
✓ Branch 3 taken 4 times.
✗ Branch 4 not taken.
4 if (it != m_force_id_to_name.end()) return it->second;
168 return force_default_rtn;
169 }
170
171 std::string ForceUtil::cp_get_name_from_id(Index idx) {
172 const std::string &default_rtn = get_name_from_id(idx);
173 return default_rtn;
174 }
175 4 void ForceUtil::create_force_id_to_name_map() {
176 4 std::map<std::string, Index>::const_iterator it;
177
2/2
✓ Branch 6 taken 10 times.
✓ Branch 7 taken 4 times.
14 for (it = m_name_to_force_id.begin(); it != m_name_to_force_id.end(); it++)
178
2/4
✓ Branch 3 taken 10 times.
✗ Branch 4 not taken.
✓ Branch 6 taken 10 times.
✗ Branch 7 not taken.
10 m_force_id_to_name[it->second] = it->first;
179 4 }
180
181 2 const ForceLimits &ForceUtil::get_limits_from_id(Index force_id) {
182 std::map<Index, ForceLimits>::const_iterator iter =
183
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
2 m_force_id_to_limits.find(force_id);
184
1/2
✗ Branch 3 not taken.
✓ Branch 4 taken 2 times.
2 if (iter == m_force_id_to_limits.end())
185 return VoidForceLimits; // Returns void instance
186 2 return iter->second;
187 }
188
189 ForceLimits ForceUtil::cp_get_limits_from_id(Index force_id) {
190 std::map<Index, ForceLimits>::const_iterator iter =
191 m_force_id_to_limits.find(force_id);
192 if (iter == m_force_id_to_limits.end())
193 return VoidForceLimits; // Returns void instance
194 return iter->second;
195 }
196
197 2 void ForceUtil::display(std::ostream &os) const {
198 2 os << "Force Id to limits " << std::endl;
199 2 for (std::map<Index, ForceLimits>::const_iterator it =
200 2 m_force_id_to_limits.begin();
201
2/2
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 2 times.
3 it != m_force_id_to_limits.end(); ++it) {
202
1/2
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
1 it->second.display(os);
203 }
204
205 2 os << "Name to force id:" << std::endl;
206 2 for (std::map<std::string, Index>::const_iterator it =
207 2 m_name_to_force_id.begin();
208
2/2
✓ Branch 3 taken 4 times.
✓ Branch 4 taken 2 times.
6 it != m_name_to_force_id.end(); ++it) {
209
5/10
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
✓ Branch 5 taken 4 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 4 times.
✗ Branch 9 not taken.
✓ Branch 12 taken 4 times.
✗ Branch 13 not taken.
✓ Branch 15 taken 4 times.
✗ Branch 16 not taken.
4 os << "(" << it->first << "," << it->second << ") ";
210 }
211 2 os << std::endl;
212
213 2 os << "Force id to Name:" << std::endl;
214 2 for (std::map<Index, std::string>::const_iterator it =
215 2 m_force_id_to_name.begin();
216
2/2
✓ Branch 3 taken 4 times.
✓ Branch 4 taken 2 times.
6 it != m_force_id_to_name.end(); ++it) {
217
5/10
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
✓ Branch 5 taken 4 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 4 times.
✗ Branch 9 not taken.
✓ Branch 12 taken 4 times.
✗ Branch 13 not taken.
✓ Branch 15 taken 4 times.
✗ Branch 16 not taken.
4 os << "(" << it->first << "," << it->second << ") ";
218 }
219 2 os << std::endl;
220
221 2 os << "Index for force sensors:" << std::endl;
222 2 os << "Left Hand (" << m_Force_Id_Left_Hand << ") ,"
223 2 << "Right Hand (" << m_Force_Id_Right_Hand << ") ,"
224 2 << "Left Foot (" << m_Force_Id_Left_Foot << ") ,"
225 2 << "Right Foot (" << m_Force_Id_Right_Foot << ") " << std::endl;
226 2 }
227
228 /**************** FromURDFToSot *************************/
229
4/8
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✓ Branch 12 taken 2 times.
✗ Branch 13 not taken.
✓ Branch 15 taken 2 times.
✗ Branch 16 not taken.
✓ Branch 19 taken 2 times.
✗ Branch 20 not taken.
2 RobotUtil::RobotUtil() {}
230
231 1 void RobotUtil::set_joint_limits_for_id(const Index &idx, const double &lq,
232 const double &uq) {
233
1/2
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
1 m_limits_map[(Index)idx] = JointLimits(lq, uq);
234 1 }
235
236 4 const JointLimits &RobotUtil::get_joint_limits_from_id(Index id) {
237
1/2
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
4 std::map<Index, JointLimits>::const_iterator iter = m_limits_map.find(id);
238
1/2
✗ Branch 3 not taken.
✓ Branch 4 taken 4 times.
4 if (iter == m_limits_map.end()) return VoidJointLimits;
239 4 return iter->second;
240 }
241 2 JointLimits RobotUtil::cp_get_joint_limits_from_id(Index id) {
242 2 const JointLimits &rtn = get_joint_limits_from_id(id);
243 2 return rtn;
244 }
245
246 1 void RobotUtil::set_name_to_id(const std::string &jointName,
247 const Index &jointId) {
248 1 m_name_to_id[jointName] = (Index)jointId;
249 1 create_id_to_name_map();
250 1 }
251
252 2 void RobotUtil::create_id_to_name_map() {
253 2 std::map<std::string, Index>::const_iterator it;
254
2/2
✓ Branch 6 taken 2 times.
✓ Branch 7 taken 2 times.
4 for (it = m_name_to_id.begin(); it != m_name_to_id.end(); it++)
255
2/4
✓ Branch 3 taken 2 times.
✗ Branch 4 not taken.
✓ Branch 6 taken 2 times.
✗ Branch 7 not taken.
2 m_id_to_name[it->second] = it->first;
256 2 }
257
258 1 const Index &RobotUtil::get_id_from_name(const std::string &name) {
259
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 std::map<std::string, Index>::const_iterator it = m_name_to_id.find(name);
260
1/2
✗ Branch 3 not taken.
✓ Branch 4 taken 1 times.
1 if (it == m_name_to_id.end()) return VoidIndex;
261 1 return it->second;
262 }
263
264 1 const std::string &RobotUtil::get_name_from_id(Index id) {
265
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 std::map<Index, std::string>::const_iterator iter = m_id_to_name.find(id);
266
1/2
✗ Branch 3 not taken.
✓ Branch 4 taken 1 times.
1 if (iter == m_id_to_name.end()) return joint_default_rtn;
267 1 return iter->second;
268 }
269
270 void RobotUtil::set_urdf_to_sot(const std::vector<Index> &urdf_to_sot) {
271 m_nbJoints = urdf_to_sot.size();
272 m_urdf_to_sot.resize(urdf_to_sot.size());
273 m_dgv_urdf_to_sot.resize(urdf_to_sot.size());
274 for (std::size_t idx = 0; idx < urdf_to_sot.size(); idx++) {
275 m_urdf_to_sot[(Index)idx] = urdf_to_sot[(Index)idx];
276 m_dgv_urdf_to_sot[(Index)idx] =
277 static_cast<double>(urdf_to_sot[(Index)idx]);
278 }
279 }
280
281 1 void RobotUtil::set_urdf_to_sot(const dg::Vector &urdf_to_sot) {
282 1 m_nbJoints = urdf_to_sot.size();
283 1 m_urdf_to_sot.resize(urdf_to_sot.size());
284
2/2
✓ Branch 1 taken 3 times.
✓ Branch 2 taken 1 times.
4 for (unsigned int idx = 0; idx < urdf_to_sot.size(); idx++) {
285 3 m_urdf_to_sot[idx] = (unsigned int)urdf_to_sot[idx];
286 }
287 1 m_dgv_urdf_to_sot = urdf_to_sot;
288 1 }
289
290 3 bool RobotUtil::joints_urdf_to_sot(ConstRefVector q_urdf, RefVector q_sot) {
291
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
3 if (m_nbJoints == 0) {
292 SEND_MSG("set_urdf_to_sot should be called", MSG_TYPE_ERROR);
293 return false;
294 }
295
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
3 assert(q_urdf.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints));
296
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
3 assert(q_sot.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints));
297
298
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 3 times.
12 for (unsigned int idx = 0; idx < m_nbJoints; idx++)
299 9 q_sot[m_urdf_to_sot[idx]] = q_urdf[idx];
300 3 return true;
301 }
302
303 3 bool RobotUtil::joints_sot_to_urdf(ConstRefVector q_sot, RefVector q_urdf) {
304
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
3 assert(q_urdf.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints));
305
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
3 assert(q_sot.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints));
306
307
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
3 if (m_nbJoints == 0) {
308 SEND_MSG("set_urdf_to_sot should be called", MSG_TYPE_ERROR);
309 return false;
310 }
311
312
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 3 times.
12 for (unsigned int idx = 0; idx < m_nbJoints; idx++)
313 9 q_urdf[idx] = q_sot[m_urdf_to_sot[idx]];
314 3 return true;
315 }
316
317 1 bool RobotUtil::velocity_urdf_to_sot(ConstRefVector q_urdf,
318 ConstRefVector v_urdf, RefVector v_sot) {
319
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 assert(q_urdf.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints + 7));
320
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 assert(v_urdf.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints + 6));
321
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 assert(v_sot.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints + 6));
322
323
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 if (m_nbJoints == 0) {
324 SEND_MSG("velocity_urdf_to_sot should be called", MSG_TYPE_ERROR);
325 return false;
326 }
327
5/10
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 1 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 1 times.
✗ Branch 11 not taken.
✓ Branch 13 taken 1 times.
✗ Branch 14 not taken.
1 const Eigen::Quaterniond q(q_urdf(6), q_urdf(3), q_urdf(4), q_urdf(5));
328
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 Eigen::Matrix3d oRb = q.toRotationMatrix();
329
4/8
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 1 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 1 times.
✗ Branch 11 not taken.
1 v_sot.head<3>() = oRb * v_urdf.head<3>();
330
4/8
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 1 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 1 times.
✗ Branch 11 not taken.
1 v_sot.segment<3>(3) = oRb * v_urdf.segment<3>(3);
331 // v_sot.head<6>() = v_urdf.head<6>();
332
5/10
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 1 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 1 times.
✗ Branch 11 not taken.
✓ Branch 13 taken 1 times.
✗ Branch 14 not taken.
1 joints_urdf_to_sot(v_urdf.tail(m_nbJoints), v_sot.tail(m_nbJoints));
333 1 return true;
334 }
335
336 1 bool RobotUtil::velocity_sot_to_urdf(ConstRefVector q_urdf,
337 ConstRefVector v_sot, RefVector v_urdf) {
338
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 assert(q_urdf.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints + 7));
339
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 assert(v_urdf.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints + 6));
340
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 assert(v_sot.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints + 6));
341
342
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 if (m_nbJoints == 0) {
343 SEND_MSG("velocity_sot_to_urdf should be called", MSG_TYPE_ERROR);
344 return false;
345 }
346 // compute rotation from world to base frame
347
5/10
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 1 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 1 times.
✗ Branch 11 not taken.
✓ Branch 13 taken 1 times.
✗ Branch 14 not taken.
1 const Eigen::Quaterniond q(q_urdf(6), q_urdf(3), q_urdf(4), q_urdf(5));
348
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 Eigen::Matrix3d oRb = q.toRotationMatrix();
349
5/10
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 1 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 1 times.
✗ Branch 11 not taken.
✓ Branch 13 taken 1 times.
✗ Branch 14 not taken.
1 v_urdf.head<3>() = oRb.transpose() * v_sot.head<3>();
350
5/10
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 1 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 1 times.
✗ Branch 11 not taken.
✓ Branch 13 taken 1 times.
✗ Branch 14 not taken.
1 v_urdf.segment<3>(3) = oRb.transpose() * v_sot.segment<3>(3);
351 // v_urdf.head<6>() = v_sot.head<6>();
352
5/10
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 1 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 1 times.
✗ Branch 11 not taken.
✓ Branch 13 taken 1 times.
✗ Branch 14 not taken.
1 joints_sot_to_urdf(v_sot.tail(m_nbJoints), v_urdf.tail(m_nbJoints));
353 1 return true;
354 }
355
356 2 bool RobotUtil::base_urdf_to_sot(ConstRefVector q_urdf, RefVector q_sot) {
357
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
2 assert(q_urdf.size() == 7);
358
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
2 assert(q_sot.size() == 6);
359
360 // ********* Quat to RPY *********
361
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
2 const double W = q_urdf[6];
362
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
2 const double X = q_urdf[3];
363
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
2 const double Y = q_urdf[4];
364
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
2 const double Z = q_urdf[5];
365
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 const Eigen::Matrix3d R = Eigen::Quaterniond(W, X, Y, Z).toRotationMatrix();
366
5/10
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 2 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 2 times.
✗ Branch 11 not taken.
✓ Branch 13 taken 2 times.
✗ Branch 14 not taken.
2 return base_se3_to_sot(q_urdf.head<3>(), R, q_sot);
367 }
368
369 2 bool RobotUtil::base_sot_to_urdf(ConstRefVector q_sot, RefVector q_urdf) {
370
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
2 assert(q_urdf.size() == 7);
371
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
2 assert(q_sot.size() == 6);
372 // ********* RPY to Quat *********
373
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
2 const double r = q_sot[3];
374
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
2 const double p = q_sot[4];
375
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
2 const double y = q_sot[5];
376
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 const Eigen::AngleAxisd rollAngle(r, Eigen::Vector3d::UnitX());
377
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 const Eigen::AngleAxisd pitchAngle(p, Eigen::Vector3d::UnitY());
378
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 const Eigen::AngleAxisd yawAngle(y, Eigen::Vector3d::UnitZ());
379
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 const Eigen::Quaternion<double> quat = yawAngle * pitchAngle * rollAngle;
380
381
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 q_urdf[0] = q_sot[0]; // BASE
382
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 q_urdf[1] = q_sot[1];
383
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 q_urdf[2] = q_sot[2];
384
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 q_urdf[3] = quat.x();
385
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 q_urdf[4] = quat.y();
386
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 q_urdf[5] = quat.z();
387
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 q_urdf[6] = quat.w();
388
389 2 return true;
390 }
391
392 1 bool RobotUtil::config_urdf_to_sot(ConstRefVector q_urdf, RefVector q_sot) {
393
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 assert(q_urdf.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints + 7));
394
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 assert(q_sot.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints + 6));
395
396
4/8
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 1 times.
✗ Branch 9 not taken.
✓ Branch 11 taken 1 times.
✗ Branch 12 not taken.
1 base_urdf_to_sot(q_urdf.head<7>(), q_sot.head<6>());
397
4/8
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 1 times.
✗ Branch 9 not taken.
✓ Branch 11 taken 1 times.
✗ Branch 12 not taken.
1 joints_urdf_to_sot(q_urdf.tail(m_nbJoints), q_sot.tail(m_nbJoints));
398
399 1 return true;
400 }
401
402 1 bool RobotUtil::config_sot_to_urdf(ConstRefVector q_sot, RefVector q_urdf) {
403
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 assert(q_urdf.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints + 7));
404
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 assert(q_sot.size() == static_cast<Eigen::VectorXd::Index>(m_nbJoints + 6));
405
4/8
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 1 times.
✗ Branch 9 not taken.
✓ Branch 11 taken 1 times.
✗ Branch 12 not taken.
1 base_sot_to_urdf(q_sot.head<6>(), q_urdf.head<7>());
406
4/8
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 1 times.
✗ Branch 9 not taken.
✓ Branch 11 taken 1 times.
✗ Branch 12 not taken.
1 joints_sot_to_urdf(q_sot.tail(m_nbJoints), q_urdf.tail(m_nbJoints));
407 1 return true;
408 }
409 1 void RobotUtil::display(std::ostream &os) const {
410 1 m_force_util.display(os);
411 1 m_foot_util.display(os);
412 1 m_hand_util.display(os);
413 1 os << "Nb of joints: " << m_nbJoints << std::endl;
414 1 os << "Urdf file name: " << m_urdf_filename << std::endl;
415
416 // Display m_urdf_to_sot
417 1 os << "Map from urdf index to the Sot Index " << std::endl;
418
2/2
✓ Branch 1 taken 3 times.
✓ Branch 2 taken 1 times.
4 for (unsigned int i = 0; i < m_urdf_to_sot.size(); i++)
419 3 os << "(" << i << " : " << m_urdf_to_sot[i] << ") ";
420 1 os << std::endl;
421
422 1 os << "Joint name to joint id:" << std::endl;
423 1 for (std::map<std::string, Index>::const_iterator it = m_name_to_id.begin();
424
2/2
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 1 times.
2 it != m_name_to_id.end(); ++it) {
425
5/10
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 1 times.
✗ Branch 9 not taken.
✓ Branch 12 taken 1 times.
✗ Branch 13 not taken.
✓ Branch 15 taken 1 times.
✗ Branch 16 not taken.
1 os << "(" << it->first << "," << it->second << ") ";
426 }
427 1 os << std::endl;
428
429 1 os << "Joint id to joint Name:" << std::endl;
430 1 for (std::map<Index, std::string>::const_iterator it = m_id_to_name.begin();
431
2/2
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 1 times.
2 it != m_id_to_name.end(); ++it) {
432
5/10
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 1 times.
✗ Branch 9 not taken.
✓ Branch 12 taken 1 times.
✗ Branch 13 not taken.
✓ Branch 15 taken 1 times.
✗ Branch 16 not taken.
1 os << "(" << it->first << "," << it->second << ") ";
433 }
434 1 os << std::endl;
435
436
2/4
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
1 boost::property_tree::write_xml(os, property_tree_);
437 1 }
438
439 1 void RobotUtil::sendMsg(const std::string &msg, MsgType t,
440 const std::string &lineId) {
441
4/8
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 1 times.
✗ Branch 9 not taken.
✓ Branch 11 taken 1 times.
✗ Branch 12 not taken.
1 logger_.stream(t, lineId) << "[RobotUtil]" << msg << '\n';
442 1 }
443
444 2 bool base_se3_to_sot(ConstRefVector pos, ConstRefMatrix R, RefVector q_sot) {
445
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
2 assert(q_sot.size() == 6);
446
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
2 assert(pos.size() == 3);
447
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
2 assert(R.rows() == 3);
448
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
2 assert(R.cols() == 3);
449 // ********* Quat to RPY *********
450 double r, p, y, m;
451 2 m = sqrt(R(2, 1) * R(2, 1) + R(2, 2) * R(2, 2));
452 2 p = atan2(-R(2, 0), m);
453
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 if (fabs(fabs(p) - M_PI / 2) < 0.001) {
454 r = 0.0;
455 y = -atan2(R(0, 1), R(1, 1));
456 } else {
457 2 y = atan2(R(1, 0), R(0, 0));
458 2 r = atan2(R(2, 1), R(2, 2));
459 }
460 // *********************************
461 2 q_sot[0] = pos[0];
462 2 q_sot[1] = pos[1];
463 2 q_sot[2] = pos[2];
464 2 q_sot[3] = r;
465 2 q_sot[4] = p;
466 2 q_sot[5] = y;
467 2 return true;
468 }
469
470 bool base_urdf_to_sot(ConstRefVector q_urdf, RefVector q_sot) {
471 assert(q_urdf.size() == 7);
472 assert(q_sot.size() == 6);
473 // ********* Quat to RPY *********
474 const double W = q_urdf[6];
475 const double X = q_urdf[3];
476 const double Y = q_urdf[4];
477 const double Z = q_urdf[5];
478 const Eigen::Matrix3d R = Eigen::Quaterniond(W, X, Y, Z).toRotationMatrix();
479 return base_se3_to_sot(q_urdf.head<3>(), R, q_sot);
480 }
481
482 bool base_sot_to_urdf(ConstRefVector q_sot, RefVector q_urdf) {
483 assert(q_urdf.size() == 7);
484 assert(q_sot.size() == 6);
485 // ********* RPY to Quat *********
486 const double r = q_sot[3];
487 const double p = q_sot[4];
488 const double y = q_sot[5];
489 const Eigen::AngleAxisd rollAngle(r, Eigen::Vector3d::UnitX());
490 const Eigen::AngleAxisd pitchAngle(p, Eigen::Vector3d::UnitY());
491 const Eigen::AngleAxisd yawAngle(y, Eigen::Vector3d::UnitZ());
492 const Eigen::Quaternion<double> quat = yawAngle * pitchAngle * rollAngle;
493
494 q_urdf[0] = q_sot[0]; // BASE
495 q_urdf[1] = q_sot[1];
496 q_urdf[2] = q_sot[2];
497 q_urdf[3] = quat.x();
498 q_urdf[4] = quat.y();
499 q_urdf[5] = quat.z();
500 q_urdf[6] = quat.w();
501
502 return true;
503 }
504
505 static std::map<std::string, RobotUtilShrPtr> sgl_map_name_to_robot_util;
506
507 std::shared_ptr<std::vector<std::string> > getListOfRobots() {
508 std::shared_ptr<std::vector<std::string> > res =
509 std::make_shared<std::vector<std::string> >();
510
511 std::map<std::string, RobotUtilShrPtr>::iterator it =
512 sgl_map_name_to_robot_util.begin();
513 while (it != sgl_map_name_to_robot_util.end()) {
514 res->push_back(it->first);
515 it++;
516 }
517
518 return res;
519 }
520
521 RobotUtilShrPtr getRobotUtil(std::string &robotName) {
522 std::map<std::string, RobotUtilShrPtr>::iterator it =
523 sgl_map_name_to_robot_util.find(robotName);
524 if (it != sgl_map_name_to_robot_util.end()) return it->second;
525 return RefVoidRobotUtil();
526 }
527
528 1 bool isNameInRobotUtil(std::string &robotName) {
529 std::map<std::string, RobotUtilShrPtr>::iterator it =
530
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 sgl_map_name_to_robot_util.find(robotName);
531
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
1 if (it != sgl_map_name_to_robot_util.end()) return true;
532 1 return false;
533 }
534
535 2 RobotUtilShrPtr createRobotUtil(std::string &robotName) {
536 std::map<std::string, RobotUtilShrPtr>::iterator it =
537
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
2 sgl_map_name_to_robot_util.find(robotName);
538
1/2
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
2 if (it == sgl_map_name_to_robot_util.end()) {
539
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 sgl_map_name_to_robot_util[robotName] = std::make_shared<RobotUtil>();
540
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
2 it = sgl_map_name_to_robot_util.find(robotName);
541 2 return it->second;
542 }
543 return RefVoidRobotUtil();
544 }
545 } // namespace sot
546 } // namespace dynamicgraph
547