tests/image/RegionGrow/Mori.cxx

   1 // =========================================================================
   2 // @author Leonardo Florez Valencia
   3 // @email florez-l@javeriana.edu.co
   4 // =========================================================================
   5
   6 #include <itkImage.h>
   7 #include <itkImageFileReader.h>
   8 #include <itkImageFileWriter.h>
   9 #include <fpa/Filters/Image/Mori.h>
  10
  11 // -------------------------------------------------------------------------
  12 const unsigned int Dim = 2;
  13 typedef unsigned char TPixel;
  14 typedef itk::Image< TPixel, Dim > TImage;
  15
  16 // -------------------------------------------------------------------------
  17 int main( int argc, char* argv[] )
  18 {
  19   // Get arguments
  20   if( argc < 9 + Dim )
  21   {
  22     std::cerr
  23       << "Usage: " << argv[ 0 ]
  24       << " input_image output_image lower upper delta seed"
  25       << std::endl;
  26     return( 1 );
  27
  28   } // fi
  29   std::string input_image_filename = argv[ 1 ];
  30   std::string output_image_filename = argv[ 2 ];
  31   TPixel lower = TPixel( std::atof( argv[ 3 ] ) );
  32   TPixel upper = TPixel( std::atof( argv[ 4 ] ) );
  33   TPixel delta = TPixel( std::atof( argv[ 5 ] ) );
  34   unsigned long signal_kernel_size = std::atoi( argv[ 6 ] );
  35   double signal_threshold = std::atof( argv[ 7 ] );
  36   double signal_influence = std::atof( argv[ 8 ] );
  37
  38   // Read image
  39   typedef itk::ImageFileReader< TImage > TImageReader;
  40   TImageReader::Pointer input_image_reader = TImageReader::New( );
  41   input_image_reader->SetFileName( input_image_filename );
  42
  43   // Prepare filter
  44   typedef fpa::Filters::Image::Mori< TImage, TImage > TFilter;
  45   TFilter::Pointer filter = TFilter::New( );
  46   filter->SetInput( input_image_reader->GetOutput( ) );
  47   filter->SetInsideValue( std::numeric_limits< TPixel >::max( ) );
  48   filter->SetOutsideValue( TPixel( 0 ) );
  49   filter->SetThresholds( lower, upper, delta );
  50   filter->SetSignalKernelSize( signal_kernel_size );
  51   filter->SetSignalThreshold( signal_threshold );
  52   filter->SetSignalInfluence( signal_influence );
  53
  54   // Configure seed
  55   TImage::IndexType seed;
  56   for( int j = 0; j < Dim; ++j )
  57     seed[ j ] = std::atoi( argv[ 9 + j ] );
  58   filter->SetSeed( seed );
  59
  60   // Execute filter
  61   filter->Update( );
  62
  63   // Save results
  64   typedef itk::ImageFileWriter< TFilter::TOutputImage > TOutputWriter;
  65   TOutputWriter::Pointer output_writer = TOutputWriter::New( );
  66   output_writer->SetInput( filter->GetOutput( ) );
  67   output_writer->SetFileName( output_image_filename );
  68   try
  69   {
  70     output_writer->Update( );
  71   }
  72   catch( std::exception& err )
  73   {
  74     std::cerr << "Error caught: " << err.what( ) << std::endl;
  75     return( 1 );
  76
  77   } // yrt
  78
  79   return( 0 );
  80 }
  81
  82 // eof - $RCSfile$