g$ddlmZmZddlmZddlmZddlm Z ddl m Z GddZ ej e dZee ZGd d Zej ed d xZZee xZZe dd dZeZdS))_apicbookN)Bbox)AxescVeZdZd ddfd ZfdZfdZfdZdZdZd Z xZ S) ParasiteAxesBaseN) viewlim_modec ||_||_||d|d<tj|j|jfi|dS)NFframeon) _parent_axestransAuxset_viewlim_modesuper__init__figure _position)self parent_axes aux_transformr kwargs __class__s [/granary/edev/quantum/lib/python3.11/site-packages/mpl_toolkits/axes_grid1/parasite_axes.pyrzParasiteAxesBase.__init__ s]'%  l+++!y+[-BMMfMMMMMcNttj|d|jj|_|jjd|j |jjd|j dS)NF)visible xlim_changed ylim_changed) rclearmartistsetp get_childrenr _get_lines callbacks_connect_picklable _sync_limsrrs rrzParasiteAxesBase.clears   T&&((%8888+6 #66 DO - - - #66 DO - - - - -rct||D]:}t|jdr#||jjvr||;dS)N parasites)rpickr"hasattrinaxesr)r mouseeventars rr*zParasiteAxesBase.picks{  Z   ""$$ # #A );77 # 1 ;;;z""" # #rc>|jt|jj|_|j|jjz|_t j|j|j|_t j|j|j|_dSt dSN) rr transAxes transData mtransformsblended_transform_factory_xaxis_transform_yaxis_transformr_set_lim_and_transformsr's rr8z(ParasiteAxesBase._set_lim_and_transforms)s = $!.8DN!]T->-HHDN$/$I%0%0D !$/$I%0%0D ! ! ! GG + + - - - - -rcDtjgd|||_dS)NNequal transformmode)r check_in_list _viewlim_mode)rr>s rrz!ParasiteAxesBase.set_viewlim_mode4s- 777dCCCC!rc|jSr1)r@)rs rget_viewlim_modez!ParasiteAxesBase.get_viewlim_mode8s !!rct|j}|}|dS|dkr|j|dS|dkrF|j||jdStjgd|dS)Nr;r<r:r=) viewLimfrozenrBset transformedrinvertedrr?)rparentviewlimr>s rr&zParasiteAxesBase._sync_lims;s.''))$$&& < D W__ L  W % % % % % [ L  W001G1G1I1IJJ K K K K K  ;;;$ G G G G G Grr1) __name__ __module__ __qualname__rrr*r8rrBr& __classcell__rs@rr r sN!%NNNNNNN----- # # # # # . . . . ."""""" H H H H H H Hrr z {}ParasiteczeZdZfdZ ddZfdZfdZfdZddZdd Z dd Z d Z d Z dfd Z xZS) HostAxesBasecHg|_tj|i|dSr1)r)rr)rargsrrs rrzHostAxesBase.__init__Ps,$)&)))))rNr;c ||j}t|}|||fd|i|}|j||jj|_|S)ab Add a parasite axes to this host. Despite this method's name, this should actually be thought of as an ``add_parasite_axes`` method. .. versionchanged:: 3.7 Defaults to same base axes class as host axes. Parameters ---------- tr : `~matplotlib.transforms.Transform` or None, default: None If a `.Transform`, the following relation will hold: ``parasite.transData = tr + host.transData``. If None, the parasite's and the host's ``transData`` are unrelated. viewlim_mode : {"equal", "transform", None}, default: "equal" How the parasite's view limits are set: directly equal to the parent axes ("equal"), equal after application of *tr* ("transform"), or independently (None). axes_class : subclass type of `~matplotlib.axes.Axes`, optional The `~.axes.Axes` subclass that is instantiated. If None, the base class of the host axes is used. kwargs Other parameters are forwarded to the parasite axes constructor. Nr )_base_axes_classparasite_axes_class_factoryr)appendremove_remove_method)rtrr axes_classrparasite_axes_classax2s r get_aux_axeszHostAxesBase.get_aux_axesTsv6  .J9*EE!! ";;#/;39;; c"""!^2 rc t|j}|}|r9|||}||d||n||}|jD]C}|||j|Dt ||j|d=dS)Nactive)which) len _childrenget_axes_locator set_position apply_aspect get_positionr)extendr"rdraw)rrendererorig_children_lenlocatorposrectaxrs rrizHostAxesBase.drawzs//''))  '$))C   c  2 2 2   c " " " "        "". 5 5B OOD ! ! ! N ! !"//"3"3 4 4 4 4  X N,-- . . .rct|jD]}|dSr1)rrr))rrors rrzHostAxesBase.clears>  .  B HHJJJJ  rct||jD]}||dSr1)rr*r)r-s rr*zHostAxesBase.picksK  Z     A FF:      rc|||}|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. )sharexrightFT)lefttopbottom_add_twin_axesaxis set_visiblerr[ros rtwinxzHostAxesBase.twinxss D 9 9 '&&u--- $$T*** '(44U;;; rc|||}|jdd|jdd|jdd|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. )shareyrvFT)rurtrwrxr|s rtwinyzHostAxesBase.twinysr D 9 9 %$$U+++ ""4((( )*66u=== rc |tj}|||d}|jdd|jdd|jdd|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. Nr<)rr )rvrtFT)rurw)r4IdentityTransformryrzr{)r aux_transr[ros rtwinzHostAxesBase.twins  #577I  ik!KK .!--e444 ++D111  !--e444 rc ||j}t||fi|}|j||j|_|S)zy Helper for `.twinx`/`.twiny`/`.twin`. *kwargs* are forwarded to the parasite axes constructor. )rUrVr)rW_remove_any_twinrY)rr[rros rryzHostAxesBase._add_twin_axessX  .J 4 ( 4 4T D DV D D b!!! 1 rcj|j|ddg}|jr|d|jr|d|jt |d|jt |dddS)NrvrtTF) ticklabelslabel)r)rX_sharex_shareyrztupler{toggle)rrorestores rrzHostAxesBase._remove_any_twins b!!!'" : " NN5 ! ! ! : $ NN7 # # # %..!--d333 %..!((E(GGGGGrTcgfd|jDt|}tjd|DS)Nc>g|]}|S))call_axes_locator) get_tightbbox).0rorrjs r z.HostAxesBase.get_tightbbox..s=(((x;LMM(((r)rbbox_extra_artistsc>g|]}|jdks |jdk|S)r)widthheight)rbs rrz.HostAxesBase.get_tightbbox..s)KKKQW\\QX]]1]]]r)r)rrrunion)rrjrrbbsrs `` rrzHostAxesBase.get_tightbboxsJ((((((((J GG ! !(4E5G " I IJ zKKcKKKLLLr)Nr;Nr1)NN)NTN)rKrLrMrr^rirr*r}rrryrrrNrOs@rrQrQOs,***** =A$$$$L/////&            HHH>B)-MMMMMMMMMMrrQz {}HostAxesrU)r[rcddlm}t|}||}||g|Ri|}|||S)aP 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.Axes` object creation. rN)matplotlib.pyplotpyplothost_axes_class_factorygcfadd_axes)r[rrSrplthost_axes_classros r host_axesrsk$#####-j99O ~  1$ 1 1 1& 1 1B OOB Ir) matplotlibrrmatplotlib.artistartistr matplotlib.transforms transformsr4rmpl_axesrr _make_class_factoryrV ParasiteAxesrQrhost_subplot_class_factoryHostAxes SubplotHostr host_subplotrrrs^""""""""######++++++&&&&&&=H=H=H=H=H=H=H=HD8e7l$$**400 SMSMSMSMSMSMSMSMnElL:LMMN400666;!%T, r