|
|
#pragma once
|
|
|
#include "afxcoll.h"
|
|
|
#include <vector>
|
|
|
#include <string>
|
|
|
#include "CxImage/CxImage/ximage.h"
|
|
|
#include "..\Xy.h"
|
|
|
|
|
|
struct ColorPoint {
|
|
|
int R;
|
|
|
int G;
|
|
|
int B;
|
|
|
double zValue;
|
|
|
};
|
|
|
|
|
|
class AFX_EXT_CLASS GenerateImageMesh
|
|
|
{
|
|
|
public:
|
|
|
// 颜色采样方式
|
|
|
enum ColorMethod {
|
|
|
HSV_M = 0, //HSV极端方法
|
|
|
UNIFORMSAMPLING_M = 1, //均匀采样方法
|
|
|
KMEANS_M = 2 //K-Means聚类方法
|
|
|
};
|
|
|
|
|
|
// 颜色排序方式
|
|
|
enum SortMethod {
|
|
|
BRIGHTNESS_SORT = 0, //亮度(升序)
|
|
|
HUE_SORT_COLD = 1, //色调冷
|
|
|
HUE_SORT_WARM = 2, //色调暖
|
|
|
HUE_SORT = 3
|
|
|
};
|
|
|
|
|
|
GenerateImageMesh();
|
|
|
~GenerateImageMesh();
|
|
|
|
|
|
//设置输入数据
|
|
|
void SetImageBase(CxImage image);
|
|
|
void SetImageBase(const std::string& path);
|
|
|
std::string GetImagePath();
|
|
|
//设置色标(默认色标)
|
|
|
void SetColorCardPath(const std::string& path);
|
|
|
std::string GetColorCardPath();
|
|
|
void SetGridNumber(int num);
|
|
|
int GetGridNumber();
|
|
|
void SetZminByZmax(double zMin, double zMax);
|
|
|
double GetZmin();
|
|
|
double GetZmax();
|
|
|
|
|
|
/**
|
|
|
* 自动获取生成色标文件组成色彩（默认由冷到暖）
|
|
|
* \param colorVectors 颜色和z值集合（z值从小到大）
|
|
|
* \param ColorMethod 颜色采样方式
|
|
|
* \param SortMethod 颜色排序方式
|
|
|
* \param numColors 提取数量
|
|
|
* \return
|
|
|
*/
|
|
|
bool CreateColorBarData(std::vector<ColorPoint>& colorPVector, GenerateImageMesh::ColorMethod type, GenerateImageMesh::SortMethod sortType, int numColors = 5);
|
|
|
|
|
|
/**
|
|
|
* 生成色标文件：
|
|
|
* \param colorVector 颜色和z值集合（z值从小到大）
|
|
|
* \param barSavePath 色标文件路径
|
|
|
* \param xmlPath 色标文件xml
|
|
|
* \param width 色标文件图片宽
|
|
|
* \param height 色标文件图片高
|
|
|
* \return
|
|
|
*/
|
|
|
bool CreateMeshColorBar(const std::vector<ColorPoint>& colorVector, const std::string& barSavePath, double minz, double maxz,
|
|
|
const std::string& xmlPath = "", int width = 1200, int height = 60);
|
|
|
|
|
|
// 生成网格图像数据
|
|
|
CMesh * CreateImageMeshProcess(int numX, int numY
|
|
|
, double dLocationX, double dLocationY
|
|
|
, double dDeltX, double dDeltY, double dmaxX, double dmaxY
|
|
|
, const std::vector<ColorPoint>& colorVector, const std::vector<ColorPoint>& colorListVector);
|
|
|
|
|
|
bool CreateImageMesh(int numX, int numY
|
|
|
, double dLocationX, double dLocationY, double dDeltX, double dDeltY , double dmaxX, double dmaxY,
|
|
|
const std::vector<ColorPoint>& colorVector);
|
|
|
|
|
|
double FindThirdValues(const std::vector<std::tuple<int, int, double>>& data,
|
|
|
int first, int second);
|
|
|
protected:
|
|
|
void Init();
|
|
|
|
|
|
bool LaodImageCV();//加载内存数据
|
|
|
private:
|
|
|
int m_num; //网格大小（10 10*10）
|
|
|
double m_zMinCV; //最开始创建的区间值
|
|
|
double m_zMaxCV;
|
|
|
|
|
|
std::string m_colorCardPath; //颜色卡
|
|
|
std::string m_imagePath; //图像
|
|
|
|
|
|
CxImage m_xImage; //输入图像内存数据
|
|
|
CMesh *p_mesh;
|
|
|
};
|