o i?8@sddlZddlmZmZmZddlmZddl m Z ddl m Z ddl ZGdddZeddd d Zee ZGd d d Zeddd dZeeZGdddZeddddZddZee ZeeZe ddddZe ddddZdS)N)artistcbook transforms)subplot_class_factory)Bbox)Axescs<eZdZddZfddZfddZfddZZS) ParasiteAxesBasecCs6dd|D}dd|jD}t|t||fS)NcSh|]}|r|qS get_visible.0ar r `/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/mpl_toolkits/axes_grid1/parasite_axes.py z6ParasiteAxesBase.get_images_artists..cSr r r rr r rrr) get_childrenimageslist)selfartistsrr r rget_images_artists sz#ParasiteAxesBase.get_images_artistsc s,||_d|d<tj|j|jfi|dS)NFframeon) _parent_axessuper__init__figure _position)r parent_axeskwargs __class__r rrszParasiteAxesBase.__init__csdttj|dd|jj|_|jr$|j d|j ddS|j d|j ddS)NF)visibleg?g@) rclamartistsetprr _get_lines _axisbelowxaxis set_zorderyaxisrr"r rr%s    zParasiteAxesBase.clacs@t||D]}t|jdr||jjvr||q dS)N parasites)rpickrhasattrinaxesr.r mouseeventrr"r rr/'s     zParasiteAxesBase.pick)__name__ __module__ __qualname__rrr%r/ __classcell__r r r"rr s   r cCs0|dur tjdddt}td|jt|fiS)N3.3zSupport for passing None to parasite_axes_class_factory is deprecated since %(since)s and will be removed %(removal)s; explicitly pass the default Axes class instead.messagez %sParasite)rwarn_deprecatedrtyper4r  axes_classr r rparasite_axes_class_factory4s r?cseZdZdfdd ZddZddZdd Zd d Zd d ZfddZ fddZ ddZ fddZ fddZ dfdd ZZS)ParasiteAxesAuxTransBaseNc s(||_||tj|fi|dSN)transAuxset_viewlim_moderr)rr aux_transform viewlim_moder!r"r rrFs z!ParasiteAxesAuxTransBase.__init__cCsB|jj|_|j|jj|_t|j|j|_t|j|j|_dSrA)r transAxesrB transData mtransformsblended_transform_factory_xaxis_transform_yaxis_transformr-r r r_set_lim_and_transformsLs  z0ParasiteAxesAuxTransBase._set_lim_and_transformscCstjgd|d||_dS)NNequal transformmode)r_check_in_list _viewlim_mode)rrQr r rrCYs z)ParasiteAxesAuxTransBase.set_viewlim_modecCs|jSrA)rSr-r r rget_viewlim_mode]sz)ParasiteAxesAuxTransBase.get_viewlim_modecCsv|jj}|}|durdS|dkr|jj|dS|dkr0|jj||jdSt j gd|ddS)NrNrOrMrP) rviewLimfrozenrTaxesset transformedrBinvertedrrR)rviewlimrQr r rupdate_viewlim`s z'ParasiteAxesAuxTransBase.update_viewlimcOst|dkr$|d}|j\}}td|}td|}t||\} } n|\} } }d|vr9|| | |fi|} | S| j} t| j| jg} |j| }|dddf | }|dddf | }||||fi|} | |j j | S)NrrgrO) lenshapenparangemeshgrid column_stackflatrBrOreshape set_transformrrG)r super_pcolorXYCr!CnynxgxgyXYmesh orig_shapexytwxyr r r_pcolorms$       z ParasiteAxesAuxTransBase._pcolorc|jtjg|Ri|SrA)rsr pcolormeshrrgr!r"r rruz#ParasiteAxesAuxTransBase.pcolormeshcrtrA)rsrpcolorrvr"r rrxrwzParasiteAxesAuxTransBase.pcolorcOst|dkr&|d}|j\}}td|}td|}t||\} } |} n|dd\} } |dd} d|vrF|| | g| Ri|} | S| j} t| j| jg}|j|}|dddf | }|dddf | }|||g| Ri|} | j D] }| |j j q}| S)NrgrOr)r]r^r_r`rarbrcrBrOrd collectionsrerrG)r super_contourXYCLr!rhrirjrkrlrmrnCLcontrprqrrcr r r_contours*        z!ParasiteAxesAuxTransBase._contourcrtrA)rrcontourrr|r!r"r rrrwz ParasiteAxesAuxTransBase.contourcrtrA)rrcontourfrr"r rrrwz!ParasiteAxesAuxTransBase.contourfcs|tdSrA)r\r apply_aspect)rpositionr"r rrsz%ParasiteAxesAuxTransBase.apply_aspectrA)r4r5r6rrLrCrTr\rsrurxrrrrr7r r r"rr@Es     r@cCsN|durtjdddt}n t|tst|}n|}td|jt|fddiS)Nr8zSupport for passing None to parasite_axes_auxtrans_class_factory is deprecated since %(since)s and will be removed %(removal)s; explicitly pass the default ParasiteAxes class instead.r9z%sParasiteAuxTransname parasite_axes) rr; ParasiteAxes issubclassr r?r<r4r@)r>parasite_axes_classr r r$parasite_axes_auxtrans_class_factorys   rcseZdZfddZdefddZdfdd Zfd d Zfd d Zfd dZ dddZ ddZ dddZ ddZ dddZ  dfdd ZZS) HostAxesBasecsg|_tj|i|dSrA)r.rr)rargsr!r"r rrszHostAxesBase.__init__rNcCs.t|}||||}|j||jj|_|SrA)rr.appendremove_remove_method)rtrrEr>rax2r r r get_aux_axess    zHostAxesBase.get_aux_axesNcs*t}|jD] }|||q|SrA)r_get_legend_handlesr.extend)rlegend_handler_map all_handlesaxr"r rrs  z HostAxesBase._get_legend_handlesc st|j}t|j}t|dr,|}|r'|||}|j|dd||n |n||}|jD]}||| \}} |j ||j | q7t |||_||_dS)Nget_axes_locatoractive)which) rrrr0r set_positionr get_positionr.rrrdraw) rrenderer orig_artists orig_imageslocatorposrectrrrr"r rrs&            zHostAxesBase.drawcs"|jD]}|qtdSrA)r.r%rrrr"r rr%s  zHostAxesBase.clacs&t||jD]}||q dSrA)rr/r.r2r"r rr/s   zHostAxesBase.pickcCn|dur|}t|}|||dd}|j||j|_|jdd|jdd|jdd|S)z Create a twin of Axes with a shared x-axis but independent y-axis. The y-axis of self will have ticks on the left and the returned axes will have ticks on the right. NF)sharexrrightT)lefttopbottom)_get_base_axesr?r.r _remove_twinxraxis set_visiblerr>rrr r rtwinx zHostAxesBase.twinxcC4|j||jdd|jdjddddS)NrTF ticklabelslabelr.rrrtogglerr r rr zHostAxesBase._remove_twinxcCr)z Create a twin of Axes with a shared y-axis but independent x-axis. The x-axis of self will have ticks on the bottom and the returned axes will have ticks on the top. NF)shareyrrT)rrr)rr?r.r _remove_twinyrrrrr r rtwiny!rzHostAxesBase.twinycCr)NrTFrrrr r rr8rzHostAxesBase._remove_twinycs|dur}t|}|dur|tdd}n||dd}j|jj|_jd d|jd d|jd dfd d }||_|S) z Create a twin of Axes with no shared axis. While self will have ticks on the left and bottom axis, the returned axes will have ticks on the top and right axis. NrN)rErOrrFT)rrcs4j|jddjdjddddS)NrTFrr)hr-r rrXrz)HostAxesBase.twin.._remove_method) rrrHIdentityTransformr.rrrrr)r aux_transr>parasite_axes_auxtrans_classrrr r-rtwin=s&    zHostAxesBase.twinTcs@gfdd|jDtj|d}tdd|DS)Ncsg|] }|jdqS))call_axes_locator) get_tightbbox)rrrrr r csz.HostAxesBase.get_tightbbox..)rbbox_extra_artistscSs$g|]}|jdks|jdkr|qS)r)widthheight)rbr r rrhs$)r.rrrunion)rrrrbbsr"rrr`s zHostAxesBase.get_tightbboxrA)NN)TN)r4r5r6rrrrrr%r/rrrrrrr7r r r"rrs       #rcs@dur tjdddtfdd}tdjtfd|iS)Nr8zSupport for passing None to host_axes_class is deprecated since %(since)s and will be removed %(removed)s; explicitly pass the default Axes class instead.r9csSrAr r-r=r rrtsz/host_axes_class_factory.._get_base_axesz %sHostAxesr)rr;rr<r4r)r>rr r=rhost_axes_class_factoryks  rcCst|}t|}|SrA)rr)r>host_axes_classsubplot_host_classr r rhost_subplot_class_factory|sr)r>rcOPddlm}t|}|dur|}||g|Ri|}||||S)aJ Create axes that can act as a hosts to parasitic axes. Parameters ---------- figure : `matplotlib.figure.Figure` Figure to which the axes will be added. Defaults to the current figure `.pyplot.gcf()`. *args, **kwargs Will be passed on to the underlying ``Axes`` object creation. rN)matplotlib.pyplotpyplotrgcfadd_axesdraw_if_interactive)r>rrr!pltrrr r r host_axes  rcOr)aQ Create a subplot that can act as a host to parasitic axes. Parameters ---------- figure : `matplotlib.figure.Figure` Figure to which the subplot will be added. Defaults to the current figure `.pyplot.gcf()`. *args, **kwargs Will be passed on to the underlying ``Axes`` object creation. rN)rrrr add_subplotr)r>rrr!rhost_subplot_classrr r r host_subplotrrrA) functools matplotlibrr&rrrHmatplotlib.axesrmatplotlib.transformsrmpl_axesrnumpyr_r lru_cacher?rr@rParasiteAxesAuxTransrrrHostAxes SubplotHostrrr r r rs.   )  o #