1 #include <vtkDataArrayTemplate.h>
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>
15 #include <vtkPointData.h>
16 #include <vtkTypeTraits.h>
17 #include <creaMessageManager.h>
22 vtkImageData* CREA_EXPORT NewVtkImageDataFromRaw( T* data,
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;
40 array = vtkCharArray::New(); break;
42 array = vtkSignedCharArray::New(); break;
43 case VTK_UNSIGNED_CHAR:
44 array = vtkUnsignedCharArray::New(); break;
46 array = vtkShortArray::New(); break;
47 case VTK_UNSIGNED_SHORT:
48 array = vtkUnsignedShortArray::New(); break;
50 array = vtkIntArray::New(); break;
51 case VTK_UNSIGNED_INT:
52 array = vtkUnsignedIntArray::New(); break;
54 array = vtkLongArray::New(); break;
55 case VTK_UNSIGNED_LONG:
56 array = vtkUnsignedLongArray::New(); break;
58 array = vtkFloatArray::New(); break;
60 array = vtkDoubleArray::New(); break;
62 creaGlobalError("NewVtkImageDataFromRaw : type "
63 <<vtkTypeTraits<T>::SizedName()
64 <<" non implemented");
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 );