|
|
#pragma once
|
|
|
#include "dllExport.h"
|
|
|
#include "XJPoint2D.h"
|
|
|
#include "XJPoint3D.h"
|
|
|
#include "Isopleth.h"
|
|
|
#include "ControlPoint2d.h"
|
|
|
#include <string>
|
|
|
#include <vector>
|
|
|
#include <limits>
|
|
|
#include <Eigen/Dense>
|
|
|
#include "RBFInterpolator.hpp"
|
|
|
#include <atomic>
|
|
|
#include "kdtree.hpp"
|
|
|
|
|
|
typedef void(*ProgressCallback)(int percent, void* userData);
|
|
|
|
|
|
enum class InterpolationMethod
|
|
|
{
|
|
|
Default = 0, // 默认
|
|
|
Kriging = 1, // 克里金
|
|
|
RFB = 2, // RFB
|
|
|
IDW = 3 // 反距离加权
|
|
|
};
|
|
|
|
|
|
class RectificationMeshWell
|
|
|
{
|
|
|
public:
|
|
|
RectificationMeshWell();
|
|
|
virtual ~RectificationMeshWell();
|
|
|
|
|
|
|
|
|
//progressCallback，C#传入
|
|
|
bool RectificationMeshWellCompute(
|
|
|
unsigned int numx, unsigned int numy, double x0, double y0, double stepX, double stepY, double* values,
|
|
|
const std::vector<ControlPoint2d>& pointList, InterpolationMethod method, const std::vector<uint8_t>& fault, double deltaz, double factor = 1,
|
|
|
ProgressCallback progressCallback = nullptr,
|
|
|
void* progressUserData = nullptr // 进度回调用户数据，可选
|
|
|
);
|
|
|
|
|
|
//终止线程
|
|
|
void requestStop() { m_stopFlag.store(true); }
|
|
|
void resetStopFlag(); // 每次计算前重置
|
|
|
bool isStopRequested() const { return m_stopFlag.load(); }
|
|
|
|
|
|
protected:
|
|
|
//Kriging
|
|
|
bool KrigingInterpolatorCompute(ProgressCallback progressCallback,
|
|
|
void* progressUserData);
|
|
|
//RFB
|
|
|
bool RFBInterpolatorCompute(ProgressCallback progressCallback,
|
|
|
void* progressUserData);
|
|
|
//IDW
|
|
|
bool IDWInterpolatorCompute(ProgressCallback progressCallback,
|
|
|
void* progressUserData);
|
|
|
|
|
|
//双线性插值函数
|
|
|
double bilinear(const Eigen::MatrixXd& grid, double xmin, double xmax, double ymin, double ymax, double x, double y);
|
|
|
|
|
|
double bicubic(
|
|
|
const Eigen::MatrixXd& grid,
|
|
|
double xmin, double xmax, double ymin, double ymax,
|
|
|
double x, double y);
|
|
|
|
|
|
public:
|
|
|
double m_factor; //步长放大倍数
|
|
|
double m_deltaZ; //网格与井点差值
|
|
|
|
|
|
uint64_t m_row; //行数
|
|
|
uint64_t m_col; //列数
|
|
|
Point2D m_minPoint; //最小点
|
|
|
Point2D m_maxPoint; //最大点
|
|
|
double m_zMin;
|
|
|
double m_zMax;
|
|
|
double m_stepX; //步长
|
|
|
double m_stepY;
|
|
|
|
|
|
std::unique_ptr<double[]> m_data; //z值数据
|
|
|
|
|
|
std::vector<ControlPoint2d> m_controlPnts; //控制点
|
|
|
std::vector<uint8_t> m_fault; //断层点
|
|
|
|
|
|
std::atomic<bool> m_stopFlag{ false };
|
|
|
};
|
|
|
|