Nektar++
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
VtkStripsToPolys.cpp
Go to the documentation of this file.
1 ///////////////////////////////////////////////////////////////////////////////
2 //
3 // File: VtkStripsToPolys.cpp
4 //
5 // For more information, please see: http://www.nektar.info
6 //
7 // The MIT License
8 //
9 // Copyright (c) 2006 Division of Applied Mathematics, Brown University (USA),
10 // Department of Aeronautics, Imperial College London (UK), and Scientific
11 // Computing and Imaging Institute, University of Utah (USA).
12 //
13 // Permission is hereby granted, free of charge, to any person obtaining a
14 // copy of this software and associated documentation files (the "Software"),
15 // to deal in the Software without restriction, including without limitation
16 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
17 // and/or sell copies of the Software, and to permit persons to whom the
18 // Software is furnished to do so, subject to the following conditions:
19 //
20 // The above copyright notice and this permission notice shall be included
21 // in all copies or substantial portions of the Software.
22 //
23 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
24 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
25 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
26 // THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
27 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
28 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
29 // DEALINGS IN THE SOFTWARE.
30 //
31 // Description:
32 //
33 ///////////////////////////////////////////////////////////////////////////////
34 
37 
38 #include <vtkCellArray.h>
39 #include <vtkPointData.h>
40 #include <vtkPoints.h>
41 #include <vtkPolyData.h>
42 #include <vtkPolyDataReader.h>
43 #include <vtkPolyDataWriter.h>
44 #include <vtkTriangle.h>
45 
46 int main(int argc, char *argv[])
47 {
48  if (argc < 2)
49  {
50  cout << "Usage: VtkStripsToPolys vtk-file" << endl;
51  exit(-1);
52  }
53 
54  vtkIdType npts;
55 #if VTK_MAJOR_VERSION >= 9 || \
56  (VTK_MAJOR_VERSION >= 8 && VTK_MINOR_VERSION >= 90)
57  const vtkIdType *pts = 0;
58 #else
59  vtkIdType *pts = 0;
60 #endif
61 
62  // Read mesh
63  vtkPolyDataReader *vtkMeshReader = vtkPolyDataReader::New();
64  vtkMeshReader->SetFileName(argv[1]);
65  vtkMeshReader->Update();
66  vtkPolyData *vtkMesh = vtkMeshReader->GetOutput();
67  vtkPoints *vtkPoints = vtkMesh->GetPoints();
68  vtkCellArray *vtkStrips = vtkMesh->GetStrips();
69 
70  // Check we found points and strips in the file.
71  ASSERTL0(vtkPoints, "ERROR: cannot get points from mesh.");
72  ASSERTL0(vtkStrips, "ERROR: cannot get triangle strips from mesh.");
73 
74  // Create new cell array for polygons
75  vtkCellArray *vtkPolys = vtkCellArray::New();
76 
77  // Generate the polygons from the triangle strips
78  vtkStrips->InitTraversal();
79  for (int i = 0; vtkStrips->GetNextCell(npts, pts); ++i)
80  {
81  for (int j = 0; j < npts - 2; ++j)
82  {
83  vtkPolys->InsertNextCell(3, &pts[j]);
84  }
85  }
86 
87  // Create the new poly data
88  vtkPolyData *vtkNewMesh = vtkPolyData::New();
89  vtkNewMesh->SetPoints(vtkPoints);
90  vtkNewMesh->SetPolys(vtkPolys);
91 
92  // Copy data across
93  for (int i = 0; i < vtkMesh->GetPointData()->GetNumberOfArrays(); ++i)
94  {
95  vtkNewMesh->GetPointData()->SetScalars(
96  vtkMesh->GetPointData()->GetArray(i));
97  }
98 
99  // Write out the new mesh
100  vtkPolyDataWriter *vtkMeshWriter = vtkPolyDataWriter::New();
101  vtkMeshWriter->SetFileName(argv[2]);
102 #if VTK_MAJOR_VERSION <= 5
103  vtkMeshWriter->SetInput(vtkNewMesh);
104 #else
105  vtkMeshWriter->SetInputData(vtkNewMesh);
106 #endif
107  vtkMeshWriter->Write();
108 }
#define ASSERTL0(condition, msg)
Definition: ErrorUtil.hpp:215
int main(int argc, char *argv[])