Files
yq_2026_duo/src/rm_utils/README.md
cyy_mac 5e1e355ffa bf
2026-03-25 12:37:08 +08:00

3.0 KiB
Raw Blame History

rm_utils

1. 介绍

该模块提供通用的工具类以及函数,包括:

  • 扩展卡尔曼滤波器 ExtendedKalmanFilter
  • PnP解算器 PnPSolver
  • 弹道补偿器 TrajectoryCompensator
  • FYT日志库
  • URL路径解析器 URLResolver
  • 心跳发布者 HeartBeatPublisher

2. 使用方法

2.0 扩展卡尔曼滤波

见armor_solver/src/armor_solver_node.cpp

2.1 PnP解算

示例:

#include "rm_utils/math/pnp_solver.hpp"

// 1. 创建PnP解算器
// 构造函数参数(cv::Mat, cv::Mat, cv::PnPMethod)相机内参矩阵、畸变系数、解算方法默认是IPPE
auto pnp_solver = PnPSolver(camera_matrix, distortion_coefficients);
// 2. 为PnPSolver设置一个物体坐标系
std::vector<cv::Point3f> object_points = {
    cv::Point3f(-1.0f, 1.0f, 0.0f),
    cv::Point3f(1.0f, 1.0f, 0.0f),
    cv::Point3f(1.0f, -1.0f, 0.0f),
    cv::Point3f(-1.0f, -1.0f, 0.0f)
};
pnp_solver.setObjectPoints(object_points, "object_frame");
// 3. PnP解算
cv::Mat rvec, tvec;
pnp_solver.solvePnP(image_points, rvec, tvec, "object_frame");

// 对于打符来说因为以前的PnP是直接返回一个表示位姿的Eigen::VectorXd所以这里也提供了一个函数用于将rvec和tvec转换为VectorXd
// 4. 获取位姿
Eigen::VectorXd pose = pnp_solver.getPose(rvec, tvec);

2.2 弹道补偿

示例:

#include "rm_utils/math/trajectory_compensator.hpp"

auto compensator = CompensatorFactory::createCompensator("ideal");
double pitch = 0;
Eigen::Vector3d p = Eigen::Vector3d(1.0, 0, 0);
double temp_pitch = pitch;
if (trajectory_compensator_->compensate(p, temp_pitch)) {
    pitch = temp_pitch;
}

2.3 Eigen和cv::Mat的相互转换

示例:

#include "rm_utils/math/utils.hpp"

cv::Mat mat = cv::Mat::eye(3, 3, CV_32FC1);
Eigen::Matrix3f eigen_mat = utils::cvToEigen(mat);
cv::Mat mat2 = utils::eigenToCv(eigen_mat);

2.4 FYT日志库

示例:

#include "rm_utils/logger/log.hpp"

// 1. 初始化
// 参数:日志名称、日志文件路径、日志级别
FYT_REGISTER_LOGGER("test_logger", "/tmp/test_logger.log", INFO);

// 2. 使用
FYT_INFO("test_logger", "This is a test log");
int a = 1;
FYT_WARN("test_logger", "a = {}", a);

2.5 URL Resolver

能够用很方便的方式类camera_info_url访问文件和或ros2包的install目录

示例:

#include "rm_utils/url_resolver.hpp"

namespace fs = std::filesystem;
fs::path model_path =
  utils::URLResolver::getResolvedPath("package://armor_detector/model/lenet.onnx");

if (fs::exists(model_path)) {
    std::cout<<model_path.string()<<std::endl;
}

fs::path file_path = utils::URLResolver::getResolvedPath("file://home/zcf/123.txt");

2.6 HearBeatPublisher

定时发布心跳数据

示例:

#include "rm_utils/heartbeat.hpp"

class TestNode : public rclcp::Node {
public:
  TestNode(const rclcpp::NodeOptions& options) : Node("test", opitions) {
    heartbeat_ = HeartBeatPublisher::create(this); // 传入Node*
  }
private:
  HeartBeatPublisher::SharedPtr heartbeat_;
};