读取nii数据，转为volume.dat二进制数据

本文主要是介绍读取nii数据，转为volume.dat二进制数据，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

代码

#include<iostream>
#include<itkImage.h>
#include"functions.h"
#include<exception>
#include<itkRescaleIntensityImageFilter.h>using GrayPixelType = unsigned char;
using OrgPixelType = short;using OrgImageType = itk::Image<OrgPixelType, 3>;
using OrgImagePointerType = OrgImageType::Pointer;using GrayImageType = itk::Image<GrayPixelType, 3>;
using GrayImagePointerType = GrayImageType::Pointer;#define WB(x,nBitCount) (((x)*nBitCount+31)/32*4)
//#define WB(x,nBitCount) ((x)*nBitCount)void normalizeImageByFilter(const OrgImagePointerType& input_image,GrayImagePointerType& out_image,const double dst_min, const double dst_max)
{using rescaleFilterType = itk::RescaleIntensityImageFilter<OrgImageType, GrayImageType>;auto rescaleFilter = rescaleFilterType::New();rescaleFilter->SetInput(input_image);rescaleFilter->SetOutputMinimum(dst_min);rescaleFilter->SetOutputMaximum(dst_max);try {rescaleFilter->Update();}catch (const itk::ExceptionObject& e) {std::cout << e.what() << std::endl;}out_image = rescaleFilter->GetOutput();
}int main()
{std::string niiPath("./45.nii.gz");OrgImagePointerType inputImage;readData<OrgImageType, OrgImagePointerType>(niiPath, inputImage);OrgImageType::SizeType size = inputImage->GetBufferedRegion().GetSize();int w = size[0], h = size[1], d = size[2];std::cout << "w = " << w << ", h = " << h << ", d = " << d<< std::endl;GrayImagePointerType normalizedImage;normalizeImageByFilter(inputImage, normalizedImage, 0, 255);GrayPixelType* inputImageBuffer = normalizedImage->GetBufferPointer();std::string volumePath("./volume.dat");FILE* fp = fopen(volumePath.c_str(), "wb");fwrite(&w, 1, sizeof(w), fp);fwrite(&h, 1, sizeof(h), fp);fwrite(&d, 1, sizeof(d), fp);int imgSize = w * h;GrayPixelType* ptrImageRaw = new GrayPixelType[imgSize];memset(ptrImageRaw, 0, imgSize);long long wb_new = WB(w, sizeof(GrayPixelType)*8);GrayPixelType* ptr_8bits = new GrayPixelType[h * wb_new];memset(ptr_8bits, 0, h * wb_new);std::cout << sizeof(GrayPixelType) << std::endl;try {int cnt = 0;for (int d_index = 0; d_index < d; ++d_index){// 取出第d_index 张切片for (int row = 0; row < h; ++row){for (int col = 0; col < w; ++col){*(ptr_8bits + row * wb_new + col) = inputImageBuffer[cnt];cnt += 1;}}// 取第d_index 张图像结束for (int row = 0; row < h; row++){memcpy(ptrImageRaw + row * w,ptr_8bits + row * wb_new, w*sizeof(GrayPixelType));}fwrite(ptrImageRaw, w * h, sizeof(GrayPixelType), fp);}fclose(fp);delete[] ptrImageRaw;delete[] ptr_8bits;}catch (...) {std::cout << "exception" << std::endl;}return 0;
}