plugins/Widgets/SeedWidget.cxx

   1 #include <plugins/Widgets/SeedWidget.h>
   2 #include <cpPlugins/DataObjects/Image.h>
   3 #include <cpPlugins/DataObjects/Mesh.h>
   4
   5 #include <vtkCommand.h>
   6 #include <vtkImageActor.h>
   7 #include <vtkImageData.h>
   8 #include <vtkProperty.h>
   9 #include <vtkRenderer.h>
  10 #include <vtkRenderWindowInteractor.h>
  11 #include <cpExtensions/QT/SimpleMPRWidget.h>
  12
  13 // -------------------------------------------------------------------------
  14 // This callback is responsible for changing update time
  15 namespace cpPluginsWidgets
  16 {
  17   /**
  18    */
  19   class SeedWidgetCallback
  20     : public vtkCommand
  21   {
  22   public:
  23     static SeedWidgetCallback* New( )
  24       { return( new SeedWidgetCallback ); }
  25     virtual void Execute( vtkObject* caller, unsigned long id, void* data )
  26       {
  27         typedef cpPluginsWidgets::SeedWidget::TImageActor _TImageActor;
  28         auto actor = dynamic_cast< _TImageActor* >( caller );
  29
  30         if( id == vtkCommand::InteractionEvent && actor != NULL )
  31         {
  32           int slice = actor->GetSliceNumber( );
  33           if( this->Data->ActualWidgetId != slice )
  34           {
  35             this->Data->Widgets[ this->Data->ActualWidgetId ]->EnabledOff( );
  36             this->Data->Widgets[ slice ]->EnabledOn( );
  37             this->Data->ActualWidgetId = slice;
  38
  39           } // fi
  40         }
  41         else if(
  42           (
  43             id == vtkCommand::CursorChangedEvent ||
  44             id == vtkCommand::PlacePointEvent
  45             ) &&
  46           this->Widget != NULL
  47           )
  48            this->Widget->Modified( );
  49       }
  50
  51   protected:
  52     SeedWidgetCallback( )
  53       : vtkCommand( ),
  54         Widget( NULL ),
  55         Data( NULL )
  56       { }
  57     virtual ~SeedWidgetCallback( ) { }
  58
  59   public:
  60     SeedWidget* Widget;
  61     SeedWidget::TWidgetData* Data;
  62   };
  63
  64 } // ecapseman
  65
  66 // -------------------------------------------------------------------------
  67 cpPluginsWidgets::SeedWidget::
  68 SeedWidget( )
  69   : Superclass( ),
  70     m_Configured( false )
  71 {
  72   typedef cpPlugins::BaseObjects::DataObject _TData;
  73   typedef cpPlugins::DataObjects::Mesh _TMesh;
  74
  75   this->_ConfigureInput< _TData >( "Input", false, false );
  76   this->_ConfigureOutput< _TMesh >( "Output" );
  77 }
  78
  79 // -------------------------------------------------------------------------
  80 cpPluginsWidgets::SeedWidget::
  81 ~SeedWidget( )
  82 {
  83   for( auto w = this->m_Widgets.begin( ); w != this->m_Widgets.end( ); ++w )
  84     delete *w;
  85   this->m_Widgets.clear( );
  86 }
  87
  88 // -------------------------------------------------------------------------
  89 void cpPluginsWidgets::SeedWidget::
  90 _GenerateData( )
  91 {
  92   static vtkPolyData* prev_pdata = NULL;
  93   auto pdata = this->_CreateVTK< vtkPolyData >( );
  94   if( prev_pdata != pdata )
  95   {
  96     pdata->SetPoints( vtkPoints::New( ) );
  97     pdata->SetVerts( vtkCellArray::New( ) );
  98     pdata->SetLines( vtkCellArray::New( ) );
  99     pdata->SetPolys( vtkCellArray::New( ) );
 100     pdata->SetStrips( vtkCellArray::New( ) );
 101     prev_pdata = pdata;
 102
 103   } // fi
 104   auto points = pdata->GetPoints( );
 105   auto verts = pdata->GetVerts( );
 106
 107   if( this->m_Configured )
 108   {
 109     if( points->GetNumberOfPoints( ) == 0 )
 110     {
 111       std::stringstream text;
 112       bool start = true;
 113       for( auto w = this->m_Widgets.begin( ); w != this->m_Widgets.end( ); ++w )
 114       {
 115         for( auto r = ( *w )->Widgets.begin( ); r != ( *w )->Widgets.end( ); ++r )
 116         {
 117           auto rep =
 118             dynamic_cast< vtkSeedRepresentation* >(
 119               ( *r )->GetRepresentation( )
 120               );
 121           if( rep != NULL )
 122           {
 123             double pos[ 3 ];
 124             for( unsigned int i = 0; i < rep->GetNumberOfSeeds( ); ++i )
 125             {
 126               rep->GetSeedWorldPosition( i, pos );
 127               if( !start )
 128                 text << "#";
 129               start = false;
 130               text << pos[ 0 ] << " " << pos[ 1 ] << " " << pos[ 2 ];
 131               points->InsertNextPoint( pos );
 132
 133             } // rof
 134
 135           } // rof
 136           ( *r )->EnabledOff( );
 137
 138         } // rof
 139
 140       } // rof
 141       this->m_Parameters.SetString( "Text", text.str( ) );
 142
 143     } // fi
 144   }
 145   else
 146   {
 147     auto init_seeds = this->m_Parameters.GetString( "Text" );
 148     std::vector< std::string > tokens;
 149     cpExtensions::Tokenize( tokens, init_seeds, "#" );
 150     for( auto tIt = tokens.begin( ); tIt != tokens.end( ); ++tIt )
 151     {
 152       std::vector< std::string > coords;
 153       cpExtensions::Tokenize( coords, *tIt, " \t" );
 154       int dim = ( coords.size( ) < 3 )? coords.size( ): 3;
 155       double pos[ 3 ];
 156       for( unsigned int d = 0; d < 3; ++d )
 157       {
 158         pos[ d ] = double( 0 );
 159         if( d < dim )
 160         {
 161           std::istringstream value( coords[ d ] );
 162           value >> pos[ d ];
 163
 164         } // fi
 165
 166       } // rof
 167       verts->InsertNextCell( 1 );
 168       verts->InsertCellPoint( points->GetNumberOfPoints( ) );
 169       points->InsertNextPoint( pos );
 170
 171     } // rof
 172     this->_Configure( );
 173     this->Modified( );
 174     this->m_Configured = true;
 175
 176   } // fi
 177   this->GetOutput( "Output" )->SetVTK( pdata );
 178 }
 179
 180 // -------------------------------------------------------------------------
 181 void cpPluginsWidgets::SeedWidget::
 182 _Configure( )
 183 {
 184   typedef cpPlugins::DataObjects::Image _TImage;
 185
 186   auto image = this->GetInput< _TImage >( "Input" );
 187   if( image != NULL )
 188   {
 189     // Update actors
 190     auto vtk_image = image->GetVTK< vtkImageData >( );
 191     auto iIt = this->m_Interactors.begin( );
 192     for( ; iIt != this->m_Interactors.end( ); ++iIt )
 193     {
 194       auto ren = ( *iIt )->GetInteractorStyle( )->GetCurrentRenderer( );
 195       if( ren != NULL )
 196       {
 197         auto props = ren->GetViewProps( );
 198         if( props != NULL )
 199         {
 200           props->InitTraversal( );
 201           while( vtkProp* prop = props->GetNextProp( ) )
 202           {
 203             auto actor = dynamic_cast< TImageActor* >( prop );
 204             if( actor != NULL )
 205               if( actor->GetImage( ) == vtk_image )
 206                 this->m_Props[ actor ] = *iIt;
 207
 208           } // elihw
 209
 210         } // fi
 211
 212       } // fi
 213
 214     } // rof
 215
 216     // Process image
 217     if( this->m_Props.size( ) > 0 )
 218     {
 219       cpPlugins_Demangle_ImageVisualDims( image->GetITK( ), _GD0_Image );
 220       else this->_Error( "Invalid input image." );
 221     }
 222     else
 223       this->_Error( "Could not create a valid widget: no actors." );
 224   }
 225   else
 226     this->_Error( "Could not create a valid widget: no input." );
 227 }
 228
 229 // -------------------------------------------------------------------------
 230 template< class _TImage >
 231 void cpPluginsWidgets::SeedWidget::
 232 _GD0_Image( _TImage* image )
 233 {
 234   for( auto p = this->m_Props.begin( ); p != this->m_Props.end( ); ++p )
 235   {
 236     TWidgetData* d =
 237       new TWidgetData(
 238         this, dynamic_cast< TImageActor* >( p->first ), p->second
 239         );
 240     this->m_Widgets.push_back( d );
 241
 242   } // rof
 243 }
 244
 245 // -------------------------------------------------------------------------
 246 cpPluginsWidgets::SeedWidget::TWidgetData::
 247 TWidgetData(
 248   SeedWidget* seedWidget,
 249   TImageActor* actor,
 250   vtkRenderWindowInteractor* iren
 251   )
 252 {
 253   auto cb = vtkSmartPointer< SeedWidgetCallback >::New( );
 254   cb->Widget = seedWidget;
 255   cb->Data = this;
 256   this->Command = cb;
 257   actor->AddObserver( vtkCommand::InteractionEvent, cb );
 258
 259   auto image = actor->GetImage( );
 260   int ori = actor->GetOrientation( );
 261   int ext[ 6 ];
 262   image->GetExtent( ext );
 263   for( int i = ext[ ori << 1 ]; i <= ext[ ( ori << 1 ) + 1 ]; ++i )
 264   {
 265     auto placer = vtkSmartPointer< _TPlacer >::New( );
 266     auto handle = vtkSmartPointer< vtkPointHandleRepresentation3D >::New( );
 267     auto rep = vtkSmartPointer< vtkSeedRepresentation >::New( );
 268     auto wdg = vtkSmartPointer< _TWidget >::New( );
 269
 270     placer->SetImageSlice( actor );
 271     handle->GetProperty( )->SetColor( 1, 0, 0 );
 272     handle->SetPointPlacer( placer );
 273     rep->SetHandleRepresentation( handle );
 274     wdg->SetRepresentation( rep );
 275     wdg->SetInteractor( iren );
 276     wdg->AddObserver( vtkCommand::PlacePointEvent, cb );
 277     wdg->AddObserver( vtkCommand::CursorChangedEvent, cb );
 278     wdg->EnabledOff( );
 279
 280     this->Widgets.push_back( wdg );
 281     this->Placers.push_back( placer );
 282     this->Handles.push_back( handle );
 283     this->Representations.push_back( rep );
 284
 285   } // rof
 286
 287   this->ActualWidgetId = actor->GetSliceNumber( );
 288   this->Widgets[ this->ActualWidgetId ]->EnabledOn( );
 289 }
 290
 291 // -------------------------------------------------------------------------
 292 cpPluginsWidgets::SeedWidget::TWidgetData::
 293 ~TWidgetData( )
 294 {
 295 }
 296
 297 // eof - $RCSfile$