plugins/Widgets/SeedWidget.cxx

   1 #include <plugins/Widgets/SeedWidget.h>
   2
   3 #include <cpPlugins/DataObjects/Image.h>
   4 #include <cpPlugins/DataObjects/Mesh.h>
   5
   6 #include <cpExtensions/Interaction/SeedWidget.h>
   7 #include <cpExtensions/Interaction/ImageSlicePointPlacer.h>
   8 #include <cpExtensions/Visualization/WindowLevelImageActor.h>
   9
  10 #include <vtkHandleWidget.h>
  11 #include <vtkImageData.h>
  12 #include <vtkPlane.h>
  13 #include <vtkPointHandleRepresentation3D.h>
  14 #include <vtkProperty.h>
  15 #include <vtkRendererCollection.h>
  16 #include <vtkRenderWindow.h>
  17 #include <vtkSeedRepresentation.h>
  18
  19 // -------------------------------------------------------------------------
  20 void cpPluginsWidgets::SeedWidget::
  21 Clear( )
  22 {
  23 }
  24
  25 // -------------------------------------------------------------------------
  26 void cpPluginsWidgets::SeedWidget::
  27 SetEnabled( bool v )
  28 {
  29   for( auto w = this->m_Widgets.begin( ); w != this->m_Widgets.end( ); ++w )
  30   {
  31     if( v ) w->second->RestartInteraction( );
  32     else    w->second->CompleteInteraction( );
  33     w->second->SetEnabled( v );
  34     w->second->Render( );
  35
  36   } // rof
  37 }
  38
  39 // -------------------------------------------------------------------------
  40 cpPluginsWidgets::SeedWidget::
  41 SeedWidget( )
  42   : Superclass( )
  43 {
  44   typedef cpPlugins::BaseObjects::DataObject _TData;
  45   typedef cpPlugins::DataObjects::Mesh       _TMesh;
  46
  47   // Create ports
  48   this->_ConfigureInput< _TData >( "Input", false, false );
  49   this->_ConfigureOutput< _TMesh >( "Output" );
  50
  51   // Create output data
  52   auto out = this->_CreateVTK< vtkPolyData >( );
  53   out->SetPoints( vtkSmartPointer< vtkPoints >::New( ) );
  54   out->SetVerts( vtkSmartPointer< vtkCellArray >::New( ) );
  55   out->SetLines( vtkSmartPointer< vtkCellArray >::New( ) );
  56   out->SetPolys( vtkSmartPointer< vtkCellArray >::New( ) );
  57   out->SetStrips( vtkSmartPointer< vtkCellArray >::New( ) );
  58   this->GetOutput( "Output" )->SetVTK( out );
  59 }
  60
  61 // -------------------------------------------------------------------------
  62 cpPluginsWidgets::SeedWidget::
  63 ~SeedWidget( )
  64 {
  65   /* TODO
  66      for( auto w = this->m_Widgets.begin( ); w != this->m_Widgets.end( ); ++w )
  67      w->second->EnabledOff( );
  68      this->m_Widgets.clear( );
  69   */
  70 }
  71
  72 // -------------------------------------------------------------------------
  73 void cpPluginsWidgets::SeedWidget::
  74 _GenerateData( )
  75 {
  76   auto image = this->GetInputData< vtkImageData >( "Input" );
  77   if( image != NULL )
  78     this->_GD0_Image( image );
  79   else
  80     this->_Error( "Invalid input image." );
  81 }
  82
  83 // -------------------------------------------------------------------------
  84 void cpPluginsWidgets::SeedWidget::
  85 _GD0_Image( vtkImageData* image )
  86 {
  87   auto seeds = this->GetOutputData< vtkPolyData >( "Output" );
  88   if( this->m_Widgets.size( ) == 0 )
  89   {
  90     this->m_Command = vtkSmartPointer< TCallback >::New( );
  91     this->m_Command->SetSeeds( seeds );
  92     for(
  93       auto inIt = this->m_Interactors.begin( );
  94       inIt != this->m_Interactors.end( );
  95       ++inIt
  96       )
  97     {
  98       auto rends = ( *inIt )->GetRenderWindow( )->GetRenderers( );
  99       std::set< TImageActor* > all_props;
 100       if( rends != NULL )
 101       {
 102         rends->InitTraversal( );
 103         while( vtkRenderer* ren = rends->GetNextItem( ) )
 104         {
 105           auto props = ren->GetViewProps( );
 106           props->InitTraversal( );
 107           while( vtkProp* p = props->GetNextProp( ) )
 108           {
 109             auto image_actor = dynamic_cast< TImageActor* >( p );
 110             if( image_actor != NULL )
 111             {
 112               if( image_actor->GetImage( ) == image )
 113                 all_props.insert( image_actor );
 114
 115             } // fi
 116
 117           } // elihw
 118
 119         } // elihw
 120
 121       } // fi
 122       if( all_props.size( ) == 1 )
 123       {
 124         if( this->m_Widgets.find( *inIt ) == this->m_Widgets.end( ) )
 125         {
 126           auto act = *( all_props.begin( ) );
 127           auto pla = vtkSmartPointer< TPlacer >::New( );
 128           auto hnd = vtkSmartPointer< THandleRep >::New( );
 129           auto rep = vtkSmartPointer< TSeedRep >::New( );
 130           auto wdg = vtkSmartPointer< TWidget >::New( );
 131
 132           pla->SetImageSlice( act );
 133           hnd->GetProperty( )->SetColor( 1, 0, 0 );
 134           hnd->SetPointPlacer( pla );
 135           rep->SetHandleRepresentation( hnd );
 136           wdg->SetRepresentation( rep );
 137           wdg->SetInteractor( *inIt );
 138           act->AddObserver( vtkCommand::InteractionEvent, this->m_Command );
 139           wdg->AddObserver( vtkCommand::PlacePointEvent, this->m_Command );
 140           wdg->AddObserver( vtkCommand::CursorChangedEvent, this->m_Command );
 141           this->m_Widgets[ *inIt ] = wdg;
 142
 143         } // fi
 144
 145       } // fi
 146
 147     } // rof
 148
 149     // Associate input text
 150     std::string text = this->m_Parameters.GetString( "Text" );
 151     std::vector< std::string > tok1, tok2;
 152     cpExtensions::Tokenize( tok1, text, "#" );
 153     for( auto t1 = tok1.begin( ); t1 != tok1.end( ); ++t1 )
 154     {
 155       if( *t1 != "" )
 156       {
 157         cpExtensions::Tokenize( tok2, *t1, " " );
 158         double x[ 3 ];
 159         for( unsigned int d = 0; d < 3; ++d )
 160         {
 161           if( d < tok2.size( ) )
 162           {
 163             std::istringstream str( tok2[ d ] );
 164             str >> x[ d ];
 165           }
 166           else
 167             x[ d ] = double( 0 );
 168
 169         } // rof
 170
 171         seeds->GetPoints( )->InsertNextPoint( x );
 172         seeds->GetVerts( )->InsertNextCell( 1 );
 173         seeds->GetVerts( )->InsertCellPoint(
 174           seeds->GetPoints( )->GetNumberOfPoints( ) - 1
 175           );
 176         seeds->Modified( );
 177
 178       } // fi
 179
 180     } // rof
 181   }
 182   else
 183   {
 184     double x[ 3 ];
 185     std::stringstream text;
 186     for( long i = 0; i < seeds->GetNumberOfPoints( ); ++i )
 187     {
 188       seeds->GetPoint( i, x );
 189       text << x[ 0 ] << " " << x[ 1 ] << " " << x[ 2 ] << "#";
 190
 191     } // rof
 192     this->m_Parameters.SetString( "Text", text.str( ) );
 193
 194   } // fi
 195 }
 196
 197 // -------------------------------------------------------------------------
 198 cpPluginsWidgets::SeedWidget::TCallback*
 199 cpPluginsWidgets::SeedWidget::TCallback::
 200 New( )
 201 {
 202   return( new TCallback );
 203 }
 204
 205 // -------------------------------------------------------------------------
 206 void cpPluginsWidgets::SeedWidget::TCallback::
 207 Execute( vtkObject* caller, unsigned long id, void* data )
 208 {
 209   static const double EPS = 1e-5;
 210   auto src_act = dynamic_cast< TImageActor* >( caller );
 211   auto src_wdg = dynamic_cast< TWidget* >( caller );
 212
 213   if( id == vtkCommand::InteractionEvent && src_act != NULL )
 214   {
 215     TWidget* widget = NULL;
 216     TSeedRep* seed_rep = NULL;
 217
 218     auto w = this->m_Widgets.begin( );
 219     while( w != this->m_Widgets.end( ) && widget == NULL )
 220     {
 221       auto rep =
 222         dynamic_cast< vtkSeedRepresentation* >(
 223           ( *w )->GetRepresentation( )
 224           );
 225       if( rep != NULL )
 226       {
 227         auto hnd = rep->GetHandleRepresentation( );
 228         if( hnd != NULL )
 229         {
 230           auto pla =
 231             dynamic_cast< SeedWidget::TPlacer* >(
 232               hnd->GetPointPlacer( )
 233               );
 234           if( pla != NULL )
 235           {
 236             auto act = pla->GetImageSlice( );
 237             if( act == src_act )
 238             {
 239               widget = *w;
 240               seed_rep = rep;
 241
 242             } // fi
 243
 244           } // fi
 245
 246         } // fi
 247
 248       } // fi
 249       w++;
 250
 251     } // elihw
 252
 253     if( widget != NULL )
 254     {
 255       // Cut through given seeds
 256       auto plane = src_act->GetSlicePlane( );
 257       double x[ 3 ];
 258       std::vector< long > ids;
 259       for( long i = 0; i < this->m_Seeds->GetNumberOfPoints( ); ++i )
 260       {
 261         this->m_Seeds->GetPoint( i, x );
 262         double d = plane->DistanceToPlane( x );
 263         if( d <= EPS )
 264           ids.push_back( i );
 265
 266       } // rof
 267
 268       // Erase seeds
 269       widget->CompleteInteraction( );
 270       widget->EnabledOff( );
 271       int nSeeds = seed_rep->GetNumberOfSeeds ();
 272       for( int i = 0; i < nSeeds; ++i )
 273       {
 274         seed_rep->RemoveLastHandle( );
 275         widget->DeleteSeed( seed_rep->GetNumberOfSeeds( ) );
 276
 277       } // rof
 278
 279       if( ids.size( ) > 0 )
 280       {
 281         auto ren = widget->GetInteractor( )->FindPokedRenderer(
 282           widget->GetInteractor( )->GetEventPosition( )[ 0 ],
 283           widget->GetInteractor( )->GetEventPosition( )[ 1 ]
 284           );
 285
 286         // Add seeds
 287         seed_rep->BuildRepresentation( );
 288         for( auto id = ids.begin( ); id != ids.end( ); ++id )
 289         {
 290           this->m_Seeds->GetPoint( *id, x );
 291           widget->ComputeWorldToDisplay(
 292             ren, x[ 0 ], x[ 1 ], x[ 2 ], x
 293             );
 294           x[ 2 ] = double( 0 );
 295           int s = seed_rep->CreateHandle( x );
 296           vtkHandleWidget* curr_hnd = widget->CreateNewHandle( );
 297           seed_rep->SetSeedDisplayPosition( s, x );
 298           curr_hnd->SetEnabled( 1 );
 299
 300         } // rof
 301
 302       } // fi
 303
 304       // Reactivate widget
 305       widget->RestartInteraction( );
 306       widget->EnabledOn( );
 307       widget->Render( );
 308
 309     } // fi
 310   }
 311   else if( id == vtkCommand::PlacePointEvent && src_wdg != NULL )
 312   {
 313     this->m_Widgets.insert( src_wdg );
 314     auto rep =
 315       dynamic_cast< vtkSeedRepresentation* >(
 316         src_wdg->GetRepresentation( )
 317         );
 318     if( rep != NULL )
 319     {
 320       unsigned long nSeeds = rep->GetNumberOfSeeds( );
 321       if( nSeeds > 0 )
 322       {
 323         double pos[ 3 ];
 324         rep->GetSeedWorldPosition( nSeeds - 1, pos );
 325
 326         this->m_Seeds->GetPoints( )->InsertNextPoint( pos );
 327         this->m_Seeds->GetVerts( )->InsertNextCell( 1 );
 328         this->m_Seeds->GetVerts( )->InsertCellPoint(
 329           this->m_Seeds->GetPoints( )->GetNumberOfPoints( ) - 1
 330           );
 331         this->m_Seeds->Modified( );
 332
 333       } // fi
 334
 335     } // fi
 336
 337   } // fi
 338 }
 339
 340 // -------------------------------------------------------------------------
 341 void cpPluginsWidgets::SeedWidget::TCallback::
 342 SetSeeds( vtkPolyData* seeds )
 343 {
 344   this->m_Seeds = seeds;
 345 }
 346
 347 // -------------------------------------------------------------------------
 348 cpPluginsWidgets::SeedWidget::TCallback::
 349 TCallback( )
 350   : vtkCommand( ),
 351     m_Seeds( NULL )
 352 {
 353 }
 354
 355 // -------------------------------------------------------------------------
 356 cpPluginsWidgets::SeedWidget::TCallback::
 357 ~TCallback( )
 358 {
 359 }
 360
 361 // eof - $RCSfile$