菜单栏与文件拾取

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 = 1MENU_SHOW = 5MENU_QUIT = 20MENU_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. 定义回调函数

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

1. 文件拾取（MENU_OPEN）
2. 退出程序（MENU_QUIT）
3. 切换显示（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)# 显示对话框self.window.show_dialog(file_picker)

这两个回调函数必须设置，选择完成时调用_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.showgui.Application.instance.menubar.set_checked(App.MENU_SHOW,self.show)self._scene.scene.show_geometry('bunny',self.show)

4. 运行结果

5. 完整代码

import open3d as o3d
import open3d.visualization.gui as gui
import open3d.visualization.rendering as renderingclass App:MENU_OPEN = 1MENU_SHOW = 5MENU_QUIT = 20MENU_ABOUT = 21show = Truedef __init__(self):gui.Application.instance.initialize()self.window = gui.Application.instance.create_window("layout",800,600)w = self.windowem = w.theme.font_size# 渲染窗口self._scene = gui.SceneWidget()self._scene.scene = rendering.Open3DScene(w.renderer)# 右侧面板self._pannel = gui.CollapsableVert('panel',0, gui.Margins(0.25*em,0.25*em,0.25*em,0.25*em))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.showgui.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_rectself._scene.frame = rpannel_width = 17*layout_context.theme.font_sizepannel_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)def run(self):gui.Application.instance.run()if __name__ == "__main__":app = App()app.run()