lib/cpExtensions/Visualization/MPRActors.cxx

   1 #include <cpExtensions/Visualization/MPRActors.h>
   2
   3 #include <vtkAlgorithmOutput.h>
   4 #include <vtkImageData.h>
   5 #include <vtkOutlineSource.h>
   6 #include <vtkProperty.h>
   7 #include <vtkRenderer.h>
   8 #include <vtkRendererCollection.h>
   9 #include <vtkRenderWindow.h>
  10
  11 // -------------------------------------------------------------------------
  12 cpExtensions::Visualization::MPRActors*
  13 cpExtensions::Visualization::MPRActors::
  14 New( )
  15 {
  16   return( new Self( ) );
  17 }
  18
  19 // -------------------------------------------------------------------------
  20 cpExtensions::Visualization::
  21 ImageSliceActors* cpExtensions::Visualization::MPRActors::
  22 GetSliceActors( const int& i ) const
  23 {
  24   if( i < 3 )
  25     return( this->Slices[ 0 ][ i ] );
  26   else
  27     return( NULL );
  28 }
  29
  30 // -------------------------------------------------------------------------
  31 int cpExtensions::Visualization::MPRActors::
  32 AddInputConnection( vtkAlgorithmOutput* aout )
  33 {
  34   int N = this->Slices[ 0 ][ 0 ]->GetNumberOfImageActors( );
  35   if( N == 0 )
  36   {
  37     this->Slices[ 0 ][ 0 ]->AddInputConnection( aout, 0 );
  38     vtkImageMapToColors* imap = this->Slices[ 0 ][ 0 ]->GetImageMap( 0 );
  39     for( unsigned int i = 0; i < 2; ++i )
  40       for( unsigned int j = 0; j < 3; ++j )
  41         if( i != 0 || j != 0 )
  42           this->Slices[ i ][ j ]->AddInputConnection(
  43             ( ( imap != NULL )? imap->GetOutputPort( ): aout ), j
  44             );
  45
  46     // Create bounding box
  47     vtkImageData* new_image = dynamic_cast< vtkImageData* >(
  48       aout->GetProducer( )->GetOutputDataObject( aout->GetIndex( ) )
  49       );
  50
  51     // Create 3D outline
  52     double bb[ 6 ];
  53     new_image->GetBounds( bb );
  54
  55     vtkSmartPointer< vtkOutlineSource > img_ol =
  56       vtkSmartPointer< vtkOutlineSource >::New( );
  57     img_ol->SetBounds( bb );
  58
  59     vtkSmartPointer< vtkPolyDataMapper > img_ol_mapper =
  60       vtkSmartPointer< vtkPolyDataMapper >::New( );
  61     img_ol_mapper->SetInputConnection( img_ol->GetOutputPort( ) );
  62     this->ImageOutlineActor->SetMapper( img_ol_mapper );
  63     this->ImageOutlineActor->GetProperty( )->SetColor( 1, 1, 1 );
  64     this->ImageOutlineActor->GetProperty( )->SetLineWidth( 1 );
  65
  66     this->AddItem( this->ImageOutlineActor );
  67   }
  68   else
  69   {
  70     // Check if the image share the same space
  71     vtkImageData* new_image = dynamic_cast< vtkImageData* >(
  72       aout->GetProducer( )->GetOutputDataObject( aout->GetIndex( ) )
  73       );
  74     if( new_image != NULL )
  75     {
  76       vtkAlgorithmOutput* ref_aout =
  77         this->Slices[ 0 ][ 0 ]->GetImageMap( 0 )->GetOutputPort( );
  78       vtkImageData* ref_image = dynamic_cast< vtkImageData* >(
  79         ref_aout->GetProducer( )->GetOutputDataObject( ref_aout->GetIndex( ) )
  80         );
  81       if( ref_image != NULL )
  82       {
  83         int ref_ext[ 6 ], new_ext[ 6 ];
  84         ref_image->GetExtent( ref_ext );
  85         new_image->GetExtent( new_ext );
  86         if(
  87           ref_ext[ 0 ] == new_ext[ 0 ] && ref_ext[ 1 ] == new_ext[ 1 ] &&
  88           ref_ext[ 2 ] == new_ext[ 2 ] && ref_ext[ 3 ] == new_ext[ 3 ] &&
  89           ref_ext[ 4 ] == new_ext[ 4 ] && ref_ext[ 5 ] == new_ext[ 5 ]
  90           )
  91         {
  92           this->Slices[ 0 ][ 0 ]->AddInputConnection( aout, 0 );
  93           vtkImageMapToColors* imap = this->Slices[ 0 ][ 0 ]->GetImageMap( N );
  94           for( unsigned int i = 0; i < 2; ++i )
  95             for( unsigned int j = 0; j < 3; ++j )
  96               if( i != 0 || j != 0 )
  97                 this->Slices[ i ][ j ]->AddInputConnection(
  98                   ( ( imap != NULL )? imap->GetOutputPort( ): aout ), j
  99                   );
 100         }
 101         else
 102           N = -1;
 103       }
 104       else
 105         N = -1;
 106     }
 107     else
 108       N = -1;
 109
 110   } // fi
 111   return( N );
 112 }
 113
 114 // -------------------------------------------------------------------------
 115 int cpExtensions::Visualization::MPRActors::
 116 AddInputData( vtkImageData* new_image )
 117 {
 118   int N = this->Slices[ 0 ][ 0 ]->GetNumberOfImageActors( );
 119   if( N == 0 )
 120   {
 121     this->Slices[ 0 ][ 0 ]->AddInputData( new_image, 0 );
 122     vtkImageMapToColors* imap = this->Slices[ 0 ][ 0 ]->GetImageMap( 0 );
 123     for( unsigned int i = 0; i < 2; ++i )
 124       for( unsigned int j = 0; j < 3; ++j )
 125         if( i != 0 || j != 0 )
 126         {
 127           if( imap != NULL )
 128             this->Slices[ i ][ j ]->AddInputConnection(
 129               imap->GetOutputPort( ), j
 130               );
 131           else
 132             this->Slices[ i ][ j ]->AddInputData( new_image, j );
 133
 134         } // fi
 135
 136     // Create 3D outline
 137     double bb[ 6 ];
 138     new_image->GetBounds( bb );
 139
 140     vtkSmartPointer< vtkOutlineSource > img_ol =
 141       vtkSmartPointer< vtkOutlineSource >::New( );
 142     img_ol->SetBounds( bb );
 143
 144     vtkSmartPointer< vtkPolyDataMapper > img_ol_mapper =
 145       vtkSmartPointer< vtkPolyDataMapper >::New( );
 146     img_ol_mapper->SetInputConnection( img_ol->GetOutputPort( ) );
 147     this->ImageOutlineActor->SetMapper( img_ol_mapper );
 148     this->ImageOutlineActor->GetProperty( )->SetColor( 1, 1, 1 );
 149     this->ImageOutlineActor->GetProperty( )->SetLineWidth( 1 );
 150
 151     this->AddItem( this->ImageOutlineActor );
 152   }
 153   else
 154   {
 155     // Check if the image share the same space
 156     vtkAlgorithmOutput* ref_aout =
 157       this->Slices[ 0 ][ 0 ]->GetImageMap( 0 )->GetOutputPort( );
 158     vtkImageData* ref_image = dynamic_cast< vtkImageData* >(
 159       ref_aout->GetProducer( )->GetOutputDataObject( ref_aout->GetIndex( ) )
 160       );
 161     if( ref_image != NULL )
 162     {
 163       int ref_ext[ 6 ], new_ext[ 6 ];
 164       ref_image->GetExtent( ref_ext );
 165       new_image->GetExtent( new_ext );
 166       if(
 167         ref_ext[ 0 ] == new_ext[ 0 ] && ref_ext[ 1 ] == new_ext[ 1 ] &&
 168         ref_ext[ 2 ] == new_ext[ 2 ] && ref_ext[ 3 ] == new_ext[ 3 ] &&
 169         ref_ext[ 4 ] == new_ext[ 4 ] && ref_ext[ 5 ] == new_ext[ 5 ]
 170         )
 171       {
 172         this->Slices[ 0 ][ 0 ]->AddInputData( new_image, 0 );
 173         vtkImageMapToColors* imap = this->Slices[ 0 ][ 0 ]->GetImageMap( N );
 174         for( unsigned int i = 0; i < 2; ++i )
 175           for( unsigned int j = 0; j < 3; ++j )
 176             if( i != 0 || j != 0 )
 177             {
 178               if( imap != NULL )
 179                 this->Slices[ i ][ j ]->AddInputConnection(
 180                   imap->GetOutputPort( ), j
 181                   );
 182               else
 183                 this->Slices[ i ][ j ]->AddInputData( new_image, j );
 184
 185             } // fi
 186       }
 187       else
 188         N = -1;
 189     }
 190     else
 191       N = -1;
 192
 193   } // fi
 194   return( N );
 195 }
 196
 197 // -------------------------------------------------------------------------
 198 unsigned int cpExtensions::Visualization::MPRActors::
 199 GetNumberOfImages( ) const
 200 {
 201   return( this->Slices[ 0 ][ 0 ]->GetNumberOfImageActors( ) );
 202 }
 203
 204 // -------------------------------------------------------------------------
 205 void cpExtensions::Visualization::MPRActors::
 206 PushActorsInto(
 207   vtkRenderWindow* x, vtkRenderWindow* y, vtkRenderWindow* z,
 208   vtkRenderWindow* w
 209   )
 210 {
 211   this->Slices[ 0 ][ 0 ]->PushActorsInto( x, true );
 212   this->Slices[ 0 ][ 1 ]->PushActorsInto( y, true );
 213   this->Slices[ 0 ][ 2 ]->PushActorsInto( z, true );
 214   this->Slices[ 1 ][ 0 ]->PushActorsInto( w, false );
 215   this->Slices[ 1 ][ 1 ]->PushActorsInto( w, false );
 216   this->Slices[ 1 ][ 2 ]->PushActorsInto( w, false );
 217
 218   vtkRenderer* xren =
 219     ( x != NULL )? x->GetRenderers( )->GetFirstRenderer( ): NULL;
 220   vtkRenderer* yren =
 221     ( y != NULL )? y->GetRenderers( )->GetFirstRenderer( ): NULL;
 222   vtkRenderer* zren =
 223     ( z != NULL )? z->GetRenderers( )->GetFirstRenderer( ): NULL;
 224   vtkRenderer* wren =
 225     ( w != NULL )? w->GetRenderers( )->GetFirstRenderer( ): NULL;
 226
 227   if( xren != NULL )
 228   {
 229     xren->AddActor( this->Slices[ 0 ][ 1 ]->GetPlaneActor( ) );
 230     xren->AddActor( this->Slices[ 0 ][ 2 ]->GetPlaneActor( ) );
 231
 232   } // fi
 233   if( yren != NULL )
 234   {
 235     yren->AddActor( this->Slices[ 0 ][ 0 ]->GetPlaneActor( ) );
 236     yren->AddActor( this->Slices[ 0 ][ 2 ]->GetPlaneActor( ) );
 237
 238   } // fi
 239   if( zren != NULL )
 240   {
 241     zren->AddActor( this->Slices[ 0 ][ 0 ]->GetPlaneActor( ) );
 242     zren->AddActor( this->Slices[ 0 ][ 1 ]->GetPlaneActor( ) );
 243
 244   } // fi
 245   if( wren != NULL )
 246     wren->AddActor( this->ImageOutlineActor );
 247 }
 248
 249 // -------------------------------------------------------------------------
 250 void cpExtensions::Visualization::MPRActors::
 251 PopActorsFrom(
 252   vtkRenderWindow* x, vtkRenderWindow* y, vtkRenderWindow* z,
 253   vtkRenderWindow* w
 254   )
 255 {
 256   this->Slices[ 0 ][ 0 ]->PopActorsFrom( x );
 257   this->Slices[ 0 ][ 1 ]->PopActorsFrom( y );
 258   this->Slices[ 0 ][ 2 ]->PopActorsFrom( z );
 259   this->Slices[ 1 ][ 0 ]->PopActorsFrom( w );
 260   this->Slices[ 1 ][ 1 ]->PopActorsFrom( w );
 261   this->Slices[ 1 ][ 2 ]->PopActorsFrom( w );
 262
 263   vtkRenderer* wren =
 264     ( w != NULL )? w->GetRenderers( )->GetFirstRenderer( ): NULL;
 265   if( wren != NULL )
 266     wren->RemoveActor( this->ImageOutlineActor );
 267
 268   this->Slices[ 0 ][ 0 ]->Clear( );
 269   this->Slices[ 0 ][ 1 ]->Clear( );
 270   this->Slices[ 0 ][ 2 ]->Clear( );
 271   this->Slices[ 1 ][ 0 ]->Clear( );
 272   this->Slices[ 1 ][ 1 ]->Clear( );
 273   this->Slices[ 1 ][ 2 ]->Clear( );
 274 }
 275
 276 // -------------------------------------------------------------------------
 277 void cpExtensions::Visualization::MPRActors::
 278 SetLookupTableAsColor( unsigned int i, double r, double g, double b )
 279 {
 280   this->Slices[ 0 ][ 0 ]->SetLookupTableAsColor( 0, r, g, b );
 281   this->Slices[ 0 ][ 1 ]->Modified( );
 282   this->Slices[ 0 ][ 2 ]->Modified( );
 283   this->Slices[ 1 ][ 0 ]->Modified( );
 284   this->Slices[ 1 ][ 1 ]->Modified( );
 285   this->Slices[ 1 ][ 2 ]->Modified( );
 286   this->Modified( );
 287 }
 288
 289 // -------------------------------------------------------------------------
 290 double cpExtensions::Visualization::MPRActors::
 291 GetWindow( )
 292 {
 293   return( this->Slices[ 0 ][ 0 ]->GetWindow( ) );
 294 }
 295
 296 // -------------------------------------------------------------------------
 297 double cpExtensions::Visualization::MPRActors::
 298 GetLevel( )
 299 {
 300   return( this->Slices[ 0 ][ 0 ]->GetLevel( ) );
 301 }
 302
 303 // -------------------------------------------------------------------------
 304 int cpExtensions::Visualization::MPRActors::
 305 GetSliceNumberMinValue( const int& axis ) const
 306 {
 307   return( this->Slices[ 0 ][ axis ]->GetSliceNumberMinValue( ) );
 308 }
 309
 310 // -------------------------------------------------------------------------
 311 int cpExtensions::Visualization::MPRActors::
 312 GetSliceNumberMaxValue( const int& axis ) const
 313 {
 314   return( this->Slices[ 0 ][ axis ]->GetSliceNumberMaxValue( ) );
 315 }
 316
 317 // -------------------------------------------------------------------------
 318 int cpExtensions::Visualization::MPRActors::
 319 GetSlice( const int& axis ) const
 320 {
 321   return( this->Slices[ 0 ][ axis ]->GetSliceNumber( ) );
 322 }
 323
 324 // -------------------------------------------------------------------------
 325 void cpExtensions::Visualization::MPRActors::
 326 SetSlice( const int& axis, const int& slice )
 327 {
 328   vtkImageData* image = this->Slices[ 0 ][ 0 ]->GetInputImage( 0 );
 329   if( image == NULL )
 330     return;
 331
 332   // Get image data extent
 333   int ext[ 6 ];
 334   image->GetExtent( ext );
 335
 336   // Check if the slice is valid
 337   int real = slice;
 338   if( slice < ext[ axis << 1 ] )
 339     real = ext[ axis << 1 ];
 340   if( ext[ ( axis << 1 ) + 1 ] < slice )
 341     real = ext[ ( axis << 1 ) + 1 ];
 342
 343   // Change slice
 344   this->Slices[ 0 ][ axis ]->SetSliceNumber( real );
 345   this->Slices[ 1 ][ axis ]->SetSliceNumber( real );
 346   this->Modified( );
 347 }
 348
 349 // -------------------------------------------------------------------------
 350 void cpExtensions::Visualization::MPRActors::
 351 SetSlice( const int& axis, const double& slice )
 352 {
 353   vtkImageData* image = this->Slices[ 0 ][ 0 ]->GetInputImage( 0 );
 354   if( image == NULL )
 355     return;
 356
 357   double x[ 3 ] = { double( 0 ) };
 358   double pcoords[ 3 ];
 359   int ijk[ 3 ];
 360
 361   x[ axis ] = slice;
 362   image->ComputeStructuredCoordinates( x, ijk, pcoords );
 363   this->SetSlice( axis, ijk[ axis ] );
 364 }
 365
 366 // -------------------------------------------------------------------------
 367 void cpExtensions::Visualization::MPRActors::
 368 ResetSlices( )
 369 {
 370   for( unsigned int i = 0; i < 2; ++i )
 371     for( unsigned int j = 0; j < 3; ++j )
 372       this->Slices[ i ][ j ]->SetSliceNumber(
 373         this->Slices[ i ][ j ]->GetSliceNumberMaxValue( )
 374         );
 375 }
 376
 377 // -------------------------------------------------------------------------
 378 cpExtensions::Visualization::MPRActors::
 379 MPRActors( )
 380   : Superclass( )
 381 {
 382   this->ImageOutlineActor = vtkSmartPointer< vtkActor >::New( );
 383   for( unsigned int i = 0; i < 2; ++i )
 384     for( unsigned int j = 0; j < 3; ++j )
 385       this->Slices[ i ][ j ] = vtkSmartPointer< ImageSliceActors >::New( );
 386
 387   this->Slices[ 0 ][ 0 ]->AssociateSlice( this->Slices[ 1 ][ 0 ] );
 388   this->Slices[ 0 ][ 1 ]->AssociateSlice( this->Slices[ 1 ][ 1 ] );
 389   this->Slices[ 0 ][ 2 ]->AssociateSlice( this->Slices[ 1 ][ 2 ] );
 390
 391   for( unsigned int i = 0; i < 3; ++ i )
 392   {
 393     this->Slices[ 0 ][ i ]->AddRenderCommand(
 394       Self::_RenderCommand, this
 395       );
 396     this->Slices[ 0 ][ i ]->AddWindowLevelCommand(
 397       Self::_WindowLevelCommand, this
 398       );
 399     // this->Slices[ 0 ][ i ]->AddSlicesCommand( Self::_SlicesCommand, this );
 400
 401   } // rof
 402 }
 403
 404 // -------------------------------------------------------------------------
 405 cpExtensions::Visualization::MPRActors::
 406 ~MPRActors( )
 407 {
 408 }
 409
 410 // -------------------------------------------------------------------------
 411 void cpExtensions::Visualization::MPRActors::
 412 _SlicesCommand( double* pos, int axis, void* data )
 413 {
 414   /*
 415     MPRActors* actors = reinterpret_cast< MPRActors* >( data );
 416     if( actors == NULL )
 417     return;
 418     for( unsigned int j = 0; j < 3; ++j )
 419     if( actors->Slices[ 0 ][ j ]->GetAxis( ) != axis )
 420     actors->Slices[ 0 ][ j ]->SetSlice( pos );
 421     actors->Modified( );
 422   */
 423 }
 424
 425 // -------------------------------------------------------------------------
 426 void cpExtensions::Visualization::MPRActors::
 427 _WindowLevelCommand( double window, double level, void* data )
 428 {
 429   MPRActors* actors = reinterpret_cast< MPRActors* >( data );
 430   if( actors == NULL )
 431     return;
 432   actors->Slices[ 0 ][ 0 ]->Render( );
 433   actors->Slices[ 0 ][ 1 ]->Render( );
 434   actors->Slices[ 0 ][ 2 ]->Render( );
 435   actors->Slices[ 1 ][ 0 ]->Render( );
 436   actors->Slices[ 1 ][ 1 ]->Render( );
 437   actors->Slices[ 1 ][ 2 ]->Render( );
 438 }
 439
 440 // -------------------------------------------------------------------------
 441 void cpExtensions::Visualization::MPRActors::
 442 _RenderCommand( void* data )
 443 {
 444   MPRActors* actors = reinterpret_cast< MPRActors* >( data );
 445   if( actors == NULL )
 446     return;
 447   actors->Slices[ 0 ][ 0 ]->Render( );
 448   actors->Slices[ 0 ][ 1 ]->Render( );
 449   actors->Slices[ 0 ][ 2 ]->Render( );
 450   actors->Slices[ 1 ][ 0 ]->Render( );
 451   actors->Slices[ 1 ][ 1 ]->Render( );
 452   actors->Slices[ 1 ][ 2 ]->Render( );
 453 }
 454
 455 // eof - $RCSfile$