#ifndef _FUNC_MSH2VTK_H
#define _FUNC_MSH2VTK_H
#include "sysDefine_msh2vtk.h"

class mshvtk
{
private:
	nodeArray nArray;
	elementArray eArray;
	nodeDataArray ndArray;
	elementDataArray edArray;
	typesMap msh2vtk_type;
	typesMap type_nodenum;
	typesMap::iterator itm;
	int nodenum;
	int elenum;

	strArray outNames;
public:
	mshvtk(){
		msh2vtk_type[1] = 3;
		msh2vtk_type[2] = 5;
		msh2vtk_type[3] = 9;
		msh2vtk_type[4] = 10;
		msh2vtk_type[5] = 12;
		msh2vtk_type[6] = 13; //注意三棱柱顶点序列gmsh和vtk貌似有点区别 上下底面的三角形顶点排序刚好相反

		type_nodenum[1] = 2;
		type_nodenum[2] = 3;
		type_nodenum[3] = 4;
		type_nodenum[4] = 4;
		type_nodenum[5] = 8;
		type_nodenum[6] = 6;
	}
	~mshvtk(){}
	int runtine(char*);
	int addname(char*);
	int readmsh(char*); //注意这里的文件名都是不带.msh后缀的
	int writevtk(char*); //注意这里的文件名需要添加.vtk后缀
};

int mshvtk::runtine(char* parachar)
{
	if(readmsh(parachar)) return -1;
	if(writevtk(parachar)) return -1;
	nArray.clear();
	ndArray.clear();
	eArray.clear();
	edArray.clear();
	return 0;
}

int mshvtk::addname(char* onename)
{
	string s = onename;
	s = "\""+s+"\"";
	outNames.push_back(s);
	return 0;
}

int mshvtk::readmsh(char* inname)
{
	int temp_int,ele_num;
	node temp_node;
	element temp_element;
	nodeData temp_node_data;
	elementData temp_element_data;
	string temp_str;
	stringstream temp_ss;
	char filename[1024];

	strcpy(filename,inname);
	strcat(filename,".msh");
	ifstream infile;
	if(open_infile(infile,filename)) return -1;

	while(getline(infile,temp_str))
	{
		if (temp_str == "$Nodes")
		{
			getline(infile,temp_str);
			temp_ss.clear(); temp_ss.str(""); temp_ss << temp_str;
			temp_ss >> nodenum;
			for (int i = 0; i < nodenum; i++)
			{
				getline(infile,temp_str);
				temp_ss.clear();
				temp_ss << temp_str;
				temp_ss >> temp_node.id >> temp_node.x >> temp_node.y >> temp_node.z;
				nArray.push_back(temp_node);
			}
		}
		else if (temp_str == "$Elements")
		{
			getline(infile,temp_str);
			temp_ss.clear(); temp_ss.str(""); temp_ss << temp_str;
			temp_ss >> ele_num;
			for (int i = 0; i < ele_num; i++)
			{
				temp_element.msh_info.clear();
				temp_element.nodes_index.clear();
				getline(infile,temp_str);
				temp_ss.clear(); temp_ss.str(""); temp_ss << temp_str;
				temp_ss >> temp_element.id >> temp_element.msh_type >> temp_element.msh_info_num;
				for (int j = 0; j < temp_element.msh_info_num; j++)
				{
					temp_ss >> temp_int;
					temp_element.msh_info.push_back(temp_int);
				}
				for (int j = 0; j < type_nodenum[temp_element.msh_type]; j++)
				{
					temp_ss >> temp_int;
					if (nArray[0].id == 1)
					{
						temp_element.nodes_index.push_back(temp_int-1);
					}
					else temp_element.nodes_index.push_back(temp_int);
				}
				temp_element.vtk_type = msh2vtk_type[temp_element.msh_type];
				eArray.push_back(temp_element);
			}
		}
		else if (temp_str == "$ElementData")
		{
			temp_element_data.ids.clear();
			temp_element_data.val.clear();

			getline(infile,temp_str);
			temp_ss.clear(); temp_ss.str(""); temp_ss << temp_str;
			temp_ss >> temp_int;
			for (int i = 0; i < temp_int; i++)
			{
				getline(infile,temp_str);
				temp_element_data.name = temp_str;
			}
			//跳过6行
			for (int i = 0; i < 6; i++)
				getline(infile,temp_str);
			temp_ss.clear(); temp_ss.str(""); temp_ss << temp_str;
			temp_ss >> temp_int;

			temp_element_data.ids.resize(temp_int);
			temp_element_data.val.resize(temp_int);
			for (int i = 0; i < temp_int; i++)
			{
				getline(infile,temp_str);
				temp_ss.clear(); temp_ss.str(""); temp_ss << temp_str;
				temp_ss >> temp_element_data.ids[i] >> temp_element_data.val[i];
			}
			edArray.push_back(temp_element_data);
		}
		else if (temp_str == "$NodeData")
		{
			temp_node_data.ids.clear();
			temp_node_data.val.clear();

			getline(infile,temp_str);
			temp_ss.clear(); temp_ss.str(""); temp_ss << temp_str;
			temp_ss >> temp_int;
			for (int i = 0; i < temp_int; i++)
			{
				getline(infile,temp_str);
				temp_node_data.name = temp_str;
			}
			//跳过6行
			for (int i = 0; i < 6; i++)
				getline(infile,temp_str);
			temp_ss.clear(); temp_ss.str(""); temp_ss << temp_str;
			temp_ss >> temp_int;

			temp_node_data.ids.resize(temp_int);
			temp_node_data.val.resize(temp_int);
			for (int i = 0; i < temp_int; i++)
			{
				getline(infile,temp_str);
				temp_ss.clear(); temp_ss.str(""); temp_ss << temp_str;
				temp_ss >> temp_node_data.ids[i] >> temp_node_data.val[i];
			}
			ndArray.push_back(temp_node_data);
		}
	}
	infile.close();
	return 0;
}

int mshvtk::writevtk(char* inname)
{
	int temp_int;
	string temp_dataName;
	int totalCellNum = 0;
	char filename[1024];
	strcpy(filename,inname);
	strcat(filename,".vtk");
	ofstream outfile;
	if(open_outfile(outfile,filename)) return -1;

	outfile<<"# vtk DataFile Version 2.0"<<endl<<"Generated by msh2vtk"<<endl<<"ASCII"<<endl<<"DATASET UNSTRUCTURED_GRID"<<endl<<"POINTS "<<nArray.size()<<" FLOAT"<<endl;
	for (int i = 0; i < nArray.size(); i++)
	{
		outfile << setprecision(16) << nArray.at(i).x << " " << nArray.at(i).y << " " << nArray.at(i).z << endl;
	}
	//计算一下CELLS的总长 注意类型名算一个 所以要加1
	for (int i = 0; i < eArray.size(); i++)
	{
		totalCellNum += type_nodenum[eArray.at(i).msh_type] + 1;
	}
	outfile<<"CELLS "<<eArray.size()<<" "<<totalCellNum<<endl;
	for (int i = 0; i < eArray.size(); i++)
	{
		outfile << type_nodenum[eArray.at(i).msh_type] << " ";
		for (int j = 0; j < eArray.at(i).nodes_index.size(); j++)
		{
			outfile << eArray.at(i).nodes_index.at(j) << " ";
		}
		outfile << endl;
	}
	outfile << "CELL_TYPES " << eArray.size() << endl;
	for (int i = 0; i < eArray.size(); i++)
	{
		outfile << eArray.at(i).vtk_type << " ";
	}
	outfile << endl;
	for (int i = 0; i < ndArray.size(); i++)
	{
		for (int n = 0; n < outNames.size(); n++)
		{
			if (outNames[n] == ndArray[i].name)
			{
				temp_dataName = replace_all(ndArray[i].name," ","_",temp_int);
				outfile<<"POINT_DATA "<<ndArray[i].val.size()<<endl<<"SCALARS " << temp_dataName << " FLOAT"<<endl<<"LOOKUP_TABLE default"<<endl;
				for (int j = 0; j < ndArray[i].val.size(); j++)
				{
					outfile<<setprecision(16)<<ndArray[i].val[j]<<endl;
				}
			}
		}
	}
	for (int i = 0; i < edArray.size(); i++)
	{
		for (int n = 0; n < outNames.size(); n++)
		{
			if (outNames[n] == edArray[i].name)
			{
				temp_dataName = replace_all(edArray[i].name," ","_",temp_int);
				outfile<<"CELL_DATA "<<edArray[i].val.size()<<endl<<"SCALARS " << temp_dataName << " FLOAT"<<endl<<"LOOKUP_TABLE default"<<endl;
				for (int j = 0; j < edArray[i].val.size(); j++)
				{
					outfile<<setprecision(16)<<edArray[i].val[j]<<endl;
				}
			}
		}
	}
	outfile.close();
	return 0;
}
#endif