Coverage for local/lib/python2.7/site-packages/sage/plot/colors.py: 68%

[(1.0, 0.0, 0.0), (1.0, 0.8571428571428571, 0.0), (0.2857142857142858, 1.0, 0.0), (0.0, 1.0, 0.5714285714285712), (0.0, 0.5714285714285716, 1.0), (0.2857142857142856, 0.0, 1.0), (1.0, 0.0, 0.8571428571428577)]

AUTHORS:

- Robert L. Miller

- Karl-Dieter Crisman (directly use :func:`hsv_to_rgb` for hues)

"""

R = []

for i in range(n):

R.append(tuple(map(float, hsv_to_rgb(i / n, 1, 1))))

if format == 'rgbtuple':

return R

elif format == 'hex':

for j in range(len(R)):

R[j] = float_to_html(*R[j])

return R

# If you change what this accepts, remember to change the documentation

# about cmap where it is used and to test these classes.

def get_cmap(cmap):

r"""

Returns a color map (actually, a matplotlib :class:`Colormap`

object), given its name or a [mixed] list/tuple of RGB list/tuples

and color names. For a list of map names, evaluate::

sage: sorted(colormaps)

[u'Accent', u'Accent_r', u'Blues', u'Blues_r', ...]

See :func:`rgbcolor` for valid list/tuple element formats.

INPUT:

- ``cmap`` - a string, list, tuple, or

:class:`matplotlib.colors.Colormap`; a string must be a valid

color map name

OUTPUT:

- a :class:`matplotlib.colors.Colormap` instance

EXAMPLES::

sage: from sage.plot.colors import get_cmap

sage: get_cmap('jet')

<matplotlib.colors.LinearSegmentedColormap object at 0x...>

sage: get_cmap(u'jet')

<matplotlib.colors.LinearSegmentedColormap object at 0x...>

sage: get_cmap([(0,0,0), (0.5,0.5,0.5), (1,1,1)])

<matplotlib.colors.ListedColormap object at 0x...>

sage: get_cmap(['green', 'lightblue', 'blue'])

<matplotlib.colors.ListedColormap object at 0x...>

sage: get_cmap(((0.5, 0.3, 0.2), [1.0, 0.0, 0.5], 'purple', Color(0.5,0.5,1, space='hsv')))

<matplotlib.colors.ListedColormap object at 0x...>

sage: get_cmap('jolies')

Traceback (most recent call last):

...

RuntimeError: Color map jolies not known (type import matplotlib.cm; matplotlib.cm.datad.keys() for valid names)

sage: get_cmap('mpl')

Traceback (most recent call last):

...

RuntimeError: Color map mpl not known (type import matplotlib.cm; matplotlib.cm.datad.keys() for valid names)

"""

# matplotlib color maps

global cm

if not cm:

from matplotlib import cm

from matplotlib.colors import ListedColormap, Colormap

if isinstance(cmap, Colormap):

return cmap

elif isinstance(cmap, six.string_types):

if not cmap in cm.datad.keys():

raise RuntimeError("Color map %s not known (type import matplotlib.cm; matplotlib.cm.datad.keys() for valid names)" % cmap)

return cm.__dict__[cmap]

elif isinstance(cmap, (list, tuple)):

cmap = [rgbcolor(_) for _ in cmap]

return ListedColormap(cmap)

class Colormaps(collections.MutableMapping):

"""

A dict-like collection of lazily-loaded matplotlib color maps.

For a list of map names, evaluate::

sage: sorted(colormaps)

[u'Accent', u'Accent_r', u'Blues', u'Blues_r', ...]

"""

def __init__(self):

"""

Constructs an empty mutable collection of color maps.

EXAMPLES::

sage: from sage.plot.colors import Colormaps

sage: maps = Colormaps()

sage: len(maps.maps)

"""

self.maps = {}

def load_maps(self):

"""

If it's necessary, loads matplotlib's color maps and adds them

to the collection.

EXAMPLES::

sage: from sage.plot.colors import Colormaps

sage: maps = Colormaps()

sage: len(maps.maps)

sage: maps.load_maps()

sage: len(maps.maps)>130

True

"""

global cm

if not cm:

from matplotlib import cm

if not self.maps:

for cmap in cm.datad.keys():

self.maps[cmap] = cm.__getattribute__(cmap)

def __dir__(self):

"""

Returns an approximate list of attribute names, including the

color map names.

OUTPUT:

- a list of strings

EXAMPLES::

sage: from sage.plot.colors import Colormaps

sage: maps = Colormaps()

sage: 'Accent' in dir(maps)

True

"""

self.load_maps()

methods = ['load_maps', '__dir__', '__len__', '__iter__',

'__contains__', '__getitem__', '__getattr__',

'__setitem__', '__delitem__']

return dir(super(Colormaps, self)) + methods + self.keys()

def __len__(self):

"""

Returns the number of color maps.

OUTPUT:

- an int

EXAMPLES::

sage: from sage.plot.colors import Colormaps

sage: maps = Colormaps()

sage: len(maps)>130

True

"""

self.load_maps()

return len(self.maps)

def __iter__(self):

"""

Returns an iterator over the color map collection.

OUTPUT:

- a dictionary key iterator instance

EXAMPLES::

sage: from sage.plot.colors import Colormaps

sage: maps = Colormaps()

sage: count = 0

sage: for m in maps: count += 1

sage: count == len(maps)

True

"""

self.load_maps()

return iter(self.maps)

def __contains__(self, name):

"""

Returns whether a map is in the color maps collection.

INPUT:

- ``name`` - a string; the name of the map to query

OUTPUT:

- a boolean

EXAMPLES::

sage: from sage.plot.colors import Colormaps

sage: maps = Colormaps()

sage: 'summer' in maps

True

sage: 'not really a color map' in maps

False

"""

self.load_maps()

return name in self.maps

def __getitem__(self, name):

"""

Gets a color map from the collection via key access.

INPUT:

- ``name`` - a string; the name of the map return

OUTPUT:

- an instance of :class:`matplotlib.colors.Colormap`

EXAMPLES::

sage: from sage.plot.colors import Colormaps

sage: maps = Colormaps()

sage: maps.get('Oranges')

<matplotlib.colors.LinearSegmentedColormap object at ...>

sage: maps['copper']

<matplotlib.colors.LinearSegmentedColormap object at ...>

sage: maps.get('not a color map')

sage: maps['not a color map']

Traceback (most recent call last):

...

KeyError: "no colormap with name 'not a color map'"

"""

self.load_maps()

try:

return self.maps[name]

except KeyError:

raise KeyError("no colormap with name '%s'" % name)

def __getattr__(self, name):

"""

Gets a color map from the collection via attribute access.

INPUT:

- ``name`` - a string; the name of the map to return

OUTPUT:

- an instance of :class:`matplotlib.colors.Colormap`

EXAMPLES::

sage: from sage.plot.colors import Colormaps

sage: maps = Colormaps()

sage: maps.pink

<matplotlib.colors.LinearSegmentedColormap object at ...>

sage: maps.punk

Traceback (most recent call last):

...

AttributeError: 'Colormaps' has no attribute or colormap punk

sage: maps['punk']

Traceback (most recent call last):

...

KeyError: "no colormap with name 'punk'"

sage: maps['bone'] == maps.bone

True

"""

try:

return self[name]

except KeyError:

raise AttributeError("'%s' has no attribute or colormap %s"%(type(self).__name__,name))

def __repr__(self):

"""

Returns a string representation of the color map collection.

OUTPUT:

- a string

EXAMPLES::

sage: from sage.plot.colors import Colormaps

sage: maps = Colormaps()

sage: maps

{...}

sage: type(repr(maps))

<... 'str'>

"""

self.load_maps()

return repr(self.maps)

def __setitem__(self, name, colormap):

"""

Adds a color map to the collection.

INPUT:

- ``name`` - a string; the name of the map to add

- ``colormap`` - an instance of

:class:`matplotlib.colors.Colormap`; the color map to add

EXAMPLES::

sage: from sage.plot.colors import Colormaps, get_cmap

sage: maps = Colormaps()

sage: count = len(maps)

sage: my_map = get_cmap(['chartreuse', '#007', (1.0, 0.0, 0.0)])

sage: maps['my_map'] = my_map

sage: 'my_map' in maps

True

sage: count + 1 == len(maps)

True

"""

self.load_maps()

self.maps[name] = colormap

def __delitem__(self, name):

"""

Removes a color map from the collection.

INPUT:

- ``name`` - a string; the name of the map to remove

EXAMPLES::

sage: from sage.plot.colors import Colormaps

sage: maps = Colormaps()

sage: count = len(maps)

sage: maps.popitem()

(u'Spectral', <matplotlib.colors.LinearSegmentedColormap object at ...>)

sage: count - 1 == len(maps)

True

"""

self.load_maps()

del self.maps[name]

colormaps = Colormaps()

Coverage for local/lib/python2.7/site-packages/sage/plot/colors.py : 68%

201 statements 137 run 64 missing 0 excluded