Message manager can now be observed (to get messages) with a boost::signal/slot mecha...

[crea.git] / src / creaVtk.txx

diff --git a/src/creaVtk.txx b/src/creaVtk.txx
index 4d39387958debb94ce4b6316ccdff53b8986af9d..bbebc1dad8e79fe1702f3559ff6d3b4847e21e37 100644 (file)
--- a/src/creaVtk.txx
+++ b/src/creaVtk.txx
@@ -1,6 +1,20 @@
 #include <vtkDataArrayTemplate.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 <typeinfo> 
+#include <vtkTypeTraits.h>
+#include <creaMessageManager.h>
 namespace crea
 {
   
@@ -10,30 +24,52 @@ namespace crea
                                                    int ny,
                                                    int nz)
   {
-    std::cout << "NV "<<nx<<" " <<ny<<" " << nz<<std::endl;
-    std::cout << typeid(T).name() << std::endl;
+    //    std::cout << "NV "<<nx<<" " <<ny<<" " << nz<<std::endl;
+    //    std::cout <<  vtkTypeTraits<T>::SizedName() << std::endl;
     vtkImageData *image = vtkImageData::New();    
     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);
-    std::cout << "alloc"<<std::endl;
     image->AllocateScalars();
-    std::cout << "alloc ok"<<std::endl;
-    vtkDataArrayTemplate<T>* array 
-      = (vtkDataArrayTemplate<T>*)vtkDataArrayTemplate<T>::New();
-    std::cout << "data array alloc ok"<<std::endl;
-    //    array->SetNumberOfComponents( 1 );
-    std::cout << "nb compo ok"<<std::endl;
-    // The last param of SetArray is set to 1 to keep the class from deleting the array when it cleans up or reallocates memory.
+    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 );
     size_t size = (long)nx*(long)ny*(long)nz;
-    std::cout << "SetArray"<<std::endl;
-    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
-   std::cout << "SetScalars"<<std::endl;
-     image->GetPointData( )->SetScalars( array );
-   std::cout << "Delete"<<std::endl;
+    // The last param of SetArray is set to 1 to keep the class from deleting the array when it cleans up or reallocates memory.
+    tarray->SetArray( data, size, 1 );
+    image->GetPointData( )->SetScalars( tarray );
     array->Delete( );
     return image;
   }