代码收藏家 Python的3D可视化库【vedo】2-5 (plotter模块) 坐标转换、场景导出、添加控件
文章目录
vedo是Python实现的一个用于辅助科学研究的3D可视化库。vedo的plotter模块是管理对象和控制3D渲染的模块。
plotter.Plotter实例可以用于显示3D图形对象、控制渲染器行为、操控相机、创建事件以及导出3D数据。几乎所有关于3D图形的操作，都是在Plotter渲染的窗口进行的。
参考：vedo官方文档
4 Plotter类的方法
按照功能把Plotter的方法分组整理，前面已经把主要的方法整理完了，剩下的是一些辅助方法，分为转换坐标、场景导出、添加附件三类。
4.7 屏幕和场景中的坐标点转换
有的方法需要以3D世界坐标系中的坐标作为参数，也有一些方法需要以屏幕坐标为参数。
这里提供了两种坐标互相转换的方法。
4.7.1 屏幕坐标转为世界坐标
plt.compute_world_coordinate()
把屏幕上的2D点转换为渲染场景中的3D点坐标。参数如下：
pos2D，2D屏幕坐标点；at=None，渲染器序号；objs=()，欲放置该点的网格对象列表；bounds=()，指定边界框；offset=None，指定偏移值；pixeltol=None，屏幕容差；worldtol=None，世界坐标容差；4.7.2 世界坐标转为屏幕坐标
plt.compute_screen_coordinates(obj, full_window=False)
给定当前渲染器或整个窗口中的3D点，找到它的屏幕像素坐标。
obj是网格对象时，会计算其所有顶点坐标在屏幕上的位置。
obj也可以是三维坐标元组组成的序列。
c = vedo.Cube()
plt = vedo.Plotter()
# 将对象的顶点转为屏幕位置
p1 = plt.compute_screen_coordinates(c)
# 将指定的3维坐标点转为屏幕位置
p2 = plt.compute_screen_coordinates([(1, 1, 1), (2, 2, 2)])
上述的两个方法可以填超出屏幕大小或是超出窗口大小的坐标。
4.7.3 屏幕坐标取颜色
plt.color_picker(xy, verbose=False)
指定屏幕上的坐标位置，提取像素点的颜色，返回RGB格式的颜色。
选取的点在渲染窗口区域之外，返回None。
verbose为真时，会输出更详细的信息。
plt = vedo.Plotter()
p = plt.color_picker([200, 100], verbose=True)
print(p)
4.8 导出
4.8.1 导出2D图片
plt.screenshot(filename='screenshot.png', scale=1, asarray=False)
对Plotter窗口截图。
scale为整数，设置图片的像素大小倍率。
asarray为真时，不生成图片，而是以numpy数组的格式返回。
c = vedo.Cube()
plt = vedo.Plotter()
plt.show(c, vedo.Text2D("scale=2", s=3), interactive=False)
plt.screenshot("kkk2.jpg", scale=2)
arr = plt.screenshot(asarray=True)
print(arr)
plt.interactive()
数组输出结果是
[[[255 255 255]
[255 255 255]
[255 255 255]
...
[255 255 255]
[255 255 255]
[255 255 255]]
...
[[255 255 255]
[255 255 255]
[255 255 255]
...
[255 255 255]
[255 255 255]
[255 255 255]]]
plt.toimage(scale=1)
从当前渲染器窗口生成一个Image对象。
4.8.2 导出3D文件
plt.export(filename='scene.npz', binary=False)
把场景导出为HTML、X3D或numpy文件，文件格式在filename参数中指定后缀即可。
4.9 添加控件
4.9.1 添加内嵌子窗口
plt.add_inset(\*objects, \*\*options)
往渲染器添加一个可拖动的嵌入空间。
at=None，渲染器编号；pos=0，嵌入框的初始位置。范围1-4，分别表示窗格的左上、右上、左下、右下。也可以填
(x,y)元组，x、y范围0-1，表示宽和高占渲染窗格大小的百分比。size=0.1，方形嵌入空间占窗格大小的百分比；draggable=True，为真时该子渲染控件可以被拖动；c="lb"，嵌入框在被拖动时的颜色；c = vedo.Cube(pos=(1, 0, 0))
s = vedo.Sphere(pos=(-1, 0, 0), res=10)
plt = vedo.Plotter(bg="lb")
# 子窗格显示立方体
plt.add_inset(c)
plt.show(c, s)
子窗口中的立方体会与主场景中的立方体同步旋转。
4.9.2 添加选择区
plt.pick_area(pos1, pos2, at=None)
由两个2D屏幕坐标点确定一个方框，选中其内的所有对象。
返回一个Mesh对象，带有可见的视图区。常用来用来选择场景中的对象。
c = vedo.Cube(pos=(1, 0, 0))
s = vedo.Sphere(pos=(-1, 0, 0), res=10)
plt = vedo.Plotter()
plt.show(c, s, axes=1, interactive=False)
pa = plt.pick_area([400, 400], [800, 800])
plt.add(pa)
plt.interactive()
要先展示对象，然后创建和添加选择区。否则会出现预料外的结果。
4.9.3 添加比例尺
plt.add_scale_indicator()
添加一个比例尺。只在平行投影模式可用。
pos=(0.7, 0.05)，在屏幕上的位置，以比例值表示；s=0.02，文本的尺寸；length=2，线的长度；lw=4，线宽；c="k1"，颜色；alpha=1，透明度；units=""，表示单位的字符串；gap=0.05，线和文本的间隔；比例尺下面的数值会随着图像的缩放自动变化数值。
4.9.4 为对象添加弹出提示
plt.add_hint()
鼠标悬浮到某个对象上时，创建一个弹出消息。
add_hint(None)可以禁用所有弹出消息，add_hint(obj, False)可以禁用指定对象的弹出消息。
参数如下：
obj，Mesh或Points的实例；text=""，消息内容；c="k"，消息文字颜色；bg="yellow9"，消息底色；font="Calco"，消息字体；size=18，消息字号；justify=0，对齐方式；angle=0，消息文字逆时针旋转的角度；delay=250，消息弹出延时多少毫秒；c = vedo.Cube()
plt = vedo.Plotter(size=(900, 900))
plt.add_hint(c, "Another Cube hint.", c="r", angle=90, size=32)
plt.show(c)
4.9.4 悬浮时显示对象说明
plt.add_hover_legend()
鼠标移到到任意对象上时，显示该对象的说明文字。
参数如下：
at=None，渲染器编号；c=None，文字颜色；默认从黑色或白色中自动选择。pos="bottom-left"，文字位置；font="Calco"，文字字体；s=0.75，文字缩放；bg="auto"，文本框的颜色；alpha=0.1，文本框的透明度；maxlength=24，每行允许的最大字符数；use_info=False，可视化obj.info的内容；c = vedo.Cube(pos=(1, 0, 0))
s = vedo.Sphere(pos=(-1, 0, 0), res=10)
plt = vedo.Plotter()
plt.add_hover_legend(s=1.75)
plt.show(c, s)
4.9.5 添加addons模块的附件
addons模块包含了各种用于标示和操控的附件，Plotter的以下8个方法都是调用的addons模块的对应方法，具体参数也一样，后续看到这个模块时详述。
plt.add_slider()
添加一个vedo.addons.Slider2D对象。
plt.add_slider3d()
添加一个3D滑块控件。
plt.add_button()
在渲染窗口添加按钮。
plt.add_spline_tool()
为当前Plotter实例添加一个曲线工具。
曲线工具上的节点可以鼠标拖动，点击这条曲线可以添加一个额外的点。选中点后按del键可以移除该点。
plt.add_icon()
在渲染器上添加一个嵌入式图标的网格对象。
plt.add_global_axes()
在场景中绘制轴。
plt.add_legend_box()
在右上角添加图例。
plt.add_renderer_frame()
为渲染器子窗口添加边框。
以上是Plotter类中进行坐标转换、场景导出和添加控件的方法。
更多关于数据可视化的内容参考：Python数据可视化笔记
作者：兔子的洋葱圈
