appli/examples/example_View2DImage.cxx

   1 #include <cstdlib>
   2 #include <iostream>
   3 #include <string>
   4
   5 #include <vtkRenderer.h>
   6 #include <vtkRenderWindow.h>
   7 #include <vtkRenderWindowInteractor.h>
   8
   9 #include <cpPlugins/Interface/Interface.h>
  10 #include <cpPlugins/Interface/ProcessObject.h>
  11 #include <cpPlugins/Interface/Image.h>
  12
  13 #include <cpExtensions/Visualization/ImageSliceActors.h>
  14
  15
  16 #include <cpExtensions/Visualization/ImageBlender.h>
  17 #include <vtkPNGWriter.h>
  18
  19
  20 // -------------------------------------------------------------------------
  21 typedef cpPlugins::Interface::Interface     TInterface;
  22 typedef cpPlugins::Interface::ProcessObject TProcessObject;
  23 typedef cpPlugins::Interface::DataObject    TDataObject;
  24 typedef cpPlugins::Interface::Image         TImage;
  25 typedef cpPlugins::Interface::Parameters    TParameters;
  26 typedef TInterface::TClasses                TClasses;
  27
  28 typedef cpExtensions::Visualization::ImageSliceActors TSliceActors;
  29
  30 // -------------------------------------------------------------------------
  31 int main( int argc, char* argv[] )
  32 {
  33   if( argc < 3 )
  34   {
  35     std::cerr
  36       << "Usage: " << argv[ 0 ]
  37       << " plugins_file input_image [widget]"
  38       << std::endl;
  39     return( 1 );
  40
  41   } // fi
  42
  43   // Create interface
  44   TInterface plugins;
  45   if( !plugins.Load( argv[ 1 ] ) )
  46   {
  47     std::cerr << "Failed to load plugins." << std::endl;
  48     return( 1 );
  49
  50   } // fi
  51
  52   // Create objects
  53   TProcessObject::Pointer reader;
  54   reader = plugins.CreateProcessObject( "cpPlugins::IO::ImageReader" );
  55   if( reader.IsNull( ) )
  56   {
  57     std::cerr
  58       << "No suitable reader found in plugins." << std::endl
  59       << "Reader: " << reader.GetPointer( ) << std::endl
  60       << std::endl;
  61     return( 1 );
  62
  63   } // fi
  64
  65   // Configure reader
  66   TParameters* reader_params = reader->GetParameters( );
  67   if( argc > 4 )
  68   {
  69     for( int i = 2; i < argc; ++i )
  70       reader_params->AddToStringList( "FileNames", argv[ i ] );
  71   }
  72   else
  73     reader_params->AddToStringList( "FileNames", argv[ 2 ] );
  74
  75   // Execute pipeline
  76   std::string err = reader->Update( );
  77   if( err != "" )
  78   {
  79     std::cerr << "ERROR: " << err << std::endl;
  80     return( 1 );
  81
  82   } // fi
  83   TImage* image = reader->GetOutput< TImage >( "Output" );
  84
  85   // Configure visualization objects
  86   vtkSmartPointer< vtkRenderer > renderer =
  87     vtkSmartPointer< vtkRenderer >::New( );
  88   renderer->SetBackground( 0.1, 0.3, 0.5 );
  89
  90   vtkSmartPointer< vtkRenderWindow > window =
  91     vtkSmartPointer< vtkRenderWindow >::New( );
  92   window->AddRenderer( renderer );
  93   window->SetSize( 600, 600 );
  94
  95   // Set up the interaction
  96   vtkSmartPointer< vtkRenderWindowInteractor > interactor =
  97     vtkSmartPointer< vtkRenderWindowInteractor >::New( );
  98   window->SetInteractor( interactor );
  99
 100   // Create slice actors
 101   vtkSmartPointer< TSliceActors > image_actors =
 102     vtkSmartPointer< TSliceActors >::New( );
 103   image_actors->AddInputData( image->GetVTK< vtkImageData >( ), 2 );
 104   image_actors->PushActorsInto( window );
 105
 106
 107
 108
 109
 110   vtkImageData* vimage = image->GetVTK< vtkImageData >( );
 111   double range[ 2 ];
 112   vimage->GetScalarRange( range );
 113
 114   vtkSmartPointer< cpExtensions::Visualization::ImageBlender > blender =
 115     vtkSmartPointer< cpExtensions::Visualization::ImageBlender >::New( );
 116   blender->AddInputData( vimage );
 117   blender->SetWindow( range[ 1 ] - range[ 0 ] );
 118   blender->SetLevel( ( range[ 1 ] + range[ 0 ] ) / double( 2 ) );
 119   blender->Update( );
 120
 121   std::cout << blender->GetWindow( ) << " " << blender->GetLevel( ) << std::endl;
 122
 123   vtkSmartPointer< vtkPNGWriter > writer =
 124     vtkSmartPointer< vtkPNGWriter >::New( );
 125   writer->SetInputConnection( blender->GetOutputPort( ) );
 126   writer->SetFileName( "leo_actor.png" );
 127   writer->Write( );
 128
 129   return( 0 );
 130
 131
 132   // Activate seed widget
 133   if( argc == 4 )
 134   {
 135     TSliceActors::TStyle* style =
 136       dynamic_cast< TSliceActors::TStyle* >( image_actors->GetStyle( ) );
 137     if( style != NULL )
 138     {
 139       if( std::string( argv[ 3 ] ) == std::string( "seed" ) )
 140         style->SeedWidgetOn( );
 141
 142     } // fi
 143
 144   } // fi
 145
 146   // Begin interaction
 147   image_actors->ResetCamera( );
 148   window->Render( );
 149   interactor->Start( );
 150
 151   return( 0 );
 152 }
 153
 154 // eof - $RCSfile$