5#include <vtkAppendFilter.h>
6#include <vtkAppendPolyData.h>
7#include <vtkCellData.h>
8#include <vtkDataSetTriangleFilter.h>
9#include <vtkFloatArray.h>
10#include <vtkGeometryFilter.h>
11#include <vtkIncrementalOctreePointLocator.h>
12#include <vtkIntArray.h>
13#include <vtkPointData.h>
14#include <vtkProbeFilter.h>
15#include <vtkRectilinearGrid.h>
16#include <vtkSmartPointer.h>
17#include <vtkTableBasedClipDataSet.h>
18#include <vtkTriangleFilter.h>
19#include <vtkUnstructuredGrid.h>
20#include <vtkXMLPolyDataWriter.h>
21#include <vtkXMLUnstructuredGridWriter.h>
23#include <hrleDenseIterator.hpp>
28#include <unordered_map>
32#ifdef LS_TO_VISUALIZATION_DEBUG
33#include <vtkXMLRectilinearGridWriter.h>
38using namespace viennacore;
46template <
class T,
int D>
class WriteVisualizationMesh {
47 typedef typename Domain<T, D>::DomainType hrleDomainType;
48 using LevelSetsType = std::vector<SmartPointer<Domain<T, D>>>;
49 LevelSetsType levelSets;
50 SmartPointer<MaterialMap> materialMap =
nullptr;
52 bool extractVolumeMesh =
true;
53 bool extractHullMesh =
false;
54 bool bottomRemoved =
false;
55 double LSEpsilon = 1e-2;
56 std::unordered_map<std::string, std::vector<double>> metaData;
61 static void removeDuplicatePoints(vtkSmartPointer<vtkPolyData> &polyData,
62 const double tolerance) {
64 vtkSmartPointer<vtkPolyData> newPolyData =
65 vtkSmartPointer<vtkPolyData>::New();
66 vtkSmartPointer<vtkIncrementalOctreePointLocator> ptInserter =
67 vtkSmartPointer<vtkIncrementalOctreePointLocator>::New();
68 ptInserter->SetTolerance(tolerance);
70 vtkSmartPointer<vtkPoints> newPoints = vtkSmartPointer<vtkPoints>::New();
74 polyData->GetBounds(gridBounds);
77 std::vector<vtkIdType> newPointIds;
78 newPointIds.reserve(polyData->GetNumberOfPoints());
79 ptInserter->InitPointInsertion(newPoints, gridBounds);
82 for (vtkIdType pointId = 0; pointId < polyData->GetNumberOfPoints();
84 vtkIdType globalPtId = 0;
85 ptInserter->InsertUniquePoint(polyData->GetPoint(pointId), globalPtId);
86 newPointIds.push_back(globalPtId);
90 newPolyData->SetPoints(newPoints);
93 vtkSmartPointer<vtkCellArray> oldCells = polyData->GetPolys();
94 vtkSmartPointer<vtkCellArray> newCells =
95 vtkSmartPointer<vtkCellArray>::New();
97 vtkSmartPointer<vtkIdList> cellPoints = vtkIdList::New();
98 oldCells->InitTraversal();
99 while (oldCells->GetNextCell(cellPoints)) {
100 for (vtkIdType pointId = 0; pointId < cellPoints->GetNumberOfIds();
102 cellPoints->SetId(pointId, newPointIds[cellPoints->GetId(pointId)]);
105 newCells->InsertNextCell(cellPoints);
108 newPolyData->SetPolys(newCells);
114 newPolyData->GetCellData()->ShallowCopy(polyData->GetCellData());
117 polyData = newPolyData;
122 static void removeDuplicatePoints(vtkSmartPointer<vtkUnstructuredGrid> &ugrid,
123 const double tolerance) {
125 vtkSmartPointer<vtkUnstructuredGrid> newGrid =
126 vtkSmartPointer<vtkUnstructuredGrid>::New();
127 vtkSmartPointer<vtkIncrementalOctreePointLocator> ptInserter =
128 vtkSmartPointer<vtkIncrementalOctreePointLocator>::New();
129 ptInserter->SetTolerance(tolerance);
131 vtkSmartPointer<vtkPoints> newPoints = vtkSmartPointer<vtkPoints>::New();
134 double gridBounds[6];
135 ugrid->GetBounds(gridBounds);
138 std::vector<vtkIdType> newPointIds;
139 newPointIds.reserve(ugrid->GetNumberOfPoints());
140 ptInserter->InitPointInsertion(newPoints, gridBounds);
143 for (vtkIdType pointId = 0; pointId < ugrid->GetNumberOfPoints();
145 vtkIdType globalPtId = 0;
146 ptInserter->InsertUniquePoint(ugrid->GetPoint(pointId), globalPtId);
147 newPointIds.push_back(globalPtId);
151 newGrid->SetPoints(newPoints);
154 for (vtkIdType cellId = 0; cellId < ugrid->GetNumberOfCells(); ++cellId) {
155 vtkSmartPointer<vtkIdList> cellPoints = vtkSmartPointer<vtkIdList>::New();
156 ugrid->GetCellPoints(cellId, cellPoints);
157 for (vtkIdType pointId = 0; pointId < cellPoints->GetNumberOfIds();
159 cellPoints->SetId(pointId, newPointIds[cellPoints->GetId(pointId)]);
162 newGrid->InsertNextCell(ugrid->GetCell(cellId)->GetCellType(),
170 newGrid->GetCellData()->ShallowCopy(ugrid->GetCellData());
178 removeDegenerateTetras(vtkSmartPointer<vtkUnstructuredGrid> &ugrid) {
179 vtkSmartPointer<vtkUnstructuredGrid> newGrid =
180 vtkSmartPointer<vtkUnstructuredGrid>::New();
183 vtkSmartPointer<vtkIntArray> materialNumberArray =
184 vtkSmartPointer<vtkIntArray>::New();
185 materialNumberArray->SetNumberOfComponents(1);
186 materialNumberArray->SetName(
"Material");
190 vtkDataArray *matArray =
191 ugrid->GetCellData()->GetArray(
"Material", arrayIndex);
192 const int &materialArrayIndex = arrayIndex;
195 for (vtkIdType cellId = 0; cellId < ugrid->GetNumberOfCells(); ++cellId) {
196 vtkSmartPointer<vtkIdList> cellPoints = vtkSmartPointer<vtkIdList>::New();
197 ugrid->GetCellPoints(cellId, cellPoints);
198 bool isDuplicate =
false;
199 for (vtkIdType pointId = 0; pointId < cellPoints->GetNumberOfIds();
201 for (vtkIdType nextId = pointId + 1;
202 nextId < cellPoints->GetNumberOfIds(); ++nextId) {
204 if (cellPoints->GetId(pointId) == cellPoints->GetId(nextId))
210 newGrid->InsertNextCell(ugrid->GetCell(cellId)->GetCellType(),
213 if (materialArrayIndex >= 0)
214 materialNumberArray->InsertNextValue(matArray->GetTuple1(cellId));
219 newGrid->SetPoints(ugrid->GetPoints());
220 newGrid->GetPointData()->ShallowCopy(ugrid->GetPointData());
222 newGrid->GetCellData()->SetScalars(materialNumberArray);
231 template <
int gr
idExtraPo
ints = 0>
232 vtkSmartPointer<vtkRectilinearGrid>
233 LS2RectiLinearGrid(SmartPointer<Domain<T, D>> levelSet,
const double LSOffset,
234 int infiniteMinimum = std::numeric_limits<int>::max(),
235 int infiniteMaximum = -std::numeric_limits<int>::max()) {
237 auto &grid = levelSet->getGrid();
238 auto &domain = levelSet->getDomain();
240 int numLayers = levelSet->getLevelSetWidth();
242 vtkSmartPointer<vtkFloatArray>
244 int gridMin = 0, gridMax = 0;
249 for (
unsigned i = 0; i <
D; ++i) {
250 coords[i] = vtkSmartPointer<vtkFloatArray>::New();
252 if (grid.getBoundaryConditions(i) ==
253 Domain<T, D>::BoundaryType::INFINITE_BOUNDARY) {
255 gridMin = std::min(domain.getMinRunBreak(i), infiniteMinimum) -
258 gridMax = std::max(domain.getMaxRunBreak(i), infiniteMaximum) + 1;
262 gridMin = grid.getMinGridPoint(i) - gridExtraPoints;
263 gridMax = grid.getMaxGridPoint(i) + gridExtraPoints;
266 for (
int x = gridMin; x <= gridMax; ++x) {
267 coords[i]->InsertNextValue(x * gridDelta);
273 coords[2] = vtkSmartPointer<vtkFloatArray>::New();
274 coords[2]->InsertNextValue(0);
277 vtkSmartPointer<vtkRectilinearGrid> rgrid =
278 vtkSmartPointer<vtkRectilinearGrid>::New();
280 rgrid->SetDimensions(coords[0]->GetNumberOfTuples(),
281 coords[1]->GetNumberOfTuples(),
282 coords[2]->GetNumberOfTuples());
283 rgrid->SetXCoordinates(coords[0]);
284 rgrid->SetYCoordinates(coords[1]);
285 rgrid->SetZCoordinates(coords[2]);
297 auto const numGridPoints = rgrid->GetNumberOfPoints();
298 vtkSmartPointer<vtkFloatArray> signedDistances =
299 vtkSmartPointer<vtkFloatArray>::New();
300 signedDistances->SetNumberOfComponents(1);
301 signedDistances->SetNumberOfTuples(numGridPoints);
302 signedDistances->SetName(
"SignedDistances");
307 viennahrle::ConstDenseIterator<typename Domain<T, D>::DomainType> it(
308 levelSet->getDomain());
311 for (vtkIdType pointId = 0; pointId < numGridPoints; ++pointId) {
315 rgrid->GetPoint(pointId, p);
317 viennahrle::Index<D> indices(grid.globalCoordinates2GlobalIndices(p));
323 if (grid.isOutsideOfDomain(indices)) {
324 indices = grid.globalIndices2LocalIndices(indices);
330 it.goToIndices(indices);
331 if (it.getValue() == Domain<T, D>::POS_VALUE) {
333 }
else if (it.getValue() == Domain<T, D>::NEG_VALUE) {
336 value = it.getValue() + LSOffset;
350 signedDistances->SetValue(pointId, value * gridDelta);
407 rgrid->GetPointData()->SetScalars(signedDistances);
412 void addMetaDataToVTK(vtkDataSet *data)
const {
413 if (metaData.empty()) {
418 vtkSmartPointer<vtkFieldData> fieldData = data->GetFieldData();
419 for (
const auto &meta : metaData) {
420 if (meta.second.empty())
423 vtkSmartPointer<vtkFloatArray> metaDataArray =
424 vtkSmartPointer<vtkFloatArray>::New();
425 metaDataArray->SetName(meta.first.c_str());
426 metaDataArray->SetNumberOfValues(meta.second.size());
427 for (
size_t i = 0; i < meta.second.size(); ++i) {
428 metaDataArray->SetValue(i, meta.second[i]);
430 fieldData->AddArray(metaDataArray);
435 WriteVisualizationMesh() =
default;
437 WriteVisualizationMesh(SmartPointer<Domain<T, D>> levelSet) {
438 levelSets.push_back(levelSet);
442 void insertNextLevelSet(SmartPointer<Domain<T, D>> levelSet) {
443 levelSets.push_back(levelSet);
446 void clearLevelSets() { levelSets.clear(); }
450 void setFileName(std::string passedFileName) {
451 fileName = std::move(passedFileName);
455 void setExtractHullMesh(
bool passedExtractHullMesh) {
456 extractHullMesh = passedExtractHullMesh;
460 void setExtractVolumeMesh(
bool passedExtractVolumeMesh) {
461 extractVolumeMesh = passedExtractVolumeMesh;
464 void setMaterialMap(SmartPointer<MaterialMap> passedMaterialMap) {
465 materialMap = passedMaterialMap;
468 void setWrappingLayerEpsilon(
double epsilon) { LSEpsilon = epsilon; }
470 void setMetaData(
const std::unordered_map<std::string, std::vector<double>>
472 metaData = passedMetaData;
475 void addMetaData(
const std::string &key,
double value) {
476 metaData[key] = std::vector<double>{value};
479 void addMetaData(
const std::string &key,
const std::vector<double> &values) {
480 metaData[key] = values;
484 const std::unordered_map<std::string, std::vector<double>> &newMetaData) {
485 for (
const auto &pair : newMetaData) {
486 metaData[pair.first] = pair.second;
492 for (
unsigned i = 0; i < levelSets.size(); ++i) {
493 if (levelSets[i]->getLevelSetWidth() < 2) {
494 Logger::getInstance()
496 "WriteVisualizationMesh: Level Set " + std::to_string(i) +
497 " should have a width greater than 1! Conversion might fail!")
502 const double gridDelta = levelSets[0]->getGrid().getGridDelta();
505 std::vector<vtkSmartPointer<vtkUnstructuredGrid>> materialMeshes;
506 std::vector<unsigned> materialIds;
508 int totalMinimum = std::numeric_limits<int>::max();
509 int totalMaximum = -std::numeric_limits<int>::max();
510 for (
auto &it : levelSets) {
511 if (it->getNumberOfPoints() == 0) {
514 auto &grid = it->getGrid();
515 auto &domain = it->getDomain();
516 for (
unsigned i = 0; i <
D; ++i) {
517 if (grid.getBoundaryConditions(i) ==
518 Domain<T, D>::BoundaryType::INFINITE_BOUNDARY) {
519 totalMinimum = std::min(totalMinimum, domain.getMinRunBreak(i));
520 totalMaximum = std::max(totalMaximum, domain.getMaxRunBreak(i));
527 vtkSmartPointer<vtkTableBasedClipDataSet> clipper =
528 vtkSmartPointer<vtkTableBasedClipDataSet>::New();
529 auto topGrid = vtkSmartPointer<vtkRectilinearGrid>::New();
532 LS2RectiLinearGrid(levelSets.back(), 0, totalMinimum, totalMaximum);
538#ifdef LS_TO_VISUALIZATION_DEBUG
540 auto gwriter = vtkSmartPointer<vtkXMLRectilinearGridWriter>::New();
541 gwriter->SetFileName(
"./grid_0.vtr");
542 gwriter->SetInputData(topGrid);
544 std::cout <<
"Wrote grid 0" << std::endl;
547 clipper->SetInputData(topGrid);
548 clipper->InsideOutOn();
549 clipper->SetValue(0.0);
550 clipper->GenerateClippedOutputOn();
553#ifdef LS_TO_VISUALIZATION_DEBUG
555 auto gwriter = vtkSmartPointer<vtkXMLUnstructuredGridWriter>::New();
556 gwriter->SetFileName(
"./clipped.vtu");
557 gwriter->SetInputData(clipper->GetClippedOutput());
559 std::cout <<
"Wrote clipped" << std::endl;
563 const bool useMaterialMap = materialMap !=
nullptr;
564 materialMeshes.emplace_back(clipper->GetOutput());
565 materialIds.push_back(useMaterialMap ? materialMap->getMaterialId(0) : 0);
567#ifdef LS_TO_VISUALIZATION_DEBUG
569 auto gwriter = vtkSmartPointer<vtkXMLUnstructuredGridWriter>::New();
570 gwriter->SetFileName(
"./probed_0.vtu");
571 gwriter->SetInputData(materialMeshes.front());
579 for (
typename LevelSetsType::const_reverse_iterator it =
580 ++levelSets.rbegin();
581 it != levelSets.rend(); ++it) {
582 if (it->get()->getNumberOfPoints() == 0)
587 vtkSmartPointer<vtkRectilinearGrid> rgrid =
588 vtkSmartPointer<vtkRectilinearGrid>::New();
590 rgrid = LS2RectiLinearGrid<1>(*it, -LSEpsilon * counter, totalMinimum,
597#ifdef LS_TO_VISUALIZATION_DEBUG
599 vtkSmartPointer<vtkXMLRectilinearGridWriter> gwriter =
600 vtkSmartPointer<vtkXMLRectilinearGridWriter>::New();
601 gwriter->SetFileName(
602 (
"./grid_" + std::to_string(counter) +
".vtr").c_str());
603 gwriter->SetInputData(rgrid);
605 std::cout <<
"Wrote grid " <<
counter << std::endl;
610 vtkSmartPointer<vtkProbeFilter> probeFilter =
611 vtkSmartPointer<vtkProbeFilter>::New();
612 probeFilter->SetInputData(materialMeshes.back());
613 probeFilter->SetSourceData(rgrid);
614 probeFilter->Update();
616#ifdef LS_TO_VISUALIZATION_DEBUG
618 vtkSmartPointer<vtkXMLUnstructuredGridWriter> gwriter =
619 vtkSmartPointer<vtkXMLUnstructuredGridWriter>::New();
620 gwriter->SetFileName(
621 (
"./probed_" + std::to_string(counter) +
".vtu").c_str());
622 gwriter->SetInputData(probeFilter->GetOutput());
624 std::cout <<
"Wrote unstructured grid " <<
counter << std::endl;
631 vtkSmartPointer<vtkTableBasedClipDataSet> insideClipper =
632 vtkSmartPointer<vtkTableBasedClipDataSet>::New();
633 insideClipper->SetInputConnection(probeFilter->GetOutputPort());
634 insideClipper->GenerateClippedOutputOn();
635 insideClipper->Update();
637 materialMeshes.back() = insideClipper->GetOutput();
638 materialMeshes.emplace_back(insideClipper->GetClippedOutput());
641 material = materialMap->getMaterialId(counter);
642 materialIds.push_back(material);
647 vtkSmartPointer<vtkAppendFilter> appendFilter =
648 vtkSmartPointer<vtkAppendFilter>::New();
650 vtkSmartPointer<vtkAppendPolyData> hullAppendFilter =
651 vtkSmartPointer<vtkAppendPolyData>::New();
653 for (
unsigned i = 0; i < materialMeshes.size(); ++i) {
656 vtkSmartPointer<vtkIntArray> materialNumberArray =
657 vtkSmartPointer<vtkIntArray>::New();
658 materialNumberArray->SetNumberOfComponents(1);
659 materialNumberArray->SetName(
"Material");
662 materialMeshes[materialMeshes.size() - 1 - i]->GetNumberOfCells();
664 materialNumberArray->InsertNextValue(materialIds[i]);
666 materialMeshes[materialMeshes.size() - 1 - i]->GetCellData()->SetScalars(
667 materialNumberArray);
673 vtkSmartPointer<vtkPointData> pointData =
674 materialMeshes[materialMeshes.size() - 1 - i]->GetPointData();
675 const int numberOfArrays = pointData->GetNumberOfArrays();
676 for (
int j = 0; j < numberOfArrays; ++j) {
677 pointData->RemoveArray(0);
682 if (extractHullMesh) {
683 vtkSmartPointer<vtkGeometryFilter> geoFilter =
684 vtkSmartPointer<vtkGeometryFilter>::New();
685 geoFilter->SetInputData(materialMeshes[materialMeshes.size() - 1 - i]);
687 hullAppendFilter->AddInputData(geoFilter->GetOutput());
690 appendFilter->AddInputData(materialMeshes[materialMeshes.size() - 1 - i]);
694 auto volumeVTK = vtkSmartPointer<vtkUnstructuredGrid>::New();
695 auto hullVTK = vtkSmartPointer<vtkPolyData>::New();
696 if (extractVolumeMesh) {
697 appendFilter->Update();
701 volumeVTK = appendFilter->GetOutput();
702#ifdef LS_TO_VISUALIZATION_DEBUG
704 std::cout <<
"Before duplicate removal: " << std::endl;
705 std::cout <<
"Points: " << volumeVTK->GetNumberOfPoints() << std::endl;
706 std::cout <<
"Cells: " << volumeVTK->GetNumberOfCells() << std::endl;
707 vtkSmartPointer<vtkXMLUnstructuredGridWriter> gwriter =
708 vtkSmartPointer<vtkXMLUnstructuredGridWriter>::New();
709 gwriter->SetFileName(
"before_removal.vtu");
710 gwriter->SetInputData(appendFilter->GetOutput());
717 removeDuplicatePoints(volumeVTK, 1e-3 * gridDelta);
719#ifdef LS_TO_VISUALIZATION_DEBUG
721 std::cout <<
"After duplicate removal: " << std::endl;
722 std::cout <<
"Points: " << volumeVTK->GetNumberOfPoints() << std::endl;
723 std::cout <<
"Cells: " << volumeVTK->GetNumberOfCells() << std::endl;
724 vtkSmartPointer<vtkXMLUnstructuredGridWriter> gwriter =
725 vtkSmartPointer<vtkXMLUnstructuredGridWriter>::New();
726 gwriter->SetFileName(
"after_removal.vtu");
727 gwriter->SetInputData(volumeVTK);
733 vtkSmartPointer<vtkDataSetTriangleFilter> triangleFilter =
734 vtkSmartPointer<vtkDataSetTriangleFilter>::New();
735 triangleFilter->SetInputData(volumeVTK);
736 triangleFilter->Update();
737 volumeVTK = triangleFilter->GetOutput();
740 removeDegenerateTetras(volumeVTK);
743 addMetaDataToVTK(volumeVTK);
745 auto writer = vtkSmartPointer<vtkXMLUnstructuredGridWriter>::New();
746 writer->SetFileName((fileName +
"_volume.vtu").c_str());
747 writer->SetInputData(volumeVTK);
752 if (extractHullMesh) {
753 hullAppendFilter->Update();
754 hullVTK = hullAppendFilter->GetOutput();
756 removeDuplicatePoints(hullVTK, 1e-3 * gridDelta);
758 vtkSmartPointer<vtkTriangleFilter> hullTriangleFilter =
759 vtkSmartPointer<vtkTriangleFilter>::New();
760 hullTriangleFilter->SetInputData(hullVTK);
761 hullTriangleFilter->Update();
763 hullVTK = hullTriangleFilter->GetOutput();
766 addMetaDataToVTK(hullVTK);
768 auto writer = vtkSmartPointer<vtkXMLPolyDataWriter>::New();
769 writer->SetFileName((fileName +
"_hull.vtp").c_str());
770 writer->SetInputData(hullVTK);
constexpr int D
Definition Epitaxy.cpp:11
double T
Definition Epitaxy.cpp:12
#define PRECOMPILE_PRECISION_DIMENSION(className)
Definition lsPreCompileMacros.hpp:24
float gridDelta
Definition AirGapDeposition.py:21
writer
Definition AirGapDeposition.py:89
int counter
Definition Deposition.py:75
Definition lsAdvect.hpp:36