z1_sdk/examples/highcmd_basic.cpp

110 lines
2.7 KiB
C++

#include "unitree_arm_sdk/control/unitreeArm.h"
using namespace UNITREE_ARM;
class Z1ARM : public unitreeArm{
public:
Z1ARM():unitreeArm(true){};
~Z1ARM(){};
void armCtrlByFSM();
void armCtrlInJointCtrl();
void armCtrlInCartesian();
void printState();
double gripper_pos = 0.0;
double joint_speed = 2.0;
double cartesian_speed = 0.5;
};
void Z1ARM::armCtrlByFSM() {
Vec6 posture[2];
gripper_pos = 0.0;
std::cout << "[TO STATE]" << std::endl;
labelRun("forward");
std::cout << "[MOVEJ]" << std::endl;
posture[0]<<0.5,0.1,0.1,0.5,-0.2,0.5;
joint_speed = 2.0;
MoveJ(posture[0], gripper_pos, joint_speed);
std::cout << "[MOVEL]" << std::endl;
posture[0] << 0,0,0,0.45,-0.2,0.2;
cartesian_speed = 0.5;
MoveL(posture[0], gripper_pos, cartesian_speed);
std::cout << "[MOVEC]" << std::endl;
posture[0] << 0,0,0,0.45,0,0.4;
posture[1] << 0,0,0,0.45,0.2,0.2;
MoveC(posture[0], posture[1], gripper_pos, cartesian_speed);
}
void Z1ARM::armCtrlInJointCtrl(){
labelRun("forward");
startTrack(ArmFSMState::JOINTCTRL);
Timer timer(_ctrlComp->dt);
for(int i(0); i<1000;i++){
directions<< 0, 0, 0, -1, 0, 0, -1;
joint_speed = 1.0;
jointCtrlCmd(directions, joint_speed);
timer.sleep();
}
}
void Z1ARM::armCtrlInCartesian(){
labelRun("forward");
startTrack(ArmFSMState::CARTESIAN);
double angular_vel = 0.3;
double linear_vel = 0.3;
Timer timer(_ctrlComp->dt);
for(int i(0); i<2000;i++){
directions<< 0, 0, 0, 0, 0, -1, -1;
cartesianCtrlCmd(directions, angular_vel, linear_vel);
timer.sleep();
}
}
void Z1ARM::printState(){
std::cout<<"------ joint State ------"<<std::endl;
std::cout<<"qState: "<<lowstate->getQ().transpose()<<std::endl;
std::cout<<"qdState: "<<lowstate->getQd().transpose()<<std::endl;
std::cout<<"tauState: "<<lowstate->getTau().transpose()<<std::endl;
std::cout<<"------ Endeffector Cartesian Posture ------"<<std::endl;
std::cout<<"roll pitch yaw x y z"<<std::endl;
std::cout<<lowstate->endPosture.transpose()<<std::endl;
}
int main() {
std::cout << std::fixed << std::setprecision(3);
Z1ARM arm;
arm.sendRecvThread->start();
arm.backToStart();
size_t demo = 3;
for(size_t demo = 1; demo < 4; demo++)
switch (demo)
{
case 1:
arm.armCtrlByFSM();
break;
case 2:
arm.armCtrlInJointCtrl();
break;
case 3:
arm.armCtrlInCartesian();
break;
default:
break;
}
arm.backToStart();
arm.setFsm(ArmFSMState::PASSIVE);
arm.sendRecvThread->shutdown();
return 0;
}