#include "stdafx.h"
#include "singlePhaseSolver.h"
#include "windows.h"
#include "iostream"
#include "fstream"
void Ng2VTKWithData(Ng_Mesh *mesh, std::vector<double> vecField, std::string filename)
{
	std::cout << "Convert Ng_Mesh to vtk legacy file version of 4.2" << std::endl;
	int i, nseg, ne, np, matnum;
	int nodes[3];
	double point[2];
	int *EleNodeNum;
	double *NodeCoors;

	np = Ng_GetNP_2D(mesh);
	ne = Ng_GetNE_2D(mesh);
	nseg = Ng_GetNSeg_2D(mesh);

	std::ofstream vtkFile;
	vtkFile.open(filename);

	if (!vtkFile.is_open())
	{
		std::cerr << "Error opening file: " << "mesh.vtk" << std::endl;
		return;
	}

	// Write VTK file header
	vtkFile << "# vtk DataFile Version 4.1" << std::endl;
	vtkFile << "Mesh data" << std::endl;
	vtkFile << "ASCII" << std::endl;
	vtkFile << "DATASET UNSTRUCTURED_GRID" << std::endl;

	// Write points
	vtkFile << "POINTS " << np << " float" << std::endl;
	for (i = 1; i <= np; i++)
	{
		Ng_GetPoint_2D(mesh, i, point);
		vtkFile << point[0] << " " << point[1] << " " << 0 << std::endl;
	}

	// Write cells
	vtkFile << "CELLS " << ne << " " << ne * 4 << std::endl;
	std::vector<int> vecMater;
	for (int i = 1; i <= ne; i++)
	{
		Ng_GetElement_2D(mesh, i, nodes, &matnum);
		vecMater.emplace_back(matnum);
		vtkFile << 3 << " " << nodes[0] - 1 << " " << nodes[1] - 1 << " " << nodes[2] - 1 << std::endl;
	}

	// Write cell types
	vtkFile << "CELL_TYPES " << ne << std::endl;
	for (int i = 1; i <= ne; i++)
	{
		vtkFile << 5 << std::endl; // Assuming getCellType is a method that returns the VTK cell type
	}

	// Write cell data
	vtkFile << "CELL_DATA " << ne << std::endl;
	vtkFile << "SCALARS Material int 1" << std::endl;
	vtkFile << "LOOKUP_TABLE default" << std::endl;
	for (int i = 0; i < vecMater.size(); i++)
	{
		vtkFile << vecMater[i] << std::endl;
	}

	// Write point data
	vtkFile << "POINT_DATA " << np << std::endl;
	vtkFile << "SCALARS Pressure double 1" << std::endl;
	vtkFile << "LOOKUP_TABLE default" << std::endl;
	for (int i = 0; i < vecField.size(); i++)
	{
		vtkFile << vecField[i] << std::endl;
	}
	// write segs
	// vtkFile << "LINES " << nseg << " " << nseg * 3 << std::endl;
	// for (i = 1; i <= nseg; i++)
	//{
	//	Ng_GetSegment_2D(mesh, i, nodes, &matnum);
	//	std::cout << i << " matnum:\t" << matnum << " " << nodes[0] - 1 << " " << nodes[1] - 1 << std::endl;
	//}
	vtkFile.close();
}

int _tmain(int argc, _TCHAR *argv[])
{
	/************************************************************************/
	/* prepare data for mesh generation                                     */
	/************************************************************************/
	// 多边形外边界，通过指定四条边来定义
	CBoundShape shape;
	shape.boundShape = POLYGON;
	shape.nNumSegs = 4;
	Segment seg1;
	seg1.p1 = Point(-1500, -1500);
	seg1.p2 = Point(1500, -1500);
	Segment seg2;
	seg2.p1 = Point(1500, -1500);
	seg2.p2 = Point(1500, 1500);
	Segment seg3;
	seg3.p1 = Point(1500, 1500);
	seg3.p2 = Point(-1500, 1500);
	Segment seg4;
	seg4.p1 = Point(-1500, 1500);
	seg4.p2 = Point(-1500, -1500);

	shape.vecSegments.push_back(seg1);
	shape.vecSegments.push_back(seg2);
	shape.vecSegments.push_back(seg3);
	shape.vecSegments.push_back(seg4);

	// shape.cCenter = Point(0, 0);
	// shape.dRadius = 1000;

	CBoundWell w1, w2;
	// 第一口井为直井（默认为直井）
	w1.cCenter = Point(0, 0); // 井中心坐标(0,0)
	w1.dRadius = 0.09144;			// 井半径0.09144

	// 第二口井为垂直裂缝井
	w2.wellType = VERTICAL_FRACTURED;
	w2.dHf = 20;							 // 裂缝半长20米
	w2.dTheata = 30;					 // 裂缝角度30°
	w2.cCenter = Point(50, 0); // 井的中心位置（50，0）
	w2.dRadius = 0.1;					 // 井的半径0.1

	// 第三口井为多段压裂水平井
	CBoundWell w3;
	w3.wellType = MULTI_FRACED_HORIZONTAL;
	w3.cCenter = Point(142, -226); // 中心点位置
	w3.dRadius = 0.1;							 // 半径
	w3.dBeta = 20;								 // 水平井角度
	w3.dLength = 200;							 // 水平井长度
	w3.nFracNum = 10;							 // 裂缝数量
	w3.dHf = 20;									 // 裂缝半长

	std::vector<CBoundWell> wells;
	wells.push_back(w1);
	wells.push_back(w2);
	wells.push_back(w3);

	CBoundPolygon limit;
	limit.boundShape = POLYGON;
	limit.nNumSegs = 5;
	limit.vecSegments.push_back(Segment(Point(-516.2335347838829, -58.37577524138442), Point(-592.72401802758, -180.9965861656542)));
	limit.vecSegments.push_back(Segment(Point(-592.72401802758, -180.9965861656542), Point(-485.12090993338666, -280.2195626173259)));
	limit.vecSegments.push_back(Segment(Point(-485.12090993338666, -280.2195626173259), Point(-328.9162066043042, -229.04373034669038)));
	limit.vecSegments.push_back(Segment(Point(-328.9162066043042, -229.04373034669038), Point(-376.36137535544276, -90.33591813134171)));
	limit.vecSegments.push_back(Segment(Point(-376.36137535544276, -90.33591813134171), Point(-516.2335347838829, -58.37577524138442)));
	limit.bAnchor = true;
	std::vector<CBoundPolygon> limits;
	limits.push_back(limit);

	// 断层
	std::vector<Frac> faults;
	faults.push_back(Frac(Point(-267.3, 120), Point(316.7, 150)));
	faults.push_back(Frac(Point(316.7, 150), Point(316.7, -50)));

	// 裂缝
	std::vector<Frac> fracs;
	fracs.push_back(Frac(Point(-326.159, 460.059), Point(165.83631, 459.085)));
	fracs.push_back(Frac(Point(50, -350), Point(358.840, -289.811)));

	/************************************************************************/
	/* load netgen mesh generation library and generate mesh                */
	/************************************************************************/
	HMODULE hMod = LoadLibrary("meshgene.dll");
	if (NULL == hMod)
	{
		std::cout << "meshgene.dll加载失败！\n";
		return 1;
	}

	typedef Ng_Mesh *(*NgMeshGen2DByIn2d)(std::vector<CBoundWell> &wells, const std::vector<CBoundPolygon> limits, const std::vector<Frac> &faults, const std::vector<Frac> &fracs, const CBoundShape &shape);
	typedef Ng_Mesh *(*NgMeshGen2D)(std::string strFileIn2D);
	typedef void *(*Ng2VTK)(Ng_Mesh *mesh);
	// Get the address of the Ng_Mesh function
	NgMeshGen2DByIn2d meshgene1 = (NgMeshGen2DByIn2d)GetProcAddress(hMod, "NgMeshGen2DByIn2d");
	NgMeshGen2D meshgene2 = (NgMeshGen2D)GetProcAddress(hMod, "NgMeshGen2D");
	if (NULL == meshgene1)
	{
		FreeLibrary(hMod);
		std::cout << "meshgene文件加载函数地址获取失败！\n";
		return 1;
	}
	Ng_Mesh *mesh = meshgene1(wells, limits, faults, fracs, shape);

	// Ng_Mesh *mesh=meshgene2("netgen-1.in2d");
	Ng2VTK ng = (Ng2VTK)GetProcAddress(hMod, "Ng2VTK");
	ng(mesh);
	return 0;
}