LemddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ddl mZmZddlmZmZdd lmZmZmZmZmZmZmZmZdd lmZmZdd lm Z dd l!m"Z"d dl#m$Z$m%Z%m&Z&d dl'm(Z(d dl)m*Z*m+Z+m,Z,ddlm-Z-m.Z.m/Z/ddl0m1Z1ddl'm2Z2m3Z3ddl)m4Z4Gdde.Z5Gdde5Z6Gdde-Z7Gdde$e5Z8Gdde2e8Z9Gd d!e8Z:Gd"d#e-Z;Gd$d%e2e;Z<Gd&d'e5e-e/Z=Gd(d)e-Z>Gd*d+e%e-e/Z?Gd,d-e&e1e-Z@Gd.d/e2e@ZAdS)0N)cm)LineCollection) DateFormatterdate2num)Version) Dimensiondimension_name)Storeabbreviated_exception) dt64_to_dt dt_to_int isdatetimeisfiniteisscalar match_specsearch_indices unique_array)HeatMapRaster)interpolate_curve)dim) AreaMixin BarsMixin SpikesMixin) PlotSelector) compute_sizesget_min_distanceget_sideplot_ranges) ColorbarPlot ElementPlot LegendPlot)PathPlot) AdjoinedPlotmpl_rc_context) mpl_versionceZdZdZdS) ChartPlotz+ Baseclass to plot Chart elements. N)__name__ __module__ __qualname____doc__> K0033G!155g?RSSI1gmmy1I1ImJJDG!-;"bB8u|T222 377c|\}}t|r|j||dfi|d}n|j||fi|d}d|iS)N-rartist)r plot_daterE)rUax plot_args plot_kwargsrYrZrcs r1 init_artistszCurvePlot.init_artists_shB b>> 7!R\"b#====a@FFRWR33{33A6F&!!r0c|jd}||||\\}}}} ||||| SNrc)handlesr_ set_xdata set_ydata) rUkeyaxisrVrWrXrcrYrZ axis_kwargss r1update_handleszCurvePlot.update_handlesgs\h''+}}Wfe'L'L$R%r0N)r+r,r-r.paramBooleanautotickObjectSelectorrJrelative_labels ClassSelectorintfloattuplepadding show_grid show_legend style_opts_nonvectorized_stylesdict _plot_methodsr_rhrqr/r0r1r3r3'sIu}U17888H)E(2M2M2M19@)***M$emE8<===O"e!(C;NOOOG e2I %-30111KYXXJ&D'''M333 """r0r3cDeZdZdZgdZedZdZdZdZ dS) ErrorPlotz ErrorPlot plots the ErrorBar Element type and supporting both horizontal and vertical error bars via the 'horizontal' plot option. ) edgecolor elinewidthcapsizecapthick barsabovelolimsuplimsxlolims erroreveryxuplimsr>rBrAmarkeredgecolormarkeredgewidthmarkerfacecolor markersizesolid_capstylesolid_joinstyledashesr?errorbarrFc|j|i|}d\}}ttdkr|\}}} |r|\}}n |\}\}}} ||| d| ddS)N)NNz2.0r)bottomstopsvertsrc)rr(r) rUre plot_datargrkrr_capsrs r1rhzErrorPlot.init_artistssz"+y8K88"  '%.. ( ($NAtU % $ (/ %A"D58uUVxXXXr0ct5|||}dddn #1swxYwY|dd}t|tjr||d<d|vr|d|d<d|vrz%ErrorPlot.get_data..1@@!'**1--@@@@@@r0r)rMxerryerrrr)r rNpop isinstancenpndarrayasarraytolistrR ValueErrorrendererbackendrMrangearray horizontalrTlenT) rUrVrWrXr?rr[rYrZerrerr_keyr^s ` r1r_zErrorPlot.get_datash " $ $ C C**7FEBBE C C C C C C C C C C C C C C C '4(( eRZ ( ( $#E(O %  #ii 44E(O %  "$*UYY{-C-C"D"D"K"K"M"ME,  IIcNN a $ $ &((   "D<@M.s1KKQW55a88KKKKKKr0rrrrr)rkr_rrrMrOrrTrmoveaxisr set_pathsrlrmset_edgecolorsset_linewidths)rUrnrorVrWrXrrrrrprYrZ neg_errorpos_idx pos_errorbxsbystxstys new_arrayss ` r1rqzErrorPlot.update_handless,y)|F# W% $ gvu E E5+KKKK%((KKKB 7--//00144!!!,,W55   *I~rCI~rCC2 >C2 >C   CCCC[C:Sz*B!C!CQJJ   ###  #   c " " "   c " " "  NN3    NN3    u    x 1 1 1 5  |!4 5 5 5r0N) r+r,r-r.r~rrrhr_rqr/r0r1rrpst 888JD +++M Y Y Y!!!@r0rceZdZejdeeefZej ddZ gdZ e Z e dZd Zd Zd S) AreaPlotr9r;Fr=r4) r? facecolorr>rrAhatchrB joinstylefillcapstyle interpolatestep fill_betweenrFct5|||}dddn #1swxYwYd}fdjD}t |g|z|ifS)Nrc:g|]}|Sr/r)rvdimrVs r1 z%AreaPlot.get_data..s' G G Gg&&t,, G G Gr0)r rNrOvdimsrzrUrVrWrXrYrZs ` r1r_zAreaPlot.get_datas " $ $ C C**7FEBBE C C C C C C C C C C C C C C C % %a ( ( G G G G G G GbT"W~~ub((s 488cD|jr|jn|j}||i|}d|iSrj)rT fill_betweenxr)rUrerrgfill_fnstacks r1rhzAreaPlot.init_artistss8&*&6K"""BO2k22%  r0N)r+r,r-rrrwrxryrzr{rsr}r~rrrr_rhr/r0r1rrs!e!(C;NOOOG%-40111K===J'D///M)))!!!!!r0rceZdZejddZejddZejdgdd Z ejdgd d Z d S) SideAreaPlotr!r!r!r( Make plot background invisible.r4rI The size of the border expressed as a fraction of the main plot.baretopbottomrztop-barez bottom-bareN Whether and where to display the xaxis, bare options allow suppressing all axis labels including ticks and xlabel. Valid options are 'top', 'bottom', 'bare', 'top-bare' and 'bottom-bare'.r5r8r6leftrightrz left-barez right-bareN Whether and where to display the yaxis, bare options allow suppressing all axis labels including ticks and ylabel. Valid options are 'left', 'right', 'bare' 'left-bare' and 'right-bare'.N) r+r,r-rr ParameterbgcolorNumber border_sizeruxaxisyaxisr/r0r1rrseol9+,,,G%,q/LMMMK !E *?*?*?E; < < *>*>D9 : : :EEEr0rc~eZdZdZejdeeefZ ej ddZ fdZ dZ d d ZxZS) SpreadPlotz3 SpreadPlot plots the Spread Element type. r9r;Fr=r4c <tj|fi|dSr)super__init__)rUrVparams __class__s r1rzSpreadPlot.__init__s(++F+++++r0ct5||||}dddn #1swxYwY|d}|d}|d}t|dkrdnd}||}|||z ||zf|ifS)Nrr!rr)r rNrOrrM) rUrVrWrXrYmeanrrrs r1r_zSpreadPlot.get_datas " $ $ C C**7FEBBE C C C C C C C C C C C C C C C  % %a ( (''**,,Q// 7--//00144!!!,,W55 DNDN3UB>>r`combinedc <t||||Sr)r* get_extents)rUrVrW range_typekwargss r1rzSpreadPlot.get_extentss$$T7FJGGGr0r)r+r,r-r.rrrwrxryrzr{rsr}rr_r __classcell__rs@r1rrs"e!(C;NOOOG%-40111K,,,,,???HHHHHHHHr0rczeZdZdZgdZgdZfdZeddZdZ dZ dfd Z d Z d Z d ZdZxZS) HistogramPlotz HistogramPlot can plot DataHistograms and ViewMaps of DataHistograms, which can be displayed as a single frame or animation. ) r>r?alignr@rrlogrerror_kwrrA)r>r r rr@rc d|_d|_tj|fi||jr gd|_n gd|_|jjd}|jr|j nd|_ dS)NF)ylabelxlabelyticks)r r xticksr!) centercyclicrrrT axis_settingshmaplast get_dimensionr cyclic_range)rU histogramsrval_dimrs r1rzHistogramPlot.__init__,s  ..v...   @!?!?!?D  !?!?!?D )...q11-4^EGMMr0Nc|jj}|jd}||j||}t ||}|dd}||\}}}} } | rU|djsHtj tj } |d t| |d<|j|j} |jr/|jdj|_|jdj|_n.|jdj|_|jdj|_t15|||| } d| vrt5d|jjd dddn #1swxYwY|jr|jnd } |j|||f|j| d d | }| |jd||| ||jd <|!|||| }|"|| |}||d <|j#|jdf||d|S)NrrrKrovminzUMapping a continuous dimension to a color on a HistogramPlot is not supported by the redge)zorderlabelrrcrM)rWrV)$rrkeyscompute_rangesrrM _process_histrLr rQr datetime64rSrrX cyclic_indexrTrkaxvline offset_linefnbarhplotfnaxhlinebarr rNrrrr}rr _update_plot_compute_ticks_process_axsettings_finalize_axis)rUrWhistrn el_rangesr[edgeshvalswidthslims is_datetimer]rXlegendbarsticks ax_settingss r1initialize_plotzHistogramPlot.initialize_plot:sy~im$$TYV<<tV,,   !$262D2DT2J2J/ufdK  KtAw3 K!1"-@I1gmmy1I1ImJJDG 4,-   3!%f!5!=D ,v.3DKK!%f!5!=D ,v.2DK " $ $  **4??E "D59]5J"D"D"D                $/7Rt{5%i 6Y_iichii!%!2!249R=$dTZ![![ X##D%>>..tT5AA $( L!"t"49R=`D``T_```s9FF"%F"c4|dj|_|jd}|j||d}|d}t j|}|d}|d}t|}|rFt jd|D}t|}td|D}t j |} |dd|| ||z|fS) zK Get data from histogram, including bin_ranges and values. rT)r1r!cdg|]-}t|tjrt|n|.Sr/)rrr#r )res r1rz/HistogramPlot._process_hist..ts5bbbWXz!R]/K/KRjmmmQRbbbr0c36K|]}t|dVdS)DN)r)rvs r1rz.HistogramPlot._process_hist..vs,99q1c**999999r0Nr) rrkdims interfacer^rOrrrrrrzdiff) rUr/xr1values hist_valsxlimylimr5r3s r1r"zHistogramPlot._process_histgs((++2 JqM%%dAT%::&&q))HV$$ zz!}}zz!}} ''  :Hbb\abbbccEUOOE99D99999DSbSz9fd4iDDr0c |jt|jtsdS|jr/|\}}}}t j|||j}d|D} n|jr|d t jdt|jt} td|dzgzfd| D} fd|D} || gS)zw Compute the ticks either as cyclic values in degrees or as roughly evenly spaced bin centers. Nc>g|]}tj|ddS)z.0f°)rrad2deg)rrDs r1rz0HistogramPlot._compute_ticks..s,KKKA;;;;KKKr0r)dtyperc g|] }| Sr/r/)rrr1s r1rz0HistogramPlot._compute_ticks..s,,,!U1X,,,r0c0g|]}j|Sr/) pprint_value)rr@rs r1rz0HistogramPlot._compute_ticks..s'999a&c&q))999r0) rrrxrrlinspacerrlist) rUrVr1r3r4x0x1rxvalslabelsindsrs ` @r1r,zHistogramPlot._compute_tickszs ; jc&B&B 4 ; :LBAqKB 44EKKUKKKFF [ :''**C;q#e**dkEEEDKK59vbz#9"::E,,,,t,,,E99995999Fvr0rc H|d}||jd\}}t|rt|dnd}t|rt |dnd}||f||jd<t |||S)Nr!softr)rnamerminmaxrr) rUrVrWrrydims0s1rs r1rzHistogramPlot.get_extentss$$Q'' "6*B#B<< .SQZZZQ#B<< .SQZZZQ%'Hty&!ww""7FJ???r0cbtt|jdd|jrdn|g}|S)z` Get axis settings options including ticks, x- and y-labels and limits. N)rziproverlaid)rUr/r4r8rs r1r-z!HistogramPlot._process_axsettingss7 S!3dD4=Cc44^c5effgg r0c|S)ze Process bars can be subclassed to manually adjust bars after being plotted. r/)rUrnr/r7r4rWs r1r+zHistogramPlot._update_plots  r0c`t|jd|||}|D]\} } } } |jr@| | | | | | N| | | | | | dS)zu Update all the artists in the histogram. Subclassable to allow updating of further artists. rcN)rarkrTset_y set_width set_heightset_x) rUrnr/r1r2r3r4rW plot_valsr*rheightwidths r1_update_artistszHistogramPlot._update_artistss  X.ufEE (1 % % $Cvu % $ f%%%u%%%% $v&&& e$$$$ % %r0c ||\}}}} } ||||| } ||| | } ||||||| || Sr)r"r,r-rl) rUrnrorVrWrXr1r2r3r4rr8r9s r1rqzHistogramPlot.update_handlessv(,(:(:7(C(C%ufdA##GUFDAA..weDD  S'5%vNNNr0rr)r+r,r-r.r~rrr'r:r"r,rr-r+rlrqrrs@r1rrs  JVUU F F F F F)a)a)a^)aXEEE&&@@@@@@%%%"r0rceZdZejddZejdddZejdd Z fd Z fd Z d Z d Z dZxZS)SideHistogramPlotrrr4皙?rr!z/ Histogram value offset for a colorbar.r5boundsr6Fz/ Whether to overlay a grid on the axis.ct|\}}}}}|j|dz}|d|jz z}||z }|dd|d|zfz}|||||fS)zC Subclassed to offset histogram by defined amount. rr!r)rr"offset) rUr/r1r2r3r4rrurs r1r"zSideHistogramPlot._process_hists271F1Ft1L1L.ufdJtAw&4;' AaCyDGf,..eVT:55r0c t|||||||||||jd||dSrj)rrlr+rk) rUnrVr1r2r3r4rWrs r1rlz!SideHistogramPlot._update_artistssP 7E5&$OOO !Wdl8&r? edgecolors facecolorsrArCsizer@r{rvmaxnorm)r>rCr{rrrr@rscatterrFc fdtdD\}}|||t5|||}dddn #1swxYwY|jr||fn||f|ifS)Nc3BK|]}|VdSrrrs r1rz%PointPlot.get_data..Qrr0r)r_compute_stylesr rNrTrs ` r1r_zPointPlot.get_dataPs@@@@uQxx@@@B Wfe444 " $ $ C C**7FEBBE C C C C C C C C C C C C C C C+9Bxx"b5"DDsA))A-0A-c||j}|dd}|dd}|rJt |t r||vst |t r|jdd}|ri|rg| |j}|j j dvr||d<n t|t||d<|||||n|||d<|d|dd|d<|d tjd }||j} | rJt |t r||vst |t r|jd d} | r}| |j} t%| |j|j|j|} | :t-|j} |j| jd | d n| |d <|dd|d<dS)Nr?r{dCannot declare style mapping for 'color' option and declare a color_index; ignoring the color_index.uifrrrrszlines.markersizez^Cannot declare style mapping for 's' option and declare a size_index; ignoring the size_index.z/ dimension is not numeric, cannot use to scale z size.)rrzrrRrstrrrrwarningrOrMkindrr _norm_kwargsmplrcParams size_indexrsize_fnscaling_factorscaling_methodrr+ pprint_label) rUrVrWrXrr?r{csrDsdimsizeseltypes r1rzPointPlot._compute_stylesXs$$T%566 '4((yy&&  j,, '1A1AjQVX[F\F\1A J  C D D DD  #D #))$*:;;Bx}%%c +B R0@0@AAc   gvud ; ; ; ;  "E'N#ii eii V6T6TUUl YYsCL);< = =$$T_55  jS)) bGmm 2s@S@Sm J  A B B BD  #,,T_==E!%t7J"&"5r;;E}g/ ""(,,#,,,----#c #ii f==lr0c |jd}||||\\}}}} |dd\} } d|| jivr.t || jd fd|D}d|| jivr.t || jd fd|D}|tj||gd|vr/|d} t| r| g} | | d|vr#| |d|dfd |vr| |d d |vr |d |_ d |vr||d d |vr||d d |vr||d dSdS)NrcrfactorscBg|]}|v|Sr/index)rrDrs r1rz,PointPlot.update_handles..*???q!w,,'--"",,,r0cBg|]}|v|Sr/r)ryrs r1rz,PointPlot.update_handles..rr0rrrrrrArr)rkr_rMrRrZrR set_offsetsr column_stackr set_sizesset_clim set_arrayrrr)rUrnrorVrWrXpathsrYrZrxdimr]rrs @r1rqzPointPlot.update_handless X&!]]7FEBBR%''))"1"- d  49b11 1 16$),Y788G????B???B  49b11 1 16$),Y788G????B???B "/2r(33444 %<<#JE  OOE " " " U?? NNE&M5=9 : : : %<< OOE#J ' ' ' U??vEJ %    {!3 4 4 4 5  |!4 5 5 5 5  |!4 5 5 5 5 5 ! r0)r+r,r-r.rrrsr|rwrrxrzrrurrrCallablerabsrr~r_disabled_optsrrr_rrqr/r0r1rr%s  e2;<<(>(>V66666r0rcbeZdZdZejddZejee fdZ ejde e e fZejdd Zejd ee fdd Zejd ee fdd ZejddZgdZgdZedZdZdZdZd S)VectorFieldPlota Renders vector fields in sheet coordinates. The vectors are expressed in polar coordinates and may be displayed according to angle alone (with some common, arbitrary arrow length) or may be true polar vectors. The color or magnitude can be mapped onto any dimension using the color_index and size_index. The length of the arrows is controlled by the 'scale' style option. The scaling of the arrows may also be controlled via the normalize_lengths and rescale_lengths plot option, which will normalize the lengths to a maximum of 1 and scale them according to the minimum distance respectively. Tz Whether or not to draw arrow heads. If arrowheads are enabled, they may be customized with the 'headlength' and 'headaxislength' style options.r4a: Dimension or dimension value transform that declares the magnitude of each vector. Magnitude is expected to be scaled between 0-1, by default the magnitudes are rescaled relative to the minimum distance between vectors, this can be disabled with the rescale_lengths option.)r<r6g?r;z| Whether the lengths will be rescaled to take into account the smallest non-zero distance between two vectors.Nzz Deprecated in favor of dimension value transform on color option, e.g. `color=dim('Magnitude')`. rzi Deprecated in favor of the magnitude option, e.g. `magnitude=dim('Magnitude')`. z Deprecated in favor of rescaling length using dimension value transforms using the magnitude option, e.g. `dim('Magnitude').norm()`.)r>r?rrrArCr@r{scale headlengthheadaxislengthpivotrk headwidthr) r>rCr{r@rrrrrquiverrFc||j}|j}|r|r|jdn/|r|}n*t |t r||}|st |tr||d}ng| |}|t|d\}}|j r |dkr||z }n!tj t|}|S)NzfCannot declare style mapping for 'magnitude' option and declare a size_index; ignoring the size_index.T)flatrr)rr magnituderrrrrrapplyrOr normalize_lengthsronesr) rUrVrXrWsize_dimmag_dim magnitudesr max_magnitudes r1_get_magnitudeszVectorFieldPlot._get_magnitudess%((99.  5 5 J  E F F F F 5GG  % % 5++G44G  '3'' <$]]7]>> $55g>> #).*A*A#B:#N =)??FE#J(()9::D   gvud ; ; ; IIgt $ $ $# $ $ C C**7FEBBE C C C C C C C C C C C C C C C T F#SVWWWXXX U??"YYv.. &0A0ABE&M %<<"YYs^^E'N %  "E'N (&'E" #UBsIIIc||||\}}}|jd}|tj|dd|d|_|d|_d|vr6||d||dd|vr| |dd|vr| |dd|vr| |d|S)Nrcrrr?rrrA) r_rkrrrUrset_facecolorsrrrr) rUrnrorVrWrXrrprs r1rqzVectorFieldPlot.update_handles*s#'==&%#H#H e[h'2?48445557h e    ! !%. 1 1 1  ! !%. 1 1 1 e     U7^ , , , U?? OOE&M * * * %    ! !% "4 5 5 5r0)r+r,r-r.rrrsrrwrrrrxryrzr{rrzrrr~rrrrr_rqr/r0r1rrs  %-3*+++K $#C:<#$$$I"e!$UE7JKKKG#emD7:;;;O &%%dC:15<    K %$T3*04;    J & d9&''' 000J ***D)))M....`r0rc >eZdZejddZejddZejddZejddZ gd Z d gZ e e je d d d d de dd dd dZdZeddZfdZdZxZS)BarPlotrpz+ Defines the padding between groups.r4TzP Whether the Bars should be grouped into a second categorical axis level.Fz6 Whether the bars should be stacked or grouped.r=) r>r?rr@rr rrr rr@)rgRQ?r~&1?rexpandr)bbox_to_anchorncollocmode borderaxespad)rgٿr~rrr:)rrc|||d\}}|j}|jdkr|ddgz}|dgd}n|jr|d}|dd|g}|jr|j} nI||vr/||jdr||d} n|dd} t| } || d}n |dgz}||d}||fS) z5 Get unique index value for each bar F) as_stringr!N)grouprrr)rcategory) _get_coordsrAndimsstackedrErZrRrOrR) rUrVrWgvalscvalsrArMrE stack_dim stack_orders r1 _get_valueszBarPlot._get_values^s''5'II u  =A  $-J$v66FF \ 9aI(D)4J A'. f$$ )?)C)CI)N)N$$Y/ : %66q%@@ {++K${;;FF$J$%88F6!!r0Nc|jj}|jd}|jd}|jd}t d d|ji|j|j}| |j||}t||}| ||||\|jd<}}|j rd|ind|i} |j |f||||gd| S) Nrrorrrcrr)rWrVrMr/)rrrrkr rrrXr$r!r _create_barsrTr.) rUrWrVrrornrXrxdimsrs r1r:zBarPlot.initialize_plotxs).}Q|F#imIIDKI4:d6G+HII$$TYV<<GV,,040A0A$QWY^0_0_- X'+'7O(F##h=O"t"3Gvw/4dmGG?EGG Gr0cd}|p|}|;tt|d\}}}t|t|f}|r|}n|r|}t||||||r|r?t||D]\} } | | dS|r?t||D]\} } | | dSdSdS)z7 Apply ticks with appropriate offsets. Nc|dS)Nrr/)rDs r1z)BarPlot._finalize_ticks..s 1r0)rn) rasortedrRr_finalize_ticksget_xticklabelsreget_yticklabelsrh) rUrorVrrzticks alignmentsr8rVtrrDrs r1rzBarPlot._finalize_tickss@  &  (+VE~~-N-N-N(O %E6:%[[$v,,/E  FF  F gvvvFFF    4 4 6 6 CCDAqGGAJJJJ  4 4 6 6 CCDAqGGAJJJJ     r0c )*||\\}}}}d*|r |d} |*n |r |d} |**r*fdt| D} ndii} ddjzz t|ddgz } jr d)d\} } }}n d)d \} } }}ggi}}}t|d dgD]\}}i}d}|H||}|g||j<|r jrd nd }| |d z||ft|ddgD]\}}|jz|| zz}|E||}|g||j<jr| || dz z|d fd }|ddgD]}| ||}|g||j<|j di|}| |j d j}t|rt|d n tj} || dz z}!||vrZ||}"|"|  |!|"|  | |"| |nt!|fi| |i}#t#5|||#}#dddn #1swxYwY| |!g| | g|| ||gd|i||<|||#|t)| r| nd z }|| |)fd|D}$|g}%d}&|r|j}&|% |n%|r#|j}&jr|% |jrzt1d|Dra|sjsXjj}'jr j|'d<j}(|(jdi|'jdd|&i|(|$||%fS)NrrcXi|]&\}}|j|'Sr/)rPrX)rrr@rU style_dims r1 z(BarPlot._create_bars..sC666!Q#//22DJqM666r0r!g@r')rrkrjrr*)rDrjrkrrg{Gzrg?rc<g|]}tdi|S)r/)getattr)rbar_specroplot_fns r1rz(BarPlot._create_bars..s4UUUx&g&&2222UUUr0rc34K|]}t|VdSr)r)rls r1rz'BarPlot._create_bars..s(#;#;qCFF#;#;#;#;#;#;r0rtitler/)r  enumerate bar_paddingrrRrTrPrZ multi_levelappendselectrOrryrnanrr rNrrrErr}any legend_specslegend_position legend_cols legend_optscopyr6)+rUrorVrWrXgdimrrrEcats style_maprkrDrwrrrVbar_datagidxgrpsel_keyr grp_labelyalignrcatxposprevstkelvalsvalxvalr bar_styler7ax_dimsr!leg_specr,rrs+`` @@r1r zBarPlot._create_barss%)%5%5gv%F%F"tTF  '?DII  *%DI  #66666%.t__666IIr IBt''(C :v0N0N,O,OO   ?GOAq!V$&r2"6::gv#>#>??( -( -ID#GE --c22 &)U ""&B4+;B tCxF;<<<&vzz*tf'E'EFF - - cD,,d5j9? --c22E*-GDI&'A tE"H}eQ&?@@@!::gv66--C $ 1 1# 6 6.1U *'22'22B..w}Q/?/DEED,/IIA%Q...26Cb=D(( (a---a,,,f ,,T2222$($K$K)--r2J2J$K$K 244VV(,(>(>r69(U(UIVVVVVVVVVVVVVVVtfa#5&4&#U+!..y9998C==7CCa7D( e,,,1- -FVUUUU8??CTCTUUU&  %%E NN4  %%E %t$$$   4#;#;F#;#;#; ; ; 4 4TM] 4()=>H DD4D&!1*//11K K  * * * * * DK 3 3e 3{ 3 3 3VW$$s L L L r)r+r,r-rrrr#rsr$rr}r~rrr$r)r r'r:rr rrs@r1rr?s%,s1.///K %-3STTTKemE09:::G %-30111KFFFJ'K4 D/   t/     L"""4 G G G^ G,[%[%[%[%[%[%[%r0rceZdZejddZejddeefdZ ej dd Z ejd ee e f Zej d d ZejdgzZdZdZdZdS) SpikesPlotsquarez The aspect ratio mode of the plot. Allows setting an explicit aspect ratio as width/height as well as 'square' and 'equal' options.r4NTzA Index of the dimension from which the color will the drawn)r5rr<r6r:D The length of each spike if Spikes object is one dimensional.r9r;rz3 The position of the lower end of each spike.r{cd|vr|d|d<d|vr1d|vr-|d|df|d<d|vrd|vr|d=t|i|}||d|iS)Nrrrrrr{rc)rradd_collection)rUrerfrg line_segmentss r1rhzSpikesPlot.init_artistss +  #.??3#7#7K [ V{%:%:"-//&"9"9;??6;R;R"RK  + % %&K*?*?F#& A[AA  -(((-((r0c,|d}t|}||dj}|j|dkrTd|vrP|ddg}||d||d} }fdt || D}n,|dg}|jfd|D}|j r d |D}|} g} |D]} t | \}} g} t|| fD]\}}tj|}t|rtjt!|dtjtj}t%|}| |t)| | |<| || tj| ||j}|d d}|rJt3|t4r||vst3|t6r|jd d}|r0|||d <|||||d|vr|ddf}nvt|dkr|ddf}n4|d|df}tA5|!|||}dddn #1swxYwY| f|d|ifS)NT)rrEr! spike_lengthrc,g|]\}}|f||zfgSr/r/)rrDrposs r1rz'SpikesPlot.get_data..&s.CCCtq!aX3q5z*CCCr0c>g|]}|df|dzfgS)rr/)rrDrjrMs r1rz'SpikesPlot.get_data..*s3BBB!adC[1Q4V"45BBBr0cRg|]$}|dddd|ddddf%S)rNrr!r/)rlines r1rz'SpikesPlot.get_data..-s9DDDtT!WTTrT]DGDDbDM2DDDr0rKr?rrrM)"rMrrrpositioncolumnsrarrKrTr"rrr rQrRrr#rrSrr%rrrzrrrrrrrOrr rN)rUrVrWrXrMrrrrYrZr[ clean_spikesspikecolsrvsr]rr? axis_dimsrjrMs @@r1r_zSpikesPlot.get_datas''d'33 J""7F33;m 199t33??Aq6**D*Q-($z!}*=BCCCCs2r{{CCCDDs##B&FBBBBBrBBBD   EDDtDDDD!!##  7 7E%[FBD"B8,, 2Xb\\b>>S ) 9 = =RU !1"-@!!I""B"1gmmy9Q9QmRRDG B    5 5 6 6 6 6$$T%566 '4((  j,, '1A1AjQVX[F\F\1A J  C D D DD  <$55d;;E'N   gvud ; ; ; T ! !!,,..q148II ##%% & &! + +!,,..q148II!,,..q173E3E3G3G3JKI " $ $ C C**7FEBBE C C C C C C C C C C C C C C C i'@@@sNN NcJ|jd}||||\\}}} ||||ddd|vr||dd|vsd|vr<||d|dd|vr#||d|dfd |vr |d |_d |vr| |d | S) Nrcr@Tr?rrrrrrA) rkr_rrrRrrrrr) rUrnrorVrWrXrcrrrps r1rqzSpikesPlot.update_handlesWs.h''+}}Wfe'L'L$599Y55666 f    ! !&/ 2 2 2 f  v   VZZCAA B B B V   OOVF^VF^< = = = V   .FK &  ! !&"5 6 6 6r0)r+r,r-rrraspectrwrrxrzrrKryrzr{rQr%r~rhr_rqr/r0r1rDrDs U_X4)***F &%%dt.13Z>DEEEK 5<2GHHHL"e!(C;NOOOGu|B-6777H$x/J ) ) )7A7A7Atr0rDceZdZejddZejddZejddZejdd Z ej d gd d Z ej d gdd Z dS)SideSpikesPlotrrr4rrr:zG The size subplots as expressed as a fraction of the main plot.r!rFrrrrrrN) r+r,r-rrrrrr subplot_sizerKrurrr/r0r1r[r[iseol9+,,,G%,q/LMMMK 5<2JKKKL 5<0GHHHL !E *?*?*?E; < < ris~ 11111144444444%%%%%%7777777788888888                    '&&&&&&&******!!!!!!6666666666GGGGGGGGGG::::::::::........  EEEEE EEERZZZZZ ZZZ|!!!!!y)!!!>:::::<:::.HHHHHHHH6]]]]]L]]]@c.c.c.c.c. mc.c.c.Ly6y6y6y6y6 <y6y6y6zYYYYYlYYYzz%z%z%z%z%izz%z%z%|iiiiih iiiX:::::\::::::r0