kindred/create
1
1
Fork 0
Code Issues 53 Pull Requests Actions Packages Projects 9 Releases 2 Wiki Activity

Fem: Import/export 1D elements from mesh to vtk - fixes #15541

Browse Source
This commit is contained in:
marioalexis
2024-07-20 20:03:03 -03:00
committed by Chris Hennes
parent 896974807b
commit 8aba0f3bb0
2 changed files with 46 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/Mod/Fem/App/FemVTKTools.cpp
View File

@@ -40,9 +40,11 @@
#include <vtkDoubleArray.h>
#include <vtkHexahedron.h>
#include <vtkIdList.h>
#include <vtkLine.h>
#include <vtkPointData.h>
#include <vtkPyramid.h>
#include <vtkQuad.h>
#include <vtkQuadraticEdge.h>
#include <vtkQuadraticHexahedron.h>
#include <vtkQuadraticPyramid.h>
#include <vtkQuadraticQuad.h>
@@ -166,6 +168,13 @@ void FemVTKTools::importVTKMesh(vtkSmartPointer<vtkDataSet> dataset, FemMesh* me
std::vector<int> ids;
fillMeshElementIds(cell, ids);
switch (cell->GetCellType()) {
// 1D edges
case VTK_LINE: // seg2
meshds->AddEdgeWithID(ids[0], ids[1], iCell + 1);
break;
case VTK_QUADRATIC_EDGE: // seg3
meshds->AddEdgeWithID(ids[0], ids[1], ids[2], iCell + 1);
break;
// 2D faces
case VTK_TRIANGLE: // tria3
meshds->AddFaceWithID(ids[0], ids[1], ids[2], iCell + 1);
@@ -282,7 +291,7 @@ void FemVTKTools::importVTKMesh(vtkSmartPointer<vtkDataSet> dataset, FemMesh* me
// not handled cases
default: {
Base::Console().Error(
"Only common 2D and 3D Cells are supported in VTK mesh import\n");
"Only common 1D, 2D and 3D Cells are supported in VTK mesh import\n");
break;
}
}
@@ -330,6 +339,36 @@ FemMesh* FemVTKTools::readVTKMesh(const char* filename, FemMesh* mesh)
return mesh;
}
void exportFemMeshEdges(vtkSmartPointer<vtkUnstructuredGrid> grid,
const SMDS_EdgeIteratorPtr& aEdgeIter)
{
Base::Console().Log(" Start: VTK mesh builder edges.\n");
vtkSmartPointer<vtkCellArray> elemArray = vtkSmartPointer<vtkCellArray>::New();
std::vector<int> types;
while (aEdgeIter->more()) {
const SMDS_MeshEdge* aEdge = aEdgeIter->next();
// edge
if (aEdge->GetEntityType() == SMDSEntity_Edge) {
fillVtkArray<vtkLine>(elemArray, types, aEdge);
}
// quadratic edge
else if (aEdge->GetEntityType() == SMDSEntity_Quad_Edge) {
fillVtkArray<vtkQuadraticEdge>(elemArray, types, aEdge);
}
else {
throw Base::TypeError("Edge not yet supported by FreeCAD's VTK mesh builder\n");
}
}
if (elemArray->GetNumberOfCells() > 0) {
grid->SetCells(types.data(), elemArray);
}
Base::Console().Log(" End: VTK mesh builder edges.\n");
}
void exportFemMeshFaces(vtkSmartPointer<vtkUnstructuredGrid> grid,
const SMDS_FaceIteratorPtr& aFaceIter)
{
@@ -450,6 +489,10 @@ void FemVTKTools::exportVTKMesh(const FemMesh* mesh,
Base::Console().Log(" Size of nodes in VTK grid: %i.\n", nNodes);
Base::Console().Log(" End: VTK mesh builder nodes.\n");
// edges
SMDS_EdgeIteratorPtr aEdgeIter = meshDS->edgesIterator();
exportFemMeshEdges(grid, aEdgeIter);
// faces
SMDS_FaceIteratorPtr aFaceIter = meshDS->facesIterator();
exportFemMeshFaces(grid, aFaceIter);

2
src/Mod/Fem/App/PreCompiled.h
View File

@@ -163,12 +163,14 @@
#include <vtkHexahedron.h>
#include <vtkIdList.h>
#include <vtkImageData.h>
#include <vtkLine.h>
#include <vtkMultiBlockDataSet.h>
#include <vtkMultiPieceDataSet.h>
#include <vtkPointData.h>
#include <vtkPolyData.h>
#include <vtkPyramid.h>
#include <vtkQuad.h>
#include <vtkQuadraticEdge.h>
#include <vtkQuadraticHexahedron.h>
#include <vtkQuadraticPyramid.h>
#include <vtkQuadraticQuad.h>