VTKWithNumpy使用 NumPy 数组来创建3D体渲染所需要的数据

一:主要的知识点

1、说明

本文只是教程内容的一小段,因博客字数限制,故进行拆分。主教程链接:vtk教程------逐行解析官网所有Python示例-CSDN博客

2、知识点纪要

本段代码主要涉及的有①vtk与numpy数组的联动,②vtkImageImport的使用

二:代码及注释

python 复制代码
import vtkmodules.vtkRenderingVolumeOpenGL2
import vtkmodules.vtkInteractionStyle
import vtkmodules.vtkRenderingOpenGL2
from vtkmodules.vtkIOImage import vtkImageImport
from vtkmodules.vtkCommonColor import vtkNamedColors
from vtkmodules.vtkRenderingCore import vtkColorTransferFunction, vtkRenderWindow, vtkRenderWindowInteractor, \
    vtkRenderer, vtkVolume, vtkVolumeProperty
import numpy as np
from vtkmodules.vtkCommonDataModel import vtkPiecewiseFunction
from vtkmodules.vtkRenderingVolume import vtkFixedPointVolumeRayCastMapper


def main():
    colors = vtkNamedColors()

    data_matrix = np.zeros([75, 75, 75], dtype=np.uint8)
    data_matrix[0:35, 0:35, 0:35] = 50
    data_matrix[25:55, 25:55, 25:55] = 100
    data_matrix[45:74, 45:74, 45:74] = 150

    """
    vtkImageImport 将外部的原始图像数据(Raw data)导入到 VTK 的图像管线中,从而变成一个 vtkImageData 对象,供后续处理或渲染使用
    这个类允许你把Python/Numpy 数组  C/C++ 原始内存数据(unsigned char*, float*, etc.)
    或者是外部图像格式(比如医学影像、数值模拟输出等)
    """
    dataImporter = vtkImageImport()
    # 会把整个数组的原始内存内容转换成 连续的字节流(bytes) 结果类似于:b'\x01\x02\x03\x04\x05\x06'
    # 因为 vtkImageImport 期望接收 原始字节数据指针,而不是 NumPy 对象
    data_string = data_matrix.tobytes()
    # CopyImportVoidPointer 复制一份数据到vtk内部
    dataImporter.CopyImportVoidPointer(data_string, len(data_string))
    dataImporter.SetDataScalarTypeToUnsignedChar()  # 将输入的数据转换为无符号8位证书类型
    dataImporter.SetNumberOfScalarComponents(1)
    """
    SetDataExtent 告诉vtkImageImport 原始数据体素在x,y,z三个方向上的索引范围
    这里的意思是
    x方向:从 0 到 74,共 75 个体素
    y方向:从 0 到 74,共 75 个体素
    z方向:从 0 到 74,共 75 个体素
    """
    dataImporter.SetDataExtent(0, 74, 0, 74, 0, 74)
    """
    SetWholeExtent 这行代码看似重复,实则是在指定整个图像(数据集)在 VTK 管线中表示的范围
    SetDataExtent(...) 是描述当前导入数据的内存范围
    SetWholeExtent(...) 是描述VTK 内部整个图像数据的逻辑范围
    """
    dataImporter.SetWholeExtent(0, 74, 0, 74, 0, 74)

    """
    如果数据是由多块volume数据拼接的
    WholeExtent 可以大于 DataExtent,用于表示你导入的是其中的一部分
    importer1 = vtkImageImport()
    importer1.CopyImportVoidPointer(block1_data, len(block1_data))
    importer1.SetDataScalarTypeToUnsignedChar()
    importer1.SetNumberOfScalarComponents(1)
    importer1.SetDataExtent(0, 511, 0, 511, 0, 511)
    importer1.SetWholeExtent(0, 1023, 0, 1023, 0, 1023)
    
    importer2 = vtkImageImport()
    importer2.CopyImportVoidPointer(block2_data, len(block2_data))
    importer2.SetDataScalarTypeToUnsignedChar()
    importer2.SetNumberOfScalarComponents(1)
    importer2.SetDataExtent(512, 1023, 0, 511, 0, 511)
    importer2.SetWholeExtent(0, 1023, 0, 1023, 0, 1023)
    
    # 然后拼接起来
    append = vtkAppendFilter()
    append.AddInputConnection(importer1.GetOutputPort())
    append.AddInputConnection(importer2.GetOutputPort())
    append.AddInputConnection(importer3.GetOutputPort())
    append.Update()
    """

    # 定义标量与颜色的对应关系
    colorFunc = vtkColorTransferFunction()
    colorFunc.AddRGBPoint(50, 1.0, 0.0, 0.0)
    colorFunc.AddRGBPoint(100, 0.0, 1.0, 0.0)
    colorFunc.AddRGBPoint(150, 0.0, 0.0, 1.0)

    alphaChannelFunc = vtkPiecewiseFunction()  # piecewise function 百度翻译 分段函数
    alphaChannelFunc.AddPoint(0, 0.0)
    alphaChannelFunc.AddPoint(50, 0.05)
    alphaChannelFunc.AddPoint(100, 0.1)
    alphaChannelFunc.AddPoint(150, 0.2)

    volumeProperty = vtkVolumeProperty()
    volumeProperty.SetColor(colorFunc)
    volumeProperty.SetScalarOpacity(alphaChannelFunc)

    volumeMapper = vtkFixedPointVolumeRayCastMapper()
    volumeMapper.SetInputConnection(dataImporter.GetOutputPort())

    volume = vtkVolume()
    volume.SetMapper(volumeMapper)
    volume.SetProperty(volumeProperty)

    renderer = vtkRenderer()
    renderer.AddVolume(volume)
    renderer.SetBackground(colors.GetColor3d("MistyRose"))

    renderWin = vtkRenderWindow()
    renderWin.AddRenderer(renderer)
    renderInteractor = vtkRenderWindowInteractor()
    renderInteractor.SetRenderWindow(renderWin)

    renderWin.SetSize(400, 400)
    renderWin.SetWindowName('VTKWithNumpy')

    """
    用于实现渲染过程的中断检查,目的是提高应用程序的响应速度
    """
    def exitCheck(obj, event):
        # 检查 VTK 渲染窗口的事件队列中是否还有未处理的操作系统或用户输入事件(比如窗口移动、鼠标点击、键盘输入等)
        if obj.GetEventPending() != 0:
            obj.SetAbortRender(1)

    renderWin.AddObserver("AbortCheckEvent", exitCheck)

    renderInteractor.Initialize()
    renderWin.Render()
    renderInteractor.Start()


if __name__ == '__main__':
    main()
相关推荐
小白快快跑哦10 分钟前
python-字符串全解(五):字符串格式化表达式以及格式化方法(2)
python·字符串格式化方法
专业程序开发源41 分钟前
springboot篮球联赛管理系统13635-计算机课程设计、毕业设计
vue.js·spring boot·后端·python·django·php·课程设计
lie..44 分钟前
30天从零开始学AI应用开发(Day 3):30 分钟搭好开发环境,Python + VSCode 一次配齐
人工智能·python·大模型
benchmark_cc1 小时前
策略临时需要一批股票的最新价格,先查行情还是先建股票池?——从量化策略执行逻辑看数据获取顺序
python·数据分析·pandas·量化交易·股票数据·quantdash
成为深度学习高手1 小时前
XLinear:用一个轻量 MLP,带外生变量做长期时间序列预测
python·深度学习·时序数据库
啊吧怪不啊吧1 小时前
LangChain之模型调用
python·langchain
溪语流沙1 小时前
Django + Vue电商项目第001讲:开篇|注册登录加增删改查,那不是电商
redis·python·mysql·docker·typescript·django·vue
夜雪一千1 小时前
如何使用BeautifulSoup库来提取HTML中的特定元素
python
IT毕设梦工厂2 小时前
计算机毕业设计选题推荐:基于大数据的城市空气污染物浓度数据分析与可视化|毕业设计选题|计算机毕设|选题推荐|毕设指导|项目定制|源码|高质量项目
大数据·python·信息可视化·数据挖掘·数据分析·课程设计·数据可视化
程序员阿黄2 小时前
基于 Django 与 Vue 3 的智能实验室预约系统设计与实现
后端·python·mysql·django·vue·毕设