tests/image/RandomWalker/Original.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/Common/OriginalRandomWalker.h>
  10 #include <fpa/Functors/Dijkstra/Image/Gaussian.h>
  11
  12 // -------------------------------------------------------------------------
  13 const unsigned int Dim = 2;
  14 typedef unsigned char TPixel;
  15 typedef unsigned char TLabel;
  16 typedef double        TScalar;
  17
  18 typedef itk::Image< TPixel, Dim > TImage;
  19 typedef itk::Image< TLabel, Dim > TLabels;
  20
  21 // -------------------------------------------------------------------------
  22 int main( int argc, char* argv[] )
  23 {
  24   // Get input arguments
  25   if( argc < 8 )
  26   {
  27     std::cerr
  28       << "Usage: " << argv[ 0 ]
  29       << " input_image input_labels output_labels output_probabilities"
  30       << " beta eps normalize"
  31       << std::endl;
  32     return( 1 );
  33
  34   } // fi
  35   std::string input_image_filename, input_labels_filename;
  36   std::string output_labels_filename, output_probabilities_filename;
  37   double beta, eps;
  38   bool normalize;
  39   input_image_filename = argv[ 1 ];
  40   input_labels_filename = argv[ 2 ];
  41   output_labels_filename = argv[ 3 ];
  42   output_probabilities_filename = argv[ 4 ];
  43   beta = std::atof( argv[ 5 ] );
  44   eps = std::atof( argv[ 6 ] );
  45   normalize = ( argv[ 7 ][ 0 ] == '1' );
  46
  47   // Read image
  48   typedef itk::ImageFileReader< TImage > TImageReader;
  49   TImageReader::Pointer input_image_reader = TImageReader::New( );
  50   input_image_reader->SetFileName( input_image_filename );
  51
  52   // Read labels
  53   typedef itk::ImageFileReader< TLabels > TLabelsReader;
  54   TLabelsReader::Pointer input_labels_reader = TLabelsReader::New( );
  55   input_labels_reader->SetFileName( input_labels_filename );
  56
  57   // Random walker function
  58   typedef fpa::Functors::Dijkstra::Image::Gaussian< TImage, TScalar > TFunction;
  59   TFunction::Pointer edge = TFunction::New( );
  60   edge->SetBeta( beta );
  61   edge->SetEpsilon( eps );
  62   edge->TreatAsWeightOff( );
  63
  64   // Random walker
  65   typedef fpa::Common::OriginalRandomWalker< TImage, TLabels, TScalar > TFilter;
  66   TFilter::Pointer rw = TFilter::New( );
  67   rw->SetInput( input_image_reader->GetOutput( ) );
  68   rw->SetInputLabels( input_labels_reader->GetOutput( ) );
  69   rw->SetEdgeFunction( edge );
  70
  71   // Save labels
  72   typedef itk::ImageFileWriter< TFilter::TLabels > TLabelsWriter;
  73   TLabelsWriter::Pointer output_labels_writer = TLabelsWriter::New( );
  74   output_labels_writer->SetFileName( output_labels_filename );
  75   output_labels_writer->SetInput( rw->GetOutput( ) );
  76
  77   // Save probabilities
  78   typedef itk::ImageFileWriter< TFilter::TScalarImage > TProbabilitesWriter;
  79   TProbabilitesWriter::Pointer output_probabilities_writer =
  80     TProbabilitesWriter::New( );
  81   output_probabilities_writer->SetFileName( output_probabilities_filename );
  82   output_probabilities_writer->SetInput( rw->GetOutputProbabilities( ) );
  83
  84   try
  85   {
  86     output_labels_writer->Update( );
  87     output_probabilities_writer->Update( );
  88   }
  89   catch( std::exception& err )
  90   {
  91     std::cerr << "Error caught: " << err.what( ) << std::endl;
  92     return( 1 );
  93
  94   } // yrt
  95   return( 0 );
  96 }
  97
  98 // eof - $RCSfile$