256 lines
6.6 KiB
C++
256 lines
6.6 KiB
C++
#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 |