VTKimage2Mat记录

最近在研究VTK，有些算法需要转到cv上操作，于是在网上搜罗了两种方法，备份使用

cv::Mat VTKimage2Mat(std::string inputFilename,  vtkSmartPointer<vtkDICOMImageReader>& reader)
{
    reader->SetFileName(inputFilename.c_str());
    reader->Update();

    vtkSmartPointer<vtkImageCast> imageCast = vtkSmartPointer<vtkImageCast>::New();
    imageCast->SetInputConnection(reader->GetOutputPort());
    imageCast->SetOutputScalarTypeToInt();
    imageCast->Update();

    cv::Mat img1 = cv::Mat(1024, 1024, CV_32SC1); //定义一个与VTK大小相同的mat矩阵

    //short* ptr = (short*)reader->GetOutput()->GetScalarPointer();
    int* ptr = (int*)imageCast->GetOutput()->GetScalarPointer();

    for (int i = 0; i < 1024; i++) {
        for (int j = 0; j < 1024; j++) {

            img1.at<int>(1023 - j, i) = *(ptr + j * 1024 + i);
        }
    }

    cv::normalize(img1, img1, 0, 255, cv::NORM_MINMAX);
    cv::imwrite("result1.bmp", img1);

    cv::Mat img2 = cv::Mat(1024, 1024, CV_32SC1);

    int dims[3];
    reader->GetOutput()->GetDimensions(dims);

    //图像的像素值
    for (int k = 0; k < dims[2]; k++)
    {
        for (int j = 0; j < dims[1]; j++)
        {
            for (int i = 0; i < dims[0]; i++)
            {
                //转换数据类型，使用imagecast转到double（或float）
                int* pixel =(int*)(imageCast->GetOutput()->GetScalarPointer(i, j, k)); // 第i列第j行的像素值
                img2.at<int>(j, i) = int(*pixel); // 第j行第i列的像素值
            }
        }
    }

    cv::flip(img2, img2, 0);
    cv::normalize(img2, img2, 0, 255, cv::NORM_MINMAX);
    cv::imwrite("result2.bmp", img2);
    return img1;
}

以上两种方法均可实现，但是方法1较快，方法2易于理解，我是新手小白，对指针不是很清楚