examples/SkeletonFilter.cxx

   1
   2 #include <chrono>
   3 #include <iomanip>
   4 #include <itkImage.h>
   5 #include <itkImageFileReader.h>
   6
   7 #include <fpa/Image/SkeletonFilter.h>
   8
   9 /* TODO
  10    #include <itkCommand.h>
  11    #include <itkImageFileWriter.h>
  12    #include <itkBinaryThresholdImageFilter.h>
  13    #include <fpa/Image/MoriRegionGrow.h>
  14 */
  15
  16 // -------------------------------------------------------------------------
  17 static const unsigned int VDim = 3;
  18 typedef unsigned char                                 TPixel;
  19 typedef double                                        TScalar;
  20 typedef itk::Image< TPixel, VDim >                    TImage;
  21 typedef itk::ImageFileReader< TImage >                TReader;
  22 typedef fpa::Image::SkeletonFilter< TImage, TScalar > TFilter;
  23
  24 /* TODO
  25    typedef itk::ImageFileWriter< TImage >                    TWriter;
  26    typedef fpa::Image::MoriRegionGrow< TImage, TImage >      TFilter;
  27    typedef itk::BinaryThresholdImageFilter< TImage, TImage > TThreshold;
  28 */
  29
  30 // -------------------------------------------------------------------------
  31 int main( int argc, char* argv[] )
  32 {
  33   // Get arguments
  34   if( argc != 3 && argc != 3 + VDim )
  35   {
  36     std::cerr
  37       << "Usage: " << argv[ 0 ]
  38       << " input_image output_skeleton [";
  39     for( unsigned int i = 0; i < VDim; ++i )
  40       std::cerr << " s_" << i;
  41     std::cerr << " ]" << std::endl;
  42     return( 1 );
  43
  44   } // fi
  45   std::string input_image_filename = argv[ 1 ];
  46   std::string output_skeleton_filename = argv[ 2 ];
  47
  48   TReader::Pointer reader = TReader::New( );
  49   reader->SetFileName( input_image_filename );
  50   try
  51   {
  52     reader->Update( );
  53   }
  54   catch( std::exception& err )
  55   {
  56     std::cerr << "ERROR: " << err.what( ) << std::endl;
  57     return( 1 );
  58
  59   } // yrt
  60
  61   TFilter::Pointer filter = TFilter::New( );
  62   filter->SetInput( reader->GetOutput( ) );
  63
  64   /* TODO
  65      filter->SetThresholdRange( lower, upper, delta );
  66      TImage::PointType pnt;
  67      for( int i = 0; i < VDim; ++i )
  68      pnt[ i ] = std::atof( argv[ i + 6 ] );
  69
  70      TImage::IndexType seed;
  71      if( !( reader->GetOutput( )->TransformPhysicalPointToIndex( pnt, seed ) ) )
  72      {
  73      std::cerr << "ERROR: seed outside image." << std::endl;
  74      return( 1 );
  75
  76      } // fi
  77      filter->AddSeed( seed );
  78
  79      std::chrono::time_point< std::chrono::system_clock > start, end;
  80      start = std::chrono::system_clock::now( );
  81      filter->Update( );
  82      end = std::chrono::system_clock::now( );
  83      std::chrono::duration< double > elapsed_seconds = end - start;
  84
  85      TThreshold::Pointer threshold = TThreshold::New( );
  86      threshold->SetInput( filter->GetOutput( ) );
  87      threshold->SetInsideValue( 255 );
  88      threshold->SetOutsideValue( 0 );
  89      threshold->SetLowerThreshold( 0 );
  90      threshold->SetUpperThreshold( filter->GetOptimumThreshold( ) );
  91
  92      TWriter::Pointer writer = TWriter::New( );
  93      writer->SetInput( threshold->GetOutput( ) );
  94      writer->SetFileName( output_image_filename );
  95      try
  96      {
  97      writer->Update( );
  98      }
  99      catch( std::exception& err )
 100      {
 101      std::cerr << "ERROR: " << err.what( ) << std::endl;
 102      return( 1 );
 103
 104      } // yrt
 105
 106      // Show data
 107      TFilter::TCurve curve = filter->GetCurve( );
 108      for( TFilter::TCurveData data: curve )
 109      {
 110      std::cout << data.XValue << " " << data.YValue << " " << data.Diff1 << std::endl;
 111      }
 112      std::cout
 113      << std::endl
 114      << "# Opt: "
 115      << curve[ filter->GetOptimumThreshold( ) ].XValue
 116      << "("
 117      << filter->GetOptimumThreshold( )
 118      << ")"
 119      << std::endl;
 120      std::cout << "Time: " << elapsed_seconds.count( ) << "s" << std::endl;
 121   */
 122   return( 0 );
 123 }
 124
 125 // eof - $RCSfile$