\e4#ddlmZddlmZddlmZddlZddlm Z m Z m Z ddl m Z mZmZmZmZmZerddlmZd2d Zd3dZd4dZd5dZd6dZd7dZd8dZd9d Zd:d$Zd:d%Zd9d&Z d;d'Z!dd+Z$d7d,Z%d?d.Z&d@d/Z'dAd0Z(dBd1Z)dS)C) annotations)chain) TYPE_CHECKINGN)check_code_arraycheck_offset_arraycheck_point_array) CLOSEPOLYLINETOMOVETO code_dtype offset_dtype point_dtypeoffsetscpy.OffsetArrayreturn cpy.CodeArrayct||d}tj|tt}t ||dd<t ||dddz <|S)zSDetermine codes from offsets, assuming they all correspond to closed polygons. dtypeN)rnpfullr r r r )rncodess /lib/python3.11/site-packages/contourpy/array.pycodes_from_offsetsrs^w A GAvZ 0 0 0E E'#2#,&E'!""+/ Lpointscpy.PointArraycXt|t|tjt |t t }t||dd<|dddz }tj||dd||kd}t|||<|S)zDetermine codes from offsets and points, using the equality of the start and end points of each line to determine if lines are closed or not. rNrraxis) rrrrlenr r r allr )rrr end_offsetscloseds rcodes_from_offsets_and_pointsr(swf GCKKz : : :E E'#2#,!""+/K VF73B3<(F;,??a H H HF!*E+f  Lrct|t|}tj|tt }t |d<tj|d|dkr t|d<|S)zDetermine codes for a single line, using the equality of the start and end points to determine if the line is closed or not. rrr) rr$rrr r r r%r )rrrs rcodes_from_pointsr*/sif F A GAvZ 0 0 0EE!H vfQi6":%&&b Lr list_of_codeslist[cpy.CodeArray]cZ|stdtj|tS)zAConcatenate a list of codes arrays into a single code array. z!Empty list passed to concat_codesr) ValueErrorr concatenater r+s r concat_codesr1=s0 ><=== >-z : : ::rlist_of_codes_or_nonelist[cpy.CodeArray | None]cpy.CodeArray | Nonec@d|D}|rt|SdS)zQConcatenate a list of codes arrays or None into a single code array or None. cg|]}||SN.0rs r z(concat_codes_or_none..IsSSSuARUARARARrN)r1)r2r+s rconcat_codes_or_noner<Fs4TS(=SSSMM***trlist_of_offsetslist[cpy.OffsetArray]cstdt}tjdDttjdgfdt |dz DRt}|S)zEConcatenate a list of offsets arrays into a single offset array. z#Empty list passed to concat_offsetscg|] }|d S)rr8r:rs rr;z"concat_offsets..WsGGGGGBKGGGrrrc3PK|] }|dzdd|zV!dS)rNr8)r:i cumulativer=s r z!concat_offsets..Ys>\\Aqs3ABB7*Q-G\\\\\\rr)r.r$rcumsumr r/range)r=rretrDs` @rconcat_offsetsrIPs @>??? OAGGGGG|\\\J>  ]\\\\\QVWXYZWZQ[Q[\\\]]C Jrlist_of_offsets_or_nonelist[cpy.OffsetArray | None]cpy.OffsetArray | Nonec@d|D}|rt|SdS)zUConcatenate a list of offsets arrays or None into a single offset array or None. cg|]}||Sr7r8rAs rr;z*concat_offsets_or_none..ds]]]7I\wI\I\I\rN)rI)rJr=s rconcat_offsets_or_nonerO_s4 ^].E]]]Oo...trlist_of_pointslist[cpy.PointArray]cZ|stdtj|tS)zBConcatenate a list of point arrays into a single point array. z"Empty list passed to concat_pointsr)r.rr/rrPs r concat_pointsrTks0 ?=>>> >. < < <.yXXXVEWfEWEWEWrN)rTrUrPs rconcat_points_or_noner]ts4 YX+AXXXN^,,,trc@d|D}|rt|SdS)zvConcatenate a list of points or None into a single point array or None, with NaNs used to separate each line. cg|]}||Sr7r8rZs rr;z2concat_points_or_none_with_nan..r[rN)concat_points_with_nanr\s rconcat_points_or_none_with_nanras4 YX+AXXXN%n555trc ,|stdt|dkr|dStjdtjt |dgt tfd|ddD}t|S)zaConcatenate a list of points into a single point array with NaNs used to separate each line. z+Empty list passed to concat_points_with_nanrr)rrc3 K|]}|fV dSr7r8)r:x nan_spacers rrEz)concat_points_with_nan..s''T'TAQ'T'T'T'T'T'TrN) r.r$rrnanrlistrrT)rPrfs @rr`r`s HFGGG >aa  WVRV;??? (+X'T'T'T'TPQPRPRAS'T'T'T UVVX^,,,rcDt|t|t|dkr|Stjtjtjgt }tj||ddtj |dS)zNInsert NaNs into a point array at locations specified by an offset array. rcrrrrr") rrr$rarrayrgrinsertastypeint64)rrrfs rinsert_nan_at_offsetsrnsfw 7||q Xrvrv.kBBB y2!5!5bh!?!?RSTTTTrrct|tjtj|tkdt |tS)zBDetermine offsets from codes using locations of MOVETO codes. r)rrappendnonzeror r$rlr )rs roffsets_from_codesrrsKU 9RZ003SZZ @ @ G G U UUrcv|stdtjdgd|DztS)z4Determine offsets from lengths of point arrays. z)Empty list passed to offsets_from_lengthsrc,g|]}t|Sr8r$)r:lines rr;z(offsets_from_lengths..sAAA$CIIAAArrr.rrFr rSs roffsets_from_lengthsrxsG FDEEE 9aSAA.AAAA V V VVrcv|stdtjdgd|DztS)zHDetermine outer offsets from codes using locations of MOVETO codes. z5Empty list passed to outer_offsets_from_list_of_codesrcHg|]}tj|tk Sr8)r count_nonzeror r9s rr;z4outer_offsets_from_list_of_codes..s'YYY%B,Uf_==YYYrrrwr0s r outer_offsets_from_list_of_codesr|sP RPQQQ 9aSYY=YYYY' ) ) ))rcv|stdtjdgd|DztS)z4Determine outer offsets from a list of offsets. z7Empty list passed to outer_offsets_from_list_of_offsetsrc2g|]}t|dz SrrurAs rr;z6outer_offsets_from_list_of_offsets..s"JJJwCLLNJJJrrrw)r=s r"outer_offsets_from_list_of_offsetsrsH TRSSS 9aSJJ/JJJJR^ _ _ __r&tuple[cpy.PointArray, cpy.OffsetArray]ct|tjtj|dddfd}t |dkr,|tjdt |gt fStj||d}|tjt |z}tj t |dzt }d|d<||dd<t ||d<||fS)z^Remove NaN from a points array, also return the offsets corresponding to the NaN removed. Nrrr"rcrr) rrrqisnanr$rjr deletearangeempty)r nan_offsetsrs r remove_nanrsf*RXfQQQTl3344Q7K ;1rxCKK 0 EEEEE6;Q777ry[!1!1222 #%8C ,<,l#S#S#S #" &kk wrct|t|t|dkrtj||ddS|gS)z]Split a code array at locations specified by an offset array into a list of code arrays. rcrr)rrr$rsplit)rrs rsplit_codes_by_offsetsrsRUw 7||axwqt}---wrct|t|t|dkrtj||ddS|gS)z_Split a point array at locations specified by an offset array into a list of point arrays. rcrr)rrr$rr)rrs rsplit_points_by_offsetsrsR fw 7||ax" ...xrc^ttjtjdddfd}t |dkrgStjdg|t gf}fdt |dd|ddDS)z>Split a points array at NaNs into a list of point arrays. Nrrc0g|]\}}|dz|Srr8)r:sers rr;z'split_points_at_nan..s)QQQ$!Qqs1u QQQrr)rrrqrr$r/zip)rrs` rsplit_points_at_nanrsf*RXfQQQTl3344Q7K ;1xnrdK#f++%GHH QQQQ#k#2#.> ABB*P*PQQQQr)rrrr)rrrr rr)rr rr)r+r,rr)r2r3rr4)r=r>rr)rJrKrrL)rPrQrr )rUrVrrW)rr rrrr )rrrr)rPrQrr)r+r,rr)rr rr)rrrrrr,)rr rrrrQ)rr rrQ)* __future__r itertoolsrtypingrnumpyrcontourpy.typecheckrrrcontourpy.typesr r r r r rcontourpy._contourpy _contourpycpyrr(r*r1r<rIrOrTr]rar`rnrrrxr|rrrrrr8rrrs_"""""" WWWWWWWWWW\\\\\\\\\\\\\\\\'&&&&&&    (    ;;;;        ====         - - - - U U U UVVVVWWWW))))````$         R R R R R Rr