plugins/Widgets/SeedWidget.cxx

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