ITK/Examples/ImageProcessing/ConvolutionImageFilter: Difference between revisions

From KitwarePublic
< ITK‎ | Examples
Jump to navigationJump to search
(Created page with "==ConvolutionImageFilter.cxx== <source lang="cpp"> #include "itkImage.h" #include "itkImageFileReader.h" #include "itkImageFileWriter.h" #include "itkRescaleIntensityImageFilter....")
 
(Use QuickView)
Line 3: Line 3:
#include "itkImage.h"
#include "itkImage.h"
#include "itkImageFileReader.h"
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include "itkRescaleIntensityImageFilter.h"
#include "itkConvolutionImageFilter.h"
#include "itkConvolutionImageFilter.h"
#include "itkImageRegionIterator.h"
#include "itkImageRegionIterator.h"


#include <itkImageToVTKImageFilter.h>
#include "QuickView.h"


#include "vtkImageViewer.h"
typedef itk::Image<float, 2> ImageType;
#include "vtkRenderWindowInteractor.h"
#include "vtkSmartPointer.h"
#include "vtkImageActor.h"
#include "vtkInteractorStyleImage.h"
#include "vtkRenderer.h"


typedef itk::Image<unsigned char, 2> ImageType;
void CreateKernel(ImageType::Pointer kernel, unsigned int width);
 
void CreateKernel(ImageType::Pointer kernel);


int main(int argc, char * argv[])
int main(int argc, char * argv[])
Line 26: Line 17:
   if( argc < 2 )
   if( argc < 2 )
     {
     {
  std::cerr << "Usage: " << std::endl;
    std::cerr << "Usage: ";
  std::cerr << argv[0] << "inputImageFile" << std::endl;
    std::cerr << argv[0] << "inputImageFile [width]" << std::endl;
  return EXIT_FAILURE;
    return EXIT_FAILURE;
     }
     }


   // Parse command line arguments
   // Parse command line arguments
   std::string inputFilename = argv[1];
   unsigned int width = 3;
  if (argc > 2)
    {
    width = atoi(argv[2]);
    }


   ImageType::Pointer kernel = ImageType::New();
   ImageType::Pointer kernel = ImageType::New();
   CreateKernel(kernel);
   CreateKernel(kernel, width);
    
    
   typedef itk::ImageFileReader<ImageType> readerType;
   typedef itk::ImageFileReader<ImageType>       ReaderType;
 
   typedef itk::ConvolutionImageFilter<ImageType> FilterType;
   typedef itk::ConvolutionImageFilter<ImageType> FilterType;


   // Create and setup a reader
   // Create and setup a reader
   readerType::Pointer reader = readerType::New();
   ReaderType::Pointer reader = ReaderType::New();
   reader->SetFileName( inputFilename.c_str() );
   reader->SetFileName( argv[1] );


   // Convolve image with kernel.
   // Convolve image with kernel.
Line 49: Line 43:
   convolutionFilter->SetInput(reader->GetOutput());
   convolutionFilter->SetInput(reader->GetOutput());
   convolutionFilter->SetImageKernelInput(kernel);
   convolutionFilter->SetImageKernelInput(kernel);
  convolutionFilter->Update();
  typedef  itk::ImageFileWriter< ImageType  > WriterType;
  WriterType::Pointer writer = WriterType::New();
  writer->SetFileName("TestOutput.jpg");
  writer->SetInput(convolutionFilter->GetOutput());
  writer->Update();
  // To visualize the derivative, we must rescale the values
  // to a reasonable range
  typedef itk::RescaleIntensityImageFilter<
  ImageType, ImageType > rescaleFilterType;
  rescaleFilterType::Pointer rescaler = rescaleFilterType::New();
  rescaler->SetOutputMinimum(0);
  rescaler->SetOutputMaximum(255);
  rescaler->SetInput( convolutionFilter->GetOutput() );
  rescaler->Update();
  typedef itk::ImageToVTKImageFilter<ImageType> ConnectorType;
  ConnectorType::Pointer originalConnector = ConnectorType::New();
  originalConnector->SetInput(reader->GetOutput());
  vtkSmartPointer<vtkImageActor> originalActor =
    vtkSmartPointer<vtkImageActor>::New();
  originalActor->SetInput(originalConnector->GetOutput());
  typedef itk::ImageToVTKImageFilter<ImageType> ConnectorType;
  ConnectorType::Pointer convolvedConnector = ConnectorType::New();
  convolvedConnector->SetInput(rescaler->GetOutput());
  vtkSmartPointer<vtkImageActor> convolvedActor =
    vtkSmartPointer<vtkImageActor>::New();
  convolvedActor->SetInput(convolvedConnector->GetOutput());
  // Define viewport ranges
  // (xmin, ymin, xmax, ymax)
  double leftViewport[4] = {0.0, 0.0, 0.5, 1.0};
  double rightViewport[4] = {0.5, 0.0, 1.0, 1.0};
  // Setup both renderers
  vtkSmartPointer<vtkRenderWindow> renderWindow =
    vtkSmartPointer<vtkRenderWindow>::New();
  renderWindow->SetSize(600,300);
  vtkSmartPointer<vtkRenderer> leftRenderer =
    vtkSmartPointer<vtkRenderer>::New();
  renderWindow->AddRenderer(leftRenderer);
  leftRenderer->SetViewport(leftViewport);
  vtkSmartPointer<vtkRenderer> rightRenderer =
    vtkSmartPointer<vtkRenderer>::New();
  renderWindow->AddRenderer(rightRenderer);
  rightRenderer->SetViewport(rightViewport);
  leftRenderer->AddActor(originalActor);
  rightRenderer->AddActor(convolvedActor);
  vtkSmartPointer<vtkRenderWindowInteractor> renderWindowInteractor =
    vtkSmartPointer<vtkRenderWindowInteractor>::New();
  vtkSmartPointer<vtkInteractorStyleImage> style =
    vtkSmartPointer<vtkInteractorStyleImage>::New();
  renderWindowInteractor->SetInteractorStyle(style);


   renderWindowInteractor->SetRenderWindow(renderWindow);
   QuickView viewer;
  renderWindowInteractor->Initialize();
  viewer.AddImage<ImageType>(
    reader->GetOutput(),true,
    itksys::SystemTools::GetFilenameName(argv[1]));


   renderWindowInteractor->Start();
   std::stringstream desc;
  desc << "ConvolutionFilter\n"
      << "Kernel Witdh = " << width;
  viewer.AddImage<ImageType>(
    convolutionFilter->GetOutput(),
    true,
    desc.str());
  viewer.Visualize();


   return EXIT_SUCCESS;
   return EXIT_SUCCESS;
}
}


void CreateKernel(ImageType::Pointer kernel)
void CreateKernel(ImageType::Pointer kernel, unsigned int width)
{
{
   ImageType::IndexType start;
   ImageType::IndexType start;
Line 128: Line 67:


   ImageType::SizeType size;
   ImageType::SizeType size;
   size.Fill(3);
   size.Fill(width);


   ImageType::RegionType region;
   ImageType::RegionType region;

Revision as of 05:04, 22 December 2010

ConvolutionImageFilter.cxx

<source lang="cpp">

  1. include "itkImage.h"
  2. include "itkImageFileReader.h"
  3. include "itkConvolutionImageFilter.h"
  4. include "itkImageRegionIterator.h"
  1. include "QuickView.h"

typedef itk::Image<float, 2> ImageType;

void CreateKernel(ImageType::Pointer kernel, unsigned int width);

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

 // Verify command line arguments
 if( argc < 2 )
   {
   std::cerr << "Usage: ";
   std::cerr << argv[0] << "inputImageFile [width]" << std::endl;
   return EXIT_FAILURE;
   }
 // Parse command line arguments
 unsigned int width = 3;
 if (argc > 2)
   {
   width = atoi(argv[2]);
   }
 ImageType::Pointer kernel = ImageType::New();
 CreateKernel(kernel, width);
 
 typedef itk::ImageFileReader<ImageType>        ReaderType;
 typedef itk::ConvolutionImageFilter<ImageType> FilterType;
 // Create and setup a reader
 ReaderType::Pointer reader = ReaderType::New();
 reader->SetFileName( argv[1] );
 // Convolve image with kernel.
 FilterType::Pointer convolutionFilter = FilterType::New();
 convolutionFilter->SetInput(reader->GetOutput());
 convolutionFilter->SetImageKernelInput(kernel);
 QuickView viewer;
 viewer.AddImage<ImageType>(
   reader->GetOutput(),true,
   itksys::SystemTools::GetFilenameName(argv[1]));  
 std::stringstream desc;
 desc << "ConvolutionFilter\n"
      << "Kernel Witdh = " << width;
 viewer.AddImage<ImageType>(
   convolutionFilter->GetOutput(),
   true,
   desc.str());
 viewer.Visualize();
 return EXIT_SUCCESS;

}

void CreateKernel(ImageType::Pointer kernel, unsigned int width) {

 ImageType::IndexType start;
 start.Fill(0);
 ImageType::SizeType size;
 size.Fill(width);
 ImageType::RegionType region;
 region.SetSize(size);
 region.SetIndex(start);
 kernel->SetRegions(region);
 kernel->Allocate();
 
 itk::ImageRegionIterator<ImageType> imageIterator(kernel, region);
 while(!imageIterator.IsAtEnd())
   {
   //imageIterator.Set(255);
   imageIterator.Set(1);
   ++imageIterator;
   }

} </source>

CMakeLists.txt

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

PROJECT(ConvolutionImageFilter)

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(ConvolutionImageFilter ConvolutionImageFilter.cxx) TARGET_LINK_LIBRARIES(ConvolutionImageFilter vtkHybrid ITKBasicFilters ITKIO ITKCommon)

</source>