VTK系列71_VTK對MHA格式檔案三維感興趣區域裁剪及MHA格式儲存

執行個體12：基于VTK對MHA格式檔案三維感興趣區域裁剪及MHA格式儲存

#include "vtkAutoInit.h" 
VTK_MODULE_INIT(vtkRenderingOpenGL2);
VTK_MODULE_INIT(vtkInteractionStyle);

#include <vtkSmartPointer.h>
#include <vtkImageData.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkInteractorStyleImage.h>
#include <vtkRenderer.h>
#include <vtkImageActor.h>
#include <vtkBMPReader.h>
#include <vtkExtractVOI.h>
#include <vtkMetaImageReader.h>//mha、mad圖像讀取類
#include <vtkMetaImageWriter.h>mha、mad圖像寫入類

//測試圖像：../data/lena.bmp
int main(int argc, char* argv[])
{
	/*vtkSmartPointer<vtkBMPReader> reader =
		vtkSmartPointer<vtkBMPReader>::New();
	reader->SetFileName("lena.bmp");
	reader->Update();*/

	vtkSmartPointer<vtkMetaImageReader> reader = vtkSmartPointer<vtkMetaImageReader>::New();
	//mhd與mha檔案其實格式是一樣的，記錄mhd對應的raw檔案應在同一目錄
	//mhd格式圖像資訊頭與實際圖像的存儲分為兩個檔案（*.mhd檔案記錄圖像資訊頭；*.raw或//*.zraw（zraw指有壓縮）記錄實際圖像）
	//mha格式将圖像資訊頭與實際的像素值等資料寫入到同一個檔案中
	//reader->SetFileName("test1.mhd");
	  //.mha和.raw檔案需要在同一個檔案夾
	reader->SetFileName("CT_6_spacing.mha");
	reader->Update();

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

	vtkSmartPointer<vtkExtractVOI> extractVOI =
		vtkSmartPointer<vtkExtractVOI>::New();
	extractVOI->SetInputConnection(reader->GetOutputPort());
	//設定感興趣區域：X_min、X_max、Y_min、Y_max、Z_min、Z_max
	extractVOI->SetVOI(dims[0] / 4., 3. * dims[0] / 4., dims[1] / 4., 3. * dims[1] / 4., 0, 5);//這裡設定0~5，表示裁剪六張切片
	extractVOI->Update();

	vtkSmartPointer<vtkImageActor> originalActor =
		vtkSmartPointer<vtkImageActor>::New();
	originalActor->SetInputData(reader->GetOutput());

	vtkSmartPointer<vtkImageActor> voiActor =
		vtkSmartPointer<vtkImageActor>::New();
	voiActor->SetInputData(extractVOI->GetOutput());

	double originalViewport[4] = { 0.0, 0.0, 0.5, 1.0 };
	double voiviewport[4] = { 0.5, 0.0, 1.0, 1.0 };

	vtkSmartPointer<vtkRenderer> originalRenderer =
		vtkSmartPointer<vtkRenderer>::New();
	originalRenderer->SetViewport(originalViewport);
	originalRenderer->AddActor(originalActor);
	originalRenderer->ResetCamera();
	originalRenderer->SetBackground(1.0, 1.0, 1.0);

	vtkSmartPointer<vtkRenderer> shiftscaleRenderer =
		vtkSmartPointer<vtkRenderer>::New();
	shiftscaleRenderer->SetViewport(voiviewport);
	shiftscaleRenderer->AddActor(voiActor);
	shiftscaleRenderer->ResetCamera();
	shiftscaleRenderer->SetBackground(1.0, 1.0, 1.0);

	vtkSmartPointer<vtkRenderWindow> renderWindow =
		vtkSmartPointer<vtkRenderWindow>::New();
	renderWindow->AddRenderer(originalRenderer);
	renderWindow->AddRenderer(shiftscaleRenderer);
	renderWindow->SetSize(900, 300);
	renderWindow->Render();
	renderWindow->SetWindowName("ExtractVOIExample");

	vtkSmartPointer<vtkRenderWindowInteractor> renderWindowInteractor =
		vtkSmartPointer<vtkRenderWindowInteractor>::New();
	vtkSmartPointer<vtkInteractorStyleImage> style =
		vtkSmartPointer<vtkInteractorStyleImage>::New();

	//儲存為mhd檔案
	//vtkMetaImageWriter* vtkWriter = vtkMetaImageWriter::New();
	//vtkWriter->SetInputConnection(extractVOI->GetOutputPort());
	以raw和mhd格式儲存，去掉此句則以zraw儲存
	//vtkWriter->SetCompression(0);
	//vtkWriter->SetFileName("BrainProtonDensity3Slices_1.mhd");
	//vtkWriter->Write();

	儲存為mha檔案
	vtkMetaImageWriter* vtkWriter = vtkMetaImageWriter::New();
	vtkWriter->SetInputConnection(extractVOI->GetOutputPort());
	//以raw和mhd格式儲存，去掉此句則以zraw儲存
	vtkWriter->SetCompression(0);
	vtkWriter->SetFileName("CT_6_spacing_cut.mha");
	vtkWriter->Write();

	renderWindowInteractor->SetInteractorStyle(style);
	renderWindowInteractor->SetRenderWindow(renderWindow);
	renderWindowInteractor->Initialize();
	renderWindowInteractor->Start();



	return EXIT_SUCCESS;
}
           

讀取的MHA源檔案:

標明區域剪切後儲存的MHA檔案：

其中前四張剪切前後效果對比：

 注：此例程MHD和NII格式檔案同樣适用。

本例程配套素材見源碼整理文章下載下傳（點選進入）

VTK系列目錄：

1 VTK基本概念

2 VTK圖像處理

3 VTK圖形處理

4 VTK體繪制