ITK/Examples/Iterators/NeighborhoodIterator

From KitwarePublic
< ITK‎ | Examples
Revision as of 14:01, 18 November 2010 by Daviddoria (talk | contribs)
Jump to navigationJump to search

This example demonstrates how to traverse an itkImage with access to a rectangular neighborhood around the center pixel of the iterator.

NeighborhoodIterator.cxx

<source lang="cpp">

  1. include "itkImage.h"
  2. include "itkImageFileReader.h"
  3. include "itkNeighborhoodIterator.h"
  1. include <itkImageToVTKImageFilter.h>
  1. include "vtkImageViewer.h"
  2. include "vtkRenderWindowInteractor.h"
  3. include "vtkSmartPointer.h"
  4. include "vtkImageActor.h"
  5. include "vtkInteractorStyleImage.h"
  6. include "vtkRenderer.h"

int main(int argc, char*argv[]) {

 if(argc < 2)
   {
   std::cerr << "Required: filename" << std::endl;
   return EXIT_FAILURE;
   }
 typedef itk::Image<unsigned char, 2>  ImageType;
 typedef itk::ImageFileReader<ImageType> ReaderType;
 ReaderType::Pointer reader = ReaderType::New();
 reader->SetFileName(argv[1]);
 reader->Update();
 ImageType::Pointer image = reader->GetOutput();
 
 ImageType::SizeType regionSize;
 regionSize[0] = 50;
 regionSize[1] = 1;
 ImageType::IndexType regionIndex;
 regionIndex[0] = 0;
 regionIndex[1] = 0;
 ImageType::RegionType region;
 region.SetSize(regionSize);
 region.SetIndex(regionIndex);
 ImageType::SizeType radius;
 radius[0] = 1;
 radius[1] = 1;
 itk::NeighborhoodIterator<ImageType> iterator(radius, image,region);


 while(!iterator.IsAtEnd())
   {
   // Set the current pixel to white
   iterator.SetCenterPixel(255);
   for(unsigned int i = 0; i < 9; i++)
     {
     ImageType::IndexType index = iterator.GetIndex(i);
     std::cout << index[0] << " " << index[1] << std::endl;
     bool IsInBounds;
     iterator.GetPixel(i, IsInBounds);
     if(IsInBounds)
       {
       iterator.SetPixel(i,255);
       }
     }
   ++iterator;
   }
 // Visualize
 typedef itk::ImageToVTKImageFilter<ImageType> ConnectorType;
 ConnectorType::Pointer connector = ConnectorType::New();
 connector->SetInput(image);
 vtkSmartPointer<vtkImageActor> actor =
   vtkSmartPointer<vtkImageActor>::New();
 actor->SetInput(connector->GetOutput());
 vtkSmartPointer<vtkRenderWindow> renderWindow =
   vtkSmartPointer<vtkRenderWindow>::New();
 vtkSmartPointer<vtkRenderWindowInteractor> interactor =
   vtkSmartPointer<vtkRenderWindowInteractor>::New();
 interactor->SetRenderWindow(renderWindow);
 vtkSmartPointer<vtkRenderer> renderer =
   vtkSmartPointer<vtkRenderer>::New();
 renderWindow->AddRenderer(renderer);
 renderer->AddActor(actor);
 renderer->ResetCamera();
 renderWindow->Render();
 vtkSmartPointer<vtkInteractorStyleImage> style =
   vtkSmartPointer<vtkInteractorStyleImage>::New();
 interactor->SetInteractorStyle(style);
 interactor->Start();
 return EXIT_SUCCESS;

}

</source>

CMakeLists.txt

<source lang="cmake"> cmake_minimum_required(VERSION 2.6)

PROJECT(NeighborhoodIterator)

include_directories(/home/doriad/src/ITK/Wrapping/WrapITK/ExternalProjects/ItkVtkGlue/src/)

FIND_PACKAGE(VTK REQUIRED) INCLUDE(${VTK_USE_FILE})

FIND_PACKAGE(ITK REQUIRED) INCLUDE(${ITK_USE_FILE})

ADD_EXECUTABLE(NeighborhoodIterator NeighborhoodIterator.cxx) TARGET_LINK_LIBRARIES(NeighborhoodIterator vtkHybrid ITKNumerics ITKBasicFilters ITKCommon ITKIO)

</source>