204 lines
4.6 KiB
C++
204 lines
4.6 KiB
C++
#ifndef _TETGEN2GMSH_H
|
||
#define _TETGEN2GMSH_H
|
||
#include "data_func.h"
|
||
|
||
class tetgen2gmsh
|
||
{
|
||
public:
|
||
tetgen2gmsh();
|
||
~tetgen2gmsh();
|
||
|
||
int call_back(string,string);//响应函数
|
||
int read_tetgen(string);//读取tetgen文件
|
||
int write_gmsh(string);//写入gmsh(.msh)文件
|
||
private:
|
||
int node_sum;//顶点数量与指针
|
||
vertex *vert;
|
||
int edge_sum;//边数量与指针
|
||
edge *edg;
|
||
int face_sum;//面数量与指针
|
||
face *fac;
|
||
int tetra_sum;//四面体数量与指针
|
||
tetra *tet;
|
||
|
||
string tetgen_name;//tetgen文件名
|
||
string gmsh_name;//gmsh文件名
|
||
};
|
||
|
||
tetgen2gmsh::tetgen2gmsh()//初始化指针
|
||
{
|
||
vert=NULL;
|
||
edg=NULL;
|
||
fac=NULL;
|
||
tet=NULL;
|
||
}
|
||
|
||
tetgen2gmsh::~tetgen2gmsh()//销毁动态数组
|
||
{
|
||
if(vert!=NULL) delete []vert;
|
||
if(edg!=NULL) delete []edg;
|
||
if(fac!=NULL) delete []fac;
|
||
if(tet!=NULL) delete []tet;
|
||
}
|
||
|
||
int tetgen2gmsh::call_back(string tetgen_name,string gmsh_name)
|
||
{
|
||
if(access((tetgen_name + ".node").c_str(),F_OK)||access((tetgen_name + ".edge").c_str(),F_OK)
|
||
||access((tetgen_name + ".face").c_str(),F_OK)||access((tetgen_name + ".ele").c_str(),F_OK))
|
||
{
|
||
cout<<"tetgen files are incomplete..."<<endl;
|
||
return 0;
|
||
}
|
||
read_tetgen(tetgen_name);
|
||
write_gmsh(gmsh_name);
|
||
return 0;
|
||
}
|
||
|
||
int tetgen2gmsh::read_tetgen(string filename)//读取tetgen文件，包括.node/.edge/.face/.ele文件,这个地方注意将点索引号置为从1开始
|
||
{
|
||
int first_no;
|
||
double temp;
|
||
string nodename,elename,facename,edgename;
|
||
|
||
nodename = filename + ".node";
|
||
elename = filename + ".ele";
|
||
facename = filename + ".face";
|
||
edgename = filename + ".edge";
|
||
const char* nodename_c = nodename.c_str();
|
||
const char* elename_c = elename.c_str();
|
||
const char* facename_c = facename.c_str();
|
||
const char* edgename_c = edgename.c_str();
|
||
|
||
ifstream node_in(nodename_c);
|
||
ifstream ele_in(elename_c);
|
||
ifstream face_in(facename_c);
|
||
ifstream edge_in(edgename_c);
|
||
|
||
node_in>>node_sum>>temp>>temp>>temp>>first_no;
|
||
node_in.seekg(-1,ios::cur);
|
||
vert = new vertex [node_sum];
|
||
if(first_no==1)
|
||
{
|
||
for(int i=0;i<node_sum;i++)
|
||
{
|
||
node_in>>vert[i].No>>vert[i].vt.x>>vert[i].vt.y>>vert[i].vt.z>>temp;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
for(int i=0;i<node_sum;i++)
|
||
{
|
||
node_in>>vert[i].No>>vert[i].vt.x>>vert[i].vt.y>>vert[i].vt.z;
|
||
vert[i].No+=1;
|
||
}
|
||
}
|
||
|
||
edge_in>>edge_sum>>temp;
|
||
edg = new edge [edge_sum];
|
||
if(first_no==1)
|
||
{
|
||
for(int i=0;i<edge_sum;i++)
|
||
{
|
||
edge_in>>edg[i].No>>edg[i].ver[0]>>edg[i].ver[1]>>temp;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
for(int i=0;i<edge_sum;i++)
|
||
{
|
||
edge_in>>edg[i].No>>edg[i].ver[0]>>edg[i].ver[1]>>temp;
|
||
edg[i].ver[0]+=1;
|
||
edg[i].ver[1]+=1;
|
||
}
|
||
}
|
||
|
||
face_in>>face_sum>>temp;
|
||
fac = new face [face_sum];
|
||
if(first_no==1)
|
||
{
|
||
for(int i=0;i<face_sum;i++)
|
||
{
|
||
face_in>>fac[i].No>>fac[i].ver[0]>>fac[i].ver[1]>>fac[i].ver[2]>>temp;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
for(int i=0;i<face_sum;i++)
|
||
{
|
||
face_in>>fac[i].No>>fac[i].ver[0]>>fac[i].ver[1]>>fac[i].ver[2]>>temp;
|
||
fac[i].ver[0]+=1;
|
||
fac[i].ver[1]+=1;
|
||
fac[i].ver[2]+=1;
|
||
}
|
||
}
|
||
|
||
ele_in>>tetra_sum>>temp>>temp;
|
||
tet = new tetra [tetra_sum];
|
||
if(first_no==1)
|
||
{
|
||
for(int i=0;i<tetra_sum;i++)
|
||
{
|
||
ele_in>>tet[i].No>>tet[i].ver[0]>>tet[i].ver[1]>>tet[i].ver[2]>>tet[i].ver[3];
|
||
}
|
||
}
|
||
else
|
||
{
|
||
for(int i=0;i<tetra_sum;i++)
|
||
{
|
||
ele_in>>tet[i].No>>tet[i].ver[0]>>tet[i].ver[1]>>tet[i].ver[2]>>tet[i].ver[3];
|
||
tet[i].ver[0]+=1;
|
||
tet[i].ver[1]+=1;
|
||
tet[i].ver[2]+=1;
|
||
tet[i].ver[3]+=1;
|
||
}
|
||
}
|
||
|
||
node_in.close();
|
||
ele_in.close();
|
||
face_in.close();
|
||
edge_in.close();
|
||
return 0;
|
||
}
|
||
|
||
int tetgen2gmsh::write_gmsh(string filename)//写入gmsh文件
|
||
{
|
||
int count = 1;
|
||
|
||
string outname = filename + ".msh";
|
||
const char* outname_c = outname.c_str();
|
||
ofstream msh_out(outname_c);
|
||
if(!msh_out)
|
||
{
|
||
cout<<"Can not create "<<outname;
|
||
return 1;
|
||
}
|
||
else
|
||
{
|
||
msh_out<<"$MeshFormat"<<endl<<"2.2 0 8"<<endl<<"$EndMeshFormat"<<endl<<"$Nodes"<<endl<<node_sum<<endl;
|
||
for(int i=0;i<node_sum;i++)
|
||
{
|
||
msh_out<<vert[i].No<<" "<<setprecision(16)<<vert[i].vt.x<<" "<<vert[i].vt.y<<" "<<vert[i].vt.z<<endl;
|
||
}
|
||
msh_out<<"$EndNodes"<<endl<<"$Elements"<<endl<<edge_sum+face_sum+tetra_sum<<endl;
|
||
for(int i=0;i<edge_sum;i++)
|
||
{
|
||
msh_out<<count<<" 1 2 0 1 "<<edg[i].ver[0]<<" "<<edg[i].ver[1]<<endl;
|
||
count++;
|
||
}
|
||
for(int i=0;i<face_sum;i++)
|
||
{
|
||
msh_out<<count<<" 2 2 0 14 "<<fac[i].ver[0]<<" "<<fac[i].ver[1]<<" "<<fac[i].ver[2]<<endl;
|
||
count++;
|
||
}
|
||
for(int i=0;i<tetra_sum;i++)
|
||
{
|
||
msh_out<<count<<" 4 2 0 26 "<<tet[i].ver[0]<<" "<<tet[i].ver[1]<<" "<<tet[i].ver[2]<<" "<<tet[i].ver[3]<<endl;
|
||
count++;
|
||
}
|
||
msh_out<<"$EndElements"<<endl;
|
||
}
|
||
msh_out.close();
|
||
return 0;
|
||
}
|
||
|
||
#endif |