:j1ddlZddlmZmZddlmZmZdZ GddZ GddeZ Gd d Z Gd d ejZ Gd dZGddZGddZy)N)ticker_api)Bbox Transformcrg}|dd|ddz }tdtdddfD]}|j|\}}|j|\}}|j|\} } |j|\} } | || kDf| || kffD]\} }g}|dd|ddz j \}||| ||z ||z||z z}|j t |||Dcgc]E\}}| |cxkr| kr5nn2| |f|tjtj|dddfGc}} |Scc}}w)a Find the points where a polyline crosses a bbox, and the crossing angles. Parameters ---------- xys : (N, 2) array The polyline coordinates. bbox : `.Bbox` The bounding box. Returns ------- list of ((float, float), float) Four separate lists of crossings, for the left, right, bottom, and top sides of the bbox, respectively. For each list, the entries are the ``((x, y), ccw_angle_in_degrees)`` of the crossing, where an angle of 0 means that the polyline is moving to the right at the crossing point. The entries are computed by linearly interpolating at each crossing between the nearest points on either side of the bbox edges. N) sliceTminmaxnonzeroappendzipnpdegreesarctan2)xysbbox crossingsdxysslusvsdusdvsuminvminumaxvmaxu0insidecrossidxsvvvdxys x/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/mpl_toolkits/axisartist/grid_finder.py_find_line_box_crossingsr)sw,I qr7S"X DT{E$b12 IrB66":SXXb\ dXXb\ d "t),tR$Y.?@ IJBECR[6!":-668EDDR"T(]c$i7#d)CCB   !"d4j1HAsTQ5F$5FabjjS2Y)?@AH I  I I HsA D3c"eZdZdZdZdZdZy)ExtremeFinderSimplezU A helper class to figure out the range of grid lines that need to be drawn. c ||_||_y)zy Parameters ---------- nx, ny : int The number of samples in each direction. N)nxny)selfr-r.s r(__init__zExtremeFinderSimple.__init__3sc |jt|dtj||||}|j|j |j |jfS)am Compute an approximation of the bounding box obtained by applying *transform_xy* to the box delimited by ``(x1, y1, x2, y2)``. The intended use is to have ``(x1, y1, x2, y2)`` in axes coordinates, and have *transform_xy* be the transform from axes coordinates to data coordinates; this method then returns the range of data coordinates that span the actual axes. The computation is done by sampling ``nx * ny`` equispaced points in the ``(x1, y1, x2, y2)`` box and finding the resulting points with extremal coordinates; then adding some padding to take into account the finite sampling. As each sampling step covers a relative range of ``1/nx`` or ``1/ny``, the padding is computed by expanding the span covered by the extremal coordinates by these fractions. N)_find_transformed_bbox_User2DTransformr from_extentsx0x1y0y1)r/ transform_xyr7r9x2y2tbboxs r(__call__zExtremeFinderSimple.__call__=sR&++ \4 0$2C2CBBPR2SUxx588UXX55r1c tjtjtj|j|j |j tj|j|j|jdj}tj}|j|j||jdd|j z zdd|jz zS)a Compute an approximation of the bounding box obtained by applying *trans* to *bbox*. See ``__call__`` for details; this method performs similar calculations, but using a different representation of the arguments and return value. )r rr@)rreshapemeshgridlinspacer6r7r-r8r9r.r rnullupdate_from_data_xy transformexpanded)r/transrgridr=s r(r3z*ExtremeFinderSimple._find_transformed_bboxTszz"++bkk$''477DGG&L&(kk$''477DGG&LN!##$1   !!%//$"78~~a!dgg+oq1tww;??r1N)__name__ __module__ __qualname____doc__r0r>r3r1r(r+r+.s6.@r1r+c6eZdZdZdxZZfdZdZdZxZ S)r4z.A transform defined by two user-set functions.r@c>t|||_||_y)z Parameters ---------- forward, backward : callable The forward and backward transforms, taking ``x`` and ``y`` as separate arguments and returning ``(tr_x, tr_y)``. N)superr0_forward _backward)r/forwardbackward __class__s r(r0z_User2DTransform.__init__js  !r1cltj|jtj|SN)r transposerR)r/valuess r(transform_non_affinez%_User2DTransform.transform_non_affinexs%||MDMM2<<+?@AAr1cNt||j|jSrX)typerSrRr/s r(invertedz_User2DTransform.inverted|stDz$..$--88r1) rJrKrLrM input_dims output_dimsr0r[r_ __classcell__rVs@r(r4r4es 8  J "B9r1r4ceZdZdZ ddZdZdZdZdZdZ eZ e jd d d Z e jd d dZdZy) GridFinderz Internal helper for `~.grid_helper_curvelinear.GridHelperCurveLinear`, with the same constructor parameters; should not be directly instantiated. Nc| tdd}| t}| t}| t}| t}||_||_||_||_||_|j|y)N) r+ MaxNLocatorFormatterPrettyPrintextreme_finder grid_locator1 grid_locator2tick_formatter1tick_formatter2 set_transform)r/rFrjrkrlrmrns r(r0zGridFinder.__init__s  !0R8N  'MM  'MM  "24O  "24O,**.. 9%r1ctj|j|jd|}t |t j r|j|S||||S)a Helper to support both standard formatters (inheriting from `.mticker.Formatter`) and axisartist-specific ones; should be called instead of directly calling ``self.tick_formatter1`` and ``self.tick_formatter2``. This method should be considered as a temporary workaround which will be removed in the future at the same time as axisartist-specific formatters. )rr@)idx)rgetitem_checkedrmrn isinstancemticker Formatter format_ticks)r/rq directionfactorlevelsfmts r( _format_tickszGridFinder._format_tickss^""$$)=)= >CI,6sGz*GridFinder.get_grid_info..sFHr1ctSrXr~)rs r(rz*GridFinder.get_grid_info..s68r1r73.11zPassing extents as separate arguments to get_grid_info is deprecated since %(since)s and support will be removed %(removal)s; pass a single bbox instead.)messager9r;r<rNgp ?extremesrlonr@lat)leftrightbottomtop)linesticksr)levellocrlabel)rselect_matching_signaturewarn_deprecatedrr5rjr3 get_transformr_rk intervalxrl intervalyrasarray_get_raw_grid_linesrGrr)rr{)#r/argskwargsparamsrr=lon_levslon_n lon_factorlat_levslat_n lat_factor lon_values lat_values lon_lines lat_lines bbox_expanded grid_inforq lon_or_latlevsrxrZrgirr&r all_crossingssidercrossingticklabelsrs# r( get_grid_infozGridFinder.get_grid_infos // ,.C DWGKWOUW 6>  ' )$$t fTlF4L&,HD&>D##::    ) ) +T3'9d&8&8%//&J#%&8d&8&8%//&J#%ZZ% 01J> ZZ% 01J> #77 JPUV 9 i; ' E8ZYGE8ZYG=  * 8CT665 "$rRK* Ij !B"%UFD!9 T Q 8m L '*:M(KTOD)$-T7 D)0051RSTT T 7  *24W+d2CD$W DD++CvtD#&r'{4'8&#A*KD%$)DM* * *( Es Hc |j}tj|j|jd}tj|j |j d}|Dcgc]<}|jtjtj|||g>}}|D cgc]<} |jtj|tj|| g>} } || fScc}wcc} w)Nd) rrrCr6r7r8r9rF column_stack full_like) r/rrrrHlonslatsrrrrs r(rzGridFinder._get_raw_grid_liness""${{477DGGS1{{477DGGS1 *,__R__bll46Mt5T%UV, ,!+,__R__dBLLsct|tr||_yt|dk(r't t t |rt||_ytd)Nr@zF'aux_trans' must be either a Transform instance or a pair of callables) rsr_aux_transformlenallmapcallabler4 TypeError)r/ aux_transs r(rozGridFinder.set_transformsM i +"+D  ^q SXy)A%B"2I">D >? ?r1c|jSrX)rr^s r(rzGridFinder.get_transforms"""r1rzgrid_finder.get_transform()) alternativecv|jjtj||gjSrX)rrFrrr r/xys r(r:zGridFinder.transform_xys,"",,R__aV-DEGGGr1z&grid_finder.get_transform().inverted()c|jjjtj||gj SrX)rr_rFrrr rs r(inv_transform_xyzGridFinder.inv_transform_xys9""++-77 OOQF #%%&Q 'r1c r|jD]$\}}|dvrt|||td|y)N)rjrkrlrmrnzUnknown update property )itemssetattr ValueError)r/rkr&s r(updatezGridFinder.updatesKLLN CDAq(( a# #;A5!ABB Cr1)NNNNN)rJrKrLrMr0r{rrrorupdate_transformr deprecatedr:rrrNr1r(reres!%##!%!% &0 53j$?#%T__V)FGHHHT__V)QR'S' Cr1rec4eZdZ dfd ZfdZxZS)rhcNt|||||||jy)N)stepsinteger symmetricprune)rQr0create_dummy_axis)r/nbinsrtrimrrrrVs r(r0zMaxNLocator.__init__s. eW#,E  ;  r1cft|||}tj|t |dfSNr)rQ tick_valuesrarrayr)r/v1v2locsrVs r(r>zMaxNLocator.__call__s-w"2r*xx~s4y!++r1) NTFFNrJrKrLr0r>rbrcs@r(rhrhs!'+  !,,r1rhceZdZdZdZy) FixedLocatorc||_yrX)_locs)r/rs r(r0zFixedLocator.__init__s  r1ct||g\}}tj|jDcgc]}||cxkr|ksnn|c}}|t |dfScc}wr)sortedrrrr)r/rrlrs r(r>zFixedLocator.__call__"sRR!BxxDJJ@q"-R-@ASY!!As AANrJrKrLr0r>rNr1r(rrs "r1rceZdZddZdZy)ricptj|d|_|jjy)NF) useMathText useOffset)rtScalarFormatter_fmtr)r/rs r(r0zFormatterPrettyPrint.__init__+s)++#u6  ##%r1c8|jj|SrX)rrv)r/rwrxrZs r(r>zFormatterPrettyPrint.__call__0syy%%f--r1N)TrrNr1r(riri*s & .r1ric&eZdZdfd ZdZxZS) DictFormatterc>t|||_||_y)zq format_dict : dictionary for format strings to be used. formatter : fall-back formatter N)rQr0 _format_dict_fallback_formatter)r/ format_dict formatterrVs r(r0zDictFormatter.__init__5s '#, r1c|jr|j|||}ndgt|z}t||Dcgc]!\}}|jj ||#c}}Scc}}w)zG factor is ignored if value is found in the dictionary )rrrrget)r/rwrxrZfallback_stringsrr&s r(r>zDictFormatter.__call__>sw  # ##7766 + !#tc&k1 (89;Aq!!%%a+; ;;s&A)rXrrcs@r(rr4s - ;r1r)numpyr matplotlibrrtrmatplotlib.transformsrrr)r+r4rerhrrirrNr1r(rsl.1$N4@4@n9y98JCJCZ ,'%% , ""..;;r1