o i@sdZddlZddlZddlZddlZddlmZ ddl m Z ddl m Z ddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZ ddl!m"Z#ddl$m%Z&ddlm'Z'e(e)Z*dZ+dZ,e+e,Z-dZ.de-e.fZ/e'j0j1e/dd d Z2Gd d d ej3Z4Gd ddej5Z6Gdddej7Z8GdddZ9ddZ:Gddde9Z;e'Gdd d e;Z?d!d"Z@dS)$a Colorbars are a visualization of the mapping from scalar values to colors. In Matplotlib they are drawn into a dedicated `~.axes.Axes`. .. note:: Colorbars are typically created through `.Figure.colorbar` or its pyplot wrapper `.pyplot.colorbar`, which use `.make_axes` and `.Colorbar` internally. As an end-user, you most likely won't have to call the methods or instantiate the classes in this module explicitly. :class:`ColorbarBase` The base class with full colorbar drawing functionality. It can be used as-is to make a colorbar for a given colormap; a mappable object (e.g., image) is not needed. :class:`Colorbar` On top of `.ColorbarBase` this connects the colorbar with a `.ScalarMappable` such as an image or contour plot. :class:`ColorbarPatch` A specialized `.Colorbar` to support hatched contour plots. :func:`make_axes` Create an `~.axes.Axes` suitable for a colorbar. This functions can be used with figures containing a single axes or with freely placed axes. :func:`make_axes_gridspec` Create a `~.SubplotBase` suitable for a colorbar. This function should be used for adding a colorbar to a `.GridSpec`. N) docstringa fraction : float, default: 0.15 Fraction of original axes to use for colorbar. shrink : float, default: 1.0 Fraction by which to multiply the size of the colorbar. aspect : float, default: 20 Ratio of long to short dimensions. a pad : float, default: 0.05 if vertical, 0.15 if horizontal Fraction of original axes between colorbar and new image axes. anchor : (float, float), optional The anchor point of the colorbar axes. Defaults to (0.0, 0.5) if vertical; (0.5, 1.0) if horizontal. panchor : (float, float), or *False*, optional The anchor point of the colorbar parent axes. If *False*, the parent axes' anchor will be unchanged. Defaults to (1.0, 0.5) if vertical; (0.5, 0.0) if horizontal. a ============ ==================================================== Property Description ============ ==================================================== *extend* {'neither', 'both', 'min', 'max'} If not 'neither', make pointed end(s) for out-of- range values. These are set for a given colormap using the colormap set_under and set_over methods. *extendfrac* {*None*, 'auto', length, lengths} If set to *None*, both the minimum and maximum triangular colorbar extensions with have a length of 5% of the interior colorbar length (this is the default setting). If set to 'auto', makes the triangular colorbar extensions the same lengths as the interior boxes (when *spacing* is set to 'uniform') or the same lengths as the respective adjacent interior boxes (when *spacing* is set to 'proportional'). If a scalar, indicates the length of both the minimum and maximum triangular colorbar extensions as a fraction of the interior colorbar length. A two-element sequence of fractions may also be given, indicating the lengths of the minimum and maximum colorbar extensions respectively as a fraction of the interior colorbar length. *extendrect* bool If *False* the minimum and maximum colorbar extensions will be triangular (the default). If *True* the extensions will be rectangular. *spacing* {'uniform', 'proportional'} Uniform spacing gives each discrete color the same space; proportional makes the space proportional to the data interval. *ticks* *None* or list of ticks or Locator If None, ticks are determined automatically from the input. *format* None or str or Formatter If None, `~.ticker.ScalarFormatter` is used. If a format string is given, e.g., '%.3f', that is used. An alternative `~.ticker.Formatter` may be given instead. *drawedges* bool Whether to draw lines at color boundaries. *label* str The label on the colorbar's long axis. ============ ==================================================== The following will probably be useful only in the context of indexed colors (that is, when the mappable has norm=NoNorm()), or other unusual circumstances. ============ =================================================== Property Description ============ =================================================== *boundaries* None or a sequence *values* None or a sequence which must be of length 1 less than the sequence of *boundaries*. For each region delimited by adjacent entries in *boundaries*, the color mapped to the corresponding value in values will be used. ============ =================================================== a} Add a colorbar to a plot. Function signatures for the :mod:`~matplotlib.pyplot` interface; all but the first are also method signatures for the `~.Figure.colorbar` method:: colorbar(**kwargs) colorbar(mappable, **kwargs) colorbar(mappable, cax=cax, **kwargs) colorbar(mappable, ax=ax, **kwargs) Parameters ---------- mappable The `matplotlib.cm.ScalarMappable` (i.e., `~matplotlib.image.AxesImage`, `~matplotlib.contour.ContourSet`, etc.) described by this colorbar. This argument is mandatory for the `.Figure.colorbar` method but optional for the `.pyplot.colorbar` function, which sets the default to the current image. Note that one can create a `.ScalarMappable` "on-the-fly" to generate colorbars not attached to a previously drawn artist, e.g. :: fig.colorbar(cm.ScalarMappable(norm=norm, cmap=cmap), ax=ax) cax : `~matplotlib.axes.Axes`, optional Axes into which the colorbar will be drawn. ax : `~matplotlib.axes.Axes`, list of Axes, optional Parent axes from which space for a new colorbar axes will be stolen. If a list of axes is given they will all be resized to make room for the colorbar axes. use_gridspec : bool, optional If *cax* is ``None``, a new *cax* is created as an instance of Axes. If *ax* is an instance of Subplot and *use_gridspec* is ``True``, *cax* is created as an instance of Subplot using the :mod:`~.gridspec` module. Returns ------- colorbar : `~matplotlib.colorbar.Colorbar` See also its base class, `~matplotlib.colorbar.ColorbarBase`. Notes ----- Additional keyword arguments are of two kinds: axes properties: %s colorbar properties: %s If *mappable* is a `~.contour.ContourSet`, its *extend* kwarg is included automatically. The *shrink* kwarg provides a simple way to scale the colorbar with respect to the axes. Note that if *cax* is specified, it determines the size of the colorbar and *shrink* and *aspect* kwargs are ignored. For more precise control, you can manually specify the positions of the axes objects in which the mappable and the colorbar are drawn. In this case, do not use any of the axes properties kwargs. It is known that some vector graphics viewers (svg and pdf) renders white gaps between segments of the colorbar. This is due to bugs in the viewers, not Matplotlib. As a workaround, the colorbar can be rendered with overlapping segments:: cbar = colorbar() cbar.solids.set_edgecolor("face") draw() However this has negative consequences in other circumstances, e.g. with semi-transparent images (alpha < 1) and colorbar extensions; therefore, this workaround is not used by default (see issue #1188). ) colorbar_doccOstddS)Nz,Use the colorbar set_ticks() method instead.)cbook_warn_external)argskwrN/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/matplotlib/colorbar.py_set_ticks_on_axis_warnsr c,eZdZdZfddZfddZZS)_ColorbarAutoLocatora AutoLocator for Colorbar This locator is just a `.MaxNLocator` except the min and max are clipped by the norm's min and max (i.e. vmin/vmax from the image/pcolor/contour object). This is necessary so ticks don't extrude into the "extend regions". cs&||_d}gd}tj||ddS)z This ticker needs to know the *colorbar* so that it can access its *vmin* and *vmax*. Otherwise it is the same as `~.ticker.AutoLocator`. auto)g@ )nbinsstepsN _colorbarsuper__init__)selfcolorbarrr __class__rr rsz_ColorbarAutoLocator.__init__csh||kr ||}}t||jjj}t||jjj}t||}||d}||||k|||k@SN绽|=)maxrnormvminminvmaxr tick_valuesrr r"ticksrtolrrr r#s  z _ColorbarAutoLocator.tick_values__name__ __module__ __qualname____doc__rr# __classcell__rrrr r s  r cs.eZdZdZdfdd ZfddZZS)_ColorbarAutoMinorLocatora* AutoMinorLocator for Colorbar This locator is just a `.AutoMinorLocator` except the min and max are clipped by the norm's min and max (i.e. vmin/vmax from the image/pcolor/contour object). This is necessary so that the minorticks don't extrude into the "extend regions". Ncs||_||_tjdddS)zo This ticker needs to know the *colorbar* so that it can access its *vmin* and *vmax*. N)n)rndivsrr)rrr.rrr rsz"_ColorbarAutoMinorLocator.__init__csF|jjj}|jjj}t}||d}||||k|||k@Sr)rrr r"r__call__r$rrr r0s    z"_ColorbarAutoMinorLocator.__call__N)r(r)r*r+rr0r,rrrr r-s  r-cr )_ColorbarLogLocatora LogLocator for Colorbarbar This locator is just a `.LogLocator` except the min and max are clipped by the norm's min and max (i.e. vmin/vmax from the image/pcolor/contour object). This is necessary so ticks don't extrude into the "extend regions". cs||_tj|i|dS)z This ticker needs to know the *colorbar* so that it can access its *vmin* and *vmax*. Otherwise it is the same as `~.ticker.LogLocator`. The ``*args`` and ``**kwargs`` are the same as `~.ticker.LogLocator`. Nr)rrrkwargsrrr r+sz_ColorbarLogLocator.__init__cs||kr ||}}t||jjj}t||jjj}t||}t |t |d}|t |t ||kt |t ||k@}|Sr) rrrr r!r"rr#nplog10r$rrr r#5s z_ColorbarLogLocator.tick_valuesr'rrrr r2!s  r2c@sfeZdZdZdZedd             dPd d ZddZddZ ddZ e dddZ ddZ ddZddZddZddZdQd d!ZdRd"d#ZdQd$d%Zd&d'Zd(d)Zd*d+Zdd,d-d.Zd/d0Zd1d2Zd3d4ZdQd5d6Zd7d8ZdSd9d:Zd;d<Zd=d>Z d?d@Z!dTdBdCZ"dDdEZ#dFdGZ$dHdIZ%dJdKZ&dLdMZ'dNdOZ(dS)U ColorbarBaseak Draw a colorbar in an existing axes. There are only some rare cases in which you would work directly with a `.ColorbarBase` as an end-user. Typically, colorbars are used with `.ScalarMappable`\s such as an `.AxesImage` generated via `~.axes.Axes.imshow`. For these cases you will use `.Colorbar` and likely create it via `.pyplot.colorbar` or `.Figure.colorbar`. The main application of using a `.ColorbarBase` explicitly is drawing colorbars that are not associated with other elements in the figure, e.g. when showing a colormap by itself. If the *cmap* kwarg is given but *boundaries* and *values* are left as None, then the colormap will be displayed on a 0-1 scale. To show the under- and over-value colors, specify the *norm* as:: norm=colors.Normalize(clip=False) To show the colors versus index instead of on the 0-1 scale, use:: norm=colors.NoNorm() Useful public methods are :meth:`set_label` and :meth:`add_lines`. Attributes ---------- ax : `~matplotlib.axes.Axes` The `~.axes.Axes` instance in which the colorbar is drawn. lines : list A list of `.LineCollection` if lines were drawn, otherwise an empty list. dividers : `.LineCollection` A LineCollection if *drawedges* is ``True``, otherwise ``None``. Parameters ---------- ax : `~matplotlib.axes.Axes` The `~.axes.Axes` instance in which the colorbar is drawn. cmap : `~matplotlib.colors.Colormap`, default: :rc:`image.cmap` The colormap to use. norm : `~matplotlib.colors.Normalize` alpha : float The colorbar transparency between 0 (transparent) and 1 (opaque). values boundaries orientation : {'vertical', 'horizontal'} ticklocation : {'auto', 'left', 'right', 'top', 'bottom'} extend : {'neither', 'both', 'min', 'max'} spacing : {'uniform', 'proportional'} ticks : `~matplotlib.ticker.Locator` or array-like of float format : str or `~matplotlib.ticker.Formatter` drawedges : bool filled : bool extendfrac extendrec label : str 23.3cmapNverticalr uniformFTcCsBtjtjdg|dtjddg|dtjgd|dtjddg| d ||_t|_|_|j d d d |dur=t }|durEt }| durTt |d rR|j} nd } ||_||_||_||_||_| |_tjtddtddtddtddd| d|_| |_||_| |_||_||_||_d|_g|_tj t!"dt#j$ddt#j$dddd|_%|&|j%|j%j dddtj t!"dt#j$dddd|_'|&|j'd|_(d|_)d|_*d|_+d|_,|dkr|dkrdnd }||_-|.||/t!0| rt1j2| t3| d!|_)n| |_)t4| t5rt16| |_*n| |_*|7dS)"N)r9r: horizontal) orientation)r leftrighttopbottom) ticklocationr; proportional)spacingF)frame_onnavigateextendneitherrr)rIbothr!rrHrraxes.edgecolornoneaxes.linewidthTr edgecolor facecolor linewidthclosedzorderclip_box clip_pathaxes.facecolor{Gz?colorrTrVr rBr@r)8r_check_isinstancecolorsColormap_check_in_listaxr set_xticks set_ytickssetcmget_cmap NormalizehasattrrHalphar9rvalues boundaries_check_getitemslice_insiderEr> drawedgesfilled extendfrac extendrectsolidslinesmpatchesPolygonr4emptymplrcParamsoutline add_artistpatchdividerslocator formatter_manual_tick_data_values_ColorbarBase__scalerC set_label_reset_locator_formatter_scaleiterableticker FixedLocatorlen isinstancestrFormatStrFormatterdraw_all)rrcr9rrkrlrmr>rCrHrEr%formatrqrrrsrtlabelrrr rs           zColorbarBase.__init__cC |jdvS)z-Return whether the lower limit is open ended.rKr!rLrrrr _extend_lower zColorbarBase._extend_lowercCr)z-Return whether the upper limit is open ended.rKrrLrrrr _extend_upperrzColorbarBase._extend_upperc Cs|||\}}|jddtjf}||||}|jdd\}}|j dd\}}|j j ||f||fd|j ||j |||jrZ||||dSdS)zs Calculate any free parameters based on the current cmap and norm, and do all the drawing. Nraxis)xlimylim)_process_values _find_range_mesh_valuesr4newaxis _config_axis_outliner!rrcrfr|set_xyr~ update_ticksrr _add_solids) rXYCxyxminyminxmaxymaxrrr rs    zColorbarBase.draw_allcCs |dSr1)rrrrr config_axiss zColorbarBase.config_axiscCs|j}|jdkr|j|j}}tjdr|n|j|j}}tjdr)||j|j|jd| g|j gdd| dS)zSet up long and short axis.r:zytick.minor.visiblezxtick.minor.visible)label_positionticks_positionT)minorN) rcr>yaxisxaxisrzr{ minorticks_onrfrC set_ticks _set_label)rrc long_axis short_axisrrr rs     zColorbarBase._config_axiscCsh|j}|j}|dur||jdurot|jtjr+t|j}dt |d}t j |dd}nQt|jtj r>|jj}t j |dd}n>t|jtjrJt|}n2t|jtjr`t jtdd|jjdd}ntjdrjt }nt|}n |j|j}t j |dd}|durt|jtjrt }nt|jtjrt j|jjd }nt }n|j}||_||_td |||fS) a Return the ``locator`` and ``formatter`` of the colorbar. If they have not been defined (i.e. are *None*), suitable formatter and locator instances will be created, attached to the respective attributes and returned. Nrrr)baseoffsetr^)subs linthreshrz_internal.classic_mode)rz locator: %r)rrrmrrr`NoNormrrintr IndexLocator BoundaryNormrLogNormr2 SymLogNormSymmetricalLogLocatorr4arangerrzr{ MaxNLocatorr _boundariesrpLogFormatterSciNotationScalarFormatter_logdebug)rrrnvrbrrr _get_ticker_locator_formatter&sJ           z*ColorbarBase._get_ticker_locator_formattercCs0|jduo |jdk}t|jtjtjfvo| S)z Return if we should use an adjustable tick locator or a fixed one. (check is used twice so factored out here...) Nr;)rmrEtyperr`rir)r contouringrrr _use_auto_colorbar_locator[sz'ColorbarBase._use_auto_colorbar_locatorcCsd|_d|_t|jtjr"|jd|jd| d|_ dS|jd|jdt |jtj ur;d|_ dSd|_ dS)z Reset the locator et al to defaults. Any user-hardcoded changes need to be re-entered if this gets called (either at init, or when the mappable normal gets changed: Colorbar.update_normal) Nloglinearmanual) rrrrr`rrc set_xscale set_yscalerrrrirrrr rds       z+ColorbarBase._reset_locator_formatter_scalecCs|j}|\}}|jdkr|jn|j}|r*td|||| |dStd| ||\}}}| || || |dS)z Force the update of the ticks and ticklabels. This must be called whenever the tick locator and/or tick formatter changes. r:z*Using auto colorbar locator %r on colorbarzUsing fixed locator on colorbarN)rcrr>rrrrrset_major_locatorset_major_formatter_tickerrset_ticklabelsget_major_formatterset_offset_string)rrcrrrr% ticklabels offset_stringrrr r{s      zColorbarBase.update_tickscCs<t|rtj|t|d|_n||_|r|d|_dS)a  Set tick locations. Parameters ---------- ticks : array-like or `~matplotlib.ticker.Locator` or None The tick positions can be hard-coded by an array of values; or they can be defined by a `.Locator`. Setting to *None* reverts to using a default locator. update_ticks : bool, default: True If True, tick locations are updated immediately. If False, the user has to call `update_ticks` later to update the ticks. r^TN)r4rrrrrrstale)rr%rrrr rs  zColorbarBase.set_tickscCs4|jdur|j}|jdkr|jn|j}|S|jS)z*Return the x ticks as a list of locations.Nr:)rrcr>rrget_majorticklocs)rrrcrrrr get_tickss zColorbarBase.get_tickscCs<t|jtjrt||_|r|ntdd|_ dS)z Set tick labels. Tick labels are updated immediately unless *update_ticks* is *False*, in which case one should call `.update_ticks` explicitly. z"set_ticks() must have been called.TN) rrrrFixedFormatterrrrrr)rrrrrr rs   zColorbarBase.set_ticklabelscCs`|j}|jdkr |jn|j}|dkr'|t|ddd|t dS|t |dS)zn Turn the minor ticks of the colorbar on without extruding into the "extend regions". r:rg$@r )rrN) rcr>rr get_scaleset_minor_locatorr2set_minor_formatterrrr-rrcrrrr rs  zColorbarBase.minorticks_oncCs.|j}|jdkr |jn|j}|tdS)z)Turn the minor ticks of the colorbar off.r:N)rcr>rrrr NullLocatorrrrr minorticks_offszColorbarBase.minorticks_offcCsF|jdkr|jj|jfi|jn |jj|jfi|jd|_dS)Nr:T)r>rc set_ylabel_label_labelkw set_xlabelrrrrr rs  zColorbarBase._set_labellocc s|jdkrdnd}|ddg}tfdd|Dr(|dur%td |d d }n |dur3tjd |}|jdkr@tjd |dntjd|d|dvrTd|<dd<n |dvr`d|<dd<||_|_| dS)z-Add a label to the long axis of the colorbar.r:yxhorizontalalignmenthacsg|]}|vqSrr).0kr3rr z*ColorbarBase.set_label..NzYSpecifying *loc* is disallowed when any of its corresponding low level keyword arguments z are also suppliedcenterz%saxis.labellocation)rBrrAr)r?rr@)r@rA?r@)r?rBr?) r>any TypeErrorrzr{rrbrrr)rrrr3_pos_xy _protected_kwrrr rs,     zColorbarBase.set_labelc Cs|jd}dd|d|dd|dd|d|d|dg }|jd|}|jd|}|jdkr:t||gSt||gS)zo Return *x*, *y* arrays of colorbar bounding polygon, taking orientation into account. rrrrJr=)shapeTreshaper>r4 column_stack)rrrNiirrrrr rs 2  zColorbarBase._outlinecsPjd}|jdkrfddtd|dDSfddtd|dDS)z;Return the separator line segments; helper for _add_solids.rr:cs"g|] }tt||qSrlistziprirrrr r"z'ColorbarBase._edges..rcs"g|] }tt||qSrrrrrr rr )rr>range)rrrrrrr _edgess  zColorbarBase._edgescCs.|jd|jdkr|dd}|jdkr|||f}nt|t|t|f}t|j|j|jdd}t d|j j |i|dd i}|j durP|j ||_ |jdur`|j d|_|jrd tjd f}tj|||tjd f|d |_|j |jdSt|j|jkr|j ddSdS)zd Draw the colors using `~.axes.Axes.pcolormesh`; optionally add separators. rNrJr:None)r9rrk edgecolorszSetting pcolormeshshadingflat?rPrNr` linewidthsT)rr>r4 transposedictr9rrkrrrc pcolormeshruremoverrqrzr{ collectionsLineCollectionr add_collectionr_y n_rasterizeset_rasterized)rrrrrrcolrrrr rs:          zColorbarBase._add_solidsc Cs,||}|jd|jdd}||jd|k||jd|k@}||}t|r3t||}t|r?t||}t|jd|jdg|\}} |jdkr^tj|| gdd} n tj| |gdd} tj | |d} |r|j r|j D]} | qvg|_ |j | | ||j| d|_dS) aQ Draw lines on the colorbar. The lines are appended to the list :attr:`lines`. Parameters ---------- levels : array-like The positions of the lines. colors : color or list of colors Either a single color applying to all lines or one color value for each line. linewidths : float or array-like Either a single linewidth applying to all lines or one linewidth for each line. erase : bool, default: True Whether to remove any previously added lines. rJrrr:r)rTN)_locaterr4rasarraymeshgridr>stackrrrvrappend set_colorrcrr) rlevelsr`reraserr&igoodrrrrlcrrr add_lines6s* $           zColorbarBase.add_linesc Cs0t|jtjr|jdur|jd|jdf}n|j|jf}|j|dd|j|dd|j ||j ||j ||j |t |}t|t jrfd}|||dd|k||dd|k@}n|d|dd}|||d|k||d|k@}||_||}||}|}|||fS)z Return the sequence of ticks (colorbar data locations), ticklabels (strings), and the corresponding offset string. NrrJ)minposrr)rrr`rrmrr r"create_dummy_axisset_view_intervalset_data_intervalr4arrayr LogLocatorrr format_ticks get_offset) rrrintvrepsr%rrrrr ras(       .$   zColorbarBase._tickercCs|dur|j}|durEtj|td|_|jdurg@rr皙?)expanderg?g?)"rmr4rfloatrrlrrrr`rastypeint16r-zerosr _uniform_yr9rrrprrrrscaledr r" mtransforms nonsingularinverse PowerNormrr)rrvbirrr r~s    (    "       zColorbarBase._process_valuescCs$|j|j}|d|_|d|_dS)z Set :attr:`vmin` and :attr:`vmax` attributes to the first and last boundary excluding extended end boundaries. rrJN)rrpr r")rrrrr rs  zColorbarBase._find_rangecCs6t|j}|jdkr|d8}|S|jdvr|d8}|S)z9Return the number of boundaries excluding end extensions.rKrr!rr)rrrH)rnbrrr _central_Ns   zColorbarBase._central_NcCs8|jjd}|jdkr|d7}|S|jdvr|d7}|S)ze Based on the colormap and extend variable, return the number of boundaries. rrKrrD)r9rrH)rrrrr _extended_Ns   zColorbarBase._extended_N皙?c Cst||g}t|tr tjdg|d||g|dd<|S|durKz||dd<t|r5t W|St t fyJ}zt d|d}~ww|S)z Return the lengths of colorbar extensions. This is a helper method for _uniform_y and _proportional_y. r )rsNzinvalid value for extendfrac) r4r-rrrrblowerisnanr ValueErrorr)rfracautominautomaxdefault extendlengtherrrrr _get_extension_lengthss"    z#ColorbarBase._get_extension_lengthscCs|jdkrtdd|}|Sd|d}}|j|j||dd}|jdkr>t|dd }d |d|d<d|d|d <n&|jd krTt|dd }d |d|d<nt|dd }d|d|d <tdd|||j<|S) zv Return colorbar data coordinates for *N* uniformly spaced boundaries, plus ends if required. rIrrrrHrOrKrdrrJr!)rHr4linspacerRrsr;rp)rrrrMrNrPrrr r< s&   zColorbarBase._uniform_ycCst|jtjr|j|jd}||jd|jd}n||j}tj|tj }|j dkrN|d|d}|d|d|}|d|d|}nC|j dkrp|d|d}|d|d|}|d|d|}n!|j d kr|d|d}|d|d|}|d|d|}|j d vr|j |j ||d d }|j d vrd|d|d<|j dvrd|d|d<||j }t|d|d}tj||tj ||j <|S)zi Return colorbar data coordinates for the boundaries of a proportional colorbar. rrJr!rrr4rr5rK)rKr!rrHrSrrrr)rrr`rrcopyr4marrnanrHrRrsrpri)rrclenrMrNrPyirrrr _proportional_y#s<       zColorbarBase._proportional_ycCst|j}|j|_|j|_tddg}|jdkr"||}n| }tdg}|j dkrA| |}| |}| |}n|j|j}|||j}|||j}|||j}||_ t ||\}}|rv|jsv||dddf<|r|js||dddf<||fS) a Return ``(X, Y)``, the coordinate arrays for the colorbar pcolormesh. These are suitable for a vertical colorbar; swapping and transposition for a horizontal colorbar are done outside this function. These are scaled between vmin and vmax. rrr;rrrNrJ)rVrr r"r4r-rEr<rFr[rr@rr rrtr)rrrrxmiddvrrrrr rJs.        zColorbarBase._meshcCst|jtjtjfr|j}|}n|j|jdd}|j|dd}|}|j}|d|dkr<|dd}|dd}|d|dkrP|dd}|dd}t |||}|S)zq Given a set of color data values, return their corresponding colorbar data coordinates. F)cliprrNrJr4) rrr`rrrrrrr4interp)rrrxnbuniqueyuniquezrrr rrs    zColorbarBase._locatecCs ||_dS)z.cSrrr)rtrrr rr)r%N) rrrrsrrrKtcolors tlinewidthsr6r(r$)rCSr%rrrrr r(s  zColorbar.add_linescCstd|j|j||_|||j|_|j|jkr%|j|_||t |jt j r;|j}|j s;| |d|_dS)a  Update solid patches, lines, etc. This is meant to be called when the norm of the image or contour plot to which this colorbar belongs changes. If the norm on the mappable is different than before, this resets the locator and formatter for the axis, so if these have been customized, they will need to be customized again. However, if the norm only changes values of *vmin*, *vmax* or *cmap* then the old formatter and locator will be preserved. zcolorbar update normal %r %rTN)rrrrqrertr9rrrrrrsrrr(rrrqrrrr r|s    zColorbar.update_normalcCs|jd|_d|_tjtdtj ddtj dddd|_ |j |j |j j ddd tjtdtj d d d d |_ |j |j d|_g|_d|_|||t|jtjrl|j}|jsn||dSdSdS)a Destroy and rebuild the colorbar. This is intended to become obsolete, and will probably be deprecated and then removed. It is not called when the pyplot.colorbar function or the Figure.colorbar method are used to create the colorbar. NrMrNrOrPTrrQrWrZr[rJr\)rcclarrrwrxr4ryrzr{r|r}rfr~rurvrr|rrrqrrrsrrr(rrrr update_bruteforces4     zColorbar.update_bruteforcecCst||jj|jjd|j_d|j_z|jj}Wn ty&YdSwz | }| }WntyH|j dd}| |YdSw||dS)z Remove this colorbar from the figure. If the colorbar was created with ``use_gridspec=True`` the previous gridspec is restored. NToriginal)r6rrq callbacksSM disconnect colorbar_cidraxesAttributeErrorget_subplotspec get_gridspecget_topmost_subplotspec get_position _set_positionset_subplotspec)rrcgs subplotspecposrrr r<s"       zColorbar.removeNrf) r(r)r*r+rrrhr~r(r|rrrrrr rms      ,rm333333?rc sgd}|dur|durtdd||dur |dur d}|dur,|dkr*dnd}tj||dd d d dd d d d dd ddd dd ddddd d}||} | d|d<||d<|d| d} |d| d} t|} t| }z|d } | j du}Wn t yd}Ynw| d}|rd}|d|}|dtfdd|Dstdtjdd|D}|}|dvr|d kr||||\}}}n|d!||d!|\}}}|d"|| |}n.|dkr||||\}}}n|d!||d!|\}}}||d"| |}d"|}t||}|D] }||jd#d$}t|}||| durF|| q'j|d%d&}|s_d}d}|j|| d'd(n!| st|d}tj ||||||d)\}}n t!||||||\}}||_ ||_"||fS)*a Create an `~.axes.Axes` suitable for a colorbar. The axes is placed in the figure of the *parents* axes, by resizing and repositioning *parents*. Parameters ---------- parents : `~.axes.Axes` or list of `~.axes.Axes` The Axes to use as parents for placing the colorbar. location : None or {'left', 'right', 'top', 'bottom'} The position, relative to *parents*, where the colorbar axes should be created. If None, the value will either come from the given ``orientation``, else it will default to 'right'. orientation : None or {'vertical', 'horizontal'} The orientation of the colorbar. Typically, this keyword shouldn't be used, as it can be derived from the ``location`` keyword. %s Returns ------- cax : `~.axes.Axes` The child axes. kw : dict The reduced keyword dictionary to be passed when creating the colorbar instance. Other Parameters ---------------- %s )r?r@rArBNz[position and orientation are mutually exclusive. Consider setting the position to any of {}z, r@r:rB)locationrrrrr6)anchorpanchorpadr>rHrrrrr=rr>rCrrrFrg{Gz?c3s|] }|uVqdSr1) get_figurerrcfigrr szmake_axes..zJUnable to create a colorbar axes as not all parents share the same figure.cSsg|] }|jddqS)Tr)rfrozenrrrr rszmake_axes..)r?r@r?rrTr rboxr adjustable)r)#rrjoinrrbpopr4r atleast_1dravelrr _layoutboxrrallrKr>Bboxunionsplitxshrunkanchoredsplity BboxTransform transformrr set_anchoradd_axes set_aspectconstrained_layoutlayoutcolorbarsinglelayoutcolorbargridspec _poslayoutbox)parentsrr>fractionshrinkaspectr locationsdefault_location_settings loc_settingsr parent_anchorparents_iterablerusing_constrained_layoutpad0r parents_bboxpbpbcb_pb1shrinking_transrcnew_posncaxlblbposrrr make_axesYs%                r)rrrcKsp|dd}d|d<d|}d|d}|||g}|} t| jtj} |dkrY|dd} d | d| } | dd || || |gd } | d d| dd |d } d}d}n0|dd} d | d| } | d d|| || |gd } | dd | dd |d } d|}d}d}| | d| | |j | ||}|j| ddd}|j||dd||fS)a Create a `~.SubplotBase` suitable for a colorbar. The axes is placed in the figure of the *parent* axes, by resizing and repositioning *parent*. This function is similar to `.make_axes`. Primary differences are - `.make_axes_gridspec` only handles the *orientation* keyword and cannot handle the *location* keyword. - `.make_axes_gridspec` should only be used with a `.SubplotBase` parent. - `.make_axes` creates an `~.axes.Axes`; `.make_axes_gridspec` creates a `.SubplotBase`. - `.make_axes` updates the position of the parent. `.make_axes_gridspec` replaces the ``grid_spec`` attribute of the parent with a new one. While this function is meant to be compatible with `.make_axes`, there could be some minor differences. Parameters ---------- parent : `~.axes.Axes` The Axes to use as parent for placing the colorbar. %s Returns ------- cax : `~.axes.SubplotBase` The child axes. kw : dict The reduced keyword dictionary to be passed when creating the colorbar instance. Other Parameters ---------------- orientation : {'vertical', 'horizontal'}, default: 'vertical' The orientation of the colorbar. %s r>r:r rCrrrrHr) subplot_specwspace width_ratiosr)rhspace height_ratiosrrrrrrrrrr)rwrr layoutboxnonetreergridspecGridSpecFromSubplotSpecrr update_paramsrfigboxrr add_subplotr)parentrrrrr>x1pad_s wh_ratiosrgs_from_subplotspecrwh_spacegs2rrrrrrr make_axes_gridspecs\ /          rc@s eZdZdZddZddZdS) ColorbarPatcha  A Colorbar that uses a list of `~.patches.Patch` instances rather than the default `~.collections.PatchCollection` created by `~.axes.Axes.pcolor`, because the latter does not allow the hatch pattern to vary among the members of the collection. cKs g|_tj|||fi|dSr1)solids_patchesrmr)rrcrqrrrr roszColorbarPatch.__init__c Cst|}|jj|}g}tt|dD]s}||d}||} t||d||dg||d||dg||dd||ddg||dd||ddgg} |jdkrg| ddddf} tjt | | | || dd|j d} |j| || q|jr|jD]} | q||_|jdur|jd|_|jrtj|||tjd fd tjd fd |_|j|jdSdS) zg Draw the colors using `~matplotlib.patches.Patch`; optionally add separators. rrr=.NrJF)rShatchrT antialiasedrkrNrrPr)rrqhatchesr r4r-r>rw PathPatchmpathPathr9rrkrc add_patchr"rrrrqrrr rzr{r) rrrr n_segmentsrpatchesrvalrrr~solidrrr rvsF            zColorbarPatch._add_solidsN)r(r)r*r+rrrrrr rgs rcKsft|tjrtdd|jDrt||fi|}n t||fi|}|jd|j }||_ ||_ |S)a1 Create a colorbar on the given axes for the given mappable. .. note:: This is a low-level function to turn an existing axes into a colorbar axes. Typically, you'll want to use `~.Figure.colorbar` instead, which automatically handles creation and placement of a suitable axes as well. Parameters ---------- cax : `~matplotlib.axes.Axes` The `~.axes.Axes` to turn into a colorbar. mappable : `~matplotlib.cm.ScalarMappable` The mappable to be described by the colorbar. **kwargs Keyword arguments are passed to the respective colorbar class. Returns ------- `.Colorbar` or `.ColorbarPatch` The created colorbar instance. `.ColorbarPatch` is only used if *mappable* is a `.ContourSet` with hatches. css|]}|duVqdSr1r)rrrrr rsz#colorbar_factory..changed) rrrrsrrrrmrconnectr|rr)rrqr3cbcidrrr colorbar_factorys r)NNrrr)Ar+rVloggingnumpyr4 matplotlibrzmatplotlib.artistartistrymatplotlib.cbookrmatplotlib.collectionsrmatplotlib.colorsr`matplotlib.contourrr matplotlib.cmrgmatplotlib.gridspecrmatplotlib.patchesrrwmatplotlib.pathpathrmatplotlib.tickerrmatplotlib.transforms transformsr>matplotlib._layoutboxrrmatplotlib._constrained_layout_constrained_layoutrr getLoggerr(r_make_axes_param_doc_make_axes_other_param_docmake_axes_kw_doccolormap_kw_docrinterpdupdater rr AutoMinorLocatorr-r.r2r6rlrm Substitutionrrrrrrrr sf!                ?MO! ]  5  )d @