]> Creatis software - cpPlugins.git/blobdiff - plugins/Widgets/SplineWidget.cxx
...
[cpPlugins.git] / plugins / Widgets / SplineWidget.cxx
index c6e711633a3052766412438ce45a96a2b0c57cb0..256ae8f334248638aa3f51275af0ee2ee1e716ab 100644 (file)
@@ -1,10 +1,48 @@
-#include <plugins/Widgets/SplineWidget.h>
+#include <Widgets/SplineWidget.h>
 #include <cpPlugins/DataObjects/Image.h>
 #include <cpPlugins/DataObjects/Mesh.h>
 #include <cpExtensions/Visualization/WindowLevelImageActor.h>
 
+#include <itkVector.h>
+
+#include <vtkImageData.h>
+#include <vtkPolyData.h>
 #include <vtkRenderer.h>
 #include <vtkSplineWidget.h>
+#include <vtkParametricSpline.h>
+
+// -------------------------------------------------------------------------
+void cpPluginsWidgets::SplineWidget::
+Clear( )
+{
+}
+
+// -------------------------------------------------------------------------
+void cpPluginsWidgets::SplineWidget::
+SetEnabled( bool v )
+{
+  auto wdg = this->GetVTK< vtkSplineWidget >( );
+  if( wdg != NULL )
+  {
+    wdg->SetEnabled( v );
+    wdg->GetInteractor( )->Render( );
+
+  } // fi
+}
+
+// -------------------------------------------------------------------------
+bool cpPluginsWidgets::SplineWidget::
+GetEnabled( ) const
+{
+  auto wdg = this->GetVTK< const vtkSplineWidget >( );
+  if( wdg != NULL )
+  {
+    vtkSplineWidget* w = const_cast< vtkSplineWidget* >( wdg );
+    return( w->GetEnabled( ) != 0 );
+  }
+  else
+    return( false );
+}
 
 // -------------------------------------------------------------------------
 cpPluginsWidgets::SplineWidget::
@@ -24,6 +62,8 @@ SplineWidget( )
   this->m_Contour->SetPolys( vtkSmartPointer< vtkCellArray >::New( ) );
   this->m_Contour->SetStrips( vtkSmartPointer< vtkCellArray >::New( ) );
   this->GetOutput( "Output" )->SetVTK( this->m_Contour );
+
+  this->m_Parameters.ConfigureAsBool( "Extend", false );
 }
 
 // -------------------------------------------------------------------------
@@ -47,8 +87,103 @@ _GenerateData( )
   auto wdg = this->_CreateVTK< vtkSplineWidget >( );
   if( this->m_Configured )
   {
-    wdg->GetPolyData( this->m_Contour.GetPointer( ) );
-    wdg->Off( );
+    if( this->m_Parameters.GetBool( "Extend" ) )
+    {
+      wdg->GetPolyData( this->m_Contour.GetPointer( ) );
+
+      typedef itk::Vector< double, 3 > _TVector;
+      typedef std::vector< _TVector > _TVectors;
+
+      unsigned long nPoints = this->m_Contour->GetNumberOfPoints( );
+      unsigned long support = nPoints / 10;
+      _TVectors pp, qp;
+      for( unsigned long i = 0; i < support; ++i )
+      {
+        double p[ 3 ], q[ 3 ];
+        this->m_Contour->GetPoint( i, p );
+        this->m_Contour->GetPoint( nPoints - 1 - i, q );
+
+        _TVector pv, qv;
+        pv[ 0 ] = p[ 0 ];
+        pv[ 1 ] = p[ 1 ];
+        pv[ 2 ] = p[ 2 ];
+        qv[ 0 ] = q[ 0 ];
+        qv[ 1 ] = q[ 1 ];
+        qv[ 2 ] = q[ 2 ];
+        pp.push_back( pv );
+        qp.push_back( qv );
+
+      } // rof
+
+      _TVectors pt, qt;
+      pt.push_back( pp[ 1 ] - pp[ 0 ] );
+      qt.push_back( qp[ 1 ] - qp[ 0 ] );
+      for( unsigned long i = 1; i < support - 1; ++i )
+      {
+        pt.push_back( pp[ i + 1 ] - pp[ i - 1 ] );
+        qt.push_back( qp[ i + 1 ] - qp[ i - 1 ] );
+
+      } // rof
+      pt.push_back( pp[ support - 1 ] - pp[ support - 2 ] );
+      qt.push_back( qp[ support - 1 ] - qp[ support - 2 ] );
+
+      _TVector t0( double( 0 ) ), t1( double( 0 ) );
+      long real0 = 0, real1 = 0;
+      for( unsigned long i = 0; i < support; ++i )
+      {
+        double n0 = pt[ i ].GetNorm( );
+        if( n0 > double( 0 ) )
+        {
+          t0 += pt[ i ] / n0;
+          real0++;
+
+        } // fi
+
+        double n1 = qt[ i ].GetNorm( );
+        if( n1 > double( 0 ) )
+        {
+          t1 += qt[ i ] / n1;
+          real1++;
+
+        } // fi
+
+      } // rof
+
+      if( real0 > 0 )
+        t0 /= double( -real0 );
+      if( real1 > 0 )
+        t1 /= double( -real1 );
+
+      t0 *= ( pp[ 0 ] - qp[ 0 ] ).GetNorm( );
+      t1 *= ( pp[ 0 ] - qp[ 0 ] ).GetNorm( );
+
+      _TVector p0 = pp[ 0 ] + t0;
+      _TVector p1 = qp[ 0 ] + t1;
+
+      vtkSmartPointer< vtkPolyData > cnt = this->m_Contour;
+      this->m_Contour = vtkSmartPointer< vtkPolyData >::New( );
+      this->m_Contour->SetPoints( vtkSmartPointer< vtkPoints >::New( ) );
+      this->m_Contour->SetVerts( vtkSmartPointer< vtkCellArray >::New( ) );
+      this->m_Contour->SetLines( vtkSmartPointer< vtkCellArray >::New( ) );
+      this->m_Contour->SetPolys( vtkSmartPointer< vtkCellArray >::New( ) );
+      this->m_Contour->SetStrips( vtkSmartPointer< vtkCellArray >::New( ) );
+      this->GetOutput( "Output" )->SetVTK( this->m_Contour );
+
+      this->m_Contour->GetPoints( )->
+        InsertNextPoint( p0[ 0 ], p0[ 1 ], p0[ 2 ] );
+      for( unsigned long i = 0; i < nPoints; ++i )
+      {
+        double p[ 3 ];
+        cnt->GetPoints( )->GetPoint( i, p );
+        this->m_Contour->GetPoints( )->InsertNextPoint( p );
+
+      } // rof
+      this->m_Contour->GetPoints( )->
+        InsertNextPoint( p1[ 0 ], p1[ 1 ], p1[ 2 ] );
+      this->m_Contour->Modified( );
+    }
+    else
+      wdg->GetPolyData( this->m_Contour.GetPointer( ) );
   }
   else
   {
@@ -110,204 +245,9 @@ _GenerateData( )
       );
     wdg->SetHandleSize( 0.005 );
     wdg->SetNumberOfHandles( 3 );
-    wdg->On( );
     this->m_Configured = true;
 
   } // fi
-
-  /* TODO
-
-     if( this->m_Configured )
-     {
-     if( points->GetNumberOfPoints( ) == 0 )
-     {
-     std::stringstream text;
-     bool start = true;
-     for( auto w = this->m_Widgets.begin( ); w != this->m_Widgets.end( ); ++w )
-     {
-     for( auto r = ( *w )->Widgets.begin( ); r != ( *w )->Widgets.end( ); ++r )
-     {
-     auto rep =
-     dynamic_cast< vtkSeedRepresentation* >(
-     ( *r )->GetRepresentation( )
-     );
-     if( rep != NULL )
-     {
-     double pos[ 3 ];
-     for( unsigned int i = 0; i < rep->GetNumberOfSeeds( ); ++i )
-     {
-     rep->GetSeedWorldPosition( i, pos );
-     if( !start )
-     text << "#";
-     start = false;
-     text << pos[ 0 ] << " " << pos[ 1 ] << " " << pos[ 2 ];
-     points->InsertNextPoint( pos );
-
-     } // rof
-
-     } // rof
-     ( *r )->EnabledOff( );
-
-     } // rof
-
-     } // rof
-     this->m_Parameters.SetString( "Text", text.str( ) );
-
-     } // fi
-     }
-     else
-     {
-     auto init_seeds = this->m_Parameters.GetString( "Text" );
-     std::vector< std::string > tokens;
-     cpExtensions::Tokenize( tokens, init_seeds, "#" );
-     for( auto tIt = tokens.begin( ); tIt != tokens.end( ); ++tIt )
-     {
-     std::vector< std::string > coords;
-     cpExtensions::Tokenize( coords, *tIt, " \t" );
-     int dim = ( coords.size( ) < 3 )? coords.size( ): 3;
-     double pos[ 3 ];
-     for( unsigned int d = 0; d < 3; ++d )
-     {
-     pos[ d ] = double( 0 );
-     if( d < dim )
-     {
-     std::istringstream value( coords[ d ] );
-     value >> pos[ d ];
-
-     } // fi
-
-     } // rof
-     verts->InsertNextCell( 1 );
-     verts->InsertCellPoint( points->GetNumberOfPoints( ) );
-     points->InsertNextPoint( pos );
-
-     } // rof
-     this->_Configure( );
-     this->Modified( );
-     this->m_Configured = true;
-
-     } // fi
-     this->GetOutput( "Output" )->SetVTK( pdata );
-  */
-}
-
-// -------------------------------------------------------------------------
-/* TODO
-void cpPluginsWidgets::SplineWidget::
-_Configure( )
-{
-  typedef cpPlugins::DataObjects::Image _TImage;
-
-  auto image = this->GetInput< _TImage >( "Input" );
-  if( image != NULL )
-  {
-    // Update actors
-    auto vtk_image = image->GetVTK< vtkImageData >( );
-    auto iIt = this->m_Interactors.begin( );
-    for( ; iIt != this->m_Interactors.end( ); ++iIt )
-    {
-      auto ren = ( *iIt )->GetInteractorStyle( )->GetCurrentRenderer( );
-      if( ren != NULL )
-      {
-        auto props = ren->GetViewProps( );
-        if( props != NULL )
-        {
-          props->InitTraversal( );
-          while( vtkProp* prop = props->GetNextProp( ) )
-          {
-            auto actor = dynamic_cast< TImageActor* >( prop );
-            if( actor != NULL )
-              if( actor->GetImage( ) == vtk_image )
-                this->m_Props[ actor ] = *iIt;
-
-          } // elihw
-
-        } // fi
-
-      } // fi
-
-    } // rof
-
-    // Process image
-    if( this->m_Props.size( ) > 0 )
-    {
-      cpPlugins_Demangle_ImageVisualDims( image->GetITK( ), _GD0_Image );
-      else this->_Error( "Invalid input image." );
-    }
-    else
-      this->_Error( "Could not create a valid widget: no actors." );
-  }
-  else
-    this->_Error( "Could not create a valid widget: no input." );
-}
-
-// -------------------------------------------------------------------------
-template< class _TImage >
-void cpPluginsWidgets::SplineWidget::
-_GD0_Image( _TImage* image )
-{
-  for( auto p = this->m_Props.begin( ); p != this->m_Props.end( ); ++p )
-  {
-    TWidgetData* d =
-      new TWidgetData(
-        this, dynamic_cast< TImageActor* >( p->first ), p->second
-        );
-    this->m_Widgets.push_back( d );
-
-  } // rof
-}
-
-// -------------------------------------------------------------------------
-cpPluginsWidgets::SplineWidget::TWidgetData::
-TWidgetData(
-  SplineWidget* seedWidget,
-  TImageActor* actor,
-  vtkRenderWindowInteractor* iren
-  )
-{
-  auto cb = vtkSmartPointer< SplineWidgetCallback >::New( );
-  cb->Widget = seedWidget;
-  cb->Data = this;
-  this->Command = cb;
-  actor->AddObserver( vtkCommand::InteractionEvent, cb );
-
-  auto image = actor->GetImage( );
-  int ori = actor->GetOrientation( );
-  int ext[ 6 ];
-  image->GetExtent( ext );
-  for( int i = ext[ ori << 1 ]; i <= ext[ ( ori << 1 ) + 1 ]; ++i )
-  {
-    auto placer = vtkSmartPointer< _TPlacer >::New( );
-    auto handle = vtkSmartPointer< vtkPointHandleRepresentation3D >::New( );
-    auto rep = vtkSmartPointer< vtkSeedRepresentation >::New( );
-    auto wdg = vtkSmartPointer< _TWidget >::New( );
-
-    placer->SetImageSlice( actor );
-    handle->GetProperty( )->SetColor( 1, 0, 0 );
-    handle->SetPointPlacer( placer );
-    rep->SetHandleRepresentation( handle );
-    wdg->SetRepresentation( rep );
-    wdg->SetInteractor( iren );
-    wdg->AddObserver( vtkCommand::PlacePointEvent, cb );
-    wdg->AddObserver( vtkCommand::CursorChangedEvent, cb );
-    wdg->EnabledOff( );
-
-    this->Widgets.push_back( wdg );
-    this->Placers.push_back( placer );
-    this->Handles.push_back( handle );
-    this->Representations.push_back( rep );
-
-  } // rof
-
-  this->ActualWidgetId = actor->GetSliceNumber( );
-  this->Widgets[ this->ActualWidgetId ]->EnabledOn( );
-}
-
-// -------------------------------------------------------------------------
-cpPluginsWidgets::SplineWidget::TWidgetData::
-~TWidgetData( )
-{
 }
-*/
 
 // eof - $RCSfile$