#pragma once
#include <string>
#include <vector>
namespace nglib
{
	#include "nglib.h"
}
using namespace nglib;

enum BOUND_TYPE
{
	CONST_PRESSURE,
	CONST_RATE,
	GEO_LIMIT
};
enum BOUND_SHAPE
{
	CIRCLE,
	RECTANGLE,
	POLYGON
};
enum WELL_TYPE
{
	VERTICAL,
	VERTICAL_FRACTURED,
	MULTI_FRACED_HORIZONTAL
};

struct Point {
	double x, y, z;	
	std::vector<double> pointData;

	Point(double x0, double y0)
	{
		x = x0;
		y = y0;
	}
	Point() {};
	// 拷贝构造函数
	Point(const Point& other) {
		//std::cout << "Copy constructor called" << std::endl;
		x = other.x; // 复制数据
		y = other.y;
		z = other.z;
		pointData=other.pointData;		
	}
	bool operator==(const Point& other) const{
		const double epsilon = 1e-6; // 误差范围
		return fabs(x - other.x) < epsilon && 
			fabs(y - other.y) < epsilon;
	}
};

struct Cell {
	int type;
	std::vector<int> pointIndices;
	std::vector<double> cellData;
	Cell(const Cell& other)
	{
		type = other.type;
		pointIndices = other.pointIndices;
		cellData = other.cellData;
	}
	Cell() {}
};
/*
class CBoundLine
{
	//	 DECLARE_SERIAL(CBoundLine)

public:

	CBoundLine() {
		nNodeNum = 8;
		nboundtype = 0;
		ddl = 100;
		bisSel = false;
	};
	virtual ~CBoundLine() {};
	Point pPre, pOri;
	int nboundtype;
	int nNodeNum;
	double ddl;
	int nNodeMin;
	int nNodeMax;
	bool bisSel;

};*/

class Line
{

public:
	Point p1;
	Point p2;
};

// 裂缝线
class Frac : public Line
{
public:
	Frac(Point p1, Point p2)
	{
		this->p1 = p1;
		this->p2 = p2;
	}
	Frac() {}
	// 交点
	int nIntersect;												// 交点数量
	std::vector<Point> vector_interPoint; // 交点坐标
};

// 定义外边界的线
class Segment : public Line
{
public:
	Segment(Point p1, Point p2)
	{
		this->p1 = p1;
		this->p2 = p2;
	}
	Segment() {}
	BOUND_TYPE boundType; // 边界线的边界条件类型
	double dConstPressure;//定义边界时的压力值
	int nNumNodes;		// 边界线上的网格点数
	std::vector<int> vecNodes;//边界线上的网格节点编号
};
// 外边界的基类
class CBoundShape
{
public:
	BOUND_SHAPE boundShape; // 定义外边界类型，0为圆形，1为四边形，2为多边形

	// 表示多边形（包括四边形）的参数
	int nNumSegs;											// 边的数量
	std::vector<Segment> vecSegments; // 边

	// 表示圆形
	Point cCenter;
	double dRadius;
	int nNodeNum;
	BOUND_TYPE boundType;
	double dConstPressure;//圆形定压边界时的压力值
	std::vector<int> vecNodes;//圆形边界上的网格点编号
};

// 多边形,内部的复合区域
class CBoundPolygon : public CBoundShape
{
public:
	int nNumSegs;											// 边的数量
	std::vector<Segment> vecSegments; // 边
	
	double dComKr;//复合区的渗透率比
	double dComW;//复合区的储能比

};
class CBoundWell
{
	//	  DECLARE_SERIAL(CWell)
public:
	CBoundWell() {
		nNodeNum = 12;
//		dRc = 0.1;
		nFlowType = 0;
		nTimeNumQ = 0;
		nTimeNumP = 0;
		dSkin = 0;
		dC = 0.01;
		dMinPress = 0.101325;
		bisSel = false;
		nDenseNum = 10;
		dDenseRadius[0] = 1; nDenseNodeNUm[0] = 18;
		dDenseRadius[1] = 10; nDenseNodeNUm[1] = 23;
		dDenseRadius[2] = 50; nDenseNodeNUm[2] = 22;
		dDenseRadius[3] = 100; nDenseNodeNUm[3] = 25;
		dDenseRadius[4] = 500; nDenseNodeNUm[4] = 26;
		dDenseRadius[5] = 1000; nDenseNodeNUm[5] = 27;
		dDenseRadius[6] = 2000; nDenseNodeNUm[6] = 32;
		dDenseRadius[7] = 3000; nDenseNodeNUm[7] = 34;
		dDenseRadius[8] = 4000; nDenseNodeNUm[8] = 35;
		dDenseRadius[9] = 5000; nDenseNodeNUm[9] = 47;
		dDenseRadius[10] = 10000; nDenseNodeNUm[10] = 48;
		dDenseRadius[11] = 20000; nDenseNodeNUm[11] = 48;
		bisAutoChoose = false;
		bisMultFlow = true;
		nInComNote = 0;		
		wellType=VERTICAL;
	};
	virtual ~CBoundWell() {};
	void SetDenseValue();
public:
	//void Serialize(CArchive& ar);
	std::string sWellname;
	Point pCenter;
	int nFlowType;
	int nNodeNum;
	//double dRc;
	double dSkin;
	double dC;
	double dMinPress;
	int nTimeNumQ;
	int nTimeNumP;
	double pdTimeQ[100];
	double pdTimeP[100];
	double pdQ[100];
	double pdP[100];
	int nNodeMin;
	int nNodeMax;
	bool bisSel;
	int nDenseNum;
	double dDenseRadius[20];
	int nDenseNodeNUm[20];
	double dS;
	bool bisAutoChoose;
	bool bisMultFlow;
	int nInComNote;
	std::vector<int> m_NodeIndex;//all the nodes index of the wellbore
	std::vector<Point> m_PressureData;//计算的井筒压力数据

	WELL_TYPE wellType;
	//直井的几何参数，wellType=VERTICAL
	Point cCenter;
	double dRadius;
	//int nNodeNum;
	BOUND_TYPE boundType;

	//直井压裂井独有的参数，wellType=VERTICAL_FRACTURED
	//直井的中心和半径仍然使用cCenter和dRaduis表示
	double dHf;//裂缝半长
	double dTheata;//裂缝角度(相对于x轴)

	//多段压裂水平井的参数，cCenter表示水平井中心点坐标，wellType=MULTI_FRACED_HORIZONTAL
	double dLength;//长度
	double dBeta;//角度
	int nFracNum;//压裂裂缝的数量
	std::vector<Frac> vecFracs;//压裂裂缝
};
/*
class CBoundCir
{
	//	DECLARE_SERIAL(CBoundCir)
public:
	CBoundCir() {
		nNodeNum = 24;
		nboundtype = 0;
		bisSel = false;
		dComKr = 10;
		dComW = 3;
	};
	virtual ~CBoundCir() {};

public:
	Point pCenter;
	double dRc;
	int nboundtype;
	int nNodeNum;
	bool bisSel;
	int nNodeMin;
	int nNodeMax;
	int nMeshMin;
	int nMeshMax;
	double dComKr;
	double dComW;
};*/

#ifdef __cplusplus
extern "C" {
#endif	

	//mesh generation API
	__declspec(dllexport) Ng_Mesh *NgMeshGen2D(std::string strFileIn2D);
	__declspec(dllexport) Ng_Mesh *NgMeshGen2DByIn2d(std::vector<CBoundWell> wells, std::vector<CBoundPolygon> limits, std::vector<CBoundPolygon> faults, std::vector<Frac> fracs, CBoundShape shape);	
	__declspec(dllexport) void Ng2VTK(Ng_Mesh *mesh);


	//singlePhase solver API
	__declspec(dllexport) int singlePhaseSolver(double* pBaseData, std::vector<Point> points, std::vector<Cell> cells, int m_nWellNum, CBoundWell* pwells, std::vector<std::vector<Point>>& vecBP, std::vector<std::pair<double, std::vector<double>>>& vecNodePre);	
	__declspec(dllexport) int singlePhaseSolverNgmesh(double* m_pBaseData, nglib::Ng_Mesh* mesh,std::vector<CBoundWell>&m_wWell,std::vector<CBoundPolygon> comPolygon,std::vector<CBoundPolygon> faultPolygon,std::vector<Frac> fracs,CBoundShape outBound, std::vector<std::pair<double,std::vector<double>>>& vecNodePre);
	__declspec(dllexport) bool logLogPre(CBoundWell* pWell, const int nSection, std::vector<Point> &logPre);
#ifdef __cplusplus
}
#endif