^jdZddlmZddlZddlmZmZddlmZ ddl m Z ddl m ZddlmZddlm Zej(d Zd ZGd d eZeeZej5eeeZeeZy) 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]i\}}d|vrtj||t nAd|vrtj |d|n$tjj||t ||<kddddd}|jD]$\}}||j}||_ |||<&t|jD]!}|j}|||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 F/opt/ringagent/.cad-venv/lib/python3.12/site-packages/matplotlib/cm.py_gen_cmap_registryr( s  FkkmL d}  * *4x @4  ! !$x.$ 7  * * 4 4T4 J t L   G!( }m$))+ u  V]]_%!}} tyy! McNeZdZdZdZdZdZdZdZdZ dd d d Z d Z d Z y)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) c2||_t||_yN)_cmapstuple_builtin_cmaps)selfrs r'__init__zColormapRegistry.__init__Zs #Elr)cntj|j|t}|j S)N)colormap _error_cls)rgetitem_checkedr.KeyErrorr)r1itemr%s r' __getitem__zColormapRegistry.__getitem__^s&##DKK$8Tyy{r)c,t|jSr-)iterr.r1s r'__iter__zColormapRegistry.__iter__bsDKK  r)c,t|jSr-)lenr.r<s r'__len__zColormapRegistry.__len__es4;;r)c8ddjd|DzS)Nz'ColormapRegistry; available colormaps: z, c3(K|] }d|d yw)'N).0rs r' z+ColormapRegistry.__str__..js7$AdV1+7s)joinr<s r'__str__zColormapRegistry.__str__hs!: 7$778 9r)ct|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__lsDzr)NF)rforcectjtj||xs |j}||vrG|st d|d||j vrt d|dtjd|d|j|j|<|j|j|k7r||j|_yy) 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 fooD9 tyy 4<!(.FGII,,, !C$(8+;"=>>   !6th?C C D!IIK D ;;t  ! !T )%)DKK  " *r)cv||jvrtd|d|jj|dy)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.unregistersB6 4&& &1$:))* * d#r)c ||tjdSt|tjr|St|t r)t 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 z image.cmaprMzrss $#B 7C5 << $Dd wd T02 3  &~6#K0r)