使用c++的itk进行nii数据的读取和写入

#pragma once

#include
#include


itk::Image::Pointer read_nii_image(const char* file_path) {
	// 设置图像类型
	using ImageType = itk::Image;
	using ReaderType = itk::ImageFileReader;
	// 设置读取的文件类型
	using ImageIoType = itk::NiftiImageIO;

	ReaderType::Pointer reader = ReaderType::New();
	ImageIoType::Pointer niftiIO = ImageIoType::New();
	reader->SetImageIO(niftiIO);

	reader->SetFileName(file_path);
	reader->Update(); // 读取文件结束
	// 得到文件
	ImageType::Pointer image = reader->GetOutput();

	return image;
}

#include
#include
#include
#include
#include
#include "funcs.h" // 自定义函数
using namespace std;


int main() {
	const char* filename = "E:/projectFiles/hips/bone_seg_4/dir_001/gu_bone.nii.gz";
	// 调用函数
	itk::Image::Pointer image = read_nii_image(filename);

	// 查看spacing
	itk::Image::SpacingType spacing = image->GetSpacing();
	cout << "spacing:n " << spacing << endl;
	// 查看direction
	itk::Image::DirectionType direction = image->GetDirection();
	cout << "direction:n " << direction << endl;
	// 查看origin
	itk::Image::PointType origin = image->GetOrigin();
	cout << "origin: n" << origin << endl;
	// 查看大小: x,y,z
	itk::Image::SizeType size = image->GetLargestPossibleRegion().GetSize();
	cout << "size: n" << size << endl;

	// 写入文件
	using WriterType = itk::ImageFileWriter>;
	using ImageIOType = itk::NiftiImageIO;
	
	WriterType::Pointer writer = WriterType::New();
	ImageIOType::Pointer niiIO = ImageIOType::New();
	const char* outputFilename = "rename.nii.gz";
	

	// casting转换,将image的输入类型转换成输出类型
	using FilterType =
		itk::CastImageFilter, itk::Image>;
		
	FilterType::Pointer filter = FilterType::New();
	
	filter->SetInput(image); // 转换完成
	// set spacing
	itk::Image::Pointer imagein = filter->GetOutput();

	writer->SetInput(imagein);
	writer->SetFileName(outputFilename);

	writer->SetImageIO(niiIO);
	try
	{
		writer->Update();
	}
	catch (const itk::ExceptionObject& err)
	{
		std::cerr << "ExceptionObject caught !" << std::endl;
		std::cerr << err << std::endl;
		return EXIT_FAILURE;
	}

	return 0;
}