GCC Code Coverage Report


Directory: ./
File: examples/casadi/casadi-rnea.cpp
Date: 2025-02-12 21:03:38
Exec Total Coverage
Lines: 39 39 100.0%
Branches: 84 166 50.6%

Line Branch Exec Source
1 #include "pinocchio/autodiff/casadi.hpp"
2
3 #include "pinocchio/multibody/sample-models.hpp"
4
5 #include "pinocchio/algorithm/rnea.hpp"
6 #include "pinocchio/algorithm/joint-configuration.hpp"
7
8 1 int main(int /*argc*/, char ** /*argv*/)
9 {
10 using namespace pinocchio;
11
12 typedef double Scalar;
13 typedef ::casadi::SX ADScalar;
14
15 typedef ModelTpl<Scalar> Model;
16 typedef Model::Data Data;
17
18 typedef ModelTpl<ADScalar> ADModel;
19 typedef ADModel::Data ADData;
20
21 // Create a random humanoid model
22
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 Model model;
23
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 buildModels::humanoidRandom(model);
24
2/4
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
1 model.lowerPositionLimit.head<3>().fill(-1.);
25
2/4
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
1 model.upperPositionLimit.head<3>().fill(1.);
26
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 Data data(model);
27
28 // Pick up random configuration, velocity and acceleration vectors.
29
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 Eigen::VectorXd q(model.nq);
30
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 q = randomConfiguration(model);
31
2/4
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
1 Eigen::VectorXd v(Eigen::VectorXd::Random(model.nv));
32
2/4
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
1 Eigen::VectorXd a(Eigen::VectorXd::Random(model.nv));
33
34 // Create CasADi model and data from model
35 typedef ADModel::ConfigVectorType ConfigVectorAD;
36 typedef ADModel::TangentVectorType TangentVectorAD;
37
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 ADModel ad_model = model.cast<ADScalar>();
38
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 ADData ad_data(ad_model);
39
40 // Create symbolic CasADi vectors
41
2/4
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
2 ::casadi::SX cs_q = ::casadi::SX::sym("q", model.nq);
42
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 ConfigVectorAD q_ad(model.nq);
43
4/8
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 8 taken 1 times.
✗ Branch 9 not taken.
✓ Branch 11 taken 1 times.
✗ Branch 12 not taken.
1 q_ad = Eigen::Map<ConfigVectorAD>(static_cast<std::vector<ADScalar>>(cs_q).data(), model.nq, 1);
44
45
2/4
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
2 ::casadi::SX cs_v = ::casadi::SX::sym("v", model.nv);
46
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 TangentVectorAD v_ad(model.nv);
47
4/8
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 8 taken 1 times.
✗ Branch 9 not taken.
✓ Branch 11 taken 1 times.
✗ Branch 12 not taken.
1 v_ad = Eigen::Map<TangentVectorAD>(static_cast<std::vector<ADScalar>>(cs_v).data(), model.nv, 1);
48
49
2/4
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
2 ::casadi::SX cs_a = ::casadi::SX::sym("a", model.nv);
50
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 TangentVectorAD a_ad(model.nv);
51
4/8
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
✓ Branch 8 taken 1 times.
✗ Branch 9 not taken.
✓ Branch 11 taken 1 times.
✗ Branch 12 not taken.
1 a_ad = Eigen::Map<TangentVectorAD>(static_cast<std::vector<ADScalar>>(cs_a).data(), model.nv, 1);
52
53 // Build CasADi function
54
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 rnea(ad_model, ad_data, q_ad, v_ad, a_ad);
55
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 ::casadi::SX tau_ad(model.nv, 1);
56
57
2/2
✓ Branch 0 taken 32 times.
✓ Branch 1 taken 1 times.
33 for (Eigen::DenseIndex k = 0; k < model.nv; ++k)
58
3/6
✓ Branch 1 taken 32 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 32 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 32 times.
✗ Branch 8 not taken.
32 tau_ad(k) = ad_data.tau[k];
59
60 ::casadi::Function eval_rnea(
61
8/16
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 6 taken 1 times.
✗ Branch 7 not taken.
✓ Branch 9 taken 1 times.
✗ Branch 10 not taken.
✓ Branch 12 taken 1 times.
✗ Branch 13 not taken.
✓ Branch 15 taken 1 times.
✗ Branch 16 not taken.
✓ Branch 19 taken 1 times.
✗ Branch 20 not taken.
✓ Branch 23 taken 1 times.
✗ Branch 24 not taken.
✓ Branch 26 taken 1 times.
✗ Branch 27 not taken.
8 "eval_rnea", ::casadi::SXVector{cs_q, cs_v, cs_a}, ::casadi::SXVector{tau_ad});
62
63 // Evaluate CasADi expression with real value
64
1/2
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
1 std::vector<double> q_vec((size_t)model.nq);
65
3/6
✓ 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.
1 Eigen::Map<Eigen::VectorXd>(q_vec.data(), model.nq, 1) = q;
66
67
1/2
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
1 std::vector<double> v_vec((size_t)model.nv);
68
3/6
✓ 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.
1 Eigen::Map<Eigen::VectorXd>(v_vec.data(), model.nv, 1) = v;
69
70
1/2
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
1 std::vector<double> a_vec((size_t)model.nv);
71
3/6
✓ 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.
1 Eigen::Map<Eigen::VectorXd>(a_vec.data(), model.nv, 1) = a;
72
73
6/12
✓ 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 11 taken 1 times.
✗ Branch 12 not taken.
✓ Branch 14 taken 1 times.
✗ Branch 15 not taken.
✓ Branch 18 taken 1 times.
✗ Branch 19 not taken.
5 ::casadi::DM tau_casadi_res = eval_rnea(::casadi::DMVector{q_vec, v_vec, a_vec})[0];
74
2/4
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
3 Data::TangentVectorType tau_casadi_vec = Eigen::Map<Data::TangentVectorType>(
75
2/4
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
2 static_cast<std::vector<double>>(tau_casadi_res).data(), model.nv, 1);
76
77 // Eval RNEA using classic Pinocchio model
78
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
1 pinocchio::rnea(model, data, q, v, a);
79
80 // Print both results
81
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 std::cout << "pinocchio double:\n" << "\ttau = " << data.tau.transpose() << std::endl;
82
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 std::cout << "pinocchio CasADi:\n" << "\ttau = " << tau_casadi_vec.transpose() << std::endl;
83 1 }
84