a f1b2@sddlZddlZddlmZddZddZddZd'd d Zd(ddZ d)ddZ GdddZ Gddde Z Gddde Z Gddde ZGddde ZGddde ZGdd d e ZGd!d"d"ZGd#d$d$eZGd%d&d&eZdS)*N)ExtremeFinderSimplecCsgd}gd}dgt|}gd}gd}t|d}dgt|}t|d}d gt|} g|||} g|||} g| ||}t| |} | | } || }| |fS) N) ? (Fxii)  -Zih?)r@ @ r)r r rr rrrr<N@ @lennparrayZ searchsorted)dvZdegree_limits_Z degree_steps_Zdegree_factorsminsec_limits_ minsec_steps_minute_limits_minute_factorssecond_limits_second_factorsZ degree_limitsZ degree_stepsnstepfactorr-Clib/python3.9/site-packages/mpl_toolkits/axisartist/angle_helper.pyselect_step_degrees  r/cCsgd}gd}dgt|}gd}gd}t|d}dgt|}t|d}d gt|} g|||} g|||} g| ||}t| |} | | } || }| |fS) N) rrrr rrr$) r r rr rr) rrrg@g@rrrrr) r r rr2r r3rr4rrrrrrrr)r#Z hour_limits_Z hour_steps_Z hour_factorsr$r%r&r'r(r)Z hour_limitsZ hour_stepsr*r+r,r-r-r.select_step_hour!s  r7cCshdtt|d}d|}d||kr0d}n0d||krBd}nd||krTd}n d}d |}||fS) Ng$@rrr g@r g@r g?)intmathlog10)r#tmpr,r+r-r-r.select_step_sub;s   r<FTrc Cs$||kr||}}|||}|r,t}d}nt}d}|d|krN||\} } nt||\} } | |} tjt|| | t|| | dtd| } t| } | dkr| d| d|krt|| }|r| dtd|dd| } n| dtd|d| } t| } t | | | fS) Ng8@v@r g?)Zdtyperr) r7r/r<r!ZarangefloorZceilr8r r") v1v2nvhour include_lastthreshold_factorr#Z _select_stepcycler+r,levsr*r-r-r. select_stepOs4       rHrcCs:|d|d}}t|||d||d\}}}|d||fS)NrTrCrDrErH)r@rArBrDrErGr*r,r-r-r. select_step24s   rKcCst|||d||dS)NFrIrJ)r@rArBrDrEr-r-r.select_step360s rLc@s eZdZdddZdddZdS) LocatorBaseTcCs||_||_dSN)nbins _include_last)selfrOrDr-r-r.__init__szLocatorBase.__init__NcCs|durt||_dSrN)r8rO)rQrOr-r-r. set_paramsszLocatorBase.set_params)T)N)__name__ __module__ __qualname__rRrSr-r-r-r.rMs rMc@seZdZddZdS) LocatorHMScCst|||j|jSrNrKrOrPrQr@rAr-r-r.__call__szLocatorHMS.__call__NrTrUrVrZr-r-r-r.rWsrWc@seZdZddZdS) LocatorHMcCst|||j|jddSNrrErXrYr-r-r.rZszLocatorHM.__call__Nr[r-r-r-r.r\sr\c@seZdZddZdS)LocatorHcCst|||j|jddSNr r^rXrYr-r-r.rZszLocatorH.__call__Nr[r-r-r-r.r_sr_c@seZdZddZdS) LocatorDMScCst|||j|jSrNrLrOrPrYr-r-r.rZszLocatorDMS.__call__Nr[r-r-r-r.rasrac@seZdZddZdS) LocatorDMcCst|||j|jddSr]rbrYr-r-r.rZszLocatorDM.__call__Nr[r-r-r-r.rcsrcc@seZdZddZdS)LocatorDcCst|||j|jddSr`rbrYr-r-r.rZszLocatorD.__call__Nr[r-r-r-r.rdsrdc@seZdZdZdZdZdedZdedZdededZded edZ deded Z d edZ d edZ d dZ ddZdS) FormatterDMSz^{\circ}z ^{\prime}z^{\prime\prime}$%d$$%d.%s$%s%d\,%02d \,%02d.%s\,%02d%02d.%scCsnd}dD]\}||krqf||}ttt|}d||kr|dkr|}|d|}||fSq||fS)N)r rrrr )r8r!r?r:)rQr,number_fractionZ thresholddZ int_log_dr-r-r._get_number_fractions z!FormatterDMS._get_number_fractioncs:t|dkrgSt|}dd|D}|\}}t|}|durvt|d|\}}d|ffdd|D}|dkr|durfddt||DSfd dt|||DSnz|d krt|d \} } |durfd dt|| | DSfd dt|| | |DSn |d kr$|ddkrPd} |ddd}|ddd}nd} d} g} t|d \} }t|d \} }|durfdd|D}nfddt||D}t|| | |D]L\}}}}j|||f}|| kr|} ||}n d||}| |q| r| dddS| Sndd||DSdS)NrcSsg|]}|dkrdndqS)r-r-.0vr-r-r. z)FormatterDMS.__call__..rz%%0%ddcsg|]}|fqSr-r-)ruf1)frac_fmtr-r.rwrxr cs$g|]\}}j|t|fqSr-)fmt_dr8)rusrvrQr-r.rwrxcs(g|] \}}}j|t||fqSr-)fmt_dsr8)rur|rvryr}r-r.rwsrcs"g|]\}}}j|||fqSr-)fmt_d_m)rus1d1m1r}r-r.rwscs&g|]\}}}}j||||fqSr-)fmt_d_ms)rur|rrryr}r-r.rws rr>TFrscsg|]}j|fqSr-) fmt_s_partial)rurr}r-r.rw rxcsg|]\}}j||fqSr-)fmt_ss_partial)rurryr}r-r.rw srgcSsg|]}dt|fqS)z $%s^{\circ}$)strrtr-r-r.rwrx) r r!Zsignrqabsdivmodzipfmt_d_m_partialappend)rQ directionr,valuesZssZsignsroZ frac_partZfrac_strZdeg_partZmin_partZ inverse_orderZl_hm_oldrZ min_part_Zsec_partZsec_strr|rrrZl_hmlr-)rzrQr.rZsf                 zFormatterDMS.__call__N)rTrUrVdeg_markmin_marksec_markr{r~rrrrrrqrZr-r-r-r.res    recseZdZdZdZdZdedZdedZdededZded edZ deded Z d edZ d edZ fd dZ ZS) FormatterHMSz ^\mathrm{h}z ^\mathrm{m}z ^\mathrm{s}rfrgrhrirjrkrlrmrncst||t|dS)Nr)superrZr!Zasarray)rQrr,r __class__r-r.rZ2szFormatterHMS.__call__)rTrUrVrrrr{r~rrrrrrZ __classcell__r-r-rr.r"s    rc@seZdZdddZddZdS) ExtremeFinderCycler=NircCs,|||_|_|||_|_||_||_dS)aB This subclass handles the case where one or both coordinates should be taken modulo 360, or be restricted to not exceed a specific range. Parameters ---------- nx, ny : int The number of samples in each direction. lon_cycle, lat_cycle : 360 or None If not None, values in the corresponding direction are taken modulo *lon_cycle* or *lat_cycle*; in theory this can be any number but the implementation actually assumes that it is 360 (if not None); other values give nonsensical results. This is done by "unwrapping" the transformed grid coordinates so that jumps are less than a half-cycle; then normalizing the span to no more than a full cycle. For example, if values are in the union of the [0, 2] and [358, 360] intervals (typically, angles measured modulo 360), the values in the second interval are normalized to [-2, 0] instead so that the values now cover [-2, 2]. If values are in a range of [5, 1000], this gets normalized to [5, 365]. lon_minmax, lat_minmax : (float, float) or None If not None, the computed bounding box is clipped to the given range in the corresponding direction. N)nxny lon_cycle lat_cycle lon_minmax lat_minmax)rQrrrrrrr-r-r.rR9s zExtremeFinderCycle.__init__cCstt|||jt|||j\}}|t|t|\}} tjdd`|jdurxt|} |d|| dk8}|j durt| } | d| | dk8} Wdn1s0Yt|t |} } t| t | }}| | | ||\} } }}|jrt | | |j} |j r2t |||j }|j durf|j d}t|| } |j d}t || } |jdur|jd}t||}|jd}t ||}| | ||fS)Nignore)Zinvalidr=gf@rr )r!ZmeshgridZlinspacerrZravelZerrstaterZnanminrZnanmaxZ_add_padminrmaxr)rQZ transform_xyZx1Zy1Zx2Zy2xyZlonZlatZlon0Zlat0Zlon_minZlon_maxZlat_minZlat_maxZmin0Zmax0r-r-r.rZ^s<    2           zExtremeFinderCycle.__call__)r=NNr)rTrUrVrRrZr-r-r-r.r6s %r)FTr)Tr)Tr)Znumpyr!r9Z#mpl_toolkits.axisartist.grid_finderrr/r7r<rHrKrLrMrWr\r_rarcrdrerrr-r-r-r.s(  1   h