1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkGrayscaleGeodesicErodeDilateImageFilterTest.cxx
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notices for more information.
=========================================================================*/
#if defined(_MSC_VER)
#pragma warning ( disable : 4786 )
#endif
#include "itkSimpleFilterWatcher.h"
#include "itkGrayscaleGeodesicDilateImageFilter.h"
#include "itkGrayscaleGeodesicErodeImageFilter.h"
#include "itkImage.h"
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include "itkShiftScaleImageFilter.h"
// This test should produce the same results as the
// itkHMaximaMinimaImageFilterTest.
int itkGrayscaleGeodesicErodeDilateImageFilterTest(int argc, char* argv [] )
{
if ( argc < 4 )
{
std::cerr << "Missing arguments" << std::endl;
std::cerr << "Usage: " << argv[0] << " Inputimage OutputImage Height" << std::endl;
return EXIT_FAILURE;
}
const int Dimension = 2;
typedef unsigned char PixelType;
typedef itk::Image< PixelType, Dimension > InputImageType;
typedef itk::Image< PixelType, Dimension > OutputImageType;
typedef itk::ImageFileReader< InputImageType >
ReaderType;
typedef itk::ImageFileWriter< OutputImageType >
WriterType;
typedef itk::ShiftScaleImageFilter <InputImageType,InputImageType>
ShiftFilterType;
typedef itk::GrayscaleGeodesicDilateImageFilter < InputImageType,
OutputImageType >
DilateFilterType;
typedef itk::GrayscaleGeodesicErodeImageFilter < InputImageType,
OutputImageType >
ErodeFilterType;
ReaderType::Pointer reader = ReaderType::New();
WriterType::Pointer writer = WriterType::New();
ShiftFilterType::Pointer shiftErode = ShiftFilterType::New();
ShiftFilterType::Pointer shiftDilate = ShiftFilterType::New();
ErodeFilterType::Pointer erode = ErodeFilterType::New();
DilateFilterType::Pointer dilate = DilateFilterType::New();
// Create the reader and writer
reader->SetFileName( argv[1] );
writer->SetFileName( argv[2] );
// Create the marker image for erosion
shiftDilate->SetInput( reader->GetOutput());
shiftDilate->SetShift( -1 * atoi(argv[3]) );
// Dilate
dilate->SetMarkerImage ( shiftDilate-> GetOutput() );
dilate->SetMaskImage ( reader->GetOutput() );
dilate->FullyConnectedOn();
// Create the marker image for erode
shiftErode->SetInput( dilate->GetOutput());
shiftErode->SetShift( atoi(argv[3]) );
// Erode
erode->SetMarkerImage ( shiftErode->GetOutput() );
erode->SetMaskImage ( dilate->GetOutput() );
erode->FullyConnectedOn();
writer->SetInput ( erode->GetOutput() );
itk::SimpleFilterWatcher watchDilate(dilate);
itk::SimpleFilterWatcher watchErode(erode);
// Execute the filter
try
{
writer->Update();
}
catch( itk::ExceptionObject & excp )
{
std::cerr << "Exception caught:" << excp << std::endl;
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}
|