VTK/Examples/Cxx/ImageData/ImageDataToPointSet

From KitwarePublic

Jump to: navigation, search

ImageDataToPointSet.cxx

#include <vtkVersion.h>
#include <vtkImageData.h>
#include <vtkPoints.h>
#include <vtkImageDataToPointSet.h>
#include <vtkSmartPointer.h>
#include <vtkXMLStructuredGridWriter.h>
#include <vtkImageReader2.h>
#include <vtkImageReader2Factory.h>
#include <vtkStructuredGrid.h>
 
static void CreateImage(vtkImageData* image);
 
int main(int argc, char*argv[])
{
  vtkSmartPointer<vtkImageData> image =
    vtkSmartPointer<vtkImageData>::New();
  CreateImage(image);
 
  vtkSmartPointer<vtkImageDataToPointSet> imageDataToPointSet =
    vtkSmartPointer<vtkImageDataToPointSet>::New();
#if VTK_MAJOR_VERSION <= 5
  imageDataToPointSet->SetInputConnection(image->GetProducerPort());
#else
  imageDataToPointSet->SetInputData(image);
#endif
  imageDataToPointSet->Update();
 
  vtkSmartPointer<vtkXMLStructuredGridWriter> writer =
    vtkSmartPointer<vtkXMLStructuredGridWriter>::New();
  writer->SetInputConnection(imageDataToPointSet->GetOutputPort());
  writer->SetFileName("Test.vts");
  writer->Write();
 
  return EXIT_SUCCESS;
}
 
void CreateImage(vtkImageData* image)
{
  // Create an image
  image->SetDimensions(3,3,1);
#if VTK_MAJOR_VERSION <= 5
  image->SetNumberOfScalarComponents(1);
  image->SetScalarTypeToUnsignedChar();
#else
  image->AllocateScalars(VTK_UNSIGNED_CHAR,1);
#endif
 
  int dims[3];
  image->GetDimensions(dims);
  for(unsigned int i = 0; i < dims[0]; i++)
    {
    for(unsigned int j = 0; j < dims[1]; j++)
      {
      unsigned char* pixel = static_cast<unsigned char*>(image->GetScalarPointer(i,j,0));
      pixel[0] = 1;
      }
    }
}

CMakeLists.txt

cmake_minimum_required(VERSION 2.8)
 
PROJECT(ImageDataToPointSet)
 
find_package(VTK REQUIRED)
include(${VTK_USE_FILE})
 
if (APPLE)
  add_executable(ImageDataToPointSet MACOSX_BUNDLE ImageDataToPointSet.cxx)
else()
  add_executable(ImageDataToPointSet ImageDataToPointSet.cxx)
endif()
 
if(VTK_LIBRARIES)
  target_link_libraries(ImageDataToPointSet ${VTK_LIBRARIES})
else()
  target_link_libraries(ImageDataToPointSet vtkHybrid )
endif()
Personal tools