如何在Mayavi中直接设置RGB / RGBA颜色

Question

我有一个带有多个顶点的mayavi对象，我想直接为这些顶点设置RGB或RGBA值，而不是将自己限制在带有标量的单个颜色图上。 如何做到这一点？

Answer 1

据我所知，没有执行此操作的文档，但是我已经找到了一种方法，只需最少的黑客攻击即可做到这一点。 下面是一个最小的例子，可能需要对不同类型的来源进行一些修改：

from tvtk.api import tvtk; from mayavi import mlab; import numpy as np
x,y,z=np.random.random((3,nr_points)) #some data
colors=np.random.randint(256,size=(100,3)) #some RGB or RGBA colors
pts=mlab.points3d(x,y,z)
sc=tvtk.UnsignedCharArray()
sc.from_array(colors)
pts.mlab_source.dataset.point_data.scalars=sc
pts.mlab_source.dataset.modified()

看起来有时您必须确保映射器指向正确的事物。 对于上述示例，这不是必需的，但对于其他来源可能是必需的

pts.actor.mapper.input=pts.mlab_source.dataset

在某些时候，mayavi API 应该得到更好的修复，以便公开一个 API 来为所有管道函数执行此操作，但事实证明这是一组相当复杂和全面的更改，我目前没有时间完成。

编辑：用户eqzx发布了另一个问题的答案（在MayaVi中为3D点指定绝对颜色），这可能更简单，特别是对于某些难以使用tvtk.UnsignedCharArray的源类型。

他的想法是创建一个跨越整个 256x256x256 RGB 值范围的 LUT。请注意，此 LUT 因此有 16，777，216 个条目。如果您想在许多vtk对象中使用它，如果不小心，可能会浪费相当多的内存。

#create direct grid as 256**3 x 4 array 
def create_8bit_rgb_lut():
    xl = numpy.mgrid[0:256, 0:256, 0:256]
    lut = numpy.vstack((xl[0].reshape(1, 256**3),
                        xl[1].reshape(1, 256**3),
                        xl[2].reshape(1, 256**3),
                        255 * numpy.ones((1, 256**3)))).T
    return lut.astype('int32')
# indexing function to above grid
def rgb_2_scalar_idx(r, g, b):
    return 256**2 *r + 256 * g + b
#N x 3 colors
colors = numpy.array([_.color for _ in points])
#N scalars
scalars = numpy.zeros((colors.shape[0],))
for (kp_idx, kp_c) in enumerate(colors):
    scalars[kp_idx] = rgb_2_scalar_idx(kp_c[0], kp_c[1], kp_c[2])
rgb_lut = create_8bit_rgb_lut()
points_mlab = mayavi.mlab.points3d(x, y, z
                                   keypoint_scalars,
                                   mode = 'point')
#magic to modify lookup table 
points_mlab.module_manager.scalar_lut_manager.lut._vtk_obj.SetTableRange(0, rgb_lut.shape[0])
points_mlab.module_manager.scalar_lut_manager.lut.number_of_colors = rgb_lut.shape[0]
points_mlab.module_manager.scalar_lut_manager.lut.table = rgb_lut