VTK/Examples/Cxx/IO/DumpXMLFile

From KitwarePublic
< VTK‎ | Examples‎ | Cxx
Jump to navigationJump to search

Dump the Contents of VTK XML and Legacy files

This example reports the cell, cell data and point data contained within a VTK XML or legacy file.

DumpXMLFile.cxx

//
// DumpXMLFile - report on the contents of an XML or legacy vtk file
//  Usage: DumpXMLFile XMLFile1 XMLFile2 ...
//         where
//         XMLFile is a vtk XML file of type .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>

#include <map>

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());
}

int main (int argc, char *argv[])
{
  if (argc < 2)
    {
    std::cerr << "Usage: " << argv[0] << " XMLFile1 XMLFile2 ..." << std::endl;
    return EXIT_FAILURE;
    }

  // Process each file on the command line
  int f = 1;
  while (f < argc)
    {
    vtkDataSet *dataSet;
    std::string extension =
      vtksys::SystemTools::GetFilenameLastExtension(argv[f]);
    // Dispatch based on the file extension
    if (extension == ".vtu")
      {
      dataSet = ReadAnXMLFile<vtkXMLUnstructuredGridReader> (argv[f]);
      }
    else if (extension == ".vtp")
      {
      dataSet = ReadAnXMLFile<vtkXMLPolyDataReader> (argv[f]);
      }
    else if (extension == ".vts")
      {
      dataSet = ReadAnXMLFile<vtkXMLStructuredGridReader> (argv[f]);
      }
    else if (extension == ".vtr")
      {
      dataSet = ReadAnXMLFile<vtkXMLRectilinearGridReader> (argv[f]);
      }
    else if (extension == ".vti")
      {
      dataSet = ReadAnXMLFile<vtkXMLImageDataReader> (argv[f]);
      }
    else if (extension == ".vto")
      {
      dataSet = ReadAnXMLFile<vtkXMLHyperOctreeReader> (argv[f]);
      }
    else if (extension == ".vtk")
      {
      dataSet = ReadAnXMLFile<vtkDataSetReader> (argv[f]);
      }
    else
      {
      std::cerr << argv[0] << " Unknown extension: " << extension << std::endl;
      return EXIT_FAILURE;
      }

    int numberOfCells = dataSet->GetNumberOfCells();
    int numberOfPoints = dataSet->GetNumberOfPoints();

    // Generate a report
    std::cout << "------------------------" << std::endl;
    std::cout << argv[f] << std::endl 
         << " contains a " << std::endl
         << dataSet->GetClassName() 
         <<  " that has " << numberOfCells << " cells"
         << " and " << numberOfPoints << " 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 << "\tCell type " 
           << vtkCellTypes::GetClassNameFromTypeId(it->first)
           << " occurs " << it->second << " times." << std::endl;
      ++it;
      }

    // Now check for point data
    vtkPointData *pd = dataSet->GetPointData();
    if (pd)
      {
      std::cout << " contains point data with " 
           << pd->GetNumberOfArrays() 
           << " arrays." << std::endl;
      for (int i = 0; i < pd->GetNumberOfArrays(); i++)
        {
        std::cout << "\tArray " << i 
             << " is named "
             << (pd->GetArrayName(i) ? pd->GetArrayName(i) : "NULL")
             << std::endl;
        }
      }
    // Now check for cell data
    vtkCellData *cd = dataSet->GetCellData();
    if (cd)
      {
      std::cout << " contains cell data with " 
           << cd->GetNumberOfArrays() 
           << " arrays." << std::endl;
      for (int i = 0; i < cd->GetNumberOfArrays(); i++)
        {
        std::cout << "\tArray " << i 
             << " is named "
             << (cd->GetArrayName(i) ? cd->GetArrayName(i) : "NULL")
             << std::endl;
        }
      }
    // Now check for field data
    if (dataSet->GetFieldData())
      {
      std::cout << " contains field data with " 
           << dataSet->GetFieldData()->GetNumberOfArrays() 
           << " arrays." << std::endl;
      for (int i = 0; i < dataSet->GetFieldData()->GetNumberOfArrays(); i++)
        {
        std::cout << "\tArray " << i 
             << " is named " << dataSet->GetFieldData()->GetArray(i)->GetName() 
             << std::endl;
        }
      }
    dataSet->Delete();
    f++;
    }
  return EXIT_SUCCESS;
}


Please try the new VTKExamples website.

CMakeLists.txt

cmake_minimum_required(VERSION 2.8)

PROJECT(DumpXMLFile)

find_package(VTK REQUIRED)
include(${VTK_USE_FILE})

add_executable(DumpXMLFile MACOSX_BUNDLE DumpXMLFile.cxx)

if(VTK_LIBRARIES)
  target_link_libraries(DumpXMLFile ${VTK_LIBRARIES})
else()
  target_link_libraries(DumpXMLFile vtkHybrid vtkWidgets)
endif()

Download and Build DumpXMLFile

Click here to download DumpXMLFile. and its CMakeLists.txt file.

Once the tarball DumpXMLFile.tar has been downloaded and extracted,

cd DumpXMLFile/build 
  • If VTK is installed:
cmake ..
  • If VTK is not installed but compiled on your system, you will need to specify the path to your VTK build:
cmake -DVTK_DIR:PATH=/home/me/vtk_build ..

Build the project:

make

and run it:

./DumpXMLFile

WINDOWS USERS PLEASE NOTE: Be sure to add the VTK bin directory to your path. This will resolve the VTK dll's at run time.