其他分享
首页 > 其他分享> > VTK生成等值线并标记值

VTK生成等值线并标记值

作者:互联网

  1 #include <vtkActor.h>
  2 #include <vtkDataSetMapper.h>
  3 #include <vtkNamedColors.h>
  4 #include <vtkNew.h>
  5 #include <vtkPoints.h>
  6 #include <vtkRenderWindow.h>
  7 #include <vtkRenderWindowInteractor.h>
  8 #include <vtkRenderer.h>
  9 #include <vtkSmartPointer.h>
 10 #include <vtkAppendFilter.h>
 11 #include <vtkProperty.h>
 12 #include <vtkUnstructuredGrid.h>
 13 #include <vtkCellDataToPointData.h>
 14 #include <vtkFloatArray.h>
 15 #include <vtkCellData.h>
 16 #include <vtkContourFilter.h>
 17 #include <vtkAutoInit.h>
 18 #include <vtkStripper.h>
 19 #include <vtkCellArrayIterator.h>
 20 #include <cstdlib>
 21 #include <vtkPointData.h>
 22 #include <vtkDoubleArray.h>
 23 #include <vtkLabeledDataMapper.h>
 24 #include <vtkPolyData.h>
 25 #include <vtkTextProperty.h>
 26 #include <vtkActor2D.h>
 27 
 28 VTK_MODULE_INIT(vtkInteractionStyle);
 29 VTK_MODULE_INIT(vtkRenderingFreeType);
 30 VTK_MODULE_INIT(vtkRenderingVolumeOpenGL2);
 31 
 32 int main(int argc, char* argv[])
 33 {
 34     vtkNew<vtkUnstructuredGrid> grid;   /* 创建一个非结构化网格对象 */
 35     /*=============================网格节点=============================*/
 36     vtkNew<vtkPoints> meshPoints; /* 用于存储网格节点 */
 37     meshPoints->InsertNextPoint(20, 0, 0);
 38     meshPoints->InsertNextPoint(17.320508109792, 10.0000000321997, 0);
 39     meshPoints->InsertNextPoint(10.0000000084796, 17.3205081212641, 0);
 40     meshPoints->InsertNextPoint(0, 20, 0);
 41     meshPoints->InsertNextPoint(-10.0000000245946, 17.3205081142845, 0);
 42     meshPoints->InsertNextPoint(-17.320508106976, 10.0000000330863, 0);
 43     meshPoints->InsertNextPoint(-20, 0, 0);
 44     meshPoints->InsertNextPoint(-17.3205081361632, -9.99999998652343, 0);
 45     meshPoints->InsertNextPoint(-10.0000000640139, -17.3205080893384, 0);
 46     meshPoints->InsertNextPoint(0, -20, 0);
 47     meshPoints->InsertNextPoint(9.99999997546481, -17.3205081119126, 0);
 48     meshPoints->InsertNextPoint(17.3205080670077, -10.000000025522, 0);
 49     meshPoints->InsertNextPoint(-7.15966651257561, -7.12683602128246, 0);
 50     meshPoints->InsertNextPoint(-2.63739358105485, -12.9978938464381, 0);
 51     meshPoints->InsertNextPoint(-9.10592684278358, -0.737849316405346, 0);
 52     meshPoints->InsertNextPoint(6.59947131139743, -9.08284206416132, 0);
 53     meshPoints->InsertNextPoint(-0.137734590259206, -2.71811814272003, 0);
 54     meshPoints->InsertNextPoint(-7.75933677195531, 5.80947235781205, 0);
 55     meshPoints->InsertNextPoint(9.11726442880688, -1.21602962482667, 0);
 56     meshPoints->InsertNextPoint(-0.484322809037511, 4.65580556604978, 0);
 57     meshPoints->InsertNextPoint(-3.48354780828141, 10.4142340529639, 0);
 58     meshPoints->InsertNextPoint(7.28614335171517, 6.31993800133739, 0);
 59     meshPoints->InsertNextPoint(3.36376652373578, 13.053010718073, 0);
 60     grid->SetPoints(meshPoints);
 61     /*=================================================================*/
 62 
 63     /*=============================设置单元=============================*/
 64     vtkNew<vtkIdList> pointsId;
 65     pointsId->SetNumberOfIds(3);
 66     pointsId->SetId(0, 9);
 67     pointsId->SetId(1, 13);
 68     pointsId->SetId(2, 8);
 69     grid->InsertNextCell(VTK_TRIANGLE, 3, pointsId->GetPointer(0));
 70 
 71     pointsId->Initialize();
 72     pointsId->SetNumberOfIds(3);
 73     pointsId->SetId(0, 14);
 74     pointsId->SetId(1, 12);
 75     pointsId->SetId(2, 16);
 76     grid->InsertNextCell(VTK_TRIANGLE, 3, pointsId->GetPointer(0));
 77 
 78     pointsId->Initialize();
 79     pointsId->SetNumberOfIds(3);
 80     pointsId->SetId(0, 11);
 81     pointsId->SetId(1, 15);
 82     pointsId->SetId(2, 10);
 83     grid->InsertNextCell(VTK_TRIANGLE, 3, pointsId->GetPointer(0));
 84 
 85     pointsId->Initialize();
 86     pointsId->SetNumberOfIds(3);
 87     pointsId->SetId(0, 15);
 88     pointsId->SetId(1, 18);
 89     pointsId->SetId(2, 16);
 90     grid->InsertNextCell(VTK_TRIANGLE, 3, pointsId->GetPointer(0));
 91 
 92     pointsId->Initialize();
 93     pointsId->SetNumberOfIds(3);
 94     pointsId->SetId(0, 17);
 95     pointsId->SetId(1, 19);
 96     pointsId->SetId(2, 20);
 97     grid->InsertNextCell(VTK_TRIANGLE, 3, pointsId->GetPointer(0));
 98 
 99     pointsId->Initialize();
100     pointsId->SetNumberOfIds(3);
101     pointsId->SetId(0, 3);
102     pointsId->SetId(1, 22);
103     pointsId->SetId(2, 2);
104     grid->InsertNextCell(VTK_TRIANGLE, 3, pointsId->GetPointer(0));
105 
106     pointsId->Initialize();
107     pointsId->SetNumberOfIds(4);
108     pointsId->SetId(0, 8);
109     pointsId->SetId(1, 13);
110     pointsId->SetId(2, 12);
111     pointsId->SetId(3, 7);
112     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
113 
114     pointsId->Initialize();
115     pointsId->SetNumberOfIds(4);
116     pointsId->SetId(0, 12);
117     pointsId->SetId(1, 14);
118     pointsId->SetId(2, 6);
119     pointsId->SetId(3, 7);
120     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
121 
122     pointsId->Initialize();
123     pointsId->SetNumberOfIds(4);
124     pointsId->SetId(0, 13);
125     pointsId->SetId(1, 9);
126     pointsId->SetId(2, 10);
127     pointsId->SetId(3, 15);
128     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
129 
130     pointsId->Initialize();
131     pointsId->SetNumberOfIds(4);
132     pointsId->SetId(0, 6);
133     pointsId->SetId(1, 14);
134     pointsId->SetId(2, 17);
135     pointsId->SetId(3, 5);
136     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
137 
138     pointsId->Initialize();
139     pointsId->SetNumberOfIds(4);
140     pointsId->SetId(0, 16);
141     pointsId->SetId(1, 12);
142     pointsId->SetId(2, 13);
143     pointsId->SetId(3, 15);
144     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
145 
146     pointsId->Initialize();
147     pointsId->SetNumberOfIds(4);
148     pointsId->SetId(0, 14);
149     pointsId->SetId(1, 16);
150     pointsId->SetId(2, 19);
151     pointsId->SetId(3, 17);
152     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
153 
154     pointsId->Initialize();
155     pointsId->SetNumberOfIds(4);
156     pointsId->SetId(0, 17);
157     pointsId->SetId(1, 20);
158     pointsId->SetId(2, 4);
159     pointsId->SetId(3, 5);
160     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
161 
162     pointsId->Initialize();
163     pointsId->SetNumberOfIds(4);
164     pointsId->SetId(0, 16);
165     pointsId->SetId(1, 18);
166     pointsId->SetId(2, 21);
167     pointsId->SetId(3, 19);
168     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
169 
170     pointsId->Initialize();
171     pointsId->SetNumberOfIds(4);
172     pointsId->SetId(0, 15);
173     pointsId->SetId(1, 11);
174     pointsId->SetId(2, 0);
175     pointsId->SetId(3, 18);
176     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
177 
178     pointsId->Initialize();
179     pointsId->SetNumberOfIds(4);
180     pointsId->SetId(0, 20);
181     pointsId->SetId(1, 19);
182     pointsId->SetId(2, 21);
183     pointsId->SetId(3, 22);
184     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
185 
186     pointsId->Initialize();
187     pointsId->SetNumberOfIds(4);
188     pointsId->SetId(0, 22);
189     pointsId->SetId(1, 3);
190     pointsId->SetId(2, 4);
191     pointsId->SetId(3, 20);
192     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
193 
194     pointsId->Initialize();
195     pointsId->SetNumberOfIds(4);
196     pointsId->SetId(0, 21);
197     pointsId->SetId(1, 18);
198     pointsId->SetId(2, 0);
199     pointsId->SetId(3, 1);
200     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
201 
202     pointsId->Initialize();
203     pointsId->SetNumberOfIds(4);
204     pointsId->SetId(0, 2);
205     pointsId->SetId(1, 22);
206     pointsId->SetId(2, 21);
207     pointsId->SetId(3, 1);
208     grid->InsertNextCell(VTK_QUAD, 4, pointsId->GetPointer(0));
209     /*=================================================================*/
210 
211     /*==============================设置单元颜色========================*/
212     vtkNew<vtkFloatArray> color;                        /* 用于设置每个单元的随机颜色 */
213     for (int i = 0; i < grid->GetNumberOfCells(); ++i)
214     {
215         color->InsertNextValue(std::rand());
216     }
217     grid->GetCellData()->SetScalars(color);             /* 设置单元格的颜色 */
218     /*=================================================================*/
219 
220     /*==============================单元——>节点========================*/
221     double rangeMin = grid->GetScalarRange()[0];
222     double rangeMax = grid->GetScalarRange()[1];
223     vtkNew<vtkCellDataToPointData> theCellDataToPointData;
224     theCellDataToPointData->SetInputData(grid);
225     theCellDataToPointData->PassCellDataOn();
226     theCellDataToPointData->Update();
227     /*=================================================================*/
228 
229     /*============================等值线================================*/
230     vtkNew<vtkContourFilter> theContourFilter;
231     theContourFilter->SetInputData(theCellDataToPointData->GetOutput());
232     theContourFilter->GenerateValues(5, rangeMin, rangeMax);
233     theContourFilter->Update();
234     /*=================================================================*/
235 
236     /*============================等值线值标记===========================*/
237     // 用于将离散的三角面片拼接为连续的等值面
238     vtkNew<vtkStripper> theStripper;
239     theStripper->SetInputData(theContourFilter->GetOutput());
240     theStripper->Update();
241     // 等值线数目
242     vtkIdType lines = theStripper->GetOutput()->GetNumberOfLines();
243     vtkPoints* points = theStripper->GetOutput()->GetPoints();
244     vtkCellArray* cells = theStripper->GetOutput()->GetLines();
245     vtkDataArray* scalars = theStripper->GetOutput()->GetPointData()->GetScalars();
246     vtkNew<vtkPolyData> labelPolyData;
247     vtkNew<vtkPoints> labelPoints;
248     vtkNew<vtkDoubleArray> labelScalars;
249     labelScalars->SetNumberOfComponents(1);
250     labelScalars->SetName("IsoValues");
251 
252     // 创建单元数组迭代器
253     auto cellIter = vtk::TakeSmartPointer(cells->NewIterator());
254     for (cellIter->GoToFirstCell(); !cellIter->IsDoneWithTraversal(); cellIter->GoToNextCell())
255     {
256         vtkIdList* cell = cellIter->GetCurrentCell();
257         const vtkIdType samplePtIdx = static_cast<vtkIdType>(vtkMath::Random(0, cell->GetNumberOfIds()));
258         vtkIdType midPointId = cell->GetId(samplePtIdx);
259 
260         double midPoint[3];
261         points->GetPoint(midPointId, midPoint);
262         labelPoints->InsertNextPoint(midPoint);
263         labelScalars->InsertNextTuple1(scalars->GetTuple1(midPointId));
264     }
265     labelPolyData->SetPoints(labelPoints);
266 
267     labelPolyData->GetPointData()->SetScalars(labelScalars);
268     /*=================================================================*/
269 
270     vtkNew<vtkDataSetMapper> contourMapper;
271     contourMapper->SetInputData(theContourFilter->GetOutput());
272     vtkNew<vtkDataSetMapper> mapper;
273     mapper->SetInputData(theCellDataToPointData->GetOutput());
274     mapper->SetScalarModeToUsePointData();
275     mapper->SetScalarRange(rangeMin, rangeMax);
276 
277     vtkNew<vtkActor> contourActor;
278     contourActor->SetMapper(contourMapper);
279 
280     vtkNew<vtkActor> actor;
281     actor->SetMapper(mapper);
282 
283     vtkNew<vtkLabeledDataMapper> labelMapper;
284     labelMapper->SetFieldDataName("IsoValues");
285     labelMapper->SetInputData(labelPolyData);
286     labelMapper->SetLabelModeToLabelScalars();
287     labelMapper->SetLabelFormat("%6.2f");
288     labelMapper->GetLabelTextProperty()->SetColor(0, 0, 1);
289 
290     vtkNew<vtkActor2D> labelActor;
291     labelActor->SetMapper(labelMapper);
292 
293     vtkNew<vtkRenderer> renderer;
294     renderer->AddActor(actor);
295     renderer->AddActor(labelActor);
296     renderer->AddActor(contourActor);
297 
298     vtkNew<vtkRenderWindowInteractor> windowInteractor;
299     vtkNew<vtkRenderWindow> renderWindow;
300 
301     windowInteractor->SetRenderWindow(renderWindow);
302     renderWindow->AddRenderer(renderer);
303 
304 
305     renderWindow->Render();
306 
307     windowInteractor->Start();
308 
309     return 0;
310 }

原文地址:https://www.cnblogs.com/liusuanyatong/p/14319699.html

标签:标记,meshPoints,等值线,VTK,pointsId,grid,InsertNextPoint,SetId,include
来源: https://www.cnblogs.com/chanyuantiandao/p/16480519.html