VTK 实例54：封闭性检测

  1 #include "vtkAutoInit.h" 
  2 VTK_MODULE_INIT(vtkRenderingOpenGL2);
  3 VTK_MODULE_INIT(vtkInteractionStyle);
  4 
  5 #include <vtkSmartPointer.h>
  6 #include <vtkSelectionNode.h>
  7 #include <vtkInformation.h>
  8 #include <vtkUnstructuredGrid.h>
  9 #include <vtkPolyData.h>
 10 #include <vtkPolyDataNormals.h>
 11 #include <vtkPointData.h>
 12 #include <vtkXMLPolyDataReader.h>
 13 #include <vtkRenderWindow.h>
 14 #include <vtkRenderWindowInteractor.h>
 15 #include <vtkRenderer.h>
 16 #include <vtkSelection.h>
 17 #include <vtkSelectionNode.h>
 18 #include <vtkSphereSource.h>
 19 #include <vtkPolyDataMapper.h>
 20 #include <vtkActor.h>
 21 #include <vtkCamera.h>
 22 #include <vtkProperty.h>
 23 #include <vtkIdTypeArray.h>
 24 #include <vtkExtractSelection.h>
 25 #include <vtkDataSetSurfaceFilter.h>
 26 #include <vtkFeatureEdges.h>
 27 #include <vtkFillHolesFilter.h>
 28 
 29 void GenerateData(vtkSmartPointer<vtkPolyData> input)
 30 {
 31     vtkSmartPointer<vtkSphereSource> sphereSource =
 32         vtkSmartPointer<vtkSphereSource>::New();
 33     sphereSource->Update();
 34 
 35     vtkSmartPointer<vtkIdTypeArray> ids =
 36         vtkSmartPointer<vtkIdTypeArray>::New();
 37     ids->SetNumberOfComponents(1);
 38     ids->InsertNextValue(2);
 39     ids->InsertNextValue(10);
 40 
 41     vtkSmartPointer<vtkSelectionNode> selectionNode =
 42         vtkSmartPointer<vtkSelectionNode>::New();
 43     selectionNode->SetFieldType(vtkSelectionNode::CELL);
 44     selectionNode->SetContentType(vtkSelectionNode::INDICES);
 45     selectionNode->SetSelectionList(ids);
 46     selectionNode->GetProperties()->Set(vtkSelectionNode::INVERSE(), 1);
 47 
 48     vtkSmartPointer<vtkSelection> selection =
 49         vtkSmartPointer<vtkSelection>::New();
 50     selection->AddNode(selectionNode);
 51 
 52     vtkSmartPointer<vtkExtractSelection> extractSelection =
 53         vtkSmartPointer<vtkExtractSelection>::New();
 54     extractSelection->SetInputData(0, sphereSource->GetOutput());
 55     extractSelection->SetInputData(1, selection);
 56     extractSelection->Update();
 57 
 58     vtkSmartPointer<vtkDataSetSurfaceFilter> surfaceFilter =
 59         vtkSmartPointer<vtkDataSetSurfaceFilter>::New();
 60     surfaceFilter->SetInputConnection(extractSelection->GetOutputPort());
 61     surfaceFilter->Update();
 62 
 63     input->ShallowCopy(surfaceFilter->GetOutput());
 64 }
 65 
 66 int main(int argc, char *argv[])
 67 {
 68     vtkSmartPointer<vtkPolyData> input =
 69         vtkSmartPointer<vtkPolyData>::New();
 70     GenerateData(input);
 71 
 72     vtkSmartPointer<vtkFeatureEdges> featureEdges =
 73         vtkSmartPointer<vtkFeatureEdges>::New();
 74     featureEdges->SetInputData(input);
 75     featureEdges->BoundaryEdgesOn();
 76     featureEdges->FeatureEdgesOff();
 77     featureEdges->ManifoldEdgesOff();
 78     featureEdges->NonManifoldEdgesOff();
 79     featureEdges->Update();
 80 
 81     int numberOfOpenEdges = featureEdges->GetOutput()->GetNumberOfCells();
 82     if(numberOfOpenEdges) 
 83     {
 84         std::cout<<"该网格模型不是封闭的..."<<std::endl;
 85     }
 86     else
 87     {
 88         std::cout<<"该网格模型是封闭的..."<<std::endl;
 89         return EXIT_SUCCESS;
 90     }
 91 
 92     vtkSmartPointer<vtkFillHolesFilter> fillHolesFilter =
 93         vtkSmartPointer<vtkFillHolesFilter>::New();
 94     fillHolesFilter->SetInputData(input);
 95     fillHolesFilter->Update();
 96 
 97     vtkSmartPointer<vtkPolyDataNormals> normals =
 98         vtkSmartPointer<vtkPolyDataNormals>::New();
 99     normals->SetInputConnection(fillHolesFilter->GetOutputPort());
100     normals->ConsistencyOn();
101     normals->SplittingOff();
102     normals->Update();
103 
104     //
105     double leftViewport[4] = {0.0, 0.0, 0.5, 1.0};
106     double rightViewport[4] = {0.5, 0.0, 1.0, 1.0};
107 
108     vtkSmartPointer<vtkPolyDataMapper> originalMapper =
109         vtkSmartPointer<vtkPolyDataMapper>::New();
110     originalMapper->SetInputData(input);
111 
112     vtkSmartPointer<vtkProperty> backfaceProp =
113         vtkSmartPointer<vtkProperty>::New();
114     backfaceProp->SetDiffuseColor(0.89,0.81,0.34);
115 
116     vtkSmartPointer<vtkActor> originalActor =
117         vtkSmartPointer<vtkActor>::New();
118     originalActor->SetMapper(originalMapper);
119     originalActor->SetBackfaceProperty(backfaceProp);
120     originalActor->GetProperty()->SetDiffuseColor(1.0, 0.3882, 0.2784);
121 
122     vtkSmartPointer<vtkPolyDataMapper> edgeMapper =
123         vtkSmartPointer<vtkPolyDataMapper>::New();
124     edgeMapper->SetInputData(featureEdges->GetOutput());
125     vtkSmartPointer<vtkActor> edgeActor =
126         vtkSmartPointer<vtkActor>::New();
127     edgeActor->SetMapper(edgeMapper);
128     edgeActor->GetProperty()->SetEdgeColor(0.,0.,1.0);
129     edgeActor->GetProperty()->SetEdgeVisibility(1);
130     edgeActor->GetProperty()->SetLineWidth(5);
131 
132     vtkSmartPointer<vtkPolyDataMapper> filledMapper =
133         vtkSmartPointer<vtkPolyDataMapper>::New();
134     filledMapper->SetInputData(normals->GetOutput());
135 
136     vtkSmartPointer<vtkActor> filledActor =
137         vtkSmartPointer<vtkActor>::New();
138     filledActor->SetMapper(filledMapper);
139     filledActor->GetProperty()->SetDiffuseColor(1.0, 0.3882, 0.2784);
140 
141     vtkSmartPointer<vtkRenderer> leftRenderer =
142         vtkSmartPointer<vtkRenderer>::New();
143     leftRenderer->SetViewport(leftViewport);
144     leftRenderer->AddActor(originalActor);
145     leftRenderer->AddActor(edgeActor);
146     leftRenderer->SetBackground(1.0, 1.0, 1.0);
147 
148     vtkSmartPointer<vtkRenderer> rightRenderer =
149         vtkSmartPointer<vtkRenderer>::New();
150     rightRenderer->SetViewport(rightViewport);
151     rightRenderer->AddActor(filledActor);
152     rightRenderer->SetBackground(1.0, 1.0, 1.0);
153 
154     vtkSmartPointer<vtkRenderWindow> renderWindow =
155         vtkSmartPointer<vtkRenderWindow>::New();
156     renderWindow->AddRenderer(leftRenderer);
157     renderWindow->AddRenderer(rightRenderer);
158     renderWindow->SetSize(640, 320);
159     renderWindow->Render();
160     renderWindow->SetWindowName("PolyDataClosed");
161 
162     vtkSmartPointer<vtkRenderWindowInteractor> renderWindowInteractor =
163         vtkSmartPointer<vtkRenderWindowInteractor>::New();
164     renderWindowInteractor->SetRenderWindow(renderWindow);
165 
166     leftRenderer->GetActiveCamera()->SetPosition(0, -1, 0);
167     leftRenderer->GetActiveCamera()->SetFocalPoint(0, 0, 0);
168     leftRenderer->GetActiveCamera()->SetViewUp(0, 0, 1);
169     leftRenderer->GetActiveCamera()->Azimuth(30);
170     leftRenderer->GetActiveCamera()->Elevation(30);
171     leftRenderer->ResetCamera();
172     rightRenderer->SetActiveCamera(leftRenderer->GetActiveCamera());
173 
174     renderWindow->Render();
175     renderWindowInteractor->Start();
176 
177     return EXIT_SUCCESS;
178 }