Pass std::string as reference

[crea.git] / src / creaVtk.txx

diff --git a/src/creaVtk.txx b/src/creaVtk.txx
index a7d6a04733b9805a2180f6232302054d85a9be92..c4e7167747d135f05c2d48269e4b445efb39bc82 100644 (file)
--- a/src/creaVtk.txx
+++ b/src/creaVtk.txx
@@ -1,25 +1,83 @@
 #include <vtkDataArrayTemplate.h>
-#include <vtkPointData.h>
 
+#include <vtkCharArray.h>
+#include <vtkSignedCharArray.h>
+#include <vtkUnsignedCharArray.h>
+#include <vtkShortArray.h>
+#include <vtkUnsignedShortArray.h>
+#include <vtkIntArray.h>
+#include <vtkUnsignedIntArray.h>
+#include <vtkLongArray.h>
+#include <vtkUnsignedLongArray.h>
+#include <vtkFloatArray.h>
+#include <vtkDoubleArray.h>
+
+#include <vtkPointData.h>
+#include <vtkTypeTraits.h>
+#include <creaMessageManager.h>
 namespace crea
-{
-  
+{  
   template <class T>
-  vtkImageData* CREA_EXPORT NewVtkImageDataFromRaw( T* data,
-                                                   size_t size)
+  /*CREA_EXPORT*/ vtkImageData* NewVtkImageDataFromRaw( T* data, 
+                                                   int nx, 
+                                                   int ny,
+                                                   int nz,
+                                                   bool do_not_desalloc)
   {
-    vtkImageData *image = vtkImageData::New();    
+    //    std::cout << "NV "<<nx<<" " <<ny<<" " << nz<<std::endl;
+    //    std::cout <<  vtkTypeTraits<T>::SizedName() << std::endl;
+    vtkImageData *image = vtkImageData::New();
+
+    // Shouldn't we pass NumberOfScalarComponents to deal with RGB, RGBA images as well? // JPR
+
     image->SetNumberOfScalarComponents(1);
-    image->SetScalarType(GetVtkIdType<T>(*data));
+
+    int vtktype = vtkTypeTraits<T>::VTKTypeID();
+    image->SetScalarType(vtktype);
+    image->SetDimensions(nx, ny ,nz);
+    image->SetSpacing(1, 1, 1);
     image->AllocateScalars();
-    vtkDataArrayTemplate<T>* array 
-      = (vtkDataArrayTemplate<T>*)vtkDataArrayTemplate<T>::New();
+    vtkDataArray* array = 0;
+    switch (vtktype)
+      {
+      case VTK_CHAR:
+       array = vtkCharArray::New(); break;
+      case VTK_SIGNED_CHAR:
+       array = vtkSignedCharArray::New(); break;
+      case VTK_UNSIGNED_CHAR:
+       array = vtkUnsignedCharArray::New(); break;
+      case VTK_SHORT:
+       array = vtkShortArray::New(); break;
+      case VTK_UNSIGNED_SHORT:
+       array = vtkUnsignedShortArray::New(); break;
+      case VTK_INT:
+       array = vtkIntArray::New(); break;      
+      case VTK_UNSIGNED_INT:
+       array = vtkUnsignedIntArray::New(); break;
+      case VTK_LONG:
+       array = vtkLongArray::New(); break;
+      case VTK_UNSIGNED_LONG:
+       array = vtkUnsignedLongArray::New(); break;
+      case VTK_FLOAT:
+       array = vtkFloatArray::New(); break;
+      case VTK_DOUBLE:
+       array = vtkDoubleArray::New(); break;
+      default:
+       creaGlobalError("NewVtkImageDataFromRaw : type "
+                 <<vtkTypeTraits<T>::SizedName()
+                 <<" non implemented");
+      }
+    vtkDataArrayTemplate<T>* tarray 
+               = dynamic_cast<vtkDataArrayTemplate<T>*>(array);
     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 );
+    size_t size = (long)nx*(long)ny*(long)nz;
+    // The last param of SetArray is set to 1 to keep the class from deleting the array 
+    // when it cleans up or reallocates memory.
+    int dndesa = 0;
+    if (do_not_desalloc)
+       dndesa = 1;
+    tarray->SetArray( data, size, dndesa );
+    image->GetPointData( )->SetScalars( tarray );
     array->Delete( );
     return image;
   }