快捷導(dǎo)航

Python?VTK映射三維模型表面距離

更新時間：2022年04月18日 13:07:41 作者：派大大大星

這篇文章主要介紹了Python?VTK映射三維模型表面距離，通過如何使用VTK讀取圖片計算兩個三維模型(stl)的表面距離，并將其距離值映射到模型上展開主題,需要的朋友可以參考一下

數(shù)據(jù)準(zhǔn)備： 需要準(zhǔn)備兩個stl文件、Python需要安裝vtk庫

步驟一：數(shù)據(jù)讀取 首先通過vtk.vtkSTLReader() 定義stl文件讀取接口，再通過reader1.GetOutput() 就可以獲得stl在vtk工作流的數(shù)據(jù)。

步驟二：去除重復(fù)點(diǎn) 通過vtk.vtkCleanPolyData() 可以去除模型中的重復(fù)點(diǎn)

步驟三：計算距離 使用 vtk.vtkDistancePolyDataFilter() ，使用上一步中過濾掉重復(fù)點(diǎn)后的數(shù)據(jù)作為輸入。如distanceFilter.SetInputConnection(1, clean1.GetOutputPort())，其中第一個參數(shù)就是輸入數(shù)據(jù)的標(biāo)號，從0開始計數(shù)；第二個參數(shù)就是輸入的數(shù)據(jù)。我們將vtkDistancePolyDataFilter的輸出到mapper就完成距離映射了。

步驟四：顏色配置 lut = vtk.vtkLookupTable() 相當(dāng)于一個調(diào)色盤函數(shù)，通過對其參數(shù)改變可以，調(diào)整最終映射的顏色范圍。 scalarBar = vtk.vtkScalarBarActor() 就是顏色條,按照前面的調(diào)色盤的結(jié)果將距離數(shù)值映射成顏色。

import vtk

input1 = vtk.vtkPolyData()
reader1 = vtk.vtkSTLReader()
reader1.SetFileName('model1.stl')
reader1.Update()
input1 = reader1.GetOutput()  # 讀取模型A

input2 = vtk.vtkPolyData()
reader2 = vtk.vtkSTLReader()
reader2.SetFileName('model2.stl')
reader2.Update()
input2 = reader2.GetOutput()  # 讀取模型B


# 數(shù)據(jù)合并，可以合并顯示兩個模型
clean1 = vtk.vtkCleanPolyData()
clean1.SetInputData(input1)

clean2 = vtk.vtkCleanPolyData()
clean2.SetInputData(input2)

distanceFilter = vtk.vtkDistancePolyDataFilter()

distanceFilter.SetInputConnection(1, clean1.GetOutputPort())
distanceFilter.SetInputConnection(0, clean2.GetOutputPort())
distanceFilter.SignedDistanceOff()
distanceFilter.Update()  # 計算距離
distanceFilter.GetOutputPort()
mapper = vtk.vtkPolyDataMapper()  # 配置mapper
mapper.SetInputConnection(distanceFilter.GetOutputPort())
mapper.SetScalarRange(  # 設(shè)置顏色映射范圍
    distanceFilter.GetOutput().GetPointData().GetScalars().GetRange()[0],
    distanceFilter.GetOutput().GetPointData().GetScalars().GetRange()[1])
actor = vtk.vtkActor()
actor.SetMapper(mapper)
actor1 = vtk.vtkActor()
actor1.SetMapper(mapper)
lut = vtk.vtkLookupTable()
lut.SetHueRange(0.2, 0.7)  # 映射的顏色變換參數(shù)（自己調(diào)顏色）
# lut.SetAlphaRange(1.0, 1.0)
# lut.SetValueRange(1.0, 1.0)
# lut.SetSaturationRange(1.0, 1.0)
# lut.SetNumberOfTableValues(256)
mapper.SetLookupTable(lut)
mapper2 = vtk.vtkPolyDataMapper()
mapper2.SetInputData((distanceFilter.GetSecondDistanceOutput()))
mapper2.SetScalarRange(  # 設(shè)置顏色映射范圍
    distanceFilter.GetSecondDistanceOutput().GetPointData().GetScalars().GetRange()[0],
    distanceFilter.GetSecondDistanceOutput().GetPointData().GetScalars().GetRange()[1])


actor2 = vtk.vtkActor()
actor2.SetMapper(mapper2)

scalarBar = vtk.vtkScalarBarActor()  # 設(shè)置color_bar
scalarBar.SetLookupTable(mapper.GetLookupTable())
scalarBar.SetTitle("SD(mm)")
scalarBar.SetNumberOfLabels(5)  # 設(shè)置要顯示的刻度標(biāo)簽數(shù)。自己設(shè)定色帶的位置
scalarBar.SetMaximumNumberOfColors(10)
# scalarBar.GetPositionCoordinate().SetCoordinateSystemToNormalizedViewport()
# scalarBar.GetPositionCoordinate().SetValue(0.01, 0.49)  # 參數(shù)越小越靠左，第二個參數(shù)越大越往上
# scalarBar.SetWidth(0.16)
# scalarBar.SetHeight(0.5)
# scalarBar.SetTextPositionToPrecedeScalarBar()  # 標(biāo)題和刻度標(biāo)記是否應(yīng)在標(biāo)量欄之前(文字會出現(xiàn)在條形左邊)
# # 設(shè)置標(biāo)題和條形之間的邊距
# scalarBar.SetVerticalTitleSeparation(10)
# # 設(shè)置標(biāo)題顏色
scalarBar.DrawTickLabelsOn()
scalarBar.GetTitleTextProperty().SetColor(0, 0, 0)
scalarBar.GetLabelTextProperty().SetColor(0, 0, 0)
arender = vtk.vtkRenderer()
arender.SetViewport(0, 0.0, 1, 1.0)
renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(arender)
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
style = vtk.vtkInteractorStyleTrackballActor()
iren.SetInteractorStyle(style)
aCamera = vtk.vtkCamera()
aCamera.SetViewUp(0, 0, -1)
aCamera.SetPosition(0, -1, 0)
aCamera.ComputeViewPlaneNormal()
aCamera.Azimuth(30.0)
aCamera.Elevation(30.0)
aCamera.Dolly(1.5)

arender.AddActor(actor)
# arender.AddActor(actor1)
arender.SetActiveCamera(aCamera)
arender.ResetCamera()
arender.SetBackground(1, 1, 1)
arender.ResetCameraClippingRange()
arender.AddActor2D(scalarBar)

renWin.Render()
iren.Initialize()
iren.Start()

結(jié)果示例：

1636355834(1).jpg