G8c\&$ddlmZmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZGddZej$ed ZeeZGd d Zej$ed d ZeeZe eZdZedddZedddZy))_apicbookN)subplot_class_factory)Bbox)AxesceZdZd ddfd ZfdZej ddZfdZfdZ d Z d Z d Z d fd Z xZS)ParasiteAxesBaseN viewlim_modec ||_||_|j|d|d<t||j |j fi|y)NFframeon) _parent_axestransAuxset_viewlim_modesuper__init__figure _position)self parent_axes aux_transformr kwargs __class__s G/usr/lib/python3/dist-packages/mpl_toolkits/axes_grid1/parasite_axes.pyrzParasiteAxesBase.__init__ sJ'%  l+!y ++[-B-BMfMct|tj|j d|j j |_y)NF)visible)rclearmartistsetp get_childrenr _get_linesrrs rrzParasiteAxesBase.clears6   T&&(%8++66rz3.5cg}g}|jD]P}|jst|tjr|j |@|j |R||fSN)r" get_visible isinstancemimage AxesImageappend)rartistsimagesas rget_images_artistsz#ParasiteAxesBase.get_images_artistssd""$ "A==?!V--. a q!  "wrct|||jD]C}t|jds||jj vs3|j|Ey)N parasites)rpickr"hasattrinaxesr1r mouseeventr.rs rr2zParasiteAxesBase.pick(s[  Z ""$ #A ));7 1 1 ; ;;z" #rc~|j|jj|_|j|jjz|_t j |j|j|_t j |j|j|_yt|%yr&) rr transAxes transData mtransformsblended_transform_factory_xaxis_transform_yaxis_transformr_set_lim_and_transformsr$s rr>z(ParasiteAxesBase._set_lim_and_transforms6s == $!..88DN!]]T->->-H-HHDN$/$I$I%0D !$/$I$I%0D ! G + -rcDtjgd|||_y)NNequal transformmode)r check_in_list _viewlim_mode)rrDs rrz!ParasiteAxesBase.set_viewlim_modeAs 7dC!rc|jSr&)rF)rs rget_viewlim_modez!ParasiteAxesBase.get_viewlim_modeEs!!!rc|jjj}|j}|y|dk(r&|jjj |y|dk(rM|jjj |j |jjytjgd|y)NrArBr@rC) rviewLimfrozenrHaxesset transformedrinvertedrrE)rviewlimrDs r_update_viewlimz ParasiteAxesBase._update_viewlimHs##++224$$& <  W_ II   ! !' * [ II   ! !##DMM$:$:$<= ?   ;$ GrcB|jt| yr&)rQr apply_aspect)rpositionrs rrSzParasiteAxesBase.apply_aspectUs  rr&)__name__ __module__ __qualname__rrr deprecatedr/r2r>rrHrQrS __classcell__rs@rr r sZN!%N7 T__U   # ."" Hrr z {}Parasitec~eZdZfdZddefdZfdZfdZfdZddZ dd Z dd Z d Z d Z dfd ZxZS) HostAxesBasec2g|_t||i|yr&)r1rr)rargsrrs rrzHostAxesBase.__init__bs $)&)rNrAct|}||||}|jj||jj|_|S)a Add a parasite axes to this host. Despite this method's name, this should actually be thought of as an ``add_parasite_axes`` method. *tr* may be `.Transform`, in which case the following relation will hold: ``parasite.transData = tr + host.transData``. Alternatively, it may be None (the default), no special relationship will hold between the parasite's and the host's ``transData``. r )parasite_axes_class_factoryr1r+remove_remove_method)rtrr axes_classparasite_axes_classax2s r get_aux_axeszHostAxesBase.get_aux_axesfsH:*E!$F c"!^^22 rct|j}|j}|r.|||}|j|d|j |n|j |j }|j D]<}|j ||jj|j>t|)||j|d=y)Nactive)which) len _childrenget_axes_locator set_positionrS get_positionr1extendr"rdraw)rrendererorig_children_lenlocatorposrectaxrs rrqzHostAxesBase.drawzs/'') $)C   c  2   c "      ".. 5B OOD ! NN ! !"//"3 4 5  X NN,- .rcd|jD]}|jt|yr&)r1rr)rrwrs rrzHostAxesBase.clears(.. B HHJ   rcht|||jD]}|j|yr&)rr2r1r5s rr2zHostAxesBase.picks0  Z  A FF:  rc|j||}|jdjd|jdjd|jdjd|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. )sharexrightFT)lefttopbottom_add_twin_axesaxis set_visiblerrdrws rtwinxzHostAxesBase.twinxsd D 9 '&&u- $$T* '(44U; rc|j||}|jdjd|jdjd|jdjd|S)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. )shareyr~FT)r}r|rrrs rtwinyzHostAxesBase.twinysc D 9 %$$U+ ""4( )*66u= rc|tj}|j||d}|jdj d|jdj d|jdj 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. rB)rr )r~r|FT)r}r)r:IdentityTransformrrr)r aux_transrdrws rtwinzHostAxesBase.twins  #557I  ik!K .!--e4 ++D1  !--e4 rc | |j}t||fi|}|jj||j|_|S)zy Helper for `.twinx`/`.twiny`/`.twin`. *kwargs* are forwarded to the parasite axes constructor. )_base_axes_classr`r1r+_remove_any_twinrb)rrdrrws rrzHostAxesBase._add_twin_axessQ  ..J 4 ( 4T DV D b! 11 rcV|jj|ddg}|jr|jd|jr|jd|jt |j d|jt |jddy)Nr~r|TF) ticklabelslabel)r1ra_sharex_shareyrtuplertoggle)rrwrestores rrzHostAxesBase._remove_any_twins b!'" :: NN5 ! :: NN7 # %.!--d3 %.!((E(Grcg|jDcgc]}|j||c}t||||}tj|Dcgc]#}|j dk7s|j dk7s"|%c}Scc}wcc}w)N)call_axes_locator)rbbox_extra_artistsr)r1 get_tightbboxrrunionwidthheight)rrrrrrwbbsbrs rrzHostAxesBase.get_tightbboxsJ(x;LM(J G !(4E5G " IJ zzcKQWW\QXX]1KLL ( LsB#B8Br&)NN)NTN)rUrVrWrrrgrqrr2rrrrrrrYrZs@rr\r\asT*#T(/&    H>B)-MMrr\z {}HostAxesrc2t|}t|}|Sr&)host_axes_class_factoryr)rdhost_axes_classsubplot_host_classs rhost_subplot_class_factoryrs-j9O.? r)rdrcddlm}t|}||j}||g|i|}|j ||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_axes)rdrr^rpltrrws r host_axesrsF$-j9O ~  1$ 1& 1B OOB Ircddlm}t|}||j}||g|i|}|j ||S)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_subplot)rdrr^rrhost_subplot_classrws r host_subplotrsI$3J? ~ F 4T 4V 4B r Ir) matplotlibrrmatplotlib.artistartistr matplotlib.imageimager)matplotlib.transforms transformsr:matplotlib.axesrrmpl_axesrr _make_class_factoryr` ParasiteAxesr\rHostAxes SubplotHostrrrrrrs"#!+1&MMd8e77l$*40 AMAMH4%33, 24 "4 (#H-  !%T,$(r