g#ddlmZddlmZmZddlmZddlZddl m Z m Z m Z ddl mZmZmZmZmZmZerddlmZd2d Zd3dZd4dZd5dZd6dZd7dZd8dZd9d Zd:d$Zd:d%Z d9d&Z!d;d'Z"dd+Z%d7d,Z&d?d.Z'd@d/Z(dAd0Z)dBd1Z*dS)C) annotations)chainpairwise) 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 E/granary/edev/quantum/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) rr rrlenr r r allr )rr r 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) r r%rrr r r r&r )r rrs 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_codesr2=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)r2)r3r,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)rr9r;rs rr<z"concat_offsets..WsGGGGGBKGGGrrrc3PK|] }|dzdd|zV!dS)rNr9)r;i cumulativer>s r z!concat_offsets..Ys>\\Aqs3ABB7*Q-G\\\\\\rr)r/r%rcumsumrr0range)r>rretrEs` @rconcat_offsetsrJPs @>??? 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|]}||Sr8r9rBs rr<z*concat_offsets_or_none..ds]]]7I\wI\I\I\rN)rJ)rKr>s rconcat_offsets_or_nonerP_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/rr0rrQs r concat_pointsrUks0 ?=>>> >. < < <.yXXXVEWfEWEWEWrN)rUrVrQs 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|]}||Sr8r9r[s rr<z2concat_points_or_none_with_nan..r\rN)concat_points_with_nanr]s rconcat_points_or_none_with_nanrbs4 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 dSr8r9)r;x nan_spacers rrFz)concat_points_with_nan..s''T'TAQ'T'T'T'T'T'TrN) r/r%rrnanrlistrrU)rQrgs @rraras 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. rdrrrrr#) r rr%rarrayrhrinsertastypeint64)r rrgs rinsert_nan_at_offsetsrosfw 7||q Xrvrv.kBBB y2!5!5bh!?!?RSTTTTrrct|tjtj|tkdt |tS)zBDetermine offsets from codes using locations of MOVETO codes. r)rrappendnonzeror r%rmr)rs roffsets_from_codesrssKU 9RZ003SZZ @ @ G G U UUrcv|stdtjdgd|DztS)z4Determine offsets from lengths of point arrays. z)Empty list passed to offsets_from_lengthsrc,g|]}t|Sr9r%)r;lines rr<z(offsets_from_lengths..sAAA$CIIAAArrr/rrGrrTs roffsets_from_lengthsrysG 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 Sr9)r count_nonzeror r:s rr<z4outer_offsets_from_list_of_codes..s'YYY%B,Uf_==YYYrrrxr1s 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 SrrvrBs rr<z6outer_offsets_from_list_of_offsets..s"JJJwCLLNJJJrrrx)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#rdrr) r rrrisnanr%rkrdeletearangeempty)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. rdrr)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. rdrr)r rr%rr)r rs rsplit_points_by_offsetsrsR fw 7||ax" ...xrc<ttjtjdddfd}t |dkrgStjdg|t gf}fdt |DS)z>Split a points array at NaNs into a list of point arrays. Nrrc0g|]\}}|dz|Srr9)r;ser s rr<z'split_points_at_nan..s)@@@$!Qqs1u @@@r)r rrrrr%r0r)r rs` rsplit_points_at_nanrsf*RXfQQQTl3344Q7K ;1xnrdK#f++%GHH @@@@(;*?*?@@@@r)rrrr)rrr r!rr)r r!rr)r,r-rr)r3r4rr5)r>r?rr)rKrLrrM)rQrRrr!)rVrWrrX)r r!rrrr!)rrrr)rQrRrr)r,r-rr)r r!rr)rrrrrr-)r r!rrrrR)r r!rrR)+ __future__r itertoolsrrtypingrnumpyrcontourpy.typecheckrrr contourpy.typesr r r r rrcontourpy._contourpy _contourpycpyrr)r+r2r=rJrPrUr^rbrarorsryr}rrrrrr9rrrse""""""%%%%%%%% WWWWWWWWWW\\\\\\\\\\\\\\\\'&&&&&&    (    ;;;;        ====         - - - - U U U UVVVVWWWW))))````$         A A A A A Ar