本篇文章将介绍如何读取fluent格式的网格和流场并显示。

Fluent是国际上比较流行的商用CFD软件包，在美国的市场占有率为60%，凡是和流体、热传递和化学反应等有关的工业均可使用。它具有丰富的物理模型、先进的数值方法和强大的前后处理功能，在航空航天、汽车设计、石油天然气和涡轮机设计等方面都有着广泛的应用。

百度百科baike.baidu.com/item/fluent/1269452

代码及算例摘抄自硫酸亚铜的博客。
文章链接: 使用VTK对Fluent计算结果进行后处理

环境

VTK : 9.1.0
Python : 3.6.5

功能

读取fluent网格及流场(cas和dat)
显示几何
显示网格
读取显示流场
切片操作

效果图

代码

try:
    import vtkmodules.all as vtk
except ImportError:
    import vtk
r = vtk.vtkFLUENTReader()
r.SetFileName(r'fluent.cas')
r.EnableAllCellArrays()
r.Update()

data = r.GetOutput().GetBlock(0).GetCellData()
arrays_number = data.GetNumberOfArrays()
for i in range(arrays_number):
    array_name = data.GetArrayName(i)
    print(array_name)
NAME = 'PRESSURE'
scalars = vtk.vtkFloatArray()
scalars = r.GetOutput().GetBlock(0).GetCellData().GetArray(NAME)
g = vtk.vtkGeometryFilter()
g.SetInputConnection(r.GetOutputPort())
lut = vtk.vtkLookupTable()
lut.SetHueRange(0.667, 0)
lut.Build()
########################################################
# 创建一个切片
plane = vtk.vtkPlane()
plane.SetOrigin(0, 0, 0)  # 切片通过的点
plane.SetNormal(1, 0, 0)  # 切片的法向量
########################################################
########################################################
# 创建一个Cutter的Filter
cutter = vtk.vtkCutter()
cutter.SetCutFunction(plane)
cutter.SetInputConnection(r.GetOutputPort())
cutter.Update()
########################################################

########################################################
# 显示几何
GeoMapper = vtk.vtkCompositePolyDataMapper()
GeoMapper.SetInputConnection(g.GetOutputPort())
GeoActor = vtk.vtkActor()
GeoActor.SetMapper(GeoMapper)
GeoRen = vtk.vtkRenderer()
GeoRen.SetViewport(0.0, 0.5, 0.5, 1.0)
GeoRen.GradientBackgroundOn()
GeoRen.SetBackground(1, 1, 1)
GeoRen.SetBackground2(0.4, 0.55, 0.75)
GeoRen.AddActor(GeoActor)
##########################################################

##########################################################
# 显示网格
# MeshMapper = vtk.vtkCompositePolyDataMapper2()
MeshMapper = vtk.vtkCompositePolyDataMapper()
MeshMapper.SetInputConnection(g.GetOutputPort())
MeshActor = vtk.vtkActor()
MeshActor.SetMapper(MeshMapper)
MeshProp = MeshActor.GetProperty()
MeshProp.EdgeVisibilityOn()
MeshRen = vtk.vtkRenderer()
MeshRen.SetViewport(0.5, 0.5, 1.0, 1.0)
MeshRen.GradientBackgroundOn()
MeshRen.SetBackground(1, 1, 1)
MeshRen.SetBackground2(0.4, 0.55, 0.75)
MeshRen.AddActor(MeshActor)
##########################################################

##########################################################
# 显示温度云图
# FluentMapper = vtk.vtkCompositePolyDataMapper2()
TempMapper = vtk.vtkCompositePolyDataMapper()
TempMapper.SetInputConnection(g.GetOutputPort())
TempMapper.SetScalarModeToUseCellFieldData()
TempMapper.SelectColorArray(NAME)
TempMapper.SetScalarRange(scalars.GetRange())
TempMapper.SetLookupTable(lut)
TempActor = vtk.vtkActor()
TempActor.SetMapper(TempMapper)
TempRen = vtk.vtkRenderer()
TempRen.SetViewport(0.0, 0.0, 0.5, 0.5)
TempRen.GradientBackgroundOn()
TempRen.SetBackground(1, 1, 1)
TempRen.SetBackground2(0.4, 0.55, 0.75)
TempRen.AddActor(TempActor)
##########################################################

##########################################################
# 显示切片云图
# SliceMapper = vtk.vtkCompositePolyDataMapper2()
SliceMapper = vtk.vtkCompositePolyDataMapper()
SliceMapper.SetInputConnection(cutter.GetOutputPort())
SliceMapper.SetScalarModeToUseCellFieldData()
SliceMapper.SelectColorArray(NAME)
SliceMapper.SetScalarRange(scalars.GetRange())
SliceMapper.SetLookupTable(lut)
SliceActor = vtk.vtkActor()
SliceActor.SetMapper(SliceMapper)
SliceRen = vtk.vtkRenderer()
SliceRen.SetViewport(0.5, 0.0, 1.0, 0.5)
SliceRen.GradientBackgroundOn()
SliceRen.SetBackground(1, 1, 1)
SliceRen.SetBackground2(0.4, 0.55, 0.75)
SliceRen.AddActor(SliceActor)
##########################################################

##########################################################
# 图例
legendBar = vtk.vtkScalarBarActor()
legendBar.SetOrientationToVertical()
legendBar.SetLookupTable(lut)
legenProp = vtk.vtkTextProperty()
legenProp.SetColor(0, 0, 0)
legenProp.SetFontSize(20)
legenProp.SetFontFamilyToArial()
legenProp.ItalicOff()
legenProp.BoldOff()
legendBar.UnconstrainedFontSizeOn()
legendBar.SetTitleTextProperty(legenProp)
legendBar.SetLabelTextProperty(legenProp)
legendBar.SetLabelFormat("%5.2f")
legendBar.SetTitle("tempereture")
TempRen.AddActor(legendBar)
SliceRen.AddActor(legendBar)
##########################################################

renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(GeoRen)
renWin.AddRenderer(MeshRen)
renWin.AddRenderer(TempRen)
renWin.AddRenderer(SliceRen)
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)

renWin.SetSize(300, 300)
renWin.Render()

iren.Initialize()
iren.Start()