Open3D-GUI系列教程（四）菜单栏与文件拾取

LeonDL168

于 2024-09-07 16:19:19 发布

阅读量23

点赞数

分类专栏： Open3D 文章标签：人工智能计算机视觉算法图像处理视觉检测

原文链接：https://blog.csdn.net/qq_31254435/article/details/122913800

版权

Open3D 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

open3d版本：0.14.1

这里继续使用（三）中的代码框架，只将右侧面板改为了gui.CollapsableVert('panel',0, gui.Margins(0.25*em,0.25*em,0.25*em,0.25*em))。

这一部分内容涉及创建创建菜单、添加项，注册菜单项的事件和文件拾取等内容。

文章目录

菜单栏与文件拾取

1. 创建菜单栏

菜单中的每一项具有一个唯一的标识ID，首先定义几个标识ID：

class App:
    MENU_OPEN = 1
    MENU_SHOW = 5
    MENU_QUIT = 20
    MENU_ABOUT = 21

 
 
 
 
  
  '
  
  
   
   
    
    运行
    
    
     
     运行

接下来通过gui.Menu()创建菜单控件，调用add_item()将ID与菜单项关联。

add_item(str, id)

str: 显示文本

id: 菜单栏标识

if gui.Application.instance.menubar is None:
            # 文件菜单栏
            file_menu = gui.Menu()
            file_menu.add_item("Open",App.MENU_OPEN)
            file_menu.add_separator()
            file_menu.add_item("Quit",App.MENU_QUIT)

# 显示菜单栏
show_menu = gui.Menu()
show_menu.add_item(“Show Geometry”,App.MENU_SHOW)
## 将该项选中
show_menu.set_checked(App.MENU_SHOW,True)

# 帮助菜单栏
help_menu = gui.Menu()
help_menu.add_item(“About”,App.MENU_ABOUT)
## 将该项禁用
help_menu.set_enabled(App.MENU_ABOUT,False)

# 菜单栏
menu = gui.Menu()
menu.add_menu(“File”,file_menu)
menu.add_menu(“Show”,show_menu)
menu.add_menu(“Help”,help_menu)

gui.Application.instance.menubar = menu

2. 注册事件

菜单栏事件通过window.set_on_menu_item_activated(id, callbable)进行注册,将ID和回调函数关联。

w.set_on_menu_item_activated(App.MENU_OPEN,self._menu_open)
w.set_on_menu_item_activated(App.MENU_QUIT,self._menu_quit)
w.set_on_menu_item_activated(App.MENU_SHOW,self._menu_show)

3. 定义回调函数

主要有三个菜单事件要完成：

文件拾取（MENU_OPEN）
退出程序（MENU_QUIT）
切换显示（MENU_SHOW）

3.1 文件拾取：

当点击菜单栏中的Open时，我们希望打开一个文件拾取器：

	def _menu_open(self):
    # 文件拾取对话框
        file_picker = gui.FileDialog(gui.FileDialog.OPEN,"Select file...",self.window.theme) 

  
  
  
  
   
   
    
    运行
    
    
     
     运行

gui.FileDialog(self, arg0, arg1, arg2)返回一个文件选择对话框。

arg0: 打开文件还是保存文件，取值为FileDialog.OPEN或FileDialog.SAVE。
arg1: 对话框的名字。
arg2: 对话框主题，通常从window.theme获得

为文件拾取器设置文件过滤类型和初始路径：

        # 文件类型过滤
        file_picker.add_filter('.obj', 'obj model files')
        file_picker.add_filter('', 'All files')
        # 初始文件路径
        file_picker.set_path('./model')

为对话框按钮设置回调函数：

        # 设置对话框按钮回调
        file_picker.set_on_cancel(self._on_cancel)
        file_picker.set_on_done(self._on_done)
    <span class="token comment"># 显示对话框</span>
    self<span class="token punctuation">.</span>window<span class="token punctuation">.</span>show_dialog<span class="token punctuation">(</span>file_picker<span class="token punctuation">)</span>

这两个回调函数必须设置，选择完成时调用_on_done()，取消选择时调用_on_cancel。

定义完成和取消时的回调函数：取消的时候直接关闭对话框，选择完成时加载并显示模型。

	def _on_cancel(self):
        # 关闭当前对话框
        self.window.close_dialog()

def _on_done(self, filename):
self.window.close_dialog()
self.load(filename)

def load(self, file):
# 读取模型文件
mesh = o3d.io.read_triangle_mesh(file)
mesh.compute_vertex_normals()
# 定义材质
material = rendering.MaterialRecord()
material.shader = ‘defaultLit’

# 向场景中添加模型
self._scene.scene.add_geometry(‘bunny’,mesh,material)
bounds = mesh.get_axis_aligned_bounding_box()
self._scene.setup_camera(60,bounds,bounds.get_center())

# 强制重绘场景
self._scene.force_redraw()

3.2 退出程序：

	# 退出应用
    def _menu_quit(self):
        self.window.close()

3.3 切换显示：

定义一个show表示当前是否显示模型，当点击菜单栏中的对应项时，改变该项的选中状态和模型的显示状态。

# 切换显示模型
    def _menu_show(self):
        self.show = not self.show
        gui.Application.instance.menubar.set_checked(App.MENU_SHOW,self.show)
        self._scene.scene.show_geometry('bunny',self.show)

4. 运行结果

resulte

5. 完整代码

import open3d as o3d
import open3d.visualization.gui as gui
import open3d.visualization.rendering as rendering

class App:

MENU_OPEN = 1

MENU_SHOW = 5

MENU_QUIT = 20
MENU_ABOUT = 21

show = True

def init(self):
gui.Application.instance.initialize()

self.window = gui.Application.instance.create_window(“layout”,800,600)
w = self.window
em = w.theme.font_size

# 渲染窗口
self._scene = gui.SceneWidget()
self._scene.scene = rendering.Open3DScene(w.renderer)

# 右侧面板
self._pannel = gui.CollapsableVert(‘panel’,0, gui.Margins(0.25em,0.25em,0.25em,0.25em))
for i in range(15):
self._pannel.add_child(gui.Label(f"{ i}-th label"))

self._button = gui.Button(“button”)

# 布局回调函数
w.set_on_layout(self._on_layout)
w.add_child(self._scene)
w.add_child(self._pannel)
w.add_child(self._button)

# ---------------Menu----------------
# 菜单栏是全局的（因为macOS上是全局的）
# 无论创建多少窗口，菜单栏只创建一次。

# ----以下只针对Windows的菜单栏创建----
if gui.Application.instance.menubar is None:
# 文件菜单栏
file_menu = gui.Menu()
file_menu.add_item(“Open”,App.MENU_OPEN)
file_menu.add_separator()
file_menu.add_item(“Quit”,App.MENU_QUIT)

# 显示菜单栏
show_menu = gui.Menu()
show_menu.add_item(“Show Geometry”,App.MENU_SHOW)
show_menu.set_checked(App.MENU_SHOW,True)

# 帮助菜单栏
help_menu = gui.Menu()
help_menu.add_item(“About”,App.MENU_ABOUT)
help_menu.set_enabled(App.MENU_ABOUT,False)

# 菜单栏
menu = gui.Menu()
menu.add_menu(“File”,file_menu)
menu.add_menu(“Show”,show_menu)
menu.add_menu(“Help”,help_menu)
gui.Application.instance.menubar = menu

#-----注册菜单栏事件------
w.set_on_menu_item_activated(App.MENU_OPEN,self._menu_open)
w.set_on_menu_item_activated(App.MENU_QUIT,self._menu_quit)
w.set_on_menu_item_activated(App.MENU_SHOW,self._menu_show)

# 打开并显示一个obj模型
def _menu_open(self):
# 文件拾取对话框
file_picker = gui.FileDialog(gui.FileDialog.OPEN,“Select file…”,self.window.theme)

# 文件类型过滤
file_picker.add_filter(‘.obj’, ‘obj model files’)
file_picker.add_filter(‘’, ‘All files’)

# 初始文件路径
file_picker.set_path(‘./model’)

# 设置对话框按钮回调
file_picker.set_on_cancel(self._on_cancel)
file_picker.set_on_done(self._on_done)

# 显示对话框
self.window.show_dialog(file_picker)

def _on_cancel(self):
# 关闭当前对话框
self.window.close_dialog()

def _on_done(self, filename):
self.window.close_dialog()
self.load(filename)

def load(self, file):
# 读取模型文件
mesh = o3d.io.read_triangle_mesh(file)
mesh.compute_vertex_normals()
# 定义材质
material = rendering.MaterialRecord()
material.shader = ‘defaultLit’

# 向场景中添加模型
self._scene.scene.add_geometry(‘bunny’,mesh,material)
bounds = mesh.get_axis_aligned_bounding_box()
self._scene.setup_camera(60,bounds,bounds.get_center())

# 重绘
self._scene.force_redraw()

# 退出应用
def _menu_quit(self):
self.window.close()

# 切换显示模型
def _menu_show(self):
self.show = not self.show
gui.Application.instance.menubar.set_checked(App.MENU_SHOW,self.show)
self._scene.scene.show_geometry(‘bunny’,self.show)

def _on_layout(self, layout_context):
# 在on_layout回调函数中应正确设置所有子对象的框架(position + size)，
# 回调结束之后才会布局孙子对象。

r = self.window.content_rect
self._scene.frame = r

pannel_width = 17*layout_context.theme.font_size
pannel_height = min(
r.height, self._pannel.calc_preferred_size(
layout_context, gui.Widget.Constraints()).height
)
self._pannel.frame = gui.Rect(r.get_right()-pannel_width,r.y,pannel_width,pannel_height)

button_pref = self._button.calc_preferred_size(
layout_context, gui.Widget.Constraints())
self._button.frame = gui.Rect(r.x,r.get_bottom()-button_pref.height, button_pref.width,button_pref.height)