l0j2dZddlmZddlZddlmZmZddlmZ ddl m Z ddl m ZddlmZddlm Zejd Zd ZGd d eZeeZeeeeZeeZdS) a| Builtin colormaps, colormap handling utilities, and the `ScalarMappable` mixin. .. seealso:: :doc:`/gallery/color/colormap_reference` for a list of builtin colormaps. :ref:`colormap-manipulation` for examples of how to make colormaps. :ref:`colormaps` an in-depth discussion of choosing colormaps. :ref:`colormapnorms` for more details about data normalization. )MappingN)_apicolors)_ScalarMappable)datad)cmaps) cmap_familiesz image.lutc*it}tjD]l\}}d|vrtj||t nDd|vrtj|d|n%tj||t ||<mddddd}|D]+\}}||}||_ |||<,t| D] }| }|||j <!|S)zw Generate a dict mapping standard colormap names to standard colormaps, as well as the reversed colormaps. redlistedgray gist_gray gist_yargGreys)grey gist_grey gist_yergGrays) cmaps_listedritemsrLinearSegmentedColormap_LUTSIZEListedColormap from_listcopynamelistvaluesreversed)cmap_drspecaliasesalias original_namecmaprmaps P/home/wildlama/miniconda3/envs/lam/lib/python3.11/site-packages/matplotlib/cm.py_gen_cmap_registryr( s6  FkmmLL d}}  *4x @ @ @4  !$x.$ 7 7 7  * 4 4T4 J J t    G!( }m$))++ u V]]__%%!!}} ty McPeZdZdZdZdZdZdZdZdZ dd d d Z d Z d Z dS)ColormapRegistrya Container for colormaps that are known to Matplotlib by name. The universal registry instance is `matplotlib.colormaps`. There should be no need for users to instantiate `.ColormapRegistry` themselves. Read access uses a dict-like interface mapping names to `.Colormap`\s:: import matplotlib as mpl cmap = mpl.colormaps['viridis'] Returned `.Colormap`\s are copies, so that their modification does not change the global definition of the colormap. Additional colormaps can be added via `.ColormapRegistry.register`:: mpl.colormaps.register(my_colormap) To get a list of all registered colormaps, you can do:: from matplotlib import colormaps list(colormaps) c<||_t||_dSN)_cmapstuple_builtin_cmaps)selfrs r'__init__zColormapRegistry.__init__Zs #Ellr)cltj|j|t}|S)N)colormap _error_cls)rgetitem_checkedr.KeyErrorr)r1itemr%s r' __getitem__zColormapRegistry.__getitem__^s)#DK$8TTTyy{{r)c*t|jSr-)iterr.r1s r'__iter__zColormapRegistry.__iter__bsDK   r)c*t|jSr-)lenr.r<s r'__len__zColormapRegistry.__len__es4;r)cFddd|DzS)Nz'ColormapRegistry; available colormaps: z, c3"K|] }d|dV dS)'N).0rs r' z+ColormapRegistry.__str__..js*77$+d+++777777r))joinr<s r'__str__zColormapRegistry.__str__hs-: 77$777778 9r)c t|S)z Return a list of the registered colormap names. This exists only for backward-compatibility in `.pyplot` which had a ``plt.colormaps()`` method. The recommended way to get this list is now ``list(colormaps)``. )rr<s r'__call__zColormapRegistry.__call__lsDzzr)NF)rforcec~tjtj||p|j}||vrI|st d|d||jvrt d|dtjd|d||j |<|j |j|kr||j |_dSdS) a Register a new colormap. The colormap name can then be used as a string argument to any ``cmap`` parameter in Matplotlib. It is also available in ``pyplot.get_cmap``. The colormap registry stores a copy of the given colormap, so that future changes to the original colormap instance do not affect the registered colormap. Think of this as the registry taking a snapshot of the colormap at registration. Parameters ---------- cmap : matplotlib.colors.Colormap The colormap to register. name : str, optional The name for the colormap. If not given, ``cmap.name`` is used. force : bool, default: False If False, a ValueError is raised if trying to overwrite an already registered name. True supports overwriting registered colormaps other than the builtin colormaps. r%zA colormap named "z" is already registered.z Re-registering the builtin cmap z is not allowed.zOverwriting the cmap z" that was already in the registry.N) rcheck_isinstancerColormapr ValueErrorr0 warn_externalrr.)r1r%rrKs r'registerzColormapRegistry.registervs2 foD9999 ty 4<< >!GGGGIII,,, "=$("="="=>>>   Ct C C C D D D!IIKK D ;t  !T ) )%)DK  " " " * )r)ct||jvrtd|d|j|ddS)a Remove a colormap from the registry. You cannot remove built-in colormaps. If the named colormap is not registered, returns with no error, raises if you try to de-register a default colormap. .. warning:: Colormap names are currently a shared namespace that may be used by multiple packages. Use `unregister` only if you know you have registered that name before. In particular, do not unregister just in case to clean the name before registering a new colormap. Parameters ---------- name : str The name of the colormap to be removed. Raises ------ ValueError If you try to remove a default built-in colormap. zcannot unregister z which is a builtin colormap.N)r0rPr.pop)r1rs r' unregisterzColormapRegistry.unregistersV6 4& & &)$)))** * d#####r)c B||tjdSt|tjr|St|t r0t jtt|||Stdd|dt|z)a Return a color map specified through *cmap*. Parameters ---------- cmap : str or `~matplotlib.colors.Colormap` or None - if a `.Colormap`, return it - if a string, look it up in ``mpl.colormaps`` - if None, return the Colormap defined in :rc:`image.cmap` Returns ------- Colormap Nz image.cmaprMzrssP $#############BBBBBB 777777CCCCCC555555 < $Dd d d d d wd d d T 0022 3 3  &&~66##K00r)