如何在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