tvtk管线技术、数据集与数据加载

from tvtk.api import tvtk
from tvtk.tools import ivtk
from pyface.api import GUI
s=tvtk.CubeSource(x_length=1.0,y_length=2.0,z_length=3.0)
m=tvtk.PolyDataMapper(input_cOnnection=s.output_port)
a=tvtk.Actor(mapper=m)
#创建一个带Crust(Python Shell)的窗口
gui=GUI()
win=ivtk.IVTKWithCrustAndBrowser()
win.open()
win.scene.add_actor(a)
"""
 运行之后左侧窗口出现空白，这个是一个小小的bug，后续将会修复它
这个bug如何产生的呢？
在vtk中引入了GUI模块。
在GUI开启的时候它的子窗口Dialogue其实是个子窗口空间，应存在于主窗口之中
因为这个bug的存在，使它脱离了这个窗口
"""
#修正错误
dialog=win.control.centralWidget().widget(0).widget(0)
from pyface.qt import QtCore
dialog.setWindowFlags(QtCore.Qt.WindowFlags(0x00000000))
dialog.show()
#开始界面消息循环
gui.start_event_loop()

from tvtk.api import tvtk
'''可以把以下两个函数分别写到一个文件中
在开头from tvtkfunc import ivtk_scene,event_loop
使代码变得简洁
'''
def ivtk_scene(actors):
    from tvtk.tools import ivtk
    #创建一个带Crust(Python Shell)的窗口
    win=ivtk.IVTKWithCrustAndBrowser()
    win.open()
    win.scene.add_actor(actors)
    #修正错误
    dialog=win.control.centralWidget().widget(0).widget(0)
    from pyface.qt import QtCore
    dialog.setWindowFlags(QtCore.Qt.WindowFlags(0x00000000))
    dialog.show()
    return win
def event_loop():
#开始界面消息循环
    from pyface.api import GUI
    gui=GUI()
    gui.start_event_loop()
s=tvtk.CubeSource(x_length=1.0,y_length=2.0,z_length=3.0)
m=tvtk.PolyDataMapper(input_cOnnection=s.output_port)
a=tvtk.Actor(mapper=m)
win=ivtk_scene(a)
win.scene.isometric_view()
event_loop(数据集(Dataset)

#构建ImageData数据集
from tvtk.api import tvtk
img=tvtk.ImageData(spacing=(1,1,1),origin=(1,2,3),dimensiOns=(3,4,5))
#img.get_point(0)获得网格中第一个点的坐标值
print("ImageData数据集:")
for n in range(6):#输出image对象的所有点
    print("%.1f,%.1f,%.1f" % img.get_point(n))
#RectilinearGrid使用
#（因为不均匀，所以网格对象上的所有点表示对象交点）
import numpy as np 
x=np.array([0,3,9,15])#实现网格的构建
y=np.array([0,1,5])
z=np.array([0,2,3])
r=tvtk.RectilinearGrid()#因为不会根据属性调整数组长度，所以需要设置
r.x_coordinates = x 
r.y_coordinates = y 
r.z_coordinates = z 
r.dimensions = len(x),len(y),len(z)
print("RectilinearGrid数据集:")
for n in range(6):
    print(r.get_point(n))

from tvtk.api import tvtk
from tvtkfunc import ivtk_scene,event_loop
#之前已经把函数ivtk_scene,event_loop封装到tvtkfunc.py中

s=tvtk.STLReader(file_name="python.stl")
m=tvtk.PolyDataMapper(input_cOnnection= s.output_port)
a=tvtk.Actor(mapper=m)

win=ivtk_scene(a)
win.scene.isometric_view()
event_loop()

'''tvtk读取Plot3D文件
    使用MultiBlock数据读取
Plot3D文件分为：
    网格文件(XYZ文件)
    空气动力学文件(Q文件)
    通用结果文件  
'''
from tvtk.api import tvtk
def read_data():#读入数据
    #MultiBolck调用形式
    plot3d=tvtk.MultiBlockPLOT3DReader(
        xyz_file_name="combxyz.bin",#网格文件
        q_file_name="combq.bin",#空气动力学结果文件
        scalar_function_number=100,#设置标量数据数量
        vector_function_number=200#设置矢量数据数量
        )
    plot3d.update()
    return plot3d 
plot3d = read_data()
grid = plot3d.output.get_block(0)#为查看细节获取网格数据集