VTK: STL (vtkPolyData) 转 Mask（vtkImageData）

STL到Mask转换技巧

最新推荐文章于 2022-12-04 23:12:52 发布

原创

最新推荐文章于 2022-12-04 23:12:52 发布 · 1.7k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #gnu #散列表

项目中需要将分割结果的STL转换成Mask，以方便用户对其进行必要的调整。

vtkPolyData 转 vtkImageData 的方法见VTK官网中有相关的例子。

需要注意的几点：

1、转换后结果的数据类型和参考数据相同，如下代码中的maskImage；

2、 vtkPolyData 的数据原点（mask转polydata时）与 maskImage必须相同；

void Stl2Mask()
{

	double spacing[3] = {0.71875,0.71875,1.0};
	double origin[3] = { -172.64,-315.14,-355 };
	int dim[3] = { 512,512,295 };

	origin[0] = -dim[0] * spacing[0] * 0.5;
	origin[1] = -dim[1] * spacing[1] * 0.5;
	origin[2] = -dim[2] * spacing[2] * 0.5;

    // 创建参考vtkImageData
	vtkNew<vtkImageData> maskImage;
	maskImage->SetSpacing(spacing);
	maskImage->SetDimensions(dim);
	maskImage->SetExtent(0, dim[0] - 1, 0, dim[1] - 1, 0, dim[2] - 1);
	maskImage->SetOrigin(origin);
	maskImage->AllocateScalars(VTK_UNSIGNED_CHAR, 1);
	void *ptrMak = maskImage->GetScalarPointer();
	memset(ptrMak, 255, dim[0] * dim[1] * dim[2]);
    
    // 读STL文件
    vtkNew<vtkSTLReader> reader;
	reader->SetFileName("P:\\肺.stl