appli/examples/example_RGBImageToYPbPrChannels.cxx

   1 #include <cstdlib>
   2 #include <iostream>
   3 #include <string>
   4
   5 #include <cpPlugins/Interface/Interface.h>
   6 #include <cpPlugins/Interface/DataObject.h>
   7 #include <cpPlugins/Interface/ProcessObject.h>
   8
   9 // -------------------------------------------------------------------------
  10 typedef cpPlugins::Interface::Interface     TInterface;
  11 typedef cpPlugins::Interface::DataObject    TDataObject;
  12 typedef TInterface::TClasses                TClasses;
  13 typedef cpPlugins::Interface::ProcessObject TProcessObject;
  14 typedef cpPlugins::Interface::Parameters    TParameters;
  15
  16 // -------------------------------------------------------------------------
  17 void SaveImage(
  18   TInterface& plugins, const std::string& fname, TDataObject* image
  19   )
  20 {
  21   TProcessObject::Pointer writer;
  22   writer = plugins.CreateProcessObject( "cpPlugins::Plugins::ImageWriter" );
  23   if( writer.IsNull( ) )
  24   {
  25     std::cerr << "No suitable writer found in plugins." << std::endl;
  26     return;
  27
  28   } // fi
  29
  30   // Configure reader
  31   TParameters writer_params = writer->GetDefaultParameters( );
  32   writer_params.SetValueAsString( "FileName", fname );
  33   writer->SetParameters( writer_params );
  34
  35   writer->SetInput( 0, image );
  36   std::string msg = writer->Update( );
  37   if( msg != "" )
  38     std::cerr << "ERROR: " << msg << std::endl;
  39 }
  40
  41 // -------------------------------------------------------------------------
  42 int main( int argc, char* argv[] )
  43 {
  44   if( argc < 8 )
  45   {
  46     std::cerr
  47       << "Usage: " << argv[ 0 ]
  48       << " plugins_file"
  49       << " input_image"
  50       << " output_hue_image output_saturation_image output_value_image"
  51       << " dimensions pixel_type" << std::endl;
  52     return( 1 );
  53
  54   } // fi
  55   std::string plugins_file = argv[ 1 ];
  56   std::string input_image_file = argv[ 2 ];
  57   std::string output_hue_image_file = argv[ 3 ];
  58   std::string output_saturation_image_file = argv[ 4 ];
  59   std::string output_value_image_file = argv[ 5 ];
  60   unsigned int dimensions = std::atoi( argv[ 6 ] );
  61   std::string pixel_type = argv[ 7 ];
  62
  63   // Create interface
  64
  65   TInterface plugins;
  66   plugins.Load( plugins_file );
  67
  68   // Create objects
  69   TProcessObject::Pointer reader;
  70   TProcessObject::Pointer filter;
  71
  72   reader = plugins.CreateProcessObject( "cpPlugins::Plugins::ImageReader" );
  73   if( reader.IsNull( ) )
  74   {
  75     std::cerr << "No suitable reader found in plugins." << std::endl;
  76     return( 1 );
  77
  78   } // fi
  79   filter = plugins.CreateProcessObject( "cpPlugins::Plugins::RGBImageToYPbPrChannelsFilter" );
  80   if( filter.IsNull( ) )
  81   {
  82     std::cerr << "No suitable filter found in plugins." << std::endl;
  83     return( 1 );
  84
  85   } // fi
  86
  87   // Configure reader
  88   TParameters reader_params = reader->GetDefaultParameters( );
  89   reader_params.AddValueToStringList( "FileNames", input_image_file );
  90   reader_params.SetValueAsString( "PixelType", pixel_type );
  91   reader_params.SetValueAsUint( "ImageDimension", dimensions );
  92   reader_params.SetValueAsUint( "IsColorImage", 1 );
  93   reader->SetParameters( reader_params );
  94
  95   // Connect pipeline
  96   TParameters filter_params = filter->GetDefaultParameters( );
  97   filter_params.SetValueAsString( "OutputPixelType", "float" );
  98   filter->SetParameters( filter_params );
  99   filter->SetInput( 0, reader->GetOutput( 0 ) );
 100
 101   // Save outputs
 102   SaveImage( plugins, output_hue_image_file, filter->GetOutput( 0 ) );
 103   SaveImage( plugins, output_saturation_image_file, filter->GetOutput( 1 ) );
 104   SaveImage( plugins, output_value_image_file, filter->GetOutput( 2 ) );
 105
 106   return( 0 );
 107 }
 108
 109 // eof - $RCSfile$