|
|
// lidar_utils.h - 雷达连接与扫描采集工具
|
|
|
#ifndef LIDAR_UTILS_H
|
|
|
#define LIDAR_UTILS_H
|
|
|
|
|
|
#include <mutex>
|
|
|
#include <string>
|
|
|
#include <thread>
|
|
|
#include <vector>
|
|
|
#include <chrono>
|
|
|
#include "PAVO2_SDK/include/pavo2_driver.h"
|
|
|
#include "common.h"
|
|
|
|
|
|
// ==================== 雷达连接 ====================
|
|
|
// 连接雷达(TCP)，配置滤波并开启数据传输；失败返回 false。
|
|
|
// 注意：Windows 下需在 WSAStartup 之后调用。
|
|
|
bool connect_lidar(pavo2::pavo2_driver& driver);
|
|
|
|
|
|
// ==================== 扫描辅助 ====================
|
|
|
// 排空 SDK 队列中积压的旧扫描包（采集 start 之前雷达已缓存的数据）。
|
|
|
// 防御性：最多排空 max_ms 毫秒（默认500），避免雷达持续产数导致卡死。
|
|
|
// 返回丢弃包数。
|
|
|
int drain_stale_scans(pavo2::pavo2_driver& driver, int max_ms = 500);
|
|
|
|
|
|
// z轴值换算（时间→z）：
|
|
|
// - cfg.scan.z_speed_type == "configuration" 且 speed.txt 有效：按速度剖面(时间→距离)变速插值，
|
|
|
// 超过表末尾时间按最后两条的平均速度外推延伸；
|
|
|
// - 否则按原逻辑：cfg.scan.z_speed>0 时为移动距离(mm)=cfg.scan.z_speed(cm/s)*10*秒；
|
|
|
// cfg.scan.z_speed=0 时 z 直接取 elapsed_ms（时间模式）。
|
|
|
double z_for_elapsed(double elapsed_ms);
|
|
|
|
|
|
// ==================== 变速速度剖面 (speed.txt) ====================
|
|
|
// speed.txt 每行两个数字: "时间(ms) 距离(mm)"（# 开头为注释，空行忽略），时间须递增。
|
|
|
// 当 cfg.scan.z_speed_type == "configuration" 时程序使用该表把采集时间换算为 z(行进距离, mm)，
|
|
|
// 从而实现变速运动识别；未找到有效表时自动回退使用 cfg.scan.z_speed 匀速。
|
|
|
// 默认路径 ./config/speed.txt（与 config/config 同目录）。
|
|
|
// 加载剖面（成功后内部替换当前剖面）。返回是否有效(≥2行)。
|
|
|
bool load_speed_profile(const std::string& path = "./config/speed.txt");
|
|
|
|
|
|
// 按当前 cfg.scan.z_speed_type 重载：configuration → 重新读 speed.txt；
|
|
|
// uniformSpeed → 清空剖面（不使用表）。配置重载(load_config)后应调用一次。
|
|
|
void reload_speed_profile();
|
|
|
|
|
|
// 当前剖面是否有效(≥2行，配置为 configuration 时 z 按表换算)。
|
|
|
bool speed_profile_loaded();
|
|
|
|
|
|
// 剖面平均速度(mm/s)；无效或非 distance 数据时返回 0。
|
|
|
double speed_profile_avg_velocity_mm_s();
|
|
|
|
|
|
// 当前 z 轴是否为距离(mm)：configuration 且剖面有效=true；
|
|
|
// uniformSpeed 时为 cfg.scan.z_speed>0。
|
|
|
bool z_is_distance_mode();
|
|
|
|
|
|
// ConfigureSpeed 标定采集：采集 seconds 秒，逐帧记录(时间, 过滤后Y最大值)，
|
|
|
// 平滑(抑制与总体趋势反向>5mm的抖动)后写入 ./config/speed.txt。
|
|
|
// 返回写入行数；失败返回 <0 并在 err 中给出原因。
|
|
|
// 注意：调用前应确保没有其他线程同时 get_scanned_data（与 ScanSession 互斥使用）。
|
|
|
int collect_speed_profile(pavo2::pavo2_driver& driver, double seconds,
|
|
|
std::string& err);
|
|
|
|
|
|
// 将一帧扫描包 vec_buff 中的点做 XY 过滤（原点噪点 + 过滤多边形）后追加到 out。
|
|
|
// z 由调用方给出（通常为 z_for_elapsed(elapsed_ms)）。
|
|
|
// 返回新增点数。
|
|
|
size_t scan_convert_points(const pavo2::pavo_packet_info_scan_t& scan_info,
|
|
|
double z_val,
|
|
|
std::vector<CountPoint>& out);
|
|
|
|
|
|
// ==================== 扫描会话 ====================
|
|
|
// 单次后台采集会话：reset() → start() 开启采集线程 → stop() 请求停止并等待线程退出。
|
|
|
// count/check 统一使用同一会话，保证同一时刻只有一个采集线程访问雷达。
|
|
|
// 线程安全；仅允许空闲时 reset()。
|
|
|
class ScanSession
|
|
|
{
|
|
|
public:
|
|
|
// 启动后台采集线程。内部先排空 SDK 积压旧包，排空完成才开始 z 计时。
|
|
|
// 已在运行或线程创建失败时返回 false。
|
|
|
bool start(pavo2::pavo2_driver& driver);
|
|
|
|
|
|
// 请求停止并等待采集线程退出，返回当前已采集点数。
|
|
|
// 会话未运行（未 start 或已 stop）时仅返回现存点数，不做其他事。
|
|
|
size_t stop();
|
|
|
|
|
|
// 拷贝当前点集快照（线程安全）。
|
|
|
std::vector<CountPoint> snapshot() const;
|
|
|
|
|
|
// 是否正在采集运行中。
|
|
|
bool running() const;
|
|
|
|
|
|
// 复位（仅空闲时调用）：清空上一次点集并复位状态，供下一次 start 使用。
|
|
|
void reset();
|
|
|
|
|
|
// 后台采集线程执行体。
|
|
|
void collect_loop(pavo2::pavo2_driver* driver);
|
|
|
|
|
|
private:
|
|
|
mutable std::mutex mtx_;
|
|
|
bool running_ = false;
|
|
|
bool stop_req_ = false;
|
|
|
std::thread th_;
|
|
|
std::vector<CountPoint> points_;
|
|
|
std::chrono::steady_clock::time_point start_tp_;
|
|
|
};
|
|
|
|
|
|
#endif // LIDAR_UTILS_H
|