Color Map to png file

For anyone wondering:

    lut = display.LookupTable.GetClientSideObject()

    dataRange = customRange
    if not dataRange:
        dataRange = lut.GetRange()

    delta = (dataRange[1] - dataRange[0]) / float(numSamples)

    colorArray = vtkUnsignedCharArray()
    colorArray.SetNumberOfComponents(3)
    colorArray.SetNumberOfTuples(numSamples)

    rgb = [ 0, 0, 0 ]
    for i in range(numSamples):
        lut.GetColor(dataRange[0] + float(i) * delta, rgb)
        r = int(round(rgb[0] * 255))
        g = int(round(rgb[1] * 255))
        b = int(round(rgb[2] * 255))
        colorArray.SetTuple3(i, r, g, b)
2 Likes