Sucan126
dbbe63c7b1
添加对运动切换命令(CheckMode/SelectMode/ReleaseMode)的处理支持 在Controller类中新增多种运动控制功能接口 更新README文档以包含新增的运动切换命令和功能说明
58 lines
1.9 KiB
C++
58 lines
1.9 KiB
C++
#pragma once
|
|
|
|
#include "controller.hpp"
|
|
#include "config.hpp"
|
|
#include "logger.hpp"
|
|
#include "mqtt.hpp"
|
|
#include "navigation.hpp"
|
|
|
|
#include <memory>
|
|
#include <atomic>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <nlohmann/json.hpp>
|
|
#include <unitree/robot/go2/robot_state/robot_state_client.hpp>
|
|
|
|
namespace custom {
|
|
|
|
class CustomRobot {
|
|
public:
|
|
explicit CustomRobot();
|
|
~CustomRobot();
|
|
|
|
bool initialize();
|
|
bool start();
|
|
|
|
Controller* getController() const { return controller_.get(); }
|
|
Navigation* getNavigation() const { return navigation_.get(); }
|
|
|
|
bool GetServiceList(std::vector<unitree::robot::go2::ServiceState>& serviceList);
|
|
bool SwitchService(const std::string& serviceName, bool enable);
|
|
bool SetReportFreq(int32_t interval, int32_t duration);
|
|
|
|
bool initializeMqtt();
|
|
void onMqttMessage(const std::string& topic, const std::string& payload);
|
|
void onMqttConnection(bool connected);
|
|
|
|
void processCmd(const nlohmann::json& message);
|
|
bool processOacCmd(const std::string& cmd, const nlohmann::json& message);
|
|
bool processSportCmd(const std::string& cmd, const nlohmann::json& message);
|
|
bool processRscCmd(const std::string& cmd, const nlohmann::json& message);
|
|
bool processNavCmd(const std::string& cmd, const nlohmann::json& message);
|
|
bool processMscCmd(const std::string& cmd, const nlohmann::json& message);
|
|
void printServiceList(const std::vector<unitree::robot::go2::ServiceState>& serviceList, int filterStatus = -1);
|
|
|
|
private:
|
|
std::string generateRandomClientId() const;
|
|
std::unique_ptr<Controller> controller_;
|
|
std::unique_ptr<Navigation> navigation_;
|
|
std::unique_ptr<unitree::robot::go2::RobotStateClient> rsc_;
|
|
std::unique_ptr<MqttClient> mqttClient_;
|
|
|
|
CustomConfig config_;
|
|
std::atomic<bool> running_;
|
|
std::atomic<bool> initialized_;
|
|
};
|
|
|
|
} // namespace custom
|