VTK循环遍历图像中的像素

#include <vtkSmartPointer.h>
#include <vtkImageData.h>

int main()
{
    // 创建一个图像数据对象
    vtkSmartPointer<vtkImageData> imageData = vtkSmartPointer<vtkImageData>::New();
    // 设置图像数据的尺寸和数据类型等属性
    imageData->SetDimensions(512, 512, 1);
    imageData->SetScalarTypeToUnsignedChar();
    imageData->AllocateScalars();

    // 获取图像数据的指针
    unsigned char* pixels = static_cast<unsigned char*>(imageData->GetScalarPointer());

    // 循环遍历图像像素
    int width = imageData->GetDimensions()[0];
    int height = imageData->GetDimensions()[1];
    for (int y = 0; y < height; y++)
    {
        for (int x = 0; x < width; x++)
        {
            // 计算当前像素的索引
            int index = (y * width + x) * imageData->GetNumberOfScalarComponents();
            // 获取当前像素的数值
            unsigned char* pixel = pixels + index;

            // 在这里可以对像素进行处理，例如修改像素值、提取像素特征等

            // 示例：将当前像素设置为红色
            pixel[0] = 255; // 红色通道
            pixel[1] = 0;   // 绿色通道
            pixel[2] = 0;   // 蓝色通道
        }
    }

    return 0;
}