appli/CTBronchi/MoriLabelling.cxx

   1 #include <chrono>
   2 #include <itkImage.h>
   3 #include <itkImageFileReader.h>
   4 #include <itkImageFileWriter.h>
   5 #include <CTBronchi/MoriLabelling.h>
   6
   7 // -------------------------------------------------------------------------
   8 const unsigned int Dim = 3;
   9 typedef short          TPixel;
  10 typedef unsigned short TLabel;
  11
  12 typedef itk::Image< TPixel, Dim > TInputImage;
  13 typedef itk::Image< TLabel, Dim > TLabelImage;
  14
  15 typedef CTBronchi::MoriLabelling< TInputImage, TLabelImage > TFilter;
  16
  17 // -------------------------------------------------------------------------
  18 int main( int argc, char* argv[] )
  19 {
  20   // Get arguments
  21   if( argc < 8 )
  22   {
  23     std::cerr
  24       << "Usage: " << argv[ 0 ] << std::endl
  25       << "   input_image label_image output_image" << std::endl
  26       << "   upper_threshold(-400)" << std::endl
  27       << "   inside_value(255)" << std::endl
  28       << "   inside_label(1) outside_label(2)"
  29       << std::endl;
  30     return( 1 );
  31
  32   } // fi
  33   std::string input_image_filename = argv[ 1 ];
  34   std::string label_image_filename = argv[ 2 ];
  35   std::string output_image_filename = argv[ 3 ];
  36   TLabel upper_threshold = std::atoi( argv[ 4 ] );
  37   TLabel inside_value = std::atoi( argv[ 5 ] );
  38   TLabel inside_label = std::atoi( argv[ 6 ] );
  39   TLabel outside_label = std::atoi( argv[ 7 ] );
  40
  41   // Read images
  42   itk::ImageFileReader< TInputImage >::Pointer input_image_reader =
  43     itk::ImageFileReader< TInputImage >::New( );
  44   input_image_reader->SetFileName( input_image_filename );
  45
  46   itk::ImageFileReader< TLabelImage >::Pointer label_image_reader =
  47     itk::ImageFileReader< TLabelImage >::New( );
  48   label_image_reader->SetFileName( label_image_filename );
  49
  50   // Prepare filter
  51   TFilter::Pointer filter = TFilter::New( );
  52   filter->SetInputLabelImage( label_image_reader->GetOutput( ) );
  53   filter->SetInputRawImage( input_image_reader->GetOutput( ) );
  54   filter->SetUpperThreshold( upper_threshold );
  55   filter->SetInsideValue( inside_value );
  56   filter->SetInsideLabel( inside_label );
  57   filter->SetOutsideLabel( outside_label );
  58
  59   // Show some information
  60   std::cout << "----------------------------------------------" << std::endl;
  61   std::cout << "Image: " << input_image_filename << std::endl;
  62
  63   // Execute pipeline
  64   std::chrono::time_point< std::chrono::high_resolution_clock > ts, te;
  65   std::chrono::duration< double > tr;
  66   try
  67   {
  68     ts = std::chrono::high_resolution_clock::now( );
  69     input_image_reader->Update( );
  70     te = std::chrono::high_resolution_clock::now( );
  71     tr = te - ts;
  72     std::cout << "Raw read time: " << tr.count( ) << " s" << std::endl;
  73
  74     ts = std::chrono::high_resolution_clock::now( );
  75     label_image_reader->Update( );
  76     te = std::chrono::high_resolution_clock::now( );
  77     tr = te - ts;
  78     std::cout << "Label read time: " << tr.count( ) << " s" << std::endl;
  79
  80     ts = std::chrono::high_resolution_clock::now( );
  81     filter->Update( );
  82     te = std::chrono::high_resolution_clock::now( );
  83     tr = te - ts;
  84     std::cout
  85       << "Labelling time: " << tr.count( ) << " s" << std::endl;
  86   }
  87   catch( std::exception& err )
  88   {
  89     std::cerr << "Error caught: " << err.what( ) << std::endl;
  90     return( 1 );
  91
  92   } // yrt
  93
  94   // Save output image
  95   itk::ImageFileWriter< TLabelImage >::Pointer output_image_writer =
  96     itk::ImageFileWriter< TLabelImage >::New( );
  97   output_image_writer->SetInput( filter->GetOutput( ) );
  98   output_image_writer->SetFileName( output_image_filename );
  99   try
 100   {
 101     ts = std::chrono::high_resolution_clock::now( );
 102     output_image_writer->Update( );
 103     te = std::chrono::high_resolution_clock::now( );
 104     tr = te - ts;
 105     std::cout << "Write time: " << tr.count( ) << " s" << std::endl;
 106   }
 107   catch( std::exception& err )
 108   {
 109     std::cerr << "Error caught: " << err.what( ) << std::endl;
 110     return( 1 );
 111
 112   } // yrt
 113   std::cout << "----------------------------------------------" << std::endl;
 114
 115   return( 0 );
 116 }
 117
 118 // eof - $RCSfile$