Compare commits
3 Commits
61d1f67b06
...
dcc99754b6
Author | SHA1 | Date | |
---|---|---|---|
dcc99754b6 | |||
5cb78e0881 | |||
80a3172c77 |
18
src/main.cc
18
src/main.cc
@ -44,10 +44,10 @@ int main(int argc, char **argv) {
|
|||||||
vtkNew<DihedralAnglesFilter> dihedralAnglesFilter;
|
vtkNew<DihedralAnglesFilter> dihedralAnglesFilter;
|
||||||
dihedralAnglesFilter->SetInputConnection(tetMeshReader->GetOutputPort());
|
dihedralAnglesFilter->SetInputConnection(tetMeshReader->GetOutputPort());
|
||||||
|
|
||||||
vtkNew<vtkXMLPolyDataReader> polyMeshReader;
|
// vtkNew<vtkXMLPolyDataReader> polyMeshReader;
|
||||||
polyMeshReader->SetFileName(argv[2]);
|
|
||||||
// vtkNew<vtkOBJReader> polyMeshReader;
|
|
||||||
// polyMeshReader->SetFileName(argv[2]);
|
// polyMeshReader->SetFileName(argv[2]);
|
||||||
|
vtkNew<vtkOBJReader> polyMeshReader;
|
||||||
|
polyMeshReader->SetFileName(argv[2]);
|
||||||
|
|
||||||
vtkNew<RemoveExternalCellsFilter> removeExternalCellsFilter;
|
vtkNew<RemoveExternalCellsFilter> removeExternalCellsFilter;
|
||||||
removeExternalCellsFilter->SetInputConnection(0,
|
removeExternalCellsFilter->SetInputConnection(0,
|
||||||
@ -55,12 +55,16 @@ int main(int argc, char **argv) {
|
|||||||
removeExternalCellsFilter->SetInputConnection(1,
|
removeExternalCellsFilter->SetInputConnection(1,
|
||||||
polyMeshReader->GetOutputPort());
|
polyMeshReader->GetOutputPort());
|
||||||
|
|
||||||
// vtkNew<ProjectSurfacePointsOnPoly> project;
|
vtkNew<SurfacePointsFilter> surfacePointsFilter;
|
||||||
// project->SetInputConnection(0, removeExternalCellsFilter->GetOutputPort());
|
surfacePointsFilter->SetInputConnection(
|
||||||
// project->SetInputConnection(1, pdReader->GetOutputPort());
|
removeExternalCellsFilter->GetOutputPort());
|
||||||
|
|
||||||
|
vtkNew<ProjectSurfacePointsOnPoly> project;
|
||||||
|
project->SetInputConnection(0, surfacePointsFilter->GetOutputPort());
|
||||||
|
project->SetInputConnection(1, polyMeshReader->GetOutputPort());
|
||||||
|
|
||||||
vtkNew<vtkXMLUnstructuredGridWriter> writer;
|
vtkNew<vtkXMLUnstructuredGridWriter> writer;
|
||||||
writer->SetInputConnection(removeExternalCellsFilter->GetOutputPort());
|
writer->SetInputConnection(project->GetOutputPort());
|
||||||
writer->SetDataModeToAscii();
|
writer->SetDataModeToAscii();
|
||||||
writer->SetFileName("out.vtu");
|
writer->SetFileName("out.vtu");
|
||||||
writer->Write();
|
writer->Write();
|
||||||
|
@ -41,27 +41,27 @@ vtkTypeBool ProjectSurfacePointsOnPoly::RequestData(
|
|||||||
vtkInformationVector **inputVector,
|
vtkInformationVector **inputVector,
|
||||||
vtkInformationVector *outputVector) {
|
vtkInformationVector *outputVector) {
|
||||||
(void) request;
|
(void) request;
|
||||||
vtkUnstructuredGrid* us = vtkUnstructuredGrid::GetData(inputVector[0]);
|
vtkUnstructuredGrid* tetMesh = vtkUnstructuredGrid::GetData(inputVector[0]);
|
||||||
vtkPolyData* pd = vtkPolyData::GetData(inputVector[1]);
|
vtkPolyData* polyMesh = vtkPolyData::GetData(inputVector[1]);
|
||||||
vtkUnstructuredGrid* output =
|
vtkUnstructuredGrid* output =
|
||||||
vtkUnstructuredGrid::GetData(outputVector);
|
vtkUnstructuredGrid::GetData(outputVector);
|
||||||
|
|
||||||
output->CopyStructure(us);
|
output->CopyStructure(tetMesh);
|
||||||
output->GetPointData()->PassData(us->GetPointData());
|
output->GetPointData()->PassData(tetMesh->GetPointData());
|
||||||
output->GetCellData()->PassData(us->GetCellData());
|
output->GetCellData()->PassData(tetMesh->GetCellData());
|
||||||
|
|
||||||
/* Generate cell links, these tell us which cell a point belongs to. */
|
/* Generate cell links, these tell us which cell a point belongs to. */
|
||||||
vtkNew<vtkStaticCellLinks> links;
|
vtkNew<vtkStaticCellLinks> links;
|
||||||
links->BuildLinks(pd);
|
links->BuildLinks(polyMesh);
|
||||||
|
|
||||||
vtkNew<vtkKdTree> kdTree;
|
vtkNew<vtkKdTree> kdTree;
|
||||||
kdTree->BuildLocatorFromPoints(pd->GetPoints());
|
kdTree->BuildLocatorFromPoints(polyMesh->GetPoints());
|
||||||
|
|
||||||
vtkIdTypeArray *externalPoints =
|
vtkIdTypeArray *surfacePoints =
|
||||||
vtkIdTypeArray::SafeDownCast(us->GetFieldData()->GetArray("external_points"));
|
vtkIdTypeArray::SafeDownCast(tetMesh->GetFieldData()->GetArray("surface_points"));
|
||||||
for (vtkIdType i = 0; i < externalPoints->GetNumberOfTuples(); i++) {
|
for (vtkIdType i = 0; i < surfacePoints->GetNumberOfTuples(); i++) {
|
||||||
double p[3];
|
double p[3];
|
||||||
us->GetPoint(externalPoints->GetTuple1(i), p);
|
tetMesh->GetPoint(surfacePoints->GetTuple1(i), p);
|
||||||
double dist2;
|
double dist2;
|
||||||
vtkIdType closest = kdTree->FindClosestPoint(p, dist2);
|
vtkIdType closest = kdTree->FindClosestPoint(p, dist2);
|
||||||
vtkIdType *cellIds = links->GetCells(closest);
|
vtkIdType *cellIds = links->GetCells(closest);
|
||||||
@ -73,7 +73,7 @@ vtkTypeBool ProjectSurfacePointsOnPoly::RequestData(
|
|||||||
for (vtkIdType j = 0; j < nCells; j++) {
|
for (vtkIdType j = 0; j < nCells; j++) {
|
||||||
vtkIdType cellId = cellIds[j];
|
vtkIdType cellId = cellIds[j];
|
||||||
double direction[3];
|
double direction[3];
|
||||||
double dist = pointTriangleDistance(p, pd->GetCell(cellId), direction);
|
double dist = pointTriangleDistance(p, polyMesh->GetCell(cellId), direction);
|
||||||
/* Find the closest one. */
|
/* Find the closest one. */
|
||||||
if (dist < min_dist) {
|
if (dist < min_dist) {
|
||||||
min_dist = dist;
|
min_dist = dist;
|
||||||
@ -85,8 +85,8 @@ vtkTypeBool ProjectSurfacePointsOnPoly::RequestData(
|
|||||||
|
|
||||||
vtkMath::MultiplyScalar(min_direction, min_dist);
|
vtkMath::MultiplyScalar(min_direction, min_dist);
|
||||||
vtkMath::Subtract(p, min_direction, p);
|
vtkMath::Subtract(p, min_direction, p);
|
||||||
us->GetPoints()->SetPoint(externalPoints->GetTuple1(i), p);
|
tetMesh->GetPoints()->SetPoint(surfacePoints->GetTuple1(i), p);
|
||||||
us->Modified();
|
tetMesh->Modified();
|
||||||
}
|
}
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
|
@ -52,7 +52,7 @@ vtkTypeBool SurfacePointsFilter::RequestData(
|
|||||||
|
|
||||||
/* Create an array to store the IDs of external points. */
|
/* Create an array to store the IDs of external points. */
|
||||||
vtkNew<vtkIdTypeArray> externalPoints;
|
vtkNew<vtkIdTypeArray> externalPoints;
|
||||||
externalPoints->SetName("external_points");
|
externalPoints->SetName("surface_points");
|
||||||
externalPoints->SetNumberOfComponents(1);
|
externalPoints->SetNumberOfComponents(1);
|
||||||
vtkNew<vtkIntArray> isExternal;
|
vtkNew<vtkIntArray> isExternal;
|
||||||
isExternal->SetName("bl");
|
isExternal->SetName("bl");
|
||||||
|
Loading…
Reference in New Issue
Block a user