【三維幾何學習】自制簡單的網格可視化軟件

自制簡單的網格可視化軟件 — Mesh Visualization

引言
一、整體框架
- 1.1 三角形網格
- 1.2 界面管理
- 1.3 VTK可視化界面
二、核心源碼
- 2.1 三角形網格：TriMesh類
- 2.2 界面Widget_Mesh_Manager
- 2.3 VTK可視化
- 2.4 main

引言

使用PyQt自制簡單的網格可視化軟件 - 視頻展示

本是為了方便輸入模型分析、網格分割結果可視化，使用PyQt做的一款小軟件，后續通過增刪變成了Mesh Visualization，主要針對三角形網格。主要功能包括：讀取網格(目前僅支持.obj)，關于網格頂點、邊、面的一些可視化小操作(是否顯示、更改顏色等)、比較簡單的燈光以及背景設置、一些基本功能(模型列表、操作日志等)。

一、整體框架

在這里插入圖片描述

MainWindow：主窗體。包含縮小、放大、關閉、菜單欄等
Mesh：三角網格，包含網格的讀取、保存、網格的一些基本屬性
Resource：資源文件夾。包含界面樣式和圖片
Widget_Mesh_Manager：界面管理，內含多個子控件。包含模型列表、網格信息顯示界面、日志界面
Widget_Vtk：模型渲染界面，就是VTK渲染窗口。包含VTK顯示代碼actor、mapper、light等

關于界面 統一采用一個界面三個文件：

xxx.ui (可編輯的界面，pyqt插件可打開)
ui_xxx.py (ui界面轉的py界面文件) 參數設置 $FileName$ -o ui_$FileNameWithoutExtension$.py -x
xxx.py (主要在這里寫代碼)

1.1 三角形網格

三角形網格Mesh文件夾中含有兩個文件：

load_and_save.py 讀取保存三角形網格，目前只支持obj文件，可用現有的庫代替 /todo
TriMesh.py 三角形網格類，保存三角形網格的文件名、格式、頂點、面片等信息，每個網格都有一個獨立vtk actor，方便操作以及顯示

1.2 界面管理

模型列表：顯示打開的每一個模型，可以進行模型間的切換顯示
模型信息：包括頂點和面片數量，可修改點、邊、面以及燈光顏色
日志顯示：記錄每一步操作，但目前只顯示部分操作 /todo

1.3 VTK可視化界面

主要用于模型渲染顯示 (self.vtk_widget = QVTKRenderWindowInteractor(self))：

mapper 映射器，將數據轉為圖形數據
renderer 渲染器，將三維圖形轉為二維圖片
light 燈光，只設置了一個燈光 /todo
交互方式 vtk.vtkInteractorStyleTrackballCamera()

其中actor每個三角形網格一個，方便單獨操作

二、核心源碼

2.1 三角形網格：TriMesh類

import ntpath
import numpy as np
from scipy.sparse import csr_matrix
from Mesh.load_and_save import load_obj, load_obj_with_edges
import vtkmodules.all as vtkclass TriMesh:# 0.文件filename = None  # 文件名path = None  # 路徑file = None  # 完整路徑format = None  # 格式# 1.基本屬性vs = None     # 頂點faces = None  # 面片name = None   # 自定義名稱actor = None  # 保存可視化的數據# 2.進階屬性point_adjacency_matrix = None  # 點的鄰接矩陣edges = Noneedge_labels = Noneedge_actor = Nonedef __init__(self, file=None, mode=None):# 賦值self.path = ntpath.split(file)[0]self.filename = ntpath.split(file)[1]self.file = fileself.format = self.filename.split('.')[-1]self.actor = vtk.vtkActor()self.edge_actor = vtk.vtkActor()# 讀取if self.format == 'obj':if mode == 1:self.vs, self.faces, self.edges, self.edge_labels = load_obj_with_edges(file)else:self.vs, self.faces = load_obj(file)else:print('Unsupported format')return# 計算點鄰接矩陣self.point_adjacency_matrix = self.computer_point_adjacency_matrix()def computer_point_adjacency_matrix(self):num = len(self.vs)row = np.hstack([self.faces[:, 0], self.faces[:, 1], self.faces[:, 2]])col = np.hstack([self.faces[:, 1], self.faces[:, 2], self.faces[:, 0]])value = 0 * row + 1point_adjacency_matrix = csr_matrix((value, (row, col)), shape=(num, num)).toarray()return point_adjacency_matrixdef boundary_edge(self):edge_cnt = self.point_adjacency_matrix + self.point_adjacency_matrix.Ttwo_point = np.where(edge_cnt == 1)return two_pointdef creat_edges(self):edge2key = dict()edges = []for face_id, face in enumerate(self.faces):faces_edges = []for i in range(3):cur_edge = (face[i], face[(i + 1) % 3])faces_edges.append(cur_edge)for idx, edge in enumerate(faces_edges):edge = tuple(sorted(list(edge)))faces_edges[idx] = edgeif edge not in edge2key:edge2key[edge] = 1edges.append(list(edge))self.edges = np.array(edges, dtype=np.int32)if __name__ == '__main__':cs = TriMesh('../00ceshi/1.obj')# print(cs.point_adjacency_matrix)point = cs.boundary_edge()print(1)

2.2 界面Widget_Mesh_Manager

from PyQt5.QtCore import *
from PyQt5.QtWidgets import *
from Widget_Mesh_Manager.ui_Widget_Mesh_Manager import Ui_Widget_Mesh_Manager
import datetime
from Mesh.TriMesh import TriMesh
import numpy as npclass Widget_Mesh_Manager(QWidget, Ui_Widget_Mesh_Manager):TriMesh_list = []mesh_show = pyqtSignal(int)def __init__(self, parent=None):super(Widget_Mesh_Manager, self).__init__(parent)self.setupUi(self)# 滾動條設置self.hSlider_pointSize.setMinimum(1)       # 點self.hSlider_pointSize.setMaximum(9)self.hSlider_pointSize.setSingleStep(1)# self.hSlider_pointSize.setTickInterval(2)     # 帶有樣式 不顯示刻度# self.hSlider_pointSize.setTickPosition(QSlider.TicksBelow)self.hSlider_edgeSize.setMinimum(1)        # 邊self.hSlider_edgeSize.setMaximum(9)self.hSlider_edgeSize.setSingleStep(1)self.hSlider_lightIntensity.setMinimum(0)  # 燈光self.hSlider_lightIntensity.setMaximum(10)self.hSlider_lightIntensity.setSingleStep(1)# 按鈕點擊函數self.initBtn()# UI布局self.initUI()def initBtn(self):passdef initUI(self):# tableWidgetself.tableWidget.setColumnCount(1)  # 列數self.tableWidget.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)  # 所有列自動拉伸，充滿界面self.tableWidget.setSelectionMode(QAbstractItemView.SingleSelection)           # 設置只能選中一行self.tableWidget.setEditTriggers(QTableView.NoEditTriggers)                    # 不可編輯self.tableWidget.setSelectionBehavior(QAbstractItemView.SelectRows)            # 設置只有行選中self.tableWidget.verticalHeader().setVisible(False)         # 隱藏列表頭self.tableWidget.horizontalHeader().setVisible(False)       # 隱藏行def addMesh(self, mesh: TriMesh):# 添加到listself.TriMesh_list.append(mesh)# 添加到uirow = self.tableWidget.rowCount()self.tableWidget.insertRow(row)item = QTableWidgetItem(mesh.filename)self.tableWidget.setItem(row, 0, item)self.tableWidget.clearFocus()self.tableWidget.selectRow(row)     # 新加入的行被選中# item.setSelected(True)# 刷新Infoself.showMesh_Info(row)def addLog(self, info):time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S: ")self.textEdit.append(time + info)def showMesh_Info(self, mesh_id):mesh = self.TriMesh_list[mesh_id]self.groupBox.setTitle(mesh.filename)self.label_vs_num.setText('頂點個數：' + str(len(mesh.vs)))self.label_face_num.setText('面片個數：' + str(len(mesh.faces)))self.mesh_show.emit(mesh_id)# 重寫tableWidget的點擊事件def on_tableWidget_cellClicked(self, row, col):self.showMesh_Info(row)pass

2.3 VTK可視化

from PyQt5.QtWidgets import *
from Widget_Vtk.ui_Widget_Vtk import Ui_Widget_Vtk
from Mesh.TriMesh import TriMesh
import numpy as np
from vtkmodules.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
import vtkmodules.all as vtkclass Widget_Vtk(QWidget, Ui_Widget_Vtk):mapper = None    # 映射器 數據actor = None     # 演員 get mapperrenderer = None  # 渲染light = None     # 燈光 只有一個def __init__(self, parent=None):super(Widget_Vtk, self).__init__(parent)self.setupUi(self)#self.vtk_widget = QVTKRenderWindowInteractor(self)self.colors = np.array([[0, 0, 255], [0, 255, 255], [255, 0, 255], [0, 255, 0],[255, 255, 0], [255, 0, 0], [100, 180, 51], [255, 150, 51]])self.colors = np.array([[80, 136, 240], [0, 255, 255], [255, 0, 255], [0, 255, 0],[255, 255, 0], [255, 0, 0], [70, 70, 220], [255, 150, 51], [0,0,0]])self.colors1 = np.array([[180, 90, 90], [121, 185, 128], [90, 90, 180], [180, 180, 0],[255, 255, 0], [255, 0, 0], [100, 255, 51], [255, 150, 51]])# 初始化renderself.renderer = vtk.vtkRenderer()self.renderer.SetBackground(1, 1, 1)self.vtk_widget.GetRenderWindow().AddRenderer(self.renderer)self.interactor = self.vtk_widget.GetRenderWindow().GetInteractor()interactor_style = vtk.vtkInteractorStyleTrackballCamera()self.interactor.SetInteractorStyle(interactor_style)self.interactor.Initialize()# 初始化lightself.light = vtk.vtkLight()self.light.SwitchOff()      # 默認關閉self.renderer.AddLight(self.light)self.vtk_widget.Render()self.vtk_widget.Start()self.renderer.ResetCamera()self.vtk_widget.update()def resizeEvent(self, e):"""重寫窗口移動事件"""self.vtk_widget.resize(self.size())def removeAllActors(self):al = self.renderer.GetActors()n = al.GetNumberOfItems()al.InitTraversal()for i in range(n):actor = al.GetNextActor()self.renderer.RemoveActor(actor)self.vtk_widget.update()def showTrimesh(self, mesh: TriMesh):# 添加點points = vtk.vtkPoints()for v in mesh.vs:points.InsertNextPoint(v)# 添加面片polys = vtk.vtkCellArray()for f in mesh.faces:polys.InsertNextCell(len(f), f)# 創建PolyDatacube = vtk.vtkPolyData()cube.SetPoints(points)cube.SetPolys(polys)# 創建 mapper 和 actormapper = vtk.vtkPolyDataMapper()mapper.ScalarVisibilityOff()mapper.SetInputData(cube)mesh.actor.SetMapper(mapper)mesh.actor.GetProperty().SetColor([0.5, 0.5, 0.5])self.renderer.AddActor(mesh.actor)self.renderer.ResetCamera()self.vtk_widget.update()def show_boundary_edge(self, mesh: TriMesh):# 添加點points = vtk.vtkPoints()for v in mesh.vs:points.InsertNextPoint(v)pa, pb = mesh.boundary_edge()# 添加邊edges = vtk.vtkCellArray()for i in range(len(pa)):edges.InsertNextCell(2, np.array([pa[i], pb[i]]))# 添加邊界mapper2 = vtk.vtkPolyDataMapper()cube1 = vtk.vtkPolyData()cube1.SetPoints(points)cube1.SetLines(edges)mapper2.SetInputData(cube1)mesh.edge_actor.SetMapper(mapper2)# mesh.edge_actor.GetProperty().SetEdgeColor(0, 0, 1)# mesh.edge_actor.GetProperty().SetEdgeVisibility(1)mesh.edge_actor.GetProperty().SetLineWidth(2)self.renderer.AddActor(mesh.edge_actor)self.vtk_widget.update()def show_mesh_color(self, mesh: TriMesh):# 添加點points = vtk.vtkPoints()for v in mesh.vs:points.InsertNextPoint(v)# 添加面片polys = vtk.vtkCellArray()cellColor = vtk.vtkFloatArray()for f in mesh.faces:polys.InsertNextCell(len(f), f)cellColor.InsertNextValue(mesh.vs[f[0]][0])# 創建PolyDatacube = vtk.vtkPolyData()cube.SetPoints(points)cube.SetPolys(polys)cube.GetCellData().SetScalars(cellColor)# 創建 mapper 和 actorself.mapper = vtk.vtkPolyDataMapper()self.mapper.SetScalarRange(min(mesh.vs[:, 0]), max(mesh.vs[:, 0]))self.mapper.SetInputData(cube)mesh.actor.SetMapper(self.mapper)mesh.actor.GetProperty().SetEdgeVisibility(1)self.vtk_widget.update()def show_point_color(self, mesh: TriMesh, seg=[]):# 添加點points = vtk.vtkPoints()pColor = vtk.vtkFloatArray()for v in mesh.vs:points.InsertNextPoint(v)# id = np.random.randint(0, 2)pColor.InsertNextValue(v[0])if len(seg) > 0:pColor = vtk.vtkFloatArray()for s in seg:pColor.InsertNextValue(s)# 添加面片polys = vtk.vtkCellArray()for f in mesh.faces:polys.InsertNextCell(len(f), f)# 創建PolyDatacube = vtk.vtkPolyData()cube.SetPoints(points)cube.SetPolys(polys)cube.GetPointData().SetScalars(pColor)mapper = vtk.vtkPolyDataMapper()mapper.SetScalarRange(min(mesh.vs[:, 2]), max(mesh.vs[:, 2]))mapper.SetInputData(cube)mesh.actor.SetMapper(mapper)mesh.actor.GetProperty().SetColor([0.5, 0.5, 0.5])self.vtk_widget.update()def show_mesh_seg(self, mesh: TriMesh, seg):# 添加點points = vtk.vtkPoints()for v in mesh.vs:points.InsertNextPoint(v)# 添加面片polys = vtk.vtkCellArray()cellColor = vtk.vtkUnsignedCharArray()cellColor.SetNumberOfComponents(3)for f in mesh.faces:polys.InsertNextCell(len(f), f)for s in seg:c = self.colors[s]cellColor.InsertNextTuple(c)# 創建PolyDatacube = vtk.vtkPolyData()cube.SetPoints(points)cube.SetPolys(polys)cube.GetCellData().SetScalars(cellColor)# 創建 mappermapper = vtk.vtkPolyDataMapper()mapper.SetColorModeToDefault()  # 需要設置為默認顏色Modemapper.SetInputData(cube)mesh.actor.SetMapper(mapper)self.vtk_widget.update()def show_points(self, mesh: TriMesh):# 添加點points = vtk.vtkPoints()for v in mesh.vs:points.InsertNextPoint(v)vs = vtk.vtkPolyData()vs.SetPoints(points)# 生成頂點vertex = vtk.vtkVertexGlyphFilter()vertex.SetInputData(vs)# 創建 mapper 和 actormapper = vtk.vtkPolyDataMapper()mapper.SetInputConnection(vertex.GetOutputPort())mesh.actor.SetMapper(mapper)mesh.actor.GetProperty().SetColor([0, 0, 0])mesh.actor.GetProperty().SetPointSize(5)self.vtk_widget.update()def saveToimage(self):from vtkmodules.vtkRenderingCore import vtkWindowToImageFilterfrom vtkmodules.vtkIOImage import (vtkBMPWriter,vtkJPEGWriter,vtkPNGWriter,vtkPNMWriter,vtkPostScriptWriter,vtkTIFFWriter)im_filter = vtkWindowToImageFilter()                   #im_filter.SetInput(self.vtk_widget.GetRenderWindow())  # QVTKRenderWindowInteractorim_filter.SetScale(3)                                  #im_filter.SetInputBufferTypeToRGBA()writer = vtkBMPWriter()#writer = vtkPostScriptWriter()writer.SetFileName('cs.bmp')#writer.SetFileName('cs.ps')writer.SetInputConnection(im_filter.GetOutputPort())writer.Write()def showTrimesh_with_edge(self, mesh: TriMesh):# 添加點points = vtk.vtkPoints()for v in mesh.vs:points.InsertNextPoint(v)# 添加面片c = 255face_color = [c, c, c]polys = vtk.vtkCellArray()cellColor = vtk.vtkUnsignedCharArray()cellColor.SetNumberOfComponents(3)for f in mesh.faces:polys.InsertNextCell(len(f), f)cellColor.InsertNextTuple(face_color)# 添加邊 和 邊的顏色edges = vtk.vtkCellArray()edge_colors = vtk.vtkUnsignedCharArray()edge_colors.SetNumberOfComponents(3)for e in mesh.edges:edges.InsertNextCell(2, e)for edge_labels in mesh.edge_labels:# edge_colors.InsertNextTuple(self.colors[edge_labels])  # colors1edge_colors.InsertNextTuple(self.colors[edge_labels])# 創建PolyDatacube = vtk.vtkPolyData()cube.SetPoints(points)cube.SetPolys(polys)# cube.SetLines(edges)cube.GetCellData().SetScalars(cellColor)# 細分loop = vtk.vtkLoopSubdivisionFilter()# loop = vtk.vtkButterflySubdivisionFilter()# loop = vtk.vtkLinearSubdivisionFilter()loop.SetInputData(cube)loop.SetNumberOfSubdivisions(0)loop.Update()print(loop.GetOutput().GetNumberOfPolys())# 平滑smoothFilter = vtk.vtkSmoothPolyDataFilter()# smoothFilter = vtk.vtkWindowedSincPolyDataFilter()smoothFilter.SetInputConnection(loop.GetOutputPort())# smoothFilter.SetInputData(cube)smoothFilter.SetNumberOfIterations(1)  # 控制平滑次數，次數越大平滑越厲害smoothFilter.Update()# 創建 mapper 和 actorself.mapper = vtk.vtkPolyDataMapper()self.mapper.SetColorModeToDefault()     # 需要設置為默認顏色Mode# self.mapper.SetInputData(cube)# self.mapper.SetInputConnection(loop.GetOutputPort())self.mapper.SetInputConnection(smoothFilter.GetOutputPort())print(smoothFilter.GetOutput().GetNumberOfPolys())self.actor = vtk.vtkActor()self.actor.SetMapper(self.mapper)# 添加邊self.mapper2 = vtk.vtkPolyDataMapper()cube1 = vtk.vtkPolyData()cube1.SetPoints(points)# cube1.SetPolys(edges)cube1.SetLines(edges)cube1.GetCellData().SetScalars(edge_colors)self.mapper2.SetInputData(cube1)self.actor2 = vtk.vtkActor()self.actor2.SetMapper(self.mapper2)# self.actor2.GetProperty().SetEdgeColor(0, 0, 1)self.actor2.GetProperty().SetEdgeVisibility(1)self.actor2.GetProperty().SetLineWidth(4)# 顯示self.vtk_widget.Render()self.vtk_widget.Start()# self.renderer.self.renderer = vtk.vtkRenderer()self.renderer.SetBackground(0.7, 0.7, 0.7)self.renderer.SetBackground2(1, 1, 1)self.renderer.SetGradientBackground(1)# 燈光myLight = vtk.vtkLight()# myLight.SetColor(1, 1, 1)  # 設置燈光的顏色，以RGB的形式指定顏色# myLight.SetPosition(100, 100, 100)  # 設燈光照位置myLight.SetLightType(vtk.VTK_LIGHT_TYPE_HEADLIGHT)#vtk.VTK_LIGHT_TYPE_SCENE_LIGHT#vtk.VTK_LIGHT_TYPE_CAMERA_LIGHT# myLight.SetFocalPoint(self.renderer.GetActiveCamera().GetFocalPoint())  # 設置燈光焦點myLight.SetIntensity(1)      # 可視化邊的時候設置為0.9# myLight.SwitchOff()self.renderer.AddLight(myLight)  # 因為renderer里可以有多個燈光，所以是AddLight() 而不是 SetLight()# self.renderer.Setself.vtk_widget.GetRenderWindow().AddRenderer(self.renderer)self.renderer.AddActor(self.actor)self.renderer.AddActor(self.actor2)self.renderer.ResetCamera()self.vtk_widget.update()def cs(self):# 三角網格vs = np.array([[-1, 1, -0.5],[-1, 0, 0],[-1, -1, -0.5],[0, 0.3, 0],[0, -0.3, 0],[0.5, 0, 0.5]], dtype=np.float32)  # 0.5 -0.5faces = np.array([[4, 1, 3], [4, 1, 2], [0, 3, 1], [3, 5, 4]], dtype=np.int16)# 顏色c = np.array([[0, 1, 1], [0.5, 1, 0.5], [0.5, 1, 0.5], [1, 1, 0.5]]) * 255# 添加點points = vtk.vtkPoints()for v in vs:points.InsertNextPoint(v)# 添加面片polys = vtk.vtkCellArray()for f in faces:polys.InsertNextCell(3, f)cellColor = vtk.vtkUnsignedCharArray()cellColor.SetNumberOfComponents(3)for tmp in c:cellColor.InsertNextTuple(tmp)# 創建PolyDatacube = vtk.vtkPolyData()cube.SetPoints(points)cube.SetPolys(polys)cube.GetCellData().SetScalars(cellColor)# 細分l = vtk.vtkLinearSubdivisionFilter()l.SetInputData(cube)l.SetNumberOfSubdivisions(1)l.Update()loop = vtk.vtkLoopSubdivisionFilter()#loop.SetInputData(l.GetOutputPort())loop.SetInputConnection(l.GetOutputPort())loop.SetNumberOfSubdivisions(5)loop.Update()# 創建Mappermapper = vtk.vtkPolyDataMapper()mapper.SetColorModeToDefault()mapper.SetInputData(cube)mapper.SetInputConnection(loop.GetOutputPort())# 創建actoractor = vtk.vtkActor()actor.SetMapper(mapper)# actor.GetProperty().SetColor([1, 1, 1])# actor.GetProperty().SetEdgeColor(0, 0, 0)actor.GetProperty().SetEdgeVisibility(0)# 燈光myLight = vtk.vtkLight()myLight.SetColor(1, 1, 1)  # 設置燈光的顏色，以RGB的形式指定顏色myLight.SetPosition(0, 0, 1)myLight.SetLightType(vtk.VTK_LIGHT_TYPE_SCENE_LIGHT)# vtk.VTK_LIGHT_TYPE_SCENE_LIGHT# vtk.VTK_LIGHT_TYPE_CAMERA_LIGHTmyLight.SetFocalPoint(self.renderer.GetActiveCamera().GetFocalPoint())  # 設置燈光焦點myLight.SetIntensity(0.5)  # 可視化邊的時候設置為0.9# myLight.SwitchOff()self.renderer.AddLight(myLight)  # 因為renderer里可以有多個燈光，所以是AddLight() 而不是 SetLight()self.renderer.AddActor(actor)self.renderer.ResetCamera()self.vtk_widget.update()

2.4 main

from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from MainWindow.MainWindow import MainWindow
import sysif __name__ == '__main__':app = QApplication(sys.argv)# 加載圖標app.setWindowIcon(QIcon('./Resource/rabbit.ico'))# 加載樣式s = './Resource/blue.css'with open(s, "r") as f:app.setPalette(QPalette(QColor('#EAF7FF')))app.setStyleSheet(f.read())# 顯示mainWindow = MainWindow()mainWindow.show()sys.exit(app.exec_())