...

[cpPlugins.git] / plugins / VTKWidgets / LineWidget.cxx

diff --git a/plugins/VTKWidgets/LineWidget.cxx b/plugins/VTKWidgets/LineWidget.cxx
index f47508473842d84486f961de7b89c6b86b444f5d..24928e614d68235fb2ea477c0a0f9098f0efd5b4 100644 (file)
--- a/plugins/VTKWidgets/LineWidget.cxx
+++ b/plugins/VTKWidgets/LineWidget.cxx
@@ -1,6 +1,6 @@
 #include <VTKWidgets/LineWidget.h>
-#include <cpInstances/Image.h>
-#include <cpInstances/Mesh.h>
+#include <cpInstances/DataObjects/Image.h>
+#include <cpInstances/DataObjects/Mesh.h>
 #include <cpExtensions/Visualization/WindowLevelImageActor.h>
 
 #include <vtkLineWidget2.h>
@@ -50,12 +50,14 @@ cpPluginsVTKWidgets::LineWidget::
 LineWidget( )
   : Superclass( )
 {
-  typedef cpPlugins::BaseObjects::DataObject _TData;
-  typedef cpInstances::Mesh       _TMesh;
+  typedef cpPlugins::Pipeline::DataObject _TData;
+  typedef cpInstances::DataObjects::Mesh       _TMesh;
 
   this->_ConfigureInput< _TData >( "Input", false, false );
   this->_ConfigureOutput< _TMesh >( "Output" );
-  auto line = vtkSmartPointer< vtkPolyData >::New( );
+
+  // Create output data
+  auto line = this->_CreateVTK< vtkPolyData >( );
   line->SetPoints( vtkSmartPointer< vtkPoints >::New( ) );
   line->SetVerts( vtkSmartPointer< vtkCellArray >::New( ) );
   line->SetLines( vtkSmartPointer< vtkCellArray >::New( ) );