ITK/Examples/ImageProcessing/AndImageFilter

From KitwarePublic

Jump to: navigation, search

AndImageFilter.cxx

#include "itkImage.h"
#include "itkSimpleFilterWatcher.h"
#include "itkAndImageFilter.h"
#include "itkImageRegionIterator.h"
#include "itkImageFileWriter.h"
 
typedef itk::Image<unsigned char, 2>  ImageType;
static void CreateImage1(ImageType::Pointer image);
static void CreateImage2(ImageType::Pointer image);
 
int main(int, char *[])
{
  ImageType::Pointer image1 = ImageType::New();
  CreateImage1(image1);
 
  typedef  itk::ImageFileWriter< ImageType  > WriterType;
  WriterType::Pointer writer = WriterType::New();
  writer->SetFileName("input1.png");
  writer->SetInput(image1);
  writer->Update();
 
  ImageType::Pointer image2 = ImageType::New();
  CreateImage2(image2);
 
  writer->SetFileName("input2.png");
  writer->SetInput(image2);
  writer->Update();
 
  typedef itk::AndImageFilter <ImageType>
          AndImageFilterType;
 
  AndImageFilterType::Pointer andFilter
          = AndImageFilterType::New();
  andFilter->SetInput(0, image1);
  andFilter->SetInput(1, image2);
  andFilter->Update();
 
  writer->SetFileName("output.png");
  writer->SetInput(andFilter->GetOutput());
  writer->Update();
 
  return EXIT_SUCCESS;
}
 
void CreateImage1(ImageType::Pointer image)
{
  ImageType::IndexType start;
  start.Fill(0);
 
  ImageType::SizeType size;
  size.Fill(100);
 
  ImageType::RegionType region;
  region.SetSize(size);
  region.SetIndex(start);
 
  image->SetRegions(region);
  image->Allocate();
 
  itk::ImageRegionIterator<ImageType> imageIterator(image,region);
 
  while(!imageIterator.IsAtEnd())
    {
    if(imageIterator.GetIndex()[0] < 70)
      {
      imageIterator.Set(255);
      }
    else
      {
      imageIterator.Set(0);
      }
 
    ++imageIterator;
    }
 
}
 
void CreateImage2(ImageType::Pointer image)
{
  ImageType::IndexType start;
  start.Fill(0);
 
  ImageType::SizeType size;
  size.Fill(100);
 
  ImageType::RegionType region;
  region.SetSize(size);
  region.SetIndex(start);
 
  image->SetRegions(region);
  image->Allocate();
 
  itk::ImageRegionIterator<ImageType> imageIterator(image,region);
 
  while(!imageIterator.IsAtEnd())
    {
    if(imageIterator.GetIndex()[0] > 30)
      {
      imageIterator.Set(255);
      }
    else
      {
      imageIterator.Set(0);
      }
 
    ++imageIterator;
    }
 
}

CMakeLists.txt

cmake_minimum_required(VERSION 2.8)
 
project(AndImageFilter)
 
find_package(ITK REQUIRED)
include(${ITK_USE_FILE})
 
add_executable(AndImageFilter AndImageFilter.cxx)
 
if( "${ITK_VERSION_MAJOR}" LESS 4 )
  target_link_libraries(AndImageFilter ITKReview ${ITK_LIBRARIES})
else( "${ITK_VERSION_MAJOR}" LESS 4 )
  target_link_libraries(AndImageFilter ${ITK_LIBRARIES})
endif( "${ITK_VERSION_MAJOR}" LESS 4 )
Personal tools