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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkCellArray.h"
#include "vtkCellArrayIterator.h"
#include "vtkLogger.h"
#include "vtkSmartPointer.h"
#include "vtkTimerLog.h"
namespace
{
void RunTest(bool use64BitStorage)
{
const vtkIdType numTris = 25000;
vtkIdType num;
auto ca = vtkSmartPointer<vtkCellArray>::New();
if (use64BitStorage)
{
cout << "\n=== Test performance of new vtkCellArray: 64-bit storage ===\n";
ca->Use64BitStorage();
}
else
{
cout << "\n=== Test performance of new vtkCellArray: 32-bit storage ===\n";
ca->Use32BitStorage();
}
vtkIdType tri[3] = { 0, 1, 2 };
auto timer = vtkSmartPointer<vtkTimerLog>::New();
vtkIdType npts;
const vtkIdType* pts;
// Insert
num = 0;
timer->StartTimer();
for (auto i = 0; i < numTris; ++i)
{
ca->InsertNextCell(3, tri);
++num;
}
timer->StopTimer();
cout << "Insert triangles: " << timer->GetElapsedTime() << "\n";
cout << " " << num << " triangles inserted\n";
cout << " Memory used: " << ca->GetActualMemorySize() << " kb\n";
// Iterate directly over cell array
num = 0;
timer->StartTimer();
for (ca->InitTraversal(); ca->GetNextCell(npts, pts);)
{
assert(npts == 3);
++num;
}
timer->StopTimer();
cout << "Traverse cell array (legacy GetNextCell()): " << timer->GetElapsedTime() << "\n";
cout << " " << num << " triangles visited\n";
// Iterate directly over cell array
num = 0;
timer->StartTimer();
vtkIdType numCells = ca->GetNumberOfCells();
for (auto cellId = 0; cellId < numCells; ++cellId)
{
ca->GetCellAtId(cellId, npts, pts);
assert(npts == 3);
++num;
}
timer->StopTimer();
cout << "Traverse cell array (new GetCellAtId()): " << timer->GetElapsedTime() << "\n";
cout << " " << num << " triangles visited\n";
// Iterate using iterator
num = 0;
timer->StartTimer();
auto iter = vtk::TakeSmartPointer(ca->NewIterator());
for (iter->GoToFirstCell(); !iter->IsDoneWithTraversal(); iter->GoToNextCell())
{
iter->GetCurrentCell(npts, pts);
assert(npts == 3);
++num;
}
timer->StopTimer();
cout << "Iterator traversal: " << timer->GetElapsedTime() << "\n";
cout << " " << num << " triangles visited\n";
} // RunTest
void RunTests()
{
// What is the size of vtkIdType?
cout << "=== vtkIdType is: " << (sizeof(vtkIdType) * 8) << " bits ===\n";
RunTest(false); // 32-bit
RunTest(true); // 64-bit
}
} // end anon namespace
int TestCellArrayTraversal(int, char*[])
{
try
{
RunTests();
}
catch (std::exception& err)
{
vtkLog(ERROR, << err.what());
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}
|