src/creaVtk.txx

   1 #include <vtkDataArrayTemplate.h>
   2
   3 #include <vtkCharArray.h>
   4 #include <vtkSignedCharArray.h>
   5 #include <vtkUnsignedCharArray.h>
   6 #include <vtkShortArray.h>
   7 #include <vtkUnsignedShortArray.h>
   8 #include <vtkIntArray.h>
   9 #include <vtkUnsignedIntArray.h>
  10 #include <vtkLongArray.h>
  11 #include <vtkUnsignedLongArray.h>
  12 #include <vtkFloatArray.h>
  13 #include <vtkDoubleArray.h>
  14
  15 #include <vtkPointData.h>
  16 #include <vtkTypeTraits.h>
  17 #include <creaMessageManager.h>
  18 namespace crea
  19 {
  20
  21   template <class T>
  22   vtkImageData* CREA_EXPORT NewVtkImageDataFromRaw( T* data,
  23                                                     int nx,
  24                                                     int ny,
  25                                                     int nz)
  26   {
  27     //    std::cout << "NV "<<nx<<" " <<ny<<" " << nz<<std::endl;
  28     //    std::cout <<  vtkTypeTraits<T>::SizedName() << std::endl;
  29     vtkImageData *image = vtkImageData::New();
  30     image->SetNumberOfScalarComponents(1);
  31     int vtktype = vtkTypeTraits<T>::VTKTypeID();
  32     image->SetScalarType(vtktype);
  33     image->SetDimensions(nx,ny,nz);
  34     image->SetSpacing(1,1,1);
  35     image->AllocateScalars();
  36     vtkDataArray* array = 0;
  37     switch (vtktype)
  38       {
  39       case VTK_CHAR:
  40         array = vtkCharArray::New(); break;
  41       case VTK_SIGNED_CHAR:
  42         array = vtkSignedCharArray::New(); break;
  43       case VTK_UNSIGNED_CHAR:
  44         array = vtkUnsignedCharArray::New(); break;
  45       case VTK_SHORT:
  46         array = vtkShortArray::New(); break;
  47       case VTK_UNSIGNED_SHORT:
  48         array = vtkUnsignedShortArray::New(); break;
  49       case VTK_INT:
  50         array = vtkIntArray::New(); break;
  51       case VTK_UNSIGNED_INT:
  52         array = vtkUnsignedIntArray::New(); break;
  53       case VTK_LONG:
  54         array = vtkLongArray::New(); break;
  55       case VTK_UNSIGNED_LONG:
  56         array = vtkUnsignedLongArray::New(); break;
  57       case VTK_FLOAT:
  58         array = vtkFloatArray::New(); break;
  59       case VTK_DOUBLE:
  60         array = vtkDoubleArray::New(); break;
  61       default:
  62         creaGlobalError("NewVtkImageDataFromRaw : type "
  63                   <<vtkTypeTraits<T>::SizedName()
  64                   <<" non implemented");
  65       }
  66     vtkDataArrayTemplate<T>* tarray
  67       = dynamic_cast<vtkDataArrayTemplate<T>*>(array);
  68     array->SetNumberOfComponents( 1 );
  69     size_t size = (long)nx*(long)ny*(long)nz;
  70     // The last param of SetArray is set to 1 to keep the class from deleting the array when it cleans up or reallocates memory.
  71     tarray->SetArray( data, size, 1 );
  72     image->GetPointData( )->SetScalars( tarray );
  73     array->Delete( );
  74     return image;
  75   }
  76
  77 }
  78