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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
// .NAME Verifies that vtkOBJReader properly handles the presence of
// groupIds
// .SECTION Description
//
#include "vtkCellData.h"
#include "vtkFloatArray.h"
#include "vtkOBJReader.h"
#include "vtkTestUtilities.h"
// Read the specified file and check for expected number of groups
// Comments directly from the data file
static bool CheckOBJGroups(const std::string& filename, const int maxExpected)
{
auto reader = vtkSmartPointer<vtkOBJReader>::New();
reader->SetFileName(filename.c_str());
reader->Update();
vtkPolyData* data = reader->GetOutput();
std::cerr << "Testing file:" << filename << std::endl
<< "Expecting " << maxExpected << " as max groupId" << std::endl;
const char* comment = reader->GetComment();
if (comment)
{
std::cerr << "Comment: " << comment << std::endl;
}
vtkFloatArray* groups =
vtkFloatArray::SafeDownCast(data->GetCellData()->GetAbstractArray("GroupIds"));
if (!groups)
{
std::cerr << "missing group id array" << std::endl;
return false;
}
const vtkIdType nTuple = groups->GetNumberOfTuples();
int maxGroupId = -1;
for (vtkIdType i = 0; i < nTuple; ++i)
{
int thisGroup = static_cast<int>(round(groups->GetTuple(i)[0]));
if (maxGroupId < thisGroup)
{
maxGroupId = thisGroup;
}
}
if (maxExpected == maxGroupId)
{
return true;
}
std::cerr << "Error: found " << maxGroupId << " as max groupId" << std::endl;
return false;
}
int TestOBJReaderGroups(int argc, char* argv[])
{
// lambda for the testing
auto doTesting = [&](int maxExpected, const char* dataName) -> bool {
char* fname = vtkTestUtilities::ExpandDataFileName(argc, argv, dataName);
std::string filename(fname);
delete[] fname;
return CheckOBJGroups(filename, maxExpected);
};
int nFailures = 0;
if (!doTesting(0, "Data/objGroup_1a.obj"))
{
++nFailures;
}
if (!doTesting(0, "Data/objGroup_1b.obj"))
{
++nFailures;
}
if (!doTesting(1, "Data/objGroup_2a.obj"))
{
++nFailures;
}
if (!doTesting(1, "Data/objGroup_2b.obj"))
{
++nFailures;
}
std::cerr << "Test with " << nFailures << std::endl;
return nFailures;
}
|