|
| 1 | +#include <Eigen/Dense> |
| 2 | +#include <fstream> |
| 3 | +#include <iostream> |
| 4 | +#include <sdu_controllers/controllers/operational_space_controller.hpp> |
| 5 | +#include <sdu_controllers/math/forward_dynamics.hpp> |
| 6 | +#include <sdu_controllers/math/inverse_dynamics_joint_space.hpp> |
| 7 | +#include <sdu_controllers/kinematics/forward_kinematics.hpp> |
| 8 | +#include <sdu_controllers/models/parameter_robot_model.hpp> |
| 9 | +#include <sdu_controllers/safety/safety_verifier.hpp> |
| 10 | +#include <sdu_controllers/utils/utility.hpp> |
| 11 | + |
| 12 | +using namespace csv; |
| 13 | +using namespace Eigen; |
| 14 | +using namespace sdu_controllers; |
| 15 | +using namespace sdu_controllers::utils; |
| 16 | + |
| 17 | +int main() |
| 18 | +{ |
| 19 | + // Setup writing of output trajectory to csv. |
| 20 | + std::ofstream output_filestream; |
| 21 | + output_filestream.open("output_cartesian.csv"); |
| 22 | + auto csv_writer = make_csv_writer(output_filestream); |
| 23 | + |
| 24 | + // Initialize robot model and parameters |
| 25 | + auto robot_model = std::make_shared<models::ParameterRobotModel>(utils::ConfigFolder::find_config_file("breeding_blanket_handling_robot_fixed.yaml")); |
| 26 | + |
| 27 | + double freq = 1000.0; |
| 28 | + double dt = 1.0 / freq; |
| 29 | + |
| 30 | + double Kp_pos_value = 10.0; |
| 31 | + double Kp_orient_value = 1; |
| 32 | + double Kd_pos_value = 20.0; |
| 33 | + double Kd_orient_value = 2; |
| 34 | + double N_value = 0; |
| 35 | + uint16_t ROBOT_DOF = robot_model->get_dof(); |
| 36 | + |
| 37 | + VectorXd Kp_pos_vec = VectorXd::Ones(3) * Kp_pos_value; |
| 38 | + VectorXd Kp_orient_vec = VectorXd::Ones(3) * Kp_orient_value; |
| 39 | + VectorXd Kd_pos_vec = VectorXd::Ones(3) * Kd_pos_value; |
| 40 | + VectorXd Kd_orient_vec = VectorXd::Ones(3) * Kd_orient_value; |
| 41 | + VectorXd N_vec = VectorXd::Ones(6) * N_value; |
| 42 | + |
| 43 | + MatrixXd Kp = MatrixXd::Zero(6, 6); |
| 44 | + Kp.setIdentity(); |
| 45 | + Kp.block<3, 3>(0,0) = Kp_pos_vec.asDiagonal(); |
| 46 | + Kp.block<3, 3>(3,3) = Kp_orient_vec.asDiagonal(); |
| 47 | + MatrixXd Kd = MatrixXd::Zero(6, 6); |
| 48 | + Kd.setIdentity(); |
| 49 | + Kd.block<3, 3>(0,0) = Kd_pos_vec.asDiagonal(); |
| 50 | + Kd.block<3, 3>(3,3) = Kd_orient_vec.asDiagonal(); |
| 51 | + |
| 52 | + controllers::OperationalSpaceController osc_controller(Kp, Kd, robot_model); |
| 53 | + osc_controller.set_kappa(1e-2); |
| 54 | + //math::InverseDynamicsJointSpace inv_dyn_jnt_space(robot_model); |
| 55 | + //math::ForwardDynamics fwd_dyn(robot_model); |
| 56 | + |
| 57 | + VectorXd x_d(6); |
| 58 | + VectorXd dx_d(6); |
| 59 | + VectorXd ddx_d(6); |
| 60 | + |
| 61 | + VectorXd q(ROBOT_DOF), q0(ROBOT_DOF); |
| 62 | + VectorXd dq(ROBOT_DOF); |
| 63 | + Vector<double, 6> he = VectorXd::Zero(6); |
| 64 | + q << 0.0, 0.0, 0.0, 0.0, 0.0, 0.0; |
| 65 | + q0 << 0.0, 0.0, 0.0, 0.0, 0.0, 0.0; |
| 66 | + dq << 0.0, 0.0, 0.0, 0.0, 0.0, 0.0; |
| 67 | + |
| 68 | + // output to csv |
| 69 | + output_filestream << "q0,q1,q2,q3,q4,q5,x,y,z,rx,ry,rz" << std::endl;; |
| 70 | + |
| 71 | + // Stay put at the initial position |
| 72 | + // Control loop for 10s |
| 73 | + for (size_t j=0; j < freq * 20.; j++) // (const std::vector<double>& trajectory_point : input_trajectory) |
| 74 | + { |
| 75 | + if (j % int(freq) == 0) |
| 76 | + std::cout << "Time: " << j * dt << std::endl; |
| 77 | + |
| 78 | + // Desired |
| 79 | + Eigen::Matrix4d T0 = robot_model->get_fk_solver().forward_kinematics(q0); |
| 80 | + VectorXd pos0 = T0.block<3, 1>(0, 3); |
| 81 | + Matrix3d rot_mat0 = T0.block<3,3>(0, 0); |
| 82 | + Vector3d rpy_zyz0 = rot_mat0.eulerAngles(2, 1, 2); // ZYZ representation |
| 83 | + |
| 84 | + Vector3d pos_desired = pos0; |
| 85 | + pos_desired[0] += 0.1; |
| 86 | + pos_desired[1] += -0.1; |
| 87 | + pos_desired[2] += 2; |
| 88 | + |
| 89 | + Vector3d rpy_zyz_desired = rpy_zyz0; |
| 90 | + rpy_zyz_desired[0] += M_PI / 8; |
| 91 | + rpy_zyz_desired[1] += 0; |
| 92 | + rpy_zyz_desired[2] += M_PI / 2; |
| 93 | + |
| 94 | + x_d << pos_desired, rpy_zyz_desired; |
| 95 | + dx_d.setZero(); |
| 96 | + ddx_d.setZero(); |
| 97 | + |
| 98 | + // std::cout << "x_d: " << x_d << std::endl; |
| 99 | + |
| 100 | + // Add noise to q and dq |
| 101 | + VectorXd q_meas = q; |
| 102 | + VectorXd dq_meas = dq; |
| 103 | + //add_noise_to_vector(q_meas, 0.0, 0.001); |
| 104 | + //add_noise_to_vector(dq_meas, 0.0, 0.001); |
| 105 | + |
| 106 | + // std::cout << "before all fk" << std::endl; |
| 107 | + // std::vector<Eigen::Matrix4d> TTT = robot_model->get_fk_solver().forward_kinematics_all(q0); |
| 108 | + // std::cout << "after all fk" << std::endl; |
| 109 | + // std::cout << TTT.at(4) << std::endl; |
| 110 | + |
| 111 | + // Controller |
| 112 | + osc_controller.step(x_d, dx_d, ddx_d, q_meas, dq_meas); |
| 113 | + VectorXd y = osc_controller.get_output(); |
| 114 | + // std::cout << "y: " << y << std::endl; |
| 115 | + VectorXd tau = robot_model->inverse_dynamics(q_meas, dq_meas, y, he); |
| 116 | + // std::cout << "tau: " << tau << std::endl; |
| 117 | + |
| 118 | + // Simulation |
| 119 | + VectorXd ddq = robot_model->forward_dynamics(q, dq, tau); |
| 120 | + // integrate to get velocity |
| 121 | + dq += ddq * dt; |
| 122 | + // integrate to get position |
| 123 | + q += dq * dt; |
| 124 | + |
| 125 | + // std::cout << "q:" << q << std::endl; |
| 126 | + MatrixXd T = robot_model->get_fk_solver().forward_kinematics(q); |
| 127 | + VectorXd pos = T.block<3, 1>(0, 3); |
| 128 | + // std::cout << "pos:" << pos << std::endl; |
| 129 | + Matrix3d rot_mat = T.block<3,3>(0, 0); |
| 130 | + Vector3d rpy_zyz = rot_mat.eulerAngles(2, 1, 2); // ZYZ representation |
| 131 | + // std::cout << "rpy_zyz:" << rpy_zyz << std::endl; |
| 132 | + VectorXd temp(q.size()+pos.size()+rpy_zyz.size()); |
| 133 | + temp << q, pos, rpy_zyz; |
| 134 | + csv_writer << eigen_to_std_vector(temp); |
| 135 | + } |
| 136 | + output_filestream.close(); |
| 137 | +} |
0 commit comments