from matplotlib.colors import LinearSegmentedColormap
from numpy import nan, inf
cm_data = [[0., 0., 0.],
[0.0117647, 0., 0.],
[0.027451, 0., 0.],
[0.0431373, 0., 0.],
[0.0588235, 0., 0.],
[0.0745098, 0., 0.],
[0.0862745, 0., 0.],
[0.101961, 0., 0.],
[0.117647, 0., 0.],
[0.133333, 0., 0.],
[0.14902, 0., 0.],
[0.160784, 0., 0.],
[0.176471, 0., 0.],
[0.192157, 0., 0.],
[0.207843, 0., 0.],
[0.223529, 0., 0.],
[0.235294, 0., 0.],
[0.25098, 0., 0.],
[0.266667, 0., 0.],
[0.282353, 0., 0.],
[0.298039, 0., 0.],
[0.309804, 0., 0.],
[0.32549, 0., 0.],
[0.341176, 0., 0.],
[0.356863, 0., 0.],
[0.372549, 0., 0.],
[0.384314, 0., 0.],
[0.4, 0., 0.],
[0.415686, 0., 0.],
[0.431373, 0., 0.],
[0.447059, 0., 0.],
[0.458824, 0., 0.],
[0.47451, 0., 0.],
[0.490196, 0., 0.],
[0.505882, 0., 0.],
[0.521569, 0., 0.],
[0.537255, 0., 0.],
[0.54902, 0., 0.],
[0.564706, 0., 0.],
[0.580392, 0., 0.],
[0.596078, 0., 0.],
[0.611765, 0., 0.],
[0.623529, 0., 0.],
[0.639216, 0., 0.],
[0.654902, 0., 0.],
[0.670588, 0., 0.],
[0.686275, 0., 0.],
[0.698039, 0., 0.],
[0.713725, 0., 0.],
[0.729412, 0., 0.],
[0.745098, 0., 0.],
[0.760784, 0., 0.],
[0.772549, 0., 0.],
[0.788235, 0., 0.],
[0.803922, 0., 0.],
[0.819608, 0., 0.],
[0.835294, 0., 0.],
[0.847059, 0., 0.],
[0.862745, 0., 0.],
[0.878431, 0., 0.],
[0.894118, 0., 0.],
[0.909804, 0., 0.],
[0.921569, 0., 0.],
[0.937255, 0., 0.],
[0.952941, 0., 0.],
[0.968627, 0.0117647, 0.],
[0.984314, 0.027451, 0.],
[1., 0.0431373, 0.],
[0.984314, 0.0588235, 0.],
[0.968627, 0.0745098, 0.],
[0.952941, 0.0901961, 0.],
[0.937255, 0.105882, 0.],
[0.921569, 0.121569, 0.],
[0.901961, 0.137255, 0.],
[0.886275, 0.152941, 0.],
[0.870588, 0.168627, 0.],
[0.854902, 0.184314, 0.],
[0.839216, 0.2, 0.],
[0.823529, 0.215686, 0.],
[0.803922, 0.231373, 0.],
[0.788235, 0.247059, 0.],
[0.772549, 0.262745, 0.],
[0.756863, 0.278431, 0.],
[0.741176, 0.294118, 0.],
[0.721569, 0.309804, 0.],
[0.705882, 0.32549, 0.],
[0.690196, 0.341176, 0.],
[0.67451, 0.356863, 0.],
[0.658824, 0.372549, 0.],
[0.643137, 0.388235, 0.],
[0.623529, 0.403922, 0.],
[0.607843, 0.419608, 0.],
[0.592157, 0.435294, 0.],
[0.576471, 0.45098, 0.],
[0.560784, 0.466667, 0.],
[0.541176, 0.482353, 0.],
[0.52549, 0.498039, 0.],
[0.509804, 0.513725, 0.],
[0.494118, 0.529412, 0.],
[0.478431, 0.545098, 0.],
[0.462745, 0.560784, 0.],
[0.443137, 0.576471, 0.],
[0.427451, 0.592157, 0.],
[0.411765, 0.607843, 0.],
[0.396078, 0.623529, 0.],
[0.380392, 0.639216, 0.],
[0.360784, 0.654902, 0.],
[0.345098, 0.670588, 0.],
[0.329412, 0.686275, 0.],
[0.313725, 0.701961, 0.],
[0.298039, 0.717647, 0.],
[0.282353, 0.733333, 0.],
[0.262745, 0.74902, 0.],
[0.247059, 0.764706, 0.],
[0.231373, 0.780392, 0.],
[0.215686, 0.796078, 0.],
[0.2, 0.811765, 0.],
[0.180392, 0.827451, 0.],
[0.164706, 0.843137, 0.],
[0.14902, 0.858824, 0.],
[0.133333, 0.87451, 0.],
[0.117647, 0.890196, 0.],
[0.101961, 0.905882, 0.],
[0.0823529, 0.921569, 0.],
[0.0666667, 0.937255, 0.],
[0.0509804, 0.952941, 0.],
[0.0352941, 0.968627, 0.],
[0.0196078, 0.984314, 0.0117647],
[0., 1., 0.027451],
[0., 0.984314, 0.0431373],
[0., 0.968627, 0.0588235],
[0., 0.952941, 0.0745098],
[0., 0.937255, 0.0901961],
[0., 0.921569, 0.105882],
[0., 0.905882, 0.117647],
[0., 0.890196, 0.133333],
[0., 0.87451, 0.14902],
[0., 0.858824, 0.164706],
[0., 0.843137, 0.180392],
[0., 0.827451, 0.196078],
[0., 0.811765, 0.211765],
[0., 0.796078, 0.223529],
[0., 0.780392, 0.239216],
[0., 0.764706, 0.254902],
[0., 0.74902, 0.270588],
[0., 0.733333, 0.286275],
[0., 0.717647, 0.301961],
[0., 0.701961, 0.317647],
[0., 0.686275, 0.333333],
[0., 0.666667, 0.345098],
[0., 0.65098, 0.360784],
[0., 0.635294, 0.376471],
[0., 0.619608, 0.392157],
[0., 0.603922, 0.407843],
[0., 0.588235, 0.423529],
[0., 0.572549, 0.439216],
[0., 0.556863, 0.45098],
[0., 0.541176, 0.466667],
[0., 0.52549, 0.482353],
[0., 0.509804, 0.498039],
[0., 0.494118, 0.513725],
[0., 0.478431, 0.529412],
[0., 0.462745, 0.545098],
[0., 0.447059, 0.556863],
[0., 0.431373, 0.572549],
[0., 0.415686, 0.588235],
[0., 0.4, 0.603922],
[0., 0.384314, 0.619608],
[0., 0.368627, 0.635294],
[0., 0.352941, 0.65098],
[0., 0.333333, 0.666667],
[0., 0.317647, 0.678431],
[0., 0.301961, 0.694118],
[0., 0.286275, 0.709804],
[0., 0.270588, 0.72549],
[0., 0.254902, 0.741176],
[0., 0.239216, 0.756863],
[0., 0.223529, 0.772549],
[0., 0.207843, 0.784314],
[0., 0.192157, 0.8],
[0., 0.176471, 0.815686],
[0., 0.160784, 0.831373],
[0., 0.145098, 0.847059],
[0., 0.129412, 0.862745],
[0., 0.113725, 0.878431],
[0., 0.0980392, 0.890196],
[0., 0.0823529, 0.905882],
[0., 0.0666667, 0.921569],
[0., 0.0509804, 0.937255],
[0., 0.0352941, 0.952941],
[0., 0.0196078, 0.968627],
[0., 0., 0.984314],
[0., 0., 1.],
[0., 0., 0.984314],
[0., 0., 0.968627],
[0., 0., 0.952941],
[0., 0., 0.937255],
[0., 0., 0.921569],
[0., 0., 0.905882],
[0., 0., 0.890196],
[0., 0., 0.87451],
[0., 0., 0.858824],
[0., 0., 0.843137],
[0., 0., 0.827451],
[0., 0., 0.811765],
[0., 0., 0.796078],
[0., 0., 0.780392],
[0., 0., 0.764706],
[0., 0., 0.74902],
[0., 0., 0.733333],
[0., 0., 0.717647],
[0., 0., 0.701961],
[0., 0., 0.686275],
[0., 0., 0.666667],
[0., 0., 0.65098],
[0., 0., 0.635294],
[0., 0., 0.619608],
[0., 0., 0.603922],
[0., 0., 0.588235],
[0., 0., 0.572549],
[0., 0., 0.556863],
[0., 0., 0.541176],
[0., 0., 0.52549],
[0., 0., 0.509804],
[0., 0., 0.494118],
[0., 0., 0.478431],
[0., 0., 0.462745],
[0., 0., 0.447059],
[0., 0., 0.431373],
[0., 0., 0.415686],
[0., 0., 0.4],
[0., 0., 0.384314],
[0., 0., 0.368627],
[0., 0., 0.352941],
[0., 0., 0.333333],
[0., 0., 0.317647],
[0., 0., 0.301961],
[0., 0., 0.286275],
[0., 0., 0.270588],
[0., 0., 0.254902],
[0., 0., 0.239216],
[0., 0., 0.223529],
[0., 0., 0.207843],
[0., 0., 0.192157],
[0., 0., 0.176471],
[0., 0., 0.160784],
[0., 0., 0.145098],
[0., 0., 0.129412],
[0., 0., 0.113725],
[0., 0., 0.0980392],
[0., 0., 0.0823529],
[0., 0., 0.0666667],
[0., 0., 0.0509804],
[0., 0., 0.0352941],
[0., 0., 0.0196078],
[0., 0., 0.]]

test_cm = LinearSegmentedColormap.from_list(__file__, cm_data)


if __name__ == "__main__":
    import matplotlib.pyplot as plt
    import numpy as np

    try:
        from pycam02ucs.cm.viscm import viscm
        viscm(test_cm)
    except ImportError:
        print("pycam02ucs not found, falling back on simple display")
        plt.imshow(np.linspace(0, 100, 256)[None, :], aspect='auto',
                   cmap=test_cm)
    plt.show()
