Some useful vtk stuff

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 0df3e2c7a513ca83cd5258a8520ca6f13bbcb9d9..b1f7cdbe6627585dcdf26b118a2d9e2ca1abdbf1 100644 (file)
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -31,6 +31,7 @@ SET( SRCS
   creaRTTI
   creawxVTKRenderWindowInteractor
   creaVtkBasicSlicer
+  creaVtk
   creaFilesFromDirectory
   )
 

diff --git a/src/creaVtk.cpp b/src/creaVtk.cpp
new file mode 100644 (file)
index 0000000..3e864e2
--- /dev/null
+++ b/src/creaVtk.cpp
@@ -0,0 +1,13 @@
+#ifdef USE_VTK
+
+#include <creaVtk.h>
+
+namespace crea
+{
+
+  
+}
+
+
+#endif // USE_VTK
+

diff --git a/src/creaVtk.h b/src/creaVtk.h
new file mode 100644 (file)
index 0000000..175769e
--- /dev/null
+++ b/src/creaVtk.h
@@ -0,0 +1,41 @@
+#ifndef __creaVtk_h_INCLUDED__
+#define __creaVtk_h_INCLUDED__
+
+#ifdef USE_VTK
+
+#include <creaSystem.h>
+#include <vtkImageData.h>
+
+namespace crea
+{
+
+  template <class T>
+  vtkImageData* CREA_EXPORT NewVtkImageDataFromRaw( T* data,
+                                                   size_t size);
+
+  template <class T>
+  vtkIdType GetVtkIdType(T&) { return VTK_VOID; }
+
+#define SPECIALIZE_GETVTKIDTYPE(T,R) \
+  template <> vtkIdType GetVtkIdType<T>(T&) { return R; }  
+
+  SPECIALIZE_GETVTKIDTYPE(char,VTK_CHAR)
+  SPECIALIZE_GETVTKIDTYPE(signed char,VTK_SIGNED_CHAR)
+  SPECIALIZE_GETVTKIDTYPE(unsigned char,VTK_UNSIGNED_CHAR)
+  SPECIALIZE_GETVTKIDTYPE(short,VTK_SHORT)
+  SPECIALIZE_GETVTKIDTYPE(unsigned short,VTK_UNSIGNED_SHORT)
+  SPECIALIZE_GETVTKIDTYPE(int,VTK_INT)
+  SPECIALIZE_GETVTKIDTYPE(unsigned int,VTK_UNSIGNED_INT)
+  SPECIALIZE_GETVTKIDTYPE(long,VTK_LONG)
+  SPECIALIZE_GETVTKIDTYPE(unsigned long,VTK_UNSIGNED_LONG)
+  SPECIALIZE_GETVTKIDTYPE(float,VTK_FLOAT)
+  SPECIALIZE_GETVTKIDTYPE(double,VTK_DOUBLE)
+
+#undef SPECIALIZE_GETVTKIDTYPE
+}
+
+#include <creaVtk.txx>
+
+
+#endif // USE_VTK
+#endif // __creaVtk_h_INCLUDED__

diff --git a/src/creaVtk.txx b/src/creaVtk.txx
new file mode 100644 (file)
index 0000000..a7d6a04
--- /dev/null
+++ b/src/creaVtk.txx
@@ -0,0 +1,28 @@
+#include <vtkDataArrayTemplate.h>
+#include <vtkPointData.h>
+
+namespace crea
+{
+  
+  template <class T>
+  vtkImageData* CREA_EXPORT NewVtkImageDataFromRaw( T* data,
+                                                   size_t size)
+  {
+    vtkImageData *image = vtkImageData::New();    
+    image->SetNumberOfScalarComponents(1);
+    image->SetScalarType(GetVtkIdType<T>(*data));
+    image->AllocateScalars();
+    vtkDataArrayTemplate<T>* array 
+      = (vtkDataArrayTemplate<T>*)vtkDataArrayTemplate<T>::New();
+    array->SetNumberOfComponents( 1 );
+    // The last param of SetArray is set to 1 to keep the class from deleting the array when it cleans up or reallocates memory.
+    array->SetArray( data, size, 1 );
+    //Pay attention in particular to the last param of SetArray:
+    //http://www.vtk.org/doc/nightly/html/classvtkUnsignedShortArray.html#z798_0
+    image->GetPointData( )->SetScalars( array );
+    array->Delete( );
+    return image;
+  }
+  
+}
+