manolas
/
vcglib
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

better MSH tetra importer

This commit is contained in:
T.Alderighi 2018-05-29 16:21:17 +02:00
parent fbda3f32f8
commit 95ecdf424e
1 changed files with 142 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

190
wrap/io_tetramesh/import_msh.h
View File

@ -62,9 +62,9 @@ class ImporterMSH
if (index < 0) if (index < 0)
return INVALID_FORMAT; return INVALID_FORMAT;
m.vert[index].P().X() = *reinterpret_cast<float *>(&data[i * lineBytes + 4]); m.vert[index].P().X() = *reinterpret_cast<double *>(&data[i * lineBytes + 4]);
m.vert[index].P().Y() = *reinterpret_cast<float *>(&data[i * lineBytes + 4 + 8]); m.vert[index].P().Y() = *reinterpret_cast<double *>(&data[i * lineBytes + 4 + 8]);
m.vert[index].P().Z() = *reinterpret_cast<float *>(&data[i * lineBytes + 4 + 2 * 8]); m.vert[index].P().Z() = *reinterpret_cast<double *>(&data[i * lineBytes + 4 + 2 * 8]);
} }
delete[] data; delete[] data;
} }
@ -103,18 +103,16 @@ class ImporterMSH
while (parsedElems < numOfElements) while (parsedElems < numOfElements)
{ {
int index, type, tags; //MSH in binary format has a elem-header 3*4 bytes: {elems_type, numElems, tagsPerElem}
fin.read((char *)&index, sizeof(int)); //followed by the list of elems under this header and eventually a new header and list.
fin.read((char *)&type, sizeof(int)); int type, elements, tags;
fin.read((char *)&tags, sizeof(int)); fin.read((char *)&type, sizeof(int));
--index; fin.read((char *)&elements, sizeof(int));
fin.read((char *)&tags, sizeof(int));
//check for tetra type //check for tetra type
if (type != 4) if (type != 4)
return NOT_IMPLEMENTED; return NOT_IMPLEMENTED;
//check index validity
if (index < 0)
return INVALID_FORMAT;
//read tags and throw them //read tags and throw them
for (size_t j = 0; j < tags; ++j) for (size_t j = 0; j < tags; ++j)
@ -123,20 +121,32 @@ class ImporterMSH
fin.read((char *)&tag, sizeof(int)); fin.read((char *)&tag, sizeof(int));
} }
//read element nodes //foreach element
TetraType t = m.tetra[index]; for (int i = 0; i < elements; ++i)
for (int i = 0; i < 4; ++i)
{ {
int nodeIndex; int index;
fin.read((char *)&nodeIndex, sizeof(int)); fin.read((char *)&index, sizeof(int));
--nodeIndex; --index;
if (nodeIndex < 0 || nodeIndex >= m.VN()) //check index validity
if (index < 0)
return INVALID_FORMAT; return INVALID_FORMAT;
t.V(i) = &m.vert[nodeIndex]; //read element nodes
TetraType * t = &m.tetra[index];
for (int i = 0; i < 4; ++i)
{
int nodeIndex;
fin.read((char *)&nodeIndex, sizeof(int));
--nodeIndex;
if (nodeIndex < 0 || nodeIndex >= m.VN())
return INVALID_FORMAT;
t->V(i) = &m.vert[nodeIndex];
}
++parsedElems;
} }
++parsedElems;
} }
} }
else else
@ -147,8 +157,6 @@ class ImporterMSH
fin >> index >> type >> tags; fin >> index >> type >> tags;
--index; --index;
// std::cerr << index << std::endl;
//check for tetra type //check for tetra type
if (type != 4) if (type != 4)
return NOT_IMPLEMENTED; return NOT_IMPLEMENTED;
@ -162,7 +170,7 @@ class ImporterMSH
fin >> tag; fin >> tag;
} }
TetraType * t = &m.tetra[index]; TetraType *t = &m.tetra[index];
for (int i = 0; i < 4; ++i) for (int i = 0; i < 4; ++i)
{ {
int nodeIndex; int nodeIndex;
@ -171,7 +179,6 @@ class ImporterMSH
if (nodeIndex < 0 || nodeIndex > m.VN()) if (nodeIndex < 0 || nodeIndex > m.VN())
return INVALID_FORMAT; return INVALID_FORMAT;
// std::cerr << nodeIndex << std::endl;
t->V(i) = &m.vert[nodeIndex]; t->V(i) = &m.vert[nodeIndex];
} }
@ -181,6 +188,106 @@ class ImporterMSH
return 0; return 0;
} }
static int parseDataField(MeshType &m, std::ifstream &fin, bool binary)
{
int numString, numReal, numInteger;
fin >> numString;
std::string *strTags = new std::string[numString];
for (int i = 0; i < numString; ++i)
{
parseWhiteSpace(fin);
fin >> strTags[i];
}
fin >> numReal;
double *doubleTags = new double[numReal];
for (int i = 0; i < numReal; ++i)
fin >> doubleTags[i];
fin >> numInteger;
if (numString <= 0 || numInteger < 3)
return INVALID_FORMAT;
int *integerTags = new int[numInteger];
for (int i = 0; i < numInteger; ++i)
fin >> integerTags[i];
std::string fieldName = strTags[0];
int fieldComponents = integerTags[1];
int fieldSize = integerTags[2];
double *fieldVec = new double[fieldComponents * fieldSize];
delete[] strTags;
delete[] doubleTags;
delete[] integerTags;
if (binary)
{
size_t totalBytes = (4 + 8 * fieldComponents) * fieldSize;
char *data = new char[totalBytes];
parseWhiteSpace(fin);
fin.read(data, totalBytes);
for (int i = 0; i < fieldSize; ++i)
{
int index = *reinterpret_cast<int *>(&data[i * (4 + fieldComponents * 8)]);
--index;
if (index < 0)
return INVALID_FORMAT;
//values
int baseIndex = i * (4 + fieldComponents * 8) + 4;
for (int j = 0; j < fieldComponents; ++j)
fieldVec[index * fieldComponents + j] = *reinterpret_cast<float *>(&data[baseIndex + j * 8]);
}
}
else
{
for (int i = 0; i < fieldSize; ++i)
{
int index;
fin >> index;
--index;
if (index < 0)
return INVALID_FORMAT;
for (int j = 0; j < fieldComponents; ++j)
fin >> fieldVec[index * fieldComponents + j];
}
}
}
static int parseNodeData(MeshType &m, std::ifstream &fin, bool binary)
{
return parseDataField(m, fin, binary);
}
static int parseElementData(MeshType &m, std::ifstream &fin, bool binary)
{
return parseDataField(m, fin, binary);
}
static int parseUnsupportedTag(std::ifstream &fin, std::string &tag)
{
std::cerr << "found unsupported tag" << std::endl;
std::string tagName = tag.substr(1, tag.size() - 1);
std::string tagEnd = tag.substr(0, 1) + "End" + tagName;
std::string buf;
while (buf != tagEnd && !fin.eof())
fin >> buf;
}
static int parseMshMesh(MeshType &m, std::string &filename) static int parseMshMesh(MeshType &m, std::string &filename)
{ {
std::ifstream fin(filename.c_str(), std::ios::in | std::ios::binary); std::ifstream fin(filename.c_str(), std::ios::in | std::ios::binary);
@ -194,7 +301,6 @@ class ImporterMSH
if (lookAhead != "$MeshFormat") if (lookAhead != "$MeshFormat")
return INVALID_FORMAT; return INVALID_FORMAT;
double version; double version;
int type, dataSize; int type, dataSize;
@ -208,7 +314,7 @@ class ImporterMSH
if (dataSize != 8) if (dataSize != 8)
return INVALID_FORMAT; return INVALID_FORMAT;
// Read in extra info from binary header. // Read endiannes info in binary header...it's a 1 used to detect endiannes.
if (binary) if (binary)
{ {
int one; int one;
@ -228,8 +334,6 @@ class ImporterMSH
if (lookAhead != "$EndMeshFormat") if (lookAhead != "$EndMeshFormat")
return INVALID_FORMAT; return INVALID_FORMAT;
std::cerr << "reading nodes and elements" << std::endl;
while (!fin.eof()) while (!fin.eof())
{ {
lookAhead.clear(); lookAhead.clear();
@ -238,13 +342,8 @@ class ImporterMSH
if (lookAhead == "$Nodes") if (lookAhead == "$Nodes")
{ {
int res = parseNodes(m, fin, binary); int res = parseNodes(m, fin, binary);
std::cerr << "reading nodes: " << res << std::endl;
if (res != 0) if (res != 0)
return res; return res;
std::cerr << "reading nodes" << std::endl;
fin >> lookAhead; fin >> lookAhead;
if (lookAhead != "$EndNodes") if (lookAhead != "$EndNodes")
@ -257,28 +356,24 @@ class ImporterMSH
if (res != 0) if (res != 0)
return res; return res;
std::cerr << "elements" << std::endl;
fin >> lookAhead; fin >> lookAhead;
if (lookAhead != "$EndElements") if (lookAhead != "$EndElements")
return INVALID_FORMAT; return INVALID_FORMAT;
} }
else if (lookAhead == "$NodeData") else if (lookAhead == "$NodeData")
{ {
return NOT_IMPLEMENTED; parseNodeData(m, fin, binary);
// parse_node_field(fin); fin >> lookAhead;
// fin >> lookAhead; if (lookAhead != "$EndNodeData")
// if (lookAhead != "$EndNodeData") return INVALID_FORMAT;
// return INVALID_FORMAT;
} }
else if (lookAhead == "$ElementData") else if (lookAhead == "$ElementData")
{ {
return NOT_IMPLEMENTED; parseElementData(m, fin, binary);
// parse_element_field(fin); fin >> lookAhead;
// fin >> lookAhead; if (lookAhead != "$EndElementData")
// if (lookAhead != "$EndElementData") return INVALID_FORMAT;
// return INVALID_FORMAT;
} }
else if (fin.eof()) else if (fin.eof())
{ {
@ -286,8 +381,7 @@ class ImporterMSH
} }
else else
{ {
return INVALID_FORMAT; parseUnsupportedTag(fin, lookAhead);
// parse_unknown_field(fin, lookAhead);
} }
} }
fin.close(); fin.close();