feat(agv_pro_bringup): add multi-lidar support and drivers

This commit is contained in:
X-lanni
2026-01-13 19:12:09 +08:00
parent b50a0c4496
commit dcb52622eb
99 changed files with 26335 additions and 39 deletions
+78
View File
@@ -0,0 +1,78 @@
cmake_minimum_required(VERSION 3.5)
project(unitree_lidar_ros2)
# Default to C99
if(NOT CMAKE_C_STANDARD)
set(CMAKE_C_STANDARD 99)
endif()
# Default to C++14
if(NOT CMAKE_CXX_STANDARD)
set(CMAKE_CXX_STANDARD 17)
endif()
if(CMAKE_COMPILER_IS_GNUCXX OR CMAKE_CXX_COMPILER_ID MATCHES "Clang")
add_compile_options(-Wall -Wextra -Wpedantic)
endif()
find_package(ament_cmake REQUIRED)
find_package(rclcpp REQUIRED)
find_package(std_msgs REQUIRED)
find_package(geometry_msgs REQUIRED)
find_package(pcl_conversions REQUIRED)
find_package(PCL REQUIRED)
find_package(tf2_ros REQUIRED)
find_package(sensor_msgs REQUIRED)
include_directories(
${PCL_INCLUDE_DIRS}
include
../unitree_lidar_sdk/include
)
link_directories(
${PCL_LIBRARY_DIRS}
../unitree_lidar_sdk/lib/${CMAKE_SYSTEM_PROCESSOR}
)
add_definitions(${PCL_DEFINITIONS})
add_executable(unitree_lidar_ros2_node src/unitree_lidar_ros2_node.cpp)
target_link_libraries( unitree_lidar_ros2_node
${Boost_SYSTEM_LIBRARY}
${PCL_LIBRARIES}
unilidar_sdk2
)
ament_target_dependencies(
unitree_lidar_ros2_node
rclcpp std_msgs
sensor_msgs
geometry_msgs
tf2_ros
pcl_conversions
)
install(TARGETS
unitree_lidar_ros2_node
DESTINATION lib/${PROJECT_NAME}
)
# install rviz file and launch file
install(DIRECTORY launch/
DESTINATION share/${PROJECT_NAME}/launch
)
install(DIRECTORY rviz/
DESTINATION share/${PROJECT_NAME}/rviz
)
# install config files if they exist
if(EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/config")
install(DIRECTORY config/
DESTINATION share/${PROJECT_NAME}/config
)
endif()
ament_package()
+1
View File
@@ -0,0 +1 @@
This is a ROS2 package for Unitree Lidar L2.
@@ -0,0 +1,230 @@
/**********************************************************************
Copyright (c) 2020-2024, Unitree Robotics.Co.Ltd. All rights reserved.
***********************************************************************/
#pragma once
#define BOOST_BIND_NO_PLACEHOLDERS
#include <memory>
#include "iostream"
#include <string>
#include <stdio.h>
#include <iterator>
#include <algorithm>
#include <chrono>
#include <pcl/io/pcd_io.h>
#include <pcl/point_types.h>
#include <pcl_conversions/pcl_conversions.h>
#include "sensor_msgs/msg/point_cloud2.hpp"
#include "sensor_msgs/msg/imu.hpp"
#include "std_msgs/msg/header.hpp"
#include "rclcpp/rclcpp.hpp"
#include "rclcpp/time.hpp"
#include "tf2_ros/transform_broadcaster.h"
#include "tf2/LinearMath/Quaternion.h"
#include "geometry_msgs/msg/transform_stamped.hpp"
// SDK
#include "unitree_lidar_sdk_pcl.h"
using std::placeholders::_1;
class UnitreeLidarSDKNode : public rclcpp::Node
{
public:
explicit UnitreeLidarSDKNode(const rclcpp::NodeOptions &options = rclcpp::NodeOptions());
~UnitreeLidarSDKNode() {};
void timer_callback();
protected:
// ROS
rclcpp::Publisher<sensor_msgs::msg::PointCloud2>::SharedPtr pub_cloud_;
rclcpp::Publisher<sensor_msgs::msg::Imu>::SharedPtr pub_imu_;
rclcpp::TimerBase::SharedPtr timer_;
std::shared_ptr<tf2_ros::TransformBroadcaster> broadcaster_;
// Unitree Lidar Reader
UnitreeLidarReader *lsdk_;
// Config params
int work_mode_;
int initialize_type_;
int local_port_;
std::string local_ip_;
int lidar_port_;
std::string lidar_ip_;
std::string serial_port_;
int baudrate_;
int cloud_scan_num_;
bool use_system_timestamp_;
double range_min_;
double range_max_;
std::string cloud_frame_;
std::string cloud_topic_;
std::string imu_frame_;
std::string imu_topic_;
};
///////////////////////////////////////////////////////////////////
UnitreeLidarSDKNode::UnitreeLidarSDKNode(const rclcpp::NodeOptions &options)
: Node("unitre_lidar_sdk_node", options)
{
// load config parameters
declare_parameter<int>("initialize_type", 2);
declare_parameter<int>("work_mode", 0);
declare_parameter<bool>("use_system_timestamp", true);
declare_parameter<double>("range_min", 0);
declare_parameter<double>("range_max", 50);
declare_parameter<int>("cloud_scan_num", 18);
declare_parameter<std::string>("serial_port", "/dev/ttyACM0");
declare_parameter<int>("baudrate", 4000000);
declare_parameter<int>("lidar_port", 6101);
declare_parameter<std::string>("lidar_ip", "192.168.1.2");
declare_parameter<int>("local_port", 6201);
declare_parameter<std::string>("local_ip", "192.168.1.62");
declare_parameter<std::string>("cloud_frame", "unilidar_lidar");
declare_parameter<std::string>("cloud_topic", "unilidar/cloud");
declare_parameter<std::string>("imu_frame", "unilidar_imu");
declare_parameter<std::string>("imu_topic", "unilidar/imu");
work_mode_ = get_parameter("work_mode").as_int();
initialize_type_ = get_parameter("initialize_type").as_int();
serial_port_ = get_parameter("serial_port").as_string();
baudrate_ = get_parameter("baudrate").as_int();
lidar_port_ = get_parameter("lidar_port").as_int();
lidar_ip_ = get_parameter("lidar_ip").as_string();
local_port_ = get_parameter("local_port").as_int();
local_ip_ = get_parameter("local_ip").as_string();
cloud_scan_num_ = get_parameter("cloud_scan_num").as_int();
use_system_timestamp_ = get_parameter("use_system_timestamp").as_bool();
range_max_ = get_parameter("range_max").as_double();
range_min_ = get_parameter("range_min").as_double();
cloud_frame_ = get_parameter("cloud_frame").as_string();
cloud_topic_ = get_parameter("cloud_topic").as_string();
imu_frame_ = get_parameter("imu_frame").as_string();
imu_topic_ = get_parameter("imu_topic").as_string();
// Initialize UnitreeLidarReader
lsdk_ = createUnitreeLidarReader();
std::cout << "initialize_type_ = " << initialize_type_ << std::endl;
if (initialize_type_ == 1)
{
lsdk_->initializeSerial(serial_port_, baudrate_,
cloud_scan_num_, use_system_timestamp_, range_min_, range_max_);
}
else if (initialize_type_ == 2)
{
lsdk_->initializeUDP(lidar_port_, lidar_ip_, local_port_, local_ip_,
cloud_scan_num_, use_system_timestamp_, range_min_, range_max_);
}
else
{
std::cout << "initialize_type is not right! exit now ...\n";
exit(0);
}
lsdk_->setLidarWorkMode(work_mode_);
// ROS2
broadcaster_ = std::make_shared<tf2_ros::TransformBroadcaster>(*this);
pub_cloud_ = this->create_publisher<sensor_msgs::msg::PointCloud2>(cloud_topic_, 10);
pub_imu_ = this->create_publisher<sensor_msgs::msg::Imu>(imu_topic_, 10);
timer_ = this->create_wall_timer(std::chrono::milliseconds(1), std::bind(&UnitreeLidarSDKNode::timer_callback, this));
}
void UnitreeLidarSDKNode::timer_callback()
{
int result = lsdk_->runParse();
static pcl::PointCloud<PointType>::Ptr cloudOut(new pcl::PointCloud<PointType>());
// RCLCPP_INFO(this->get_logger(), "result = %d", result);
if (result == LIDAR_IMU_DATA_PACKET_TYPE)
{
LidarImuData imu;
lsdk_->getImuData(imu);
if (lsdk_->getImuData(imu))
{
// publish imu message
rclcpp::Time timestamp(imu.info.stamp.sec, imu.info.stamp.nsec);
sensor_msgs::msg::Imu imuMsg;
imuMsg.header.frame_id = imu_frame_;
imuMsg.header.stamp = timestamp;
imuMsg.orientation.x = imu.quaternion[0];
imuMsg.orientation.y = imu.quaternion[1];
imuMsg.orientation.z = imu.quaternion[2];
imuMsg.orientation.w = imu.quaternion[3];
imuMsg.angular_velocity.x = imu.angular_velocity[0];
imuMsg.angular_velocity.y = imu.angular_velocity[1];
imuMsg.angular_velocity.z = imu.angular_velocity[2];
imuMsg.linear_acceleration.x = imu.linear_acceleration[0];
imuMsg.linear_acceleration.y = imu.linear_acceleration[1];
imuMsg.linear_acceleration.z = imu.linear_acceleration[2];
pub_imu_->publish(imuMsg);
// publish tf from imu to lidar
geometry_msgs::msg::TransformStamped transformStamped;
transformStamped.header.stamp = timestamp; // 使用IMU数据的时间戳保持同步
transformStamped.header.frame_id = imu_frame_; // 父坐标系
transformStamped.child_frame_id = cloud_frame_; // 子坐标系
transformStamped.transform.translation.x = 0.007698;
transformStamped.transform.translation.y = 0.014655;
transformStamped.transform.translation.z = -0.00667;
transformStamped.transform.rotation.x = 0;
transformStamped.transform.rotation.y = 0;
transformStamped.transform.rotation.z = 0;
transformStamped.transform.rotation.w = 1;
broadcaster_->sendTransform(transformStamped);
}
}
else if (result == LIDAR_POINT_DATA_PACKET_TYPE)
{
// RCLCPP_INFO(this->get_logger(), "POINT_CLOUD");
PointCloudUnitree cloud;
if (lsdk_->getPointCloud(cloud))
{
transformUnitreeCloudToPCL(cloud, cloudOut);
rclcpp::Time timestamp(
static_cast<int32_t>(cloud.stamp),
static_cast<uint32_t>((cloud.stamp - static_cast<int32_t>(cloud.stamp)) * 1e9));
sensor_msgs::msg::PointCloud2 cloud_msg;
pcl::toROSMsg(*cloudOut, cloud_msg);
cloud_msg.header.frame_id = cloud_frame_;
cloud_msg.header.stamp = timestamp;
pub_cloud_->publish(cloud_msg);
}
}
}
+69
View File
@@ -0,0 +1,69 @@
import os
from launch.substitutions import LaunchConfiguration
from launch import LaunchDescription
from launch.actions import DeclareLaunchArgument
from launch_ros.actions import Node
from launch.conditions import IfCondition
from ament_index_python.packages import get_package_share_directory
def generate_launch_description():
use_rviz = LaunchConfiguration('use_rviz')
declare_use_rviz = DeclareLaunchArgument(
'use_rviz',
default_value='false',
description='Whether to start RViz'
)
pkg_share = get_package_share_directory('unitree_lidar_ros2')
rviz_config_file = os.path.join(pkg_share,
'rviz',
'view.rviz')
# Run unitree lidar
node1 = Node(
package='unitree_lidar_ros2',
executable='unitree_lidar_ros2_node',
name='unitree_lidar_ros2_node',
output='screen',
parameters= [
{'initialize_type': 1},
{'work_mode': 8},
{'use_system_timestamp': True},
{'range_min': 0.0},
{'range_max': 100.0},
{'cloud_scan_num': 18},
{'serial_port': '/dev/ttyACM1'},
{'baudrate': 4000000},
{'lidar_port': 6101},
{'lidar_ip': '192.168.1.62'},
{'local_port': 6201},
{'local_ip': '192.168.1.2'},
{'cloud_frame': "unilidar_lidar"},
{'cloud_topic': "unilidar/cloud"},
{'imu_frame': "unilidar_imu"},
{'imu_topic': "unilidar/imu"},
]
)
# Run Rviz
rviz_node = Node(
package='rviz2',
executable='rviz2',
name='rviz2',
arguments=['-d', rviz_config_file],
condition=IfCondition(use_rviz),
output='log'
)
return LaunchDescription(
[
declare_use_rviz,
node1,
rviz_node,
]
)
+28
View File
@@ -0,0 +1,28 @@
<?xml version="1.0"?>
<?xml-model href="http://download.ros.org/schema/package_format3.xsd" schematypens="http://www.w3.org/2001/XMLSchema"?>
<package format="3">
<name>unitree_lidar_ros2</name>
<version>0.0.0</version>
<description>unitree lidar ros2</description>
<maintainer email="lukelbmeng@gmail.com">jo</maintainer>
<license>GPLv3</license>
<buildtool_depend>ament_cmake</buildtool_depend>
<test_depend>ament_lint_auto</test_depend>
<test_depend>ament_lint_common</test_depend>
<depend>rclcpp</depend>
<depend>std_msgs</depend>
<depend>sensor_msgs</depend>
<depend>pcl_conversions</depend>
<depend>pcl</depend>
<depend>geometry_msgs</depend>
<depend>tf2_ros</depend>
<export>
<build_type>ament_cmake</build_type>
</export>
</package>
+182
View File
@@ -0,0 +1,182 @@
Panels:
- Class: rviz_common/Displays
Help Height: 78
Name: Displays
Property Tree Widget:
Expanded:
- /Global Options1
- /Status1
- /PointCloud21
- /TF1
- /TF1/Frames1
Splitter Ratio: 0.5
Tree Height: 607
- Class: rviz_common/Selection
Name: Selection
- Class: rviz_common/Tool Properties
Expanded:
- /2D Goal Pose1
- /Publish Point1
Name: Tool Properties
Splitter Ratio: 0.5886790156364441
- Class: rviz_common/Views
Expanded:
- /Current View1
Name: Views
Splitter Ratio: 0.5
Visualization Manager:
Class: ""
Displays:
- Alpha: 0.5
Cell Size: 1
Class: rviz_default_plugins/Grid
Color: 160; 160; 164
Enabled: true
Line Style:
Line Width: 0.029999999329447746
Value: Lines
Name: Grid
Normal Cell Count: 0
Offset:
X: 0
Y: 0
Z: 0
Plane: XY
Plane Cell Count: 10
Reference Frame: <Fixed Frame>
Value: true
- Alpha: 1
Autocompute Intensity Bounds: true
Autocompute Value Bounds:
Max Value: 3.4900152683258057
Min Value: 0
Value: true
Axis: Z
Channel Name: intensity
Class: rviz_default_plugins/PointCloud2
Color: 255; 255; 255
Color Transformer: AxisColor
Decay Time: 1
Enabled: true
Invert Rainbow: false
Max Color: 255; 255; 255
Max Intensity: 85
Min Color: 0; 0; 0
Min Intensity: 0
Name: PointCloud2
Position Transformer: XYZ
Selectable: true
Size (Pixels): 3
Size (m): 0.009999999776482582
Style: Points
Topic:
Depth: 5
Durability Policy: Volatile
History Policy: Keep Last
Reliability Policy: Reliable
Value: /unilidar/cloud
Use Fixed Frame: true
Use rainbow: true
Value: true
- Class: rviz_default_plugins/TF
Enabled: true
Frame Timeout: 15
Frames:
All Enabled: true
unilidar_imu:
Value: true
unilidar_imu_initial:
Value: true
unilidar_lidar:
Value: true
Marker Scale: 1
Name: TF
Show Arrows: true
Show Axes: true
Show Names: false
Tree:
unilidar_imu_initial:
unilidar_imu:
unilidar_lidar:
{}
Update Interval: 0
Value: true
Enabled: true
Global Options:
Background Color: 48; 48; 48
Fixed Frame: unilidar_lidar
Frame Rate: 30
Name: root
Tools:
- Class: rviz_default_plugins/Interact
Hide Inactive Objects: true
- Class: rviz_default_plugins/MoveCamera
- Class: rviz_default_plugins/Select
- Class: rviz_default_plugins/FocusCamera
- Class: rviz_default_plugins/Measure
Line color: 128; 128; 0
- Class: rviz_default_plugins/SetInitialPose
Topic:
Depth: 5
Durability Policy: Volatile
History Policy: Keep Last
Reliability Policy: Reliable
Value: /initialpose
- Class: rviz_default_plugins/SetGoal
Topic:
Depth: 5
Durability Policy: Volatile
History Policy: Keep Last
Reliability Policy: Reliable
Value: /goal_pose
- Class: rviz_default_plugins/PublishPoint
Single click: true
Topic:
Depth: 5
Durability Policy: Volatile
History Policy: Keep Last
Reliability Policy: Reliable
Value: /clicked_point
Transformation:
Current:
Class: rviz_default_plugins/TF
Value: true
Views:
Current:
Class: rviz_default_plugins/Orbit
Distance: 14.384671211242676
Enable Stereo Rendering:
Stereo Eye Separation: 0.05999999865889549
Stereo Focal Distance: 1
Swap Stereo Eyes: false
Value: false
Focal Point:
X: 0
Y: 0
Z: 0
Focal Shape Fixed Size: true
Focal Shape Size: 0.05000000074505806
Invert Z Axis: false
Name: Current View
Near Clip Distance: 0.009999999776482582
Pitch: 0.5603978037834167
Target Frame: <Fixed Frame>
Value: Orbit (rviz)
Yaw: 0.785398006439209
Saved: ~
Window Geometry:
Displays:
collapsed: false
Height: 836
Hide Left Dock: false
Hide Right Dock: true
QMainWindow State: 000000ff00000000fd000000040000000000000156000002eafc0200000008fb0000001200530065006c0065006300740069006f006e00000001e10000009b0000005c00fffffffb0000001e0054006f006f006c002000500072006f007000650072007400690065007302000001ed000001df00000185000000a3fb000000120056006900650077007300200054006f006f02000001df000002110000018500000122fb000000200054006f006f006c002000500072006f0070006500720074006900650073003203000002880000011d000002210000017afb000000100044006900730070006c006100790073010000003d000002ea000000c900fffffffb0000002000730065006c0065006300740069006f006e00200062007500660066006500720200000138000000aa0000023a00000294fb00000014005700690064006500530074006500720065006f02000000e6000000d2000003ee0000030bfb0000000c004b0069006e0065006300740200000186000001060000030c00000261000000010000010f000002eafc0200000003fb0000001e0054006f006f006c002000500072006f00700065007200740069006500730100000041000000780000000000000000fb0000000a00560069006500770073000000003d000002ea000000a400fffffffb0000001200530065006c0065006300740069006f006e010000025a000000b200000000000000000000000200000490000000a9fc0100000001fb0000000a00560069006500770073030000004e00000080000002e10000019700000003000004420000003efc0100000002fb0000000800540069006d00650100000000000004420000000000000000fb0000000800540069006d006501000000000000045000000000000000000000049c000002ea00000004000000040000000800000008fc0000000100000002000000010000000a0054006f006f006c00730100000000ffffffff0000000000000000
Selection:
collapsed: false
Tool Properties:
collapsed: false
Views:
collapsed: true
Width: 1528
X: 72
Y: 27
@@ -0,0 +1,14 @@
/**********************************************************************
Copyright (c) 2020-2024, Unitree Robotics.Co.Ltd. All rights reserved.
***********************************************************************/
#include "unitree_lidar_ros2.h"
int main(int argc, char *argv[])
{
const rclcpp::NodeOptions options;
rclcpp::init(argc, argv);
rclcpp::spin(std::make_shared<UnitreeLidarSDKNode>());
rclcpp::shutdown();
return 0;
}