vtkImageData圖像遍歷

 

#ifndef MAINWINDOW_H
#define MAINWINDOW_H

#include <QMainWindow>
#include <vtkAutoInit.h>
VTK_MODULE_INIT(vtkRenderingOpenGL2)
VTK_MODULE_INIT(vtkInteractionStyle)
#include <vtkSmartPointer.h>
#include <vtkImageActor.h>
#include <vtkRenderer.h>
#include <vtkJPEGReader.h>

namespace Ui {
class MainWindow;
}

class MainWindow : public QMainWindow
{
    Q_OBJECT

public:
    explicit MainWindow(QWidget *parent = nullptr);
    ~MainWindow();
private slots:
    void openFileSlot();
private:
    vtkSmartPointer<vtkRenderer>   pRenderer0;
    vtkSmartPointer<vtkRenderer>   pRenderer1;
    vtkSmartPointer<vtkRenderer>   pRenderer2;

private:
    Ui::MainWindow *ui;
};

#endif // MAINWINDOW_H

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include <QDebug>
#include <QFileDialog>
#include <vtkRenderWindow.h>
#include <vtkPNGReader.h>
#include <vtkImageCanvasSource2D.h>
#include <vtkImageBlend.h>
#include <vtkImageData.h>
#include <vtkImageViewer.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkInteractorStyleImage.h>
#include <vtkImageStencil.h>
#include <vtkImageStencilData.h>
#include <vtkImageLuminance.h>
#include <vtkImageChangeInformation.h>
#include <vtkImageIterator.h>

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    ui->setupUi(this);

    pRenderer0 = vtkSmartPointer<vtkRenderer>::New();
    pRenderer0->SetBackground(1, 0, 0);
    pRenderer0->SetViewport(0, 0, 0.33, 1);

    pRenderer1 = vtkSmartPointer<vtkRenderer>::New();
    pRenderer1->SetBackground(0, 1, 0);
    pRenderer1->SetViewport(0.33, 0, 0.66, 1);

    pRenderer2 = vtkSmartPointer<vtkRenderer>::New();
    pRenderer2->SetBackground(0, 0, 1);
    pRenderer2->SetViewport(0.66, 0, 1, 1);

    ui->qvtkWidget->GetRenderWindow()->AddRenderer(pRenderer0);
    ui->qvtkWidget->GetRenderWindow()->AddRenderer(pRenderer1);
    ui->qvtkWidget->GetRenderWindow()->AddRenderer(pRenderer2);

    connect(ui->pushButton, SIGNAL(clicked(bool)), this, SLOT(openFileSlot()));
}

MainWindow::~MainWindow()
{
    delete ui;
}

void MainWindow::openFileSlot()
{
#if 1
    QString selectFilePath = QFileDialog::getOpenFileName(this, QString("選擇圖像文件"), QString(""), QString("圖像(*.jpg)"));
#else
    QString selectFilePath = QFileDialog::getOpenFileName(this, QString("選擇圖像文件"), QString(""), QString("圖像(*.png)"));
#endif
    if(selectFilePath.isEmpty())
    {
        ui->textBrowser->append("選擇圖像路徑爲空!");
        return ;
    }

    // 原始圖像
#if 1
    vtkSmartPointer<vtkJPEGReader> pJpegReader =  vtkSmartPointer<vtkJPEGReader>::New();
#else
    vtkSmartPointer<vtkPNGReader> pJpegReader =  vtkSmartPointer<vtkPNGReader>::New();
#endif
    pJpegReader->SetFileName(selectFilePath.toStdString().c_str());
    pJpegReader->Update();

    ui->textBrowser->append(QString("載入圖像:") + selectFilePath + QString(" 成功 !"));

    // vtkImageData
    {
        // 獲取圖像信息
        int dims[3];
        pJpegReader->GetOutput()->GetDimensions(dims);
        ui->textBrowser->append(QString("圖像維數：") + QString::number(dims[0]) + ", " + QString::number(dims[1]) + ", " + QString::number(dims[2]));

        // 獲取圖像原點
        double origin[3];
        pJpegReader->GetOutput()->GetOrigin(origin);
        ui->textBrowser->append(QString("圖像原點：") + QString::number(origin[0]) + ", " + QString::number(origin[1]) + ", " + QString::number(origin[2]));

        // 獲取圖像的間隔
        double spaceing[3];
        pJpegReader->GetOutput()->GetSpacing(spaceing);
        ui->textBrowser->append(QString("圖像像素點間隔：") + QString::number(spaceing[0]) + ", " + QString::number(spaceing[1]) + ", " + QString::number(spaceing[2]));

        vtkSmartPointer<vtkImageActor> pJpegActor = vtkSmartPointer<vtkImageActor>::New();
        pJpegActor->SetInputData(pJpegReader->GetOutput());
        pRenderer0->AddActor(pJpegActor);
    }

    // vtkChangeImageInformation
    {
        // 圖像信息進行修改
        vtkSmartPointer<vtkImageChangeInformation> pJpegChange = vtkSmartPointer<vtkImageChangeInformation>::New();
        pJpegChange->SetInputData(pJpegReader->GetOutput());
        pJpegChange->SetOutputOrigin(100, 100, 0);
        pJpegChange->SetOutputSpacing(5, 5, 1);
        pJpegChange->SetCenterImage(1);
        pJpegChange->Update();


        // 獲取圖像信息
        int dims[3];
        pJpegChange->GetOutput()->GetDimensions(dims);
        ui->textBrowser->append(QString("修改後圖像維數：") + QString::number(dims[0]) + ", " + QString::number(dims[1]) + ", " + QString::number(dims[2]));

        // 獲取圖像原點
        double origin[3];
        pJpegChange->GetOutput()->GetOrigin(origin);
        ui->textBrowser->append(QString("修改後圖像原點：") + QString::number(origin[0]) + ", " + QString::number(origin[1]) + ", " + QString::number(origin[2]));

        // 獲取圖像的間隔
        double spaceing[3];
        pJpegChange->GetOutput()->GetSpacing(spaceing);
        ui->textBrowser->append(QString("修改後圖像像素點間隔：") + QString::number(spaceing[0]) + ", " + QString::number(spaceing[1]) + ", " + QString::number(spaceing[2]));
    }

    // 訪問圖像點
    {
        int dims[3];
        pJpegReader->GetOutput()->GetDimensions(dims);

        // 深拷貝圖像
        vtkSmartPointer<vtkImageData> pCopyImage = vtkSmartPointer<vtkImageData>::New();
        pCopyImage->DeepCopy(pJpegReader->GetOutput());

        // 非管線，不需要update
        for(int k = 0; k < dims[2]; ++k)
        {
            for(int i = 0; i < dims[0]; ++i)
            {
                for(int j = 0; j < dims[1]; ++j)
                {
                    if(i < 100 && j < 100)
                    {
                        unsigned char* pPixel = (unsigned char*)(pCopyImage->GetScalarPointer(i, j, k));
                        *(pPixel + 0) = 0;
                        *(pPixel + 1) = 0;
                        *(pPixel + 2) = 0;
                    }
                }
            }
        }

        vtkSmartPointer<vtkImageActor> pCopyActor = vtkSmartPointer<vtkImageActor>::New();
        pCopyActor->SetInputData(pCopyImage);
        pRenderer1->AddActor(pCopyActor);
    }

    // 使用迭代器遍歷圖像點
    {
        int dims[3];
        pJpegReader->GetOutput()->GetDimensions(dims);

        // 深拷貝圖像
        vtkSmartPointer<vtkImageData> pCopyImage = vtkSmartPointer<vtkImageData>::New();
        pCopyImage->DeepCopy(pJpegReader->GetOutput());

        // 非管線，不需要update
        int region[] = {dims[0] / 4, dims[0] * 3 / 4, dims[1] / 4, dims[1] * 3 / 4};
        vtkImageIterator<unsigned char> pImageIterator(pCopyImage, region);

        while(!pImageIterator.IsAtEnd())
        {
            auto iterBegin = pImageIterator.BeginSpan();
            auto iterEnd   = pImageIterator.EndSpan();
            for(auto iter = iterBegin; iter != iterEnd; ++iter)
            {
                *iter = 255 - *iter;
            }
            pImageIterator.NextSpan();
        }

        vtkSmartPointer<vtkImageActor> pCopyActor = vtkSmartPointer<vtkImageActor>::New();
        pCopyActor->SetInputData(pCopyImage);
        pRenderer2->AddActor(pCopyActor);
    }

    // 不能在MainWidow的構造函數中
    vtkSmartPointer<vtkInteractorStyleImage> style = vtkSmartPointer<vtkInteractorStyleImage>::New();
    ui->qvtkWidget->GetRenderWindow()->GetInteractor()->SetInteractorStyle(style);

    pRenderer0->ResetCamera();
    pRenderer1->ResetCamera();
    pRenderer2->ResetCamera();
    ui->qvtkWidget->GetRenderWindow()->Render();
}

 QT UI 文件

<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
 <class>MainWindow</class>
 <widget class="QMainWindow" name="MainWindow">
  <property name="geometry">
   <rect>
    <x>0</x>
    <y>0</y>
    <width>400</width>
    <height>300</height>
   </rect>
  </property>
  <property name="windowTitle">
   <string>MainWindow</string>
  </property>
  <widget class="QWidget" name="centralWidget">
   <layout class="QHBoxLayout" name="horizontalLayout" stretch="2,5">
    <item>
     <layout class="QVBoxLayout" name="verticalLayout">
      <item>
       <widget class="QPushButton" name="pushButton">
        <property name="text">
         <string>選擇圖像</string>
        </property>
       </widget>
      </item>
      <item>
       <widget class="QTextBrowser" name="textBrowser"/>
      </item>
     </layout>
    </item>
    <item>
     <widget class="QVTKWidget" name="qvtkWidget"/>
    </item>
   </layout>
  </widget>
  <widget class="QMenuBar" name="menuBar">
   <property name="geometry">
    <rect>
     <x>0</x>
     <y>0</y>
     <width>400</width>
     <height>22</height>
    </rect>
   </property>
  </widget>
  <widget class="QToolBar" name="mainToolBar">
   <attribute name="toolBarArea">
    <enum>TopToolBarArea</enum>
   </attribute>
   <attribute name="toolBarBreak">
    <bool>false</bool>
   </attribute>
  </widget>
  <widget class="QStatusBar" name="statusBar"/>
 </widget>
 <layoutdefault spacing="6" margin="11"/>
 <customwidgets>
  <customwidget>
   <class>QVTKWidget</class>
   <extends>QWidget</extends>
   <header>QVTKWidget.h</header>
  </customwidget>
 </customwidgets>
 <resources/>
 <connections/>
</ui>