Mayavi-鼠标悬停互动并从不同角度绘制同一场景

时间:2019-03-25 17:58:01

标签: python opencv gtk pygtk mayavi

我目前正在努力查看如何以适当的方式与mayavi渲染的场景进行交互。

我有一个激光雷达点云,通过函数points3d()进行绘制,现在我在点云之间另外设置了一个围绕汽车的边界框,我想更改内部的点的颜色。当我将鼠标悬停在边界框上时,立即单击该框。 您能告诉我如何选择bbox中的点并更改其颜色吗?

第二个问题是,如何在3d视图和鸟瞰视图中同时显示点云的同一场景?

非常感谢:]

1 个答案:

答案 0 :(得分:0)

我找到了有关颜色问题的解决方案-我不知道这是否是最佳实践。但是我仍然需要帮助来确定边界框内的点。我还想创建一个GUI,使用户可以修改边界框的大小和方向。 [但这是另一个主题]

import numpy as np
from mayavi.mlab import draw, points3d
from tvtk.api import tvtk

# Primitives
N = 3000 # Number of points
ones = np.ones(N) #np.hstack((np.tile(np.array([1]), int(N/2)).T,  np.tile(np.array([4000]),  int(N/2)).T))
scalars = ones #np.arange(N)  # Key point: set an integer for each point

# Define color table (including alpha), which must be uint8 and [0,255]
colors = np.vstack((np.tile(np.array([[255],[255],[0]]), int(N/2)).T,  np.tile(np.array([[0],[0],[255]]),  int(N/2)).T))
# Define coordinates and points
x, y, z = (np.random.random((N, 3))*255).astype(np.uint8).T # Assign x, y, z values to match color
pts = points3d(x, y, z, scale_factor=10) # Create points
#pts.glyph.color_mode = 'color_by_vector' # Color by scalar
# Set look-up table and redraw
#pts.module_manager.scalar_lut_manager.lut.table = colors
pts.glyph.scale_mode = 'scale_by_vector'
sc=tvtk.UnsignedCharArray()
sc.from_array(colors)
pts.mlab_source.dataset.point_data.scalars = sc
draw()