3D Slicer可视化代码

以下是一个使用3D Slicer进行可视化的简单示例代码：

import numpy as np
import slicer

# 创建一个 vtkMRMLScene 对象
scene = slicer.mrmlScene

# 创建一个 vtkImageData 对象
imageData = vtk.vtkImageData()
imageData.SetDimensions(100, 100, 100)
imageData.AllocateScalars(vtk.VTK_DOUBLE, 1)

# 将图像数据填充为一个球体
for i in range(100):
    for j in range(100):
        for k in range(100):
            x = i - 50
            y = j - 50
            z = k - 50
            if x**2 + y**2 + z**2 < 2500:
                imageData.SetScalarComponentFromDouble(i, j, k, 0, 1.0)

# 创建一个 vtkMRMLScalarVolumeNode 对象，并将图像数据添加到该节点中
volumeNode = slicer.vtkMRMLScalarVolumeNode()
volumeNode.SetName('Sphere')
volumeNode.SetSpacing(1.0, 1.0, 1.0)
volumeNode.SetImageData(imageData)
scene.AddNode(volumeNode)

# 创建一个 vtkMRMLSliceNode 对象，并将其添加到场景中
sliceNode = slicer.vtkMRMLSliceNode()
sliceNode.SetName('Slice')
sliceNode.SetOrientationToAxial()
sliceNode.SetSliceOffset(50)
scene.AddNode(sliceNode)

# 创建一个 vtkMRMLSliceCompositeNode 对象，并将其添加到场景中
compositeNode = slicer.vtkMRMLSliceCompositeNode()
compositeNode.SetName('Composite')
compositeNode.SetForegroundOpacity(0.5)
scene.AddNode(compositeNode)

# 创建一个 vtkMRMLSliceCompositeWidget 对象，并将其添加到界面中
sliceWidget = slicer.modules.slice.widgetRepresentation().self()
sliceWidget.setMRMLScene(scene)
sliceWidget.setMRMLSliceNode(sliceNode)
sliceWidget.setMRMLSliceCompositeNode(compositeNode)

# 将体积节点添加到切片中
displayNode = slicer.vtkMRMLScalarVolumeDisplayNode()
displayNode.SetAutoWindowLevel(0)
displayNode.SetWindowLevelMinMax(-1000, 1000)
scene.AddNode(displayNode)
volumeNode.SetAndObserveDisplayNodeID(displayNode.GetID())
sliceNode.SetAndObserveVolumeNodeID(volumeNode.GetID())

这段代码创建了一个球体，并将其添加到3D Slicer中。它还创建了一个切片窗口，将球体添加到该窗口中，以便用户可以查看球体的切片。注意，该代码仅为示例，可以根据实际需求进行修改和扩展。