1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186
|
/*=========================================================================
*
* Copyright NumFOCUS
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#include <iostream>
#include "itkMesh.h"
#include "itkPolyLineCell.h"
#include "itkHexahedronCell.h"
#include "itkTetrahedronCell.h"
#include "itkQuadraticTriangleCell.h"
#include "itkPolygonCell.h"
/**
* Define a mesh type that stores a PixelType of "int". Use the defaults
* for the other template parameters.
*/
using MeshType = itk::Mesh<int>;
using CellTraits = MeshType::CellTraits;
using CellInterfaceType = itk::CellInterface<int, CellTraits>;
/**
* Typedef the generic cell type for the mesh. It is an abstract class,
* so we can only use information from it, like get its pointer type.
*/
using CellType = MeshType::CellType;
using CellAutoPointer = CellType::CellAutoPointer;
// Test the cell interface
template <typename TCell>
int
TestCellInterface(std::string name, TCell * aCell)
{
CellAutoPointer cell(aCell, true);
std::cout << "-------- " << name << " (" << aCell->GetNameOfClass() << ')' << std::endl;
std::cout << " Type: " << static_cast<int>(cell->GetType()) << std::endl;
std::cout << " Dimension: " << cell->GetDimension() << std::endl;
std::cout << " NumberOfPoints: " << cell->GetNumberOfPoints() << std::endl;
std::cout << " NumberOfBoundaryFeatures:" << std::endl;
for (unsigned int i = 0; i < cell->GetDimension(); ++i)
{
std::cout << " " << i << ": " << cell->GetNumberOfBoundaryFeatures(i) << std::endl;
for (unsigned int j = 0; j < cell->GetNumberOfBoundaryFeatures(i); ++j)
{
CellAutoPointer feature;
cell->GetBoundaryFeature(i, j, feature);
}
}
std::cout << " Iterator test: PointIds for empty cell: ";
typename TCell::PointIdIterator pointId = cell->PointIdsBegin();
typename TCell::PointIdIterator endId = cell->PointIdsEnd();
while (pointId != endId)
{
std::cout << *pointId << ", ";
pointId++;
}
std::cout << std::endl;
std::cout << " ConstIterator test: PointIds for empty cell: ";
typename TCell::PointIdConstIterator cpointId = cell->PointIdsBegin();
typename TCell::PointIdConstIterator cendId = cell->PointIdsEnd();
while (cpointId != cendId)
{
std::cout << *cpointId << ", ";
cpointId++;
}
std::cout << std::endl;
// Add point ids
std::cout << " SetPointIds" << std::endl;
using PointIdentifier = typename TCell::PointIdentifier;
auto * pointIds = new PointIdentifier[cell->GetNumberOfPoints() * 2];
for (unsigned int i = 0; i < cell->GetNumberOfPoints() * 2; ++i)
{
pointIds[i] = i;
}
cell->SetPointIds(pointIds);
if (cell->GetNumberOfPoints() > 0)
{
cell->SetPointId(0, 100);
}
std::cout << " ConstIterator test: PointIds for populated cell: ";
typename TCell::PointIdConstIterator ppointId = cell->PointIdsBegin();
typename TCell::PointIdConstIterator pendId = cell->PointIdsEnd();
while (ppointId != pendId)
{
std::cout << *ppointId << ", ";
ppointId++;
}
std::cout << std::endl;
cell->SetPointIds(&pointIds[cell->GetNumberOfPoints()], &pointIds[cell->GetNumberOfPoints() * 2]);
std::cout << " Iterator test: PointIds for populated cell: ";
typename TCell::PointIdIterator pxpointId = cell->PointIdsBegin();
typename TCell::PointIdIterator pxendId = cell->PointIdsEnd();
while (pxpointId != pxendId)
{
std::cout << *pxpointId << ", ";
pxpointId++;
}
std::cout << std::endl;
// Make a copy
CellAutoPointer copyOfCell;
cell->MakeCopy(copyOfCell);
std::cout << " PointIds for copied cell: ";
typename TCell::PointIdConstIterator xpointId = copyOfCell->PointIdsBegin();
typename TCell::PointIdConstIterator xendId = copyOfCell->PointIdsEnd();
while (xpointId != xendId)
{
std::cout << *xpointId << ", ";
xpointId++;
}
std::cout << std::endl;
delete[] pointIds;
return EXIT_SUCCESS;
}
int
itkCellInterfaceTest(int, char *[])
{
int status = EXIT_SUCCESS;
using VertexCellType = itk::VertexCell<CellInterfaceType>;
status |= TestCellInterface("Vertex", new VertexCellType);
using LineCellType = itk::LineCell<CellInterfaceType>;
status |= TestCellInterface("Line", new LineCellType);
using PolyLineCellType = itk::PolyLineCell<CellInterfaceType>;
status |= TestCellInterface("PolyLine", new PolyLineCellType);
using PolyLineCellType = itk::PolyLineCell<CellInterfaceType>;
status |= TestCellInterface("PolyLineCellType with 7 vertices", new PolyLineCellType(7));
using TriangleCellType = itk::TriangleCell<CellInterfaceType>;
status |= TestCellInterface("Triangle", new TriangleCellType);
using HexahedronCellType = itk::HexahedronCell<CellInterfaceType>;
status |= TestCellInterface("HexahedronCell", new HexahedronCellType);
using TetrahedronCellType = itk::TetrahedronCell<CellInterfaceType>;
status |= TestCellInterface("TetrahedronCell", new TetrahedronCellType);
using QuadraticEdgeCellType = itk::QuadraticEdgeCell<CellInterfaceType>;
status |= TestCellInterface("QuadraticEdgeCell", new QuadraticEdgeCellType);
using QuadraticTriangleCellType = itk::QuadraticTriangleCell<CellInterfaceType>;
status |= TestCellInterface("QuadraticTriangleCell", new QuadraticTriangleCellType);
using QuadrilateralCellType = itk::QuadrilateralCell<CellInterfaceType>;
status |= TestCellInterface("QuadrilateralCell", new QuadrilateralCellType);
using PolygonCellType = itk::PolygonCell<CellInterfaceType>;
status |= TestCellInterface("PolygonCell with 0 vertices", new PolygonCellType());
using PolygonCellType = itk::PolygonCell<CellInterfaceType>;
status |= TestCellInterface("PolygonCell with 5 vertices", new PolygonCellType(5));
std::cout << "Test finished." << std::endl;
return status;
}
|