232 lines
7.4 KiB
C++
232 lines
7.4 KiB
C++
#pragma once
|
||
|
||
#if defined(SG_API_LIBRARY)
|
||
# define SG_APISHARED_EXPORT __declspec(dllexport)
|
||
#else
|
||
# define SG_APISHARED_EXPORT __declspec(dllimport)
|
||
#endif
|
||
|
||
#include "SG_baseDataType.h"
|
||
#include <vector>
|
||
#include <opencv2/opencv.hpp>
|
||
|
||
//滤除离群点:z跳变门限方法（大于门限视为不连续，根据连续段点数量判断噪声）
|
||
void sg_lineDataRemoveOutlier(
|
||
SVzNL3DPosition* lineData,
|
||
int dataSize,
|
||
SSG_outlierFilterParam filterParam,
|
||
std::vector<SVzNL3DPosition>& filerData,
|
||
std::vector<int>& noisePts);
|
||
//滤除离群点：z跳变门限方法,改变原始数据
|
||
void sg_lineDataRemoveOutlier_changeOriginData(
|
||
SVzNL3DPosition* lineData,
|
||
int dataSize,
|
||
SSG_outlierFilterParam filterParam);
|
||
|
||
//滤除离群点：点距离门限方法（大于门限视为不连续，根据连续段点数量判断噪声）
|
||
void sg_lineDataRemoveOutlier_ptDistMethod(
|
||
SVzNL3DPosition* lineData,
|
||
int dataSize,
|
||
SSG_outlierFilterParam filterParam,
|
||
std::vector<SVzNL3DPosition>& filerData,
|
||
std::vector<int>& noisePts);
|
||
//平滑
|
||
void sg_lineDataSmoothing(
|
||
std::vector<SVzNL3DPosition>& input,
|
||
int smoothWin,
|
||
std::vector<SVzNL3DPosition>& output);
|
||
|
||
//VZ_APISHARED_EXPORT void sg_getLineMeanVar();
|
||
void sg_getLineLVFeature(
|
||
SVzNL3DPosition* lineData,
|
||
int dataSize,
|
||
int lineIdx,
|
||
const SSG_slopeParam slopeParam,
|
||
const SSG_VFeatureParam valleyPara,
|
||
SSG_lineFeature* line_features);
|
||
|
||
//获取扫描线拐点特征
|
||
void sg_getLineCornerFeature(
|
||
SVzNL3DPosition* lineData,
|
||
int dataSize,
|
||
int lineIdx,
|
||
const SSG_cornerParam cornerPara, //scale通常取bagH的1/4
|
||
SSG_lineFeature* line_features);
|
||
|
||
/// <summary>
|
||
/// 提取激光线上的极值点（极大值点和极小值点）
|
||
///
|
||
/// </summary>
|
||
void sg_getLineLocalPeaks(
|
||
SVzNL3DPosition* lineData,
|
||
int dataSize,
|
||
int lineIdx,
|
||
const double scaleWin,
|
||
std::vector< SSG_basicFeature1D>& localMax,
|
||
std::vector< SSG_basicFeature1D>& localMin);
|
||
|
||
void sg_getLineDownJumps(
|
||
SVzNL3DPosition* lineData,
|
||
int dataSize,
|
||
int lineIdx,
|
||
const double jumpTh,
|
||
std::vector< SSG_basicFeature1D>& downJumps);
|
||
|
||
//对feature进行生长
|
||
void sg_getFeatureGrowingTrees(
|
||
std::vector<SSG_lineFeature>& lineFeatures,
|
||
std::vector<SSG_featureTree>& trees,
|
||
SSG_treeGrowParam growParam);
|
||
|
||
//对ending进行生长
|
||
void sg_getEndingGrowingTrees(
|
||
std::vector<SSG_2DValueI>& lineEndings,
|
||
SVzNL3DLaserLine* laser3DPoints,
|
||
bool isVScan,
|
||
int featureType,
|
||
std::vector<SSG_featureTree>& trees,
|
||
SSG_treeGrowParam growParam);
|
||
|
||
//对扫描线上的
|
||
void sg_LVFeatureGrowing(
|
||
std::vector<SSG_lineFeature>& lineFeatures,
|
||
std::vector<SSG_featureTree>& trees,
|
||
SSG_bagParam bagParam,
|
||
SSG_treeGrowParam growParam,
|
||
std::vector<SSG_2DValueI>& edgePts_0,
|
||
std::vector<SSG_2DValueI>& edgePts_1);
|
||
|
||
void sg_peakFeatureGrowing(
|
||
std::vector<std::vector< SSG_basicFeature1D>>& lineFeatures,
|
||
std::vector<SSG_featureTree>& trees,
|
||
SSG_treeGrowParam growParam);
|
||
|
||
SSG_meanVar _computeMeanVar(double* data, int size);
|
||
|
||
///搜索局部最高点（z最小点）。
|
||
///搜索方法：每次步进搜索窗口长度的一半。对局部最高点进行标记，防止被重复记录。
|
||
void sg_getLocalPeaks(
|
||
SVzNL3DLaserLine* scanLines,
|
||
int lineNum,
|
||
std::vector<SSG_2DValueI>& peaks,
|
||
SSG_localPkParam searchWin);
|
||
|
||
/// <summary>
|
||
/// 区域生长法：以局部最高点作为生长种子进行生长
|
||
/// 生长方法与一般的区域生长不同：以种子点为圆心作圆周扫描，记录扫描到的边界
|
||
/// </summary>
|
||
//void sg_peakPolarScan(cv::Mat& edgeMask, SVzNL2DPoint a_peak, SSG_polarScanParam polarScanParam, std::vector< SSG_2DValueI>& rgnContour);
|
||
//从Peak点进行水平垂直扫描得到区域边界
|
||
void sg_peakXYScan(
|
||
SVzNL3DLaserLine* laser3DPoints,
|
||
int lineNum,
|
||
cv::Mat& featureEdgeMask,
|
||
SSG_2DValueI a_peak,
|
||
SSG_treeGrowParam growParam,
|
||
SSG_bagParam bagParam,
|
||
bool rgnPtAsEdge,
|
||
std::vector< SSG_lineConotours>& topContour,
|
||
std::vector< SSG_lineConotours>& bottomContour,
|
||
std::vector< SSG_lineConotours>& leftContour,
|
||
std::vector< SSG_lineConotours>& rightContour,
|
||
int* maxEdgeId_top,
|
||
int* maxEdgeId_btm,
|
||
int* maxEdgeId_left,
|
||
int* maxEdgeId_right);
|
||
|
||
//取出与给定edgeId相同的边界点
|
||
void sg_getContourPts(
|
||
std::vector< SSG_lineConotours>& contour_all,
|
||
int vldEdgeId,
|
||
std::vector< SSG_2DValueI>& contourFilter,
|
||
int* lowLevelFlag);
|
||
|
||
//从边界点对中取出与给定edgeId相同的边界点
|
||
void sg_getPairingContourPts(
|
||
std::vector<SSG_conotourPair>& contourPairs,
|
||
std::vector<SSG_intPair>& idPairs,
|
||
std::vector< SSG_conotourPair>& contourFilter,
|
||
SVzNLRangeD range,
|
||
bool isTBDir,
|
||
int* lowLevelFlag_0,
|
||
int* lowLevelFlag_1);
|
||
|
||
//取出最短距离的边界点
|
||
void sg_contourPostProc(
|
||
std::vector< SSG_contourPtInfo>& contour,
|
||
int maxEdgeIdx,
|
||
double sameConturDistTh,
|
||
std::vector< SSG_2DValueI>& contourFilter,
|
||
int sideID,
|
||
int* blockFlag);
|
||
|
||
//距离变换
|
||
//input, output均为float型
|
||
void sg_distanceTrans(const cv::Mat input, cv::Mat& output, int distType);
|
||
|
||
/// <summary>
|
||
/// 以5x5方式寻找localPeaks
|
||
/// </summary>
|
||
/// <param name="input"></param>
|
||
/// <param name="peaks"></param>
|
||
void sg_getLocalPeaks_distTransform(
|
||
cv::Mat& input,
|
||
std::vector<SSG_2DValueI>& peaks,
|
||
SSG_localPkParam searchWin);
|
||
|
||
/// <summary>
|
||
/// 使用模板法提取直角特征
|
||
/// 水平向下直角特征：拐点左侧deltaZ在一个很小的范围内；拐点右侧deltaY在一个很小的范围内
|
||
/// </summary>
|
||
void sg_getLineRigthAngleFeature(
|
||
SVzNL3DPosition* lineData,
|
||
int dataSize,
|
||
int lineIdx,
|
||
const SSG_lineRightAngleParam templatePara_HF,
|
||
const SSG_lineRightAngleParam templatePara_FH,
|
||
const SSG_lineRightAngleParam templatePara_HR,
|
||
const SSG_lineRightAngleParam templatePara_RH,
|
||
SSG_lineFeature* line_features);
|
||
|
||
//计算扫描ROI
|
||
SVzNL3DRangeD sg_getScanDataROI(
|
||
SVzNL3DLaserLine* laser3DPoints,
|
||
int lineNum);
|
||
|
||
//XY平面直线拟合
|
||
void lineFitting(
|
||
std::vector< SVzNL3DPoint>& inliers,
|
||
double* _k,
|
||
double* _b);
|
||
|
||
//Bresenham算法
|
||
void drawLine(
|
||
int x0,
|
||
int y0,
|
||
int x1,
|
||
int y1,
|
||
std::vector<SVzNL2DPoint>& pts);
|
||
|
||
/// <summary>
|
||
/// 两步法标注
|
||
/// </summary>
|
||
/// <param name="bwImg"> 目标点为“1”， 空白点为“0”</param>
|
||
/// <param name="labImg"> 标注结果。每个点为rgnID, ID从2开始 </param>
|
||
/// <param name="labelRgns"></param>
|
||
void SG_TwoPassLabel(
|
||
const cv::Mat& bwImg,
|
||
cv::Mat& labImg,
|
||
std::vector<SSG_Region>& labelRgns,
|
||
int connectivity = 4);
|
||
|
||
//计算一个平面调平参数。
|
||
//数据输入中可以有一个地平面和参考调平平面，以最高的平面进行调平
|
||
//旋转矩阵为调平参数，即将平面法向调整为垂直向量的参数
|
||
SSG_planeCalibPara sg_getPlaneCalibPara(
|
||
SVzNL3DLaserLine* laser3DPoints,
|
||
int lineNum);
|
||
|
||
// 从旋转矩阵计算欧拉角（Z-Y-X顺序）
|
||
SSG_EulerAngles rotationMatrixToEulerZYX(const double R[3][3]);
|
||
// 从欧拉角计算旋转矩阵（Z-Y-X顺序）
|
||
void eulerToRotationMatrixZYX(const SSG_EulerAngles& angles, double R[3][3]); |