对于vtk XML格式的文件: .vtu,.vtp,.vts,.vtr,.vti,.vto, 解析其存储单元、单元数据。
#include <vtkSmartPointer.h>
#include <vtkXMLReader.h>
#include <vtkXMLUnstructuredGridReader.h>
#include <vtkXMLPolyDataReader.h>
#include <vtkXMLStructuredGridReader.h>
#include <vtkXMLRectilinearGridReader.h>
#include <vtkXMLHyperOctreeReader.h>
#include <vtkXMLCompositeDataReader.h>
#include <vtkXMLStructuredGridReader.h>
#include <vtkXMLImageDataReader.h>
#include <vtkDataSetReader.h>
#include <vtkDataSet.h>
#include <vtkUnstructuredGrid.h>
#include <vtkRectilinearGrid.h>
#include <vtkHyperOctree.h>
#include <vtkImageData.h>
#include <vtkPolyData.h>
#include <vtkStructuredGrid.h>
#include <vtkPointData.h>
#include <vtkCellData.h>
#include <vtkFieldData.h>
#include <vtkCellTypes.h>
#include <vtksys\SystemTools.hxx>
template<class TReader> vtkDataSet *ReadAnXMLFile(const char*fileName);
int main()
{
char *argv[5];
argv[0] = @H_502_66@"cow.vtp";
argv[1] = @H_502_66@"cth.vtr";
argv[2] = @H_502_66@"chombo3d_1.vti";
argv[3] = @H_502_66@"multicomb_1.vts";
argv[4] = @H_502_66@"quadraticTetra01.vtu";
int f = 0;
while (f < 5)
{
vtkDataSet *dataSet;
std::string extension =
vtksys::SystemTools::GetFilenameLastExtension(argv[f]);
if (extension == @H_502_66@".vtu")
{
dataSet = ReadAnXMLFile<vtkXMLUnstructuredGridReader>(argv[f]);
}
else if (extension == @H_502_66@".vtp")
{
dataSet = ReadAnXMLFile<vtkXMLPolyDataReader>(argv[f]);
}
else if (extension == @H_502_66@".vts")
{
dataSet = ReadAnXMLFile<vtkXMLStructuredGridReader>(argv[f]);
}
else if (extension == @H_502_66@".vtr")
{
dataSet = ReadAnXMLFile<vtkXMLRectilinearGridReader>(argv[f]);
}
else if (extension == @H_502_66@".vti")
{
dataSet = ReadAnXMLFile<vtkXMLImageDataReader>(argv[f]);
}
else if (extension == @H_502_66@".vto")
{
dataSet = ReadAnXMLFile<vtkXMLHyperOctreeReader>(argv[f]);
}
else if (extension == @H_502_66@".vtk")
{
dataSet = ReadAnXMLFile<vtkDataSetReader>(argv[f]);
}
else
{
std::cerr << argv[0] << @H_502_66@" Unknown extension: " << extension << std::endl;
return EXIT_FAILURE;
}
int numberOfCells = dataSet->GetNumberOfCells();
int numberOfPoints = dataSet->GetNumberOfPoints();
std::cout << @H_502_66@"------------------------" << std::endl;
std::cout << argv[f] << std::endl
<< @H_502_66@" contains a " << std::endl
<< dataSet->GetClassName()
<< @H_502_66@" that has " << numberOfCells << @H_502_66@" cells"
<< @H_502_66@" and " << numberOfPoints << @H_502_66@" points." << std::endl;
typedef std::map<int,int> CellContainer;
CellContainer cellMap;
for (int i = 0; i < numberOfCells; i++)
{
cellMap[dataSet->GetCellType(i)]++;
}
CellContainer::const_iterator it = cellMap.begin();
while (it != cellMap.end())
{
std::cout << @H_502_66@"\tCell type "
<< vtkCellTypes::GetClassNameFromTypeId(it->first)
<< @H_502_66@" occurs " << it->second << @H_502_66@" times." << std::endl;
++it;
}
vtkPointData *pd = dataSet->GetPointData();
if (pd)
{
std::cout << @H_502_66@" contains point data with "
<< pd->GetNumberOfArrays()
<< @H_502_66@" arrays." << std::endl;
for (int i = 0; i < pd->GetNumberOfArrays(); i++)
{
std::cout << @H_502_66@"\tArray " << i
<< @H_502_66@" is named "
<< (pd->GetArrayName(i) ? pd->GetArrayName(i) : @H_502_66@"NULL")
<< std::endl;
}
}
vtkCellData *cd = dataSet->GetCellData();
if (cd)
{
std::cout << @H_502_66@" contains cell data with "
<< cd->GetNumberOfArrays()
<< @H_502_66@" arrays." << std::endl;
for (int i = 0; i < cd->GetNumberOfArrays(); i++)
{
std::cout << @H_502_66@"\tArray " << i
<< @H_502_66@" is named "
<< (cd->GetArrayName(i) ? cd->GetArrayName(i) : @H_502_66@"NULL")
<< std::endl;
}
}
// Now check for field data
if (dataSet->GetFieldData())
{
std::cout << @H_502_66@" contains field data with "
<< dataSet->GetFieldData()->GetNumberOfArrays()
<< @H_502_66@" arrays." << std::endl;
for (int i = 0; i < dataSet->GetFieldData()->GetNumberOfArrays(); i++)
{
std::cout << @H_502_66@"\tArray " << i
<< @H_502_66@" is named " << dataSet->GetFieldData()->GetArray(i)->GetName()
<< std::endl;
}
}
dataSet->Delete();
f++;
}
return 0;
}
template<class TReader> vtkDataSet *ReadAnXMLFile(const char*fileName)
{
vtkSmartPointer<TReader> reader =
vtkSmartPointer<TReader>::New();
reader->SetFileName(fileName);
reader->Update();
reader->GetOutput()->Register(reader);
return vtkDataSet::SafeDownCast(reader->GetOutput());
}