VTK笔记-使用vtkImageImport类将内存数组转换为vtkImageData

vtkImageImport类

vtkImageImport提供了从独立于 VTK 的数据源（例如简单的C数组或第三方管道）导入图像数据所需的方法。请注意，VTK约定是将图像体素索引(0,0,0)设为图像的左下角，而大多数2D图像格式使用左上角。在图像加载到VTK 后，可以使用vtkImageFlip来校正方向。
请注意，也可以从 Python 字符串而不是从C数组导入原始数据。
该数组仅适用于标量点数据，不适用于单元数据。

void SetImportVoidPointer(void* ptr);

设置输入图像数据的指针；
将输入的ptr指针内联到vtk中，如果ptr被销毁，则GetOutput函数获取的vtkImageData会失效报错；
此时，vtk不会制作自己的数据副本，会直接从提供的数组访问数据。Vtk也不会删除数据或者修改数据；
如果需要vtk将ptr数组内容进行拷贝出一个副本使用，就需要使用函数CopyImportVoidPointer；

void CopyImportVoidPointer(void* ptr, vtkIdType size);

导入数据并制作内部数据副本；
如果您不想让VTK复制数据，请改用 SetImportVoidPointer（不要同时使用两者）。以字节为单位给出数据数组的大小。

void SetImportVoidPointer(void* ptr, int save);

设置输入图像数据的指针；
save用来控制ptr的生存期与vtkImageImport类对象的关系，如果期望在vtkImageImport类对象销毁时删除并释放ptr内存，可以将save设置为0；设置为1时，在vtkImageImport类对象销毁时不会删除并释放ptr内存；

void* GetImportVoidPointer() { return this->ImportVoidPointer; }

设置标量数据类型；

vtkSetMacro(DataScalarType, int);
void SetDataScalarTypeToDouble() { this->SetDataScalarType(VTK_DOUBLE); }
void SetDataScalarTypeToFloat() { this->SetDataScalarType(VTK_FLOAT); }
void SetDataScalarTypeToInt() { this->SetDataScalarType(VTK_INT); }
void SetDataScalarTypeToShort() { this->SetDataScalarType(VTK_SHORT); }
void SetDataScalarTypeToUnsignedShort() { this->SetDataScalarType(VTK_UNSIGNED_SHORT); }
void SetDataScalarTypeToUnsignedChar() { this->SetDataScalarType(VTK_UNSIGNED_CHAR); }
vtkGetMacro(DataScalarType, int);
const char* GetDataScalarTypeAsString(){
     return vtkImageScalarTypeNameMacro(this->DataScalarType);
}

设置标量数据的组成通道个数；

vtkSetMacro(NumberOfScalarComponents, int);
vtkGetMacro(NumberOfScalarComponents, int);

设置三维尺寸、像素间距、数据起点；

vtkSetVector6Macro(DataExtent, int);
vtkGetVector6Macro(DataExtent, int);
void SetDataExtentToWholeExtent() { this->SetDataExtent(this->GetWholeExtent()); }
vtkSetVector3Macro(DataSpacing, double);
vtkGetVector3Macro(DataSpacing, double);  
vtkSetVector3Macro(DataOrigin, double);
vtkGetVector3Macro(DataOrigin, double);
vtkSetVector6Macro(WholeExtent, int);
vtkGetVector6Macro(WholeExtent, int);

设置三维体数据的方向向量，默认为3*3的单位矩阵； (1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0)

vtkSetVectorMacro(DataDirection, double, 9);
vtkGetVectorMacro(DataDirection, double, 9);

示例代码

#include "pch.h"
#include <vtkImageActor.h>
#include <vtkImageData.h>
#include <vtkImageImport.h>
#include <vtkInteractorStyleImage.h>
#include <vtkNamedColors.h>
#include <vtkNew.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkXMLImageDataWriter.h>

#include "vtkAutoInit.h" 
VTK_MODULE_INIT(vtkRenderingOpenGL2);
VTK_MODULE_INIT(vtkRenderingVolumeOpenGL2);
VTK_MODULE_INIT(vtkRenderingFreeType);
VTK_MODULE_INIT(vtkInteractionStyle);
int main(){
    vtkNew<vtkNamedColors> colors;

    // Create a c-style image
    constexpr int width = 4;
    constexpr int height = 4;

    unsigned char cImage[width * height];
    unsigned char value = 0;
    for (unsigned int row = 0; row < height; ++row) {
        for (unsigned int col = 0; col < width; ++col) {
            cImage[row * width + col] = value;
            value += 10;
        }
    }

    // Convert the c-style image to a vtkImageData
    vtkNew<vtkImageImport> imageImport;
    imageImport->SetDataSpacing(1, 1, 1);
    imageImport->SetDataOrigin(0, 0, 0);
    imageImport->SetWholeExtent(0, width - 1, 0, height - 1, 0, 0);
    imageImport->SetDataExtentToWholeExtent();
    imageImport->SetDataScalarTypeToUnsignedChar();
    imageImport->SetNumberOfScalarComponents(1);
    imageImport->SetImportVoidPointer(cImage);
    imageImport->Update();

    // Create an actor
    vtkNew<vtkImageActor> actor;
    actor->SetInputData(imageImport->GetOutput());

    // Setup renderer
    vtkNew<vtkRenderer> renderer;
    renderer->AddActor(actor);
    renderer->ResetCamera();
    renderer->SetBackground(colors->GetColor3d("SaddleBrown").GetData());

    // Setup render window
    vtkNew<vtkRenderWindow> renderWindow;
    renderWindow->AddRenderer(renderer);
    renderWindow->SetWindowName("ImageImport");

    // Setup render window interactor
    vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
    vtkNew<vtkInteractorStyleImage> style;

    renderWindowInteractor->SetInteractorStyle(style);

    // Render and start interaction
    renderWindowInteractor->SetRenderWindow(renderWindow);
    renderWindow->Render();
    renderWindowInteractor->Initialize();

    renderWindowInteractor->Start();
    return 0;
}