examples/RegionGrow_Mori.cxx

   1 #include <itkCommand.h>
   2 #include <itkImage.h>
   3 #include <itkImageFileReader.h>
   4 #include <itkImageFileWriter.h>
   5
   6 #include <fpa/Image/MoriRegionGrow.h>
   7
   8 // -------------------------------------------------------------------------
   9 static const unsigned int VDim = 3;
  10 typedef short                          TPixel;
  11 typedef itk::Image< TPixel, VDim >     TImage;
  12 typedef itk::ImageFileReader< TImage > TReader;
  13 typedef itk::ImageFileWriter< TImage > TWriter;
  14 typedef fpa::Image::MoriRegionGrow< TImage, TImage > TFilter;
  15 typedef itk::ImageFileWriter< TFilter::TAuxiliaryImage > TAuxWriter;
  16
  17 // -------------------------------------------------------------------------
  18 int main( int argc, char* argv[] )
  19 {
  20   // Get arguments
  21   if( argc < 7 + VDim )
  22   {
  23     std::cerr
  24       << "Usage: " << argv[ 0 ]
  25       << " input_image output_image auxiliary_image lower upper delta";
  26     for( unsigned int i = 0; i < VDim; ++i )
  27       std::cerr << " s_" << i;
  28     std::cerr << std::endl;
  29     return( 1 );
  30
  31   } // fi
  32   std::string input_image_filename = argv[ 1 ];
  33   std::string output_image_filename = argv[ 2 ];
  34   std::string auxiliary_image_filename = argv[ 3 ];
  35   TPixel lower = std::atof( argv[ 4 ] );
  36   TPixel upper = std::atof( argv[ 5 ] );
  37   TPixel delta = std::atof( argv[ 6 ] );
  38
  39   TReader::Pointer reader = TReader::New( );
  40   reader->SetFileName( input_image_filename );
  41
  42   TFilter::Pointer filter = TFilter::New( );
  43   filter->SetInput( reader->GetOutput( ) );
  44   filter->SetThresholdRange( lower, upper, delta );
  45   TImage::IndexType seed;
  46   for( int i = 0; i < VDim; ++i )
  47     seed[ i ] = std::atoi( argv[ i + 7 ] );
  48   filter->SetSeed( seed );
  49   filter->SetInsideValue( 255 );
  50   filter->SetOutsideValue( 0 );
  51
  52   TWriter::Pointer writer = TWriter::New( );
  53   writer->SetInput( filter->GetOutput( ) );
  54   writer->SetFileName( output_image_filename );
  55   try
  56   {
  57     writer->Update( );
  58   }
  59   catch( std::exception& err )
  60   {
  61     std::cerr << "ERROR: " << err.what( ) << std::endl;
  62     return( 1 );
  63
  64   } // yrt
  65
  66   TAuxWriter::Pointer aux_writer = TAuxWriter::New( );
  67   aux_writer->SetInput( filter->GetAuxiliaryImage( ) );
  68   aux_writer->SetFileName( auxiliary_image_filename );
  69   try
  70   {
  71     aux_writer->Update( );
  72   }
  73   catch( std::exception& err )
  74   {
  75     std::cerr << "ERROR: " << err.what( ) << std::endl;
  76     return( 1 );
  77
  78   } // yrt
  79   return( 0 );
  80 }
  81
  82 // eof - $RCSfile$