U \ ¸a.ìã@sØdZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZddlmZddlmZddlmZe e¡Z dZ!dZ"dZ#ej$j%de &e!d ¡e &e"d ¡e#fd ej$j'd Z(e#Z)e!e"Z*d d „Z+Gdd„de j,ƒZ-Gdd„de j.ƒZ/Gdd„de j0ƒZ1Gdd„dej2ƒZ3Gdd„dƒZ4dd„Z5Gdd„de4ƒZ6dd„Z7e 8e!e"¡d,d!d"„ƒZ9e 8e!e"¡ddddd d#œd$d%„ƒZ:ej;d&dd'Gd(d)„d)e6ƒƒZs ÿz_ColorbarSpine.__init__NcCstjj||dS)N)Úrenderer)ÚmpatchesÚPatchÚget_window_extent)rr@r r rrCBsz _ColorbarSpine.get_window_extentcCstj|dd|_d|_dS)NTr:)r<r=Ú_pathÚstale)rÚxyr r rÚset_xyHsz_ColorbarSpine.set_xycCstj ||¡}d|_|S)NF)rArBÚdrawrE)rr@Úretr r rrHLsz_ColorbarSpine.draw)N)r,r-r.rrCrGrHr0r r rrr8=s  r8c@s&eZdZdZdZe dd¡dHd d „ƒZdd„Zdd„Z dd„Z e  d¡dd„ƒZ dd„Z dd„Zdd„Zdd„Zdd„ZdId d!„ZdJd"d#„ZdKd$d%„Zd&d'„Zd(d)„Zdd*œd+d,„Zd-d.„Zd/d0„Zd1d2„ZdLd3d4„Zd5d6„ZdMd7d8„ZdNd:d;„Zdd?„Z d@dA„Z!dBdC„Z"dDdE„Z#dFdG„Z$dS)OÚ ColorbarBaseaJ 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` (empty if no lines were drawn). dividers : `.LineCollection` A LineCollection (empty if *drawedges* is ``False``). 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 é2ú3.3ÚcmapNÚverticalrÚuniformFTÚcCshtjtjdg|dtjddg|dtjddddd g|d tjd d g| d ||_t|_|_|j dd|dkr~t   ¡}|dkrŽt  ¡}| dkr¬t |dƒr¨|j} nd} ||_||_||_||_||_| |_tjtddƒtddƒtddƒtddƒdœ| d|_| |_||_| |_||_||_||_d|_g|_g|_|j  ¡D]}| !d¡qBt"|ƒ|_#|j d<t$j%t& 'd¡t(j)dddd|_*| +|j*¡t,j-gt(j)dgdt(j)dgd|_.|j /|j.¡d|_0d|_1d|_2d|_3|dkrü|dkrød nd}||_4| 5|¡| 6¡t& 7| ¡r6t8j9| t:| ƒd |_0n| |_0t;| t<ƒrVt8 =| ¡|_1n| |_1| >¡dS)!N)rMrNÚ horizontal©Ú orientationrÚleftÚrightÚtopÚbottom)Ú ticklocationrOZ proportional)ÚspacingF)ZnavigateÚextendÚneitherrréÿÿÿÿ)r[Úbothr%r"©rZÚoutliner9úaxes.facecolorç{®Gáz„?©ÚcolorÚ linewidthZzorderzaxes.edgecolorçà?zaxes.linewidth)rÚ linewidths©r)?rZcheck_isinstancerZColormapÚ check_in_listÚaxrZ set_xticksZ set_yticksÚsetrZget_cmapÚ NormalizeÚhasattrrZÚalpharMr#ÚvaluesÚ boundariesÚ check_getitemÚsliceÚ_insiderYrSÚ drawedgesÚfilledÚ extendfracÚ extendrectÚsolidsÚsolids_patchesÚlinesÚspinesÚ set_visibler8r_rAÚPolygonr6r>ÚmplÚrcParamsÚpatchÚ add_artistrÚLineCollectionÚdividersÚadd_collectionÚlocatorÚ formatterÚ_manual_tick_data_valuesÚ_ColorbarBase__scalerXÚ set_labelÚ_reset_locator_formatter_scaleÚiterablerÚ FixedLocatorÚlenÚ isinstanceÚstrZFormatStrFormatterÚdraw_all)rrirMr#rmrnrorSrXrZrYr)ÚformatrsrtrurvÚlabelÚspiner r rržsœÿ þÿ   ÿýþ  ý    zColorbarBase.__init__cCs |jdkS)z-Return whether the lower limit is open ended.©r]r%r^©rr r rÚ _extend_lowerszColorbarBase._extend_lowercCs |jdkS)z-Return whether the upper limit is open ended.©r]r"r^r”r r rÚ _extend_upperszColorbarBase._extend_upperc Cs(| ¡| ¡|j|jddg\|_|_| ¡\}}t |ddddgdf|dddddgdfg¡}t |ddddgdf|dddddgdfg¡}t  ||g¡}|j dd\}}|j dd\}} |j j ||f|| fd|j |¡|j |¡| ¡|jr$| |||jdd…tjf¡dS)zs Calculate any free parameters based on the current cmap and norm, and do all the drawing. rr\réþÿÿÿ©Zaxis)ZxlimZylimN)Ú _config_axisÚ_process_valuesÚ _boundariesrrr$r&Ú_meshr6Z concatenateZ column_stackr%r"rirjr_rGrÚ update_ticksrtÚ _add_solidsÚ_valuesZnewaxis) rÚXÚYÚxÚyrFZxminZyminZxmaxZymaxr r rr s 00  zColorbarBase.draw_allcCs | ¡dS©N)ršr”r r rÚ config_axis%szColorbarBase.config_axiscCs†|j}|jdkr2|j|j}}tjdrR| ¡n |j|j}}tjdrR| ¡|j|j|jd|  g¡|j gddd|_ dS)zSet up long and short axis.rNzytick.minor.visiblezxtick.minor.visible)Zlabel_positionZticks_positionT)ÚminorN) rirSÚyaxisÚxaxisr}r~Ú minorticks_onrjrXÚ set_ticksrE)rriÚ long_axisZ short_axisr r rrš)s    ÿ zColorbarBase._config_axiscCsn|j}|j}|dkrø|jdkrÞt|jtjƒrVt|jƒ}dt |dƒ}t j |dd}qøt|jtj ƒr||jj}t j |dd}qøt|jtjƒr”t|ƒ}qøt|jtjƒrÀt jt dd¡|jjdd}qøtjdrÔt  ¡}qøt|ƒ}n|j|j}t j |dd}|dkrHt|jtjƒrt  ¡}n*t|jtjƒr>t j|jjd }nt  ¡}n|j}||_||_t d |¡||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)ÚbaseÚoffsetrg)ÚsubsÚ linthreshr­z_internal.classic_mode)r°z locator: %r)r„r…rorr#rÚNoNormrŒr ÚintrZ IndexLocatorÚ BoundaryNormr‹ÚLogNormr4Z SymLogNormZSymmetricalLogLocatorr6Úaranger°r}r~Ú MaxNLocatorrrœrrÚLogFormatterSciNotationZScalarFormatterÚ_logÚdebug)rr„r…Únvr­Úbr r rÚ_get_ticker_locator_formatter:sJ    ý      ÿ  z*ColorbarBase._get_ticker_locator_formattercCs0|jdk o|jdk}t|jƒtjtjfko.| S)z‹ Return if we should use an adjustable tick locator or a fixed one. (check is used twice so factored out here...) NrO)rorYÚtyper#rrkr´)rZ contouringr r rÚ_use_auto_colorbar_locatorosÿz'ColorbarBase._use_auto_colorbar_locatorcCs|d|_d|_t|jtjƒrB|j d¡|j d¡|  ¡d|_ n6|j d¡|j d¡t |jƒtj krrd|_ nd|_ 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) NÚlogZlinearÚmanual) r„r…rr#rr´riZ set_xscaleZ set_yscalerªr‡r½rkr”r r rr‰xs    z+ColorbarBase._reset_locator_formatter_scalecCs”|j}| ¡\}}|jdkr"|jn|j}| ¡rRt d|¡| |¡|  |¡n>t d¡|  ||¡\}}}|  |¡|  |¡|  ¡ |¡dS)z” Force the update of the ticks and ticklabels. This must be called whenever the tick locator and/or tick formatter changes. rNz*Using auto colorbar locator %r on colorbarzUsing fixed locator on colorbarN)rir¼rSr¨r©r¾r¸r¹Zset_major_locatorZset_major_formatterÚ_tickerr«Úset_ticklabelsZget_major_formatterZset_offset_string)rrir„r…r¬r)Ú ticklabelsÚ offset_stringr r rržs       zColorbarBase.update_tickscCs<t |¡r tj|t|ƒd|_n||_|r2| ¡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. rgTN)r6rŠrr‹rŒr„ržrE)rr)ržr r rr«£s  zColorbarBase.set_tickscCs8|jdkr.|j}|jdkr |jn|j}| ¡S|jSdS)z*Return the x ticks as a list of locations.NrN)r†rirSr¨r©Zget_majorticklocs)rr§rir¬r r rÚ get_ticks¼s  ÿzColorbarBase.get_tickscCs<t|jtjƒr(t |¡|_|r2| ¡n t d¡d|_ 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) rr„rr‹ZFixedFormatterr…ržrr rE)rrÃržr r rrÂÈs    zColorbarBase.set_ticklabelscCs^|j}|jdkr|jn|j}| ¡dkrL| t|ddd¡| t  ¡¡n| t |ƒ¡dS)zn Turn the minor ticks of the colorbar on without extruding into the "extend regions". rNr¿g$@r)r­r¯N) rirSr¨r©Z get_scaleÚset_minor_locatorr4Zset_minor_formatterrr·r1©rrir¬r r rrª×s  ÿzColorbarBase.minorticks_oncCs.|j}|jdkr|jn|j}| t ¡¡dS)z)Turn the minor ticks of the colorbar off.rNN)rirSr¨r©rÆrZ NullLocatorrÇr r rÚminorticks_offæszColorbarBase.minorticks_off)ÚloccKsF|jdkr$|jj|fd|i|—Žn|jj|fd|i|—Žd|_dS)a¤ Add a label to the long axis of the colorbar. Parameters ---------- label : str The label text. loc : str, optional The location of the label. - For horizontal orientation one of {'left', 'center', 'right'} - For vertical orientation one of {'bottom', 'center', 'top'} Defaults to :rc:`xaxis.labellocation` or :rc:`yaxis.labellocation` depending on the orientation. **kwargs Keyword arguments are passed to `~.Axes.set_xlabel` / `~.Axes.set_ylabel`. Supported keywords are *labelpad* and `.Text` properties. rNrÉTN)rSriZ set_ylabelZ set_xlabelrE)rr‘rÉr5r r rrˆìs zColorbarBase.set_labelcCs |jdk r|j ¡|jD] }| ¡qt|ddƒ}t|tjƒrftdd„|jDƒƒrf|  ||||¡n|  |||¡|j   |j r–t ||g¡dd…ng¡dS)z+Draw the colors; optionally add separators.NÚmappablecss|]}|dk VqdSr¥r )Ú.0Úhatchr r rÚ sz+ColorbarBase._add_solids..rr\)rwÚremoverxÚgetattrrrÚ ContourSetÚanyÚhatchesÚ_add_solids_patchesÚ_add_solids_pcolormeshr‚Z set_segmentsrsr6Zdstack)rr¡r¢ÚCZsolidrÊr r rrŸs      ÿÿzColorbarBase._add_solidsc Cstt d¡|jd|jdkr*|dd…}|jj||||j|j|jddd|_|j spt |j ƒ|j krp|j  d¡dS)NzSetting pcolormeshrr\ÚnoneZflat)rMr#rmZ edgecolorsZshadingT)r¸r¹ÚshaperiZ pcolormeshrMr#rmrwrsrŒÚ_yÚ n_rasterizeZset_rasterized)rr¡r¢rÕr r rrÔs   þz#ColorbarBase._add_solids_pcolormeshc Csð|jt|ƒ}g}tt|ƒdƒD]Â}t ||df||dfg||df||dfg||ddf||ddfg||ddf||ddfgg¡}tjt |¡|  |  ||d¡¡||dd|j d} |j   | ¡| | ¡q"||_dS)NrrF)Z facecolorrÌrdZ antialiasedrm)rÒrŒÚranger6ÚarrayrAZ PathPatchr<r=rMr#rmriZ add_patchÚappendrx) rr¡r¢rÕrÊrÒÚpatchesÚirFrr r rrÓ%s$ý ý  z ColorbarBase._add_solids_patchesc Cs0| |¡}|jd|jdd}||jd|k||jd|k@}||}t |¡rft |¡|}t |¡r~t |¡|}t |jd|jdg|¡\}} |jdkr¼tj|| gdd} ntj| |gdd} tj | |d} |r|j r|j D] } |   ¡qðg|_ |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. r\rr!rNr™)rfTN)Ú_locaterØr6rŠÚasarrayÚmeshgridrSÚstackrrryrÎrÜZ set_colorrirƒrE) rÚlevelsrrfÚeraser¤r*Zigoodr¡r¢rFÚcolÚlcr r rÚ add_lines5s* $         zColorbarBase.add_linesc Cs0t|jtjƒr.|jdkr.|jd|jdf}n |j|jf}|j|dd|j|dd|j |Ž|j |Ž|j |Ž|j |Žt   |ƒ¡}t|t jƒrÌd}|||dd|k||dd|k@}n8|d|dd}|||d|k||d|k@}||_| |¡}| |¡}| ¡}|||fS)zŒ Return the sequence of ticks (colorbar data locations), ticklabels (strings), and the corresponding offset string. Nrr\)Zminposr!r)rr#rr±ror r$r&Zcreate_dummy_axisZset_view_intervalZset_data_intervalr6rÛrÚ LogLocatorr†rßZ format_ticksZ get_offset) rr„r…Zintvr»Zepsr)rÃrÄr r rrÁ`s(       .$  zColorbarBase._tickercCs”|dkr|j}|dk rˆtj|td|_|jdkrvd|jdd…|jdd…|_t|jt j ƒr„|jd  tj ¡|_nt  |j¡|_dS|jdk r6t  |j¡|_|jdkr$t t|jƒd¡}d|jdd…|jdd…|dd…<d|d|d|d <d|d |d |d<||_dSt  |j¡|_dSt|jt j ƒrÎ| |jjd¡|jjd}tjt|ƒdtj d}tj|jjtj d||j<| ¡r¨d|d <| ¡r¾|jj|d<||_||_dSt|jt jƒr¬t|jjƒ}| ¡r|d dg|}| ¡r"||ddg}t  |¡}t t|ƒd¡}|jj}d|dd…|dd…||j<| ¡r‚|d d|d <| ¡rœ|dd|d<||_||_dS|j ¡sÈd |j_d|j_tj|jj|jjd d \|j_|j_|j | |jjd¡¡}t|jt jt j fƒrR| ¡r6d|d |d <| ¡r†d|d|d<n4| ¡rl|d d|d <| ¡r†|dd|d<| !|¡dS)zÆ Set the :attr:`_boundaries` and :attr:`_values` attributes based on the input boundaries and values. Input boundaries can be *self.boundaries* or the argument *b*. N)Zdtyperer\rgñh㈵øä>g@rrr˜éýÿÿÿçš™™™™™¹?)ZexpandergÍÌÌÌÌÌì?gš™™™™™ñ?)"ror6ràÚfloatrœrnr rr#rr±ZastypeZint16rÛÚzerosrŒÚ _uniform_yrMÚNrµrrr•r—r³ÚlistZscaledr$r&Ú mtransformsZ nonsingularÚinverseZ PowerNormr´r›)rr»ÚvZbir r rr›}sŒ  ÿ  (       "   ý    zColorbarBase._process_valuesçš™™™™™©?c Cs t ||g¡}t|tƒr>tjdg| ¡d||g|dd…<n^|dk rœz$||dd…<t |¡ ¡rht ƒ‚Wn0t t fk rš}zt dƒ|‚W5d}~XYnX|S)z Return the lengths of colorbar extensions. This is a helper method for _uniform_y and _proportional_y. r)ruNzinvalid value for extendfrac) r6rÛrrŽrrhÚlowerÚisnanrÑÚ ValueErrorÚ TypeError)rÚfracÚautominÚautomaxÚdefaultÚ extendlengthÚerrr r rÚ_get_extension_lengthsÒs   z#ColorbarBase._get_extension_lengthscCsÞ|jdkrt dd|¡}nÀd|d}}|j|j||dd}|jdkrzt |dd ¡}d |d|d<d|d|d <nL|jd kr¦t |dd ¡}d |d|d<n t |dd ¡}d|d|d <t dd|¡||j<|S) zv Return colorbar data coordinates for *N* uniformly spaced boundaries, plus ends if required. r[rrçð?ró©rûr]rÚdçr\r%)rZr6Zlinspacerþrurìrr)rrîr¤rùrúrür r rríës& þ  zColorbarBase._uniform_ycCs¸t|jtjƒr8|j|jd}||jd|jd}n | |j ¡¡}tj |tj ¡}|j dkrœ|d|d}|d|d|}|d|d|}nˆ|j dkrà|d|d}|d|d|}|d|d|}nD|j d kr$|d|d}|d|d|}|d|d|}|j d krD|j |j ||d d }|j d kr`d|d|d<|j dkr|d|d|d<||j }t |d|d¡}tj ||ƒtj ¡||j <|S)zi Return colorbar data coordinates for the boundaries of a proportional colorbar. rr\r%rrr˜r"rér])r]r%r"rórr“rr–rÿ)rr#rr³rœÚcopyr6ÚmartÚnanrZrþrurrrk)rr¤ÚclenrùrúrüÚyir#r r rÚ_proportional_ys>    þ   zColorbarBase._proportional_yc Cs6t |j¡}|j|_|j|_t ddg¡}|jdkrPt|j|j ƒ}|  |¡}n|  ¡}t dg¡}|j dkrŽ|  |¡}|  |¡}|  |¡}n6|j|j}|||j}|||j}|||j}||_t ||¡\}}| ¡rø|jsø||ddd…f<| ¡r|js||ddd…f<|jd kr.||fS||fS) zµ Return the coordinate arrays for the colorbar pcolormesh/patches. These are scaled between vmin and vmax, and already handle colorbar orientation. rrÿrOrerÀrNr\rN)rr#r$r&r6rÛrYrŒrœrrrírr‡rñrØrár•rvr—rS) rr#r£Zn_boundaries_no_extensionsr¤ZxmidZdvr¡r¢r r rr+s0         zColorbarBase._meshcCs²t|jtjtjfƒr |j}|}n&|j|jdd ¡}|j|dd ¡}|}|j}|d|dkrx|dd…}|dd…}|d|dkr |dd…}|dd…}t  |||¡}|S)zq Given a set of color data values, return their corresponding colorbar data coordinates. F)ZcliprrNr\r˜) rr#rr±r³rœrtrØr6Zinterp)rr£r»ZxnZbuniqueZyuniqueÚzr r rrßSs    zColorbarBase._locatecCs ||_dS)z  úÿ  ÿzColorbar.__init__rLr©Ú alternativecCst d¡| |¡dS)zÙ Update this colorbar to match the mappable's properties. Typically this is automatically registered as an event handler by :func:`colorbar_factory` and should not be called manually. zcolorbar mappable changedN)r¸r¹r)rrÊr r rÚon_mappable_changed³s zColorbar.on_mappable_changedTcsTt|tjƒr|jrtdƒ‚dd„|jDƒ}dd„|jDƒ}tƒj|j |||ddS)av Add the lines from a non-filled `~.contour.ContourSet` to the colorbar. Parameters ---------- CS : `~.contour.ContourSet` The line positions are taken from the ContourSet levels. The ContourSet must not be filled. erase : bool, default: True Whether to remove any previously added lines. z+add_lines is only for a ContourSet of linescSsg|] }|d‘qS©rr )rËÚcr r rÚ Ìsz&Colorbar.add_lines..cSsg|] }|d‘qSrr )rËÚtr r rr!Ís)räN) rrrÐrtröÚtcolorsÚ tlinewidthsrrçrã)rÚCSrär#r$rr rrç¾s zColorbar.add_linescCs€t d|j|j¡||_| | ¡¡|j|_|j|jkrJ|j|_| ¡| ¡t |jt j ƒrv|j}|j sv|  |¡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)r¸r¹r#rÊr rrMr‰rrrrÐrtrçrE)rrÊr%r r rrÑs   zColorbar.update_normalcCs¾|j ¡d|_d|_|jj ¡D]}| d¡q"t|jƒ|_|jjd<t j t   d¡t jdddd|_|j |j¡d|_g|_| |¡| ¡t|jtjƒrº|j}|jsº| |¡dS) 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. NFr_r9r`rar\rb)riZclar„r…rzrnr{r8r_rAr|r6r>r}r~rr€rwryrrrrÊrrÐrtrç)rrÊr’r%r r rÚupdate_bruteforceís*  þ zColorbar.update_bruteforcecs¢tƒ ¡|jj |jj¡d|j_d|j_z |jj}Wntk rNYdSXz|  ¡  ¡}|  ¡}Wn*tk r’|j dd}|  |¡Yn X| |¡dS)z¢ Remove this colorbar from the figure. If the colorbar was created with ``use_gridspec=True`` the previous gridspec is restored. NT©Úoriginal)rrÎrÊrZ disconnectrrr?ÚAttributeErrorÚget_subplotspecZ get_gridspecZget_topmost_subplotspecÚ get_positionÚ _set_positionÚset_subplotspec)rriÚgsZ subplotspecÚposrr rrÎs     zColorbar.remove)T) r,r-r.r/rrr rrçrr&rÎr0r r rrr†s #    (rc Cs†|dkrtjddddœ|d}tjddddd d œddddd d œd d ddd d œdd dddd œdœ|d}|dk r‚||dkr‚tdƒ‚|S)NrUrW)NrNrQrRrTrN)rÿre)rrerê)ÚlocationrSÚanchorÚpanchorÚpadrórVrQ)rer)rerÿç333333Ã?)rTrUrVrW)r0rSz/location and orientation are mutually exclusive)rrpr÷)r0rSÚ loc_settingsr r rÚ_normalize_location_orientation3s@ þÿÿÿÿù ÷ r6r4rÿéc s*t||ƒ}|d|d<|d}|d<| d|d¡}| d|d¡} t |¡} t |¡ ¡}|d ¡‰ˆ ¡rvdn|d} | d| ¡} t‡fd d „|Dƒƒs¨t d ƒ‚t j   d d „|Dƒ¡} | }|dkr |dkrì|  ||| ¡\}}}n|  d|| d|¡\}}}| d|¡ ||¡}n\|dkrB| ||| ¡\}}}n| d|| d|¡\}}}| |d¡ ||¡}d|}t  | |¡}|D]@}| |jdd¡}t   |¡}| |¡| dk rŒ| | ¡qŒˆj|dd}|D]}|j|g7_qàt||||| ||| d|_|j||dd||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. %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 rSr0rXr1r2rrór3c3s|]}ˆ| ¡kVqdSr¥)Ú get_figure©rËri©Úfigr rrÍvszmake_axes..zJUnable to create a colorbar axes as not all parents share the same figure.cSsg|]}|jdd ¡‘qS)Tr')r+Úfrozenr9r r rr!|szmake_axes..©rTrUrTrrÿrWTr'Fú ©r‘)r0ÚparentsÚshrinkr1r2ÚfractionÚaspectr3Úbox©r1Z adjustable)r6Úpopr6rŠZ atleast_1dZravelr8Zget_constrained_layoutÚallrörðZBboxÚunionZsplitxZshrunkZanchoredZsplityZ BboxTransformZ transformr+r,Ú set_anchorZadd_axesZ _colorbarsÚdictZ_colorbar_infoÚ set_aspect)r@r0rSrBrArCr r5r1Z parent_anchorZparents_iterableZpad0r3Z parents_bboxZpbZpbcbÚ_Zpb1Zshrinking_transriZnew_posnÚcaxÚar r:rÚ make_axesHs`      ÿ      ø rO)r0rSrBrArCcKs>t||ƒ}|d|d<|d}|d<| d|d¡}| d|d¡} | d|d¡} d| d| } |d kr*d|dd|||dd|g} |d krä| ¡jdd| |d|| gd } | d}| d jd dd | dd}nD| ¡jdd| d|| |gd } | d }| djd dd | dd}nÊ|d d||d|d d|g}|dkr¨| ¡jdd| d|| |gd} | d }| djdd d |d d}d|}nL| ¡jdd| |d|| gd} | d}| d jdd d |d d}d|}| |¡| |d¡| ¡}|j|dd}|j||ddd||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` 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 ---------------- %s rSr0rXr1r2r3rrr=rT)ZwspaceÚ width_ratiosré)ZhspaceÚ height_ratiosrWr>r?rDrE) r6rFr*Z subgridspecr-rIr8Z add_subplotrK)Úparentr0rSrBrArCr r5r1r2r3Zwh_spacerRr.Zss_mainZss_cbrPr;rMr r rÚmake_axes_gridspec®s¬)   ÿþÿÿþÿÿÿ þÿÿ þÿÿ rTz3.4rc@s eZdZdS)Ú ColorbarPatchN)r,r-r.r r r rrUsrUcKst||f|ŽS)aÈ 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` The created colorbar instance. )r)rMrÊr5r r rÚcolorbar_factorysrV)NNr4rÿr7)>r/rÚloggingÚtextwrapZnumpyr6Z matplotlibr}rrrrrrZmatplotlib.artistZartistrZmatplotlib.patchesrÝrAZmatplotlib.pathÚpathr<Zmatplotlib.spinesrzZmspinesZmatplotlib.transformsZ transformsrðrÚ getLoggerr,r¸Z_make_axes_param_docZ_make_axes_other_param_docZ_colormap_kw_docZinterpdÚupdateÚindentÚparamsr Zcolormap_kw_docZmake_axes_kw_docrr¶rZAutoMinorLocatorr1rèr4ZSpiner8rJrrr6Z SubstitutionrOrTr rUrVr r r rÚsp         ?C  þ½H ! - . ÿ e ÿc