Vf4#DddlmZddlmZddlmZddlZddlm Z m Z m Z ddl m Z mZmZmZmZmZerddlmZddZ ddZdd Zdd Zdd Zd d Z d!d Zd"dZ d#dZ d#dZd"dZ d$dZ!d%dZ"d&dZ#d'dZ$d dZ%d(dZ&d)dZ' d*dZ(d+dZ)y),) annotations)chain) TYPE_CHECKINGN)check_code_arraycheck_offset_arraycheck_point_array) CLOSEPOLYLINETOMOVETO code_dtype offset_dtype point_dtypect||d}tj|tt}t ||dd<t ||dddz <|S)zSDetermine codes from offsets, assuming they all correspond to closed polygons. dtypeN)rnpfullr r r r )offsetsncodess X/var/www/html/software/conda/envs/catlas/lib/python3.12/site-packages/contourpy/array.pycodes_from_offsetsrsPw A GGAvZ 0E E'#2,&E'!"+/ Lc t|t|tjt |t t }t||dd<|dddz }tj||dd||k(d}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 )rpointsr end_offsetscloseds rcodes_from_offsets_and_pointsr$swf GGCKz :E E'#2,!"+/K VVF73B<(F;,??a HF!*E+f  Lrct|t|}tj|tt }t |d<tj|d|dk(r 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) rrrrr r r r r )r!rrs rcodes_from_pointsr&/sWf F A GGAvZ 0EE!H vvfQi6":%&b LrcR|s tdtj|tS)zAConcatenate a list of codes arrays into a single code array. z!Empty list passed to concat_codesr) ValueErrorr concatenater ) list_of_codess r concat_codesr+=s$ <== >>-z ::rcL|Dcgc]}|| }}|r t|Sycc}w)zQConcatenate a list of codes arrays or None into a single code array or None. N)r+)list_of_codes_or_nonerr*s rconcat_codes_or_noner.Fs3)>SuARUSMSM** T!!c s tdt}tjDcgc]}|d c}ttj dgfdt |dz Dt}|Scc}w)zEConcatenate a list of offsets arrays into a single offset array. z#Empty list passed to concat_offsetsrrrc3@K|]}|dzdd|zyw)rN).0i cumulativelist_of_offsetss r z!concat_offsets..Ys*\Aqs3AB7*Q-G\sr)r(rrcumsumr r)range)r6rrretr5s` @rconcat_offsetsr;Ps~ >?? OAGGGBKG|\J>>  ]\QVWXYZWZQ[\]C J Hs BcL|Dcgc]}|| }}|r t|Sycc}w)zUConcatenate a list of offsets arrays or None into a single offset array or None. N)r;)list_of_offsets_or_nonerr6s rconcat_offsets_or_noner>_s4 /F]7I\w]O]o.. ^r/cR|s tdtj|tS)zBConcatenate a list of point arrays into a single point array. z"Empty list passed to concat_pointsr)r(rr)r)list_of_pointss r concat_pointsrAks$ =>> >>. <.s'TAQ'TsN) r(rrrnanrlistrrA)r@rMs @rrGrGs FGG >aa  WWVRVV;? (+X'TPQPRAS'T UVX^,,rc2t|t|t|dkr|Stjtj tj gt }tj||ddjtj|dS)zNInsert NaNs into a point array at locations specified by an offset array. rJrrrrr) rrrrarrayrNrinsertastypeint64)r!rrMs rinsert_nan_at_offsetsrUsmfw 7|q XXrvvrvv.kB yy2!5!5bhh!?RSTTrct|tjtj|tk(dt |j tS)zBDetermine offsets from codes using locations of MOVETO codes. r)rrappendnonzeror rrSr )rs roffsets_from_codesrYs>U 99RZZ03SZ @ G G UUrc |s tdtjdg|Dcgc] }t|c}ztScc}w)z4Determine offsets from lengths of point arrays. z)Empty list passed to offsets_from_lengthsrrr(rr8rr )r@lines roffsets_from_lengthsr]s= DEE 99aS.A$CIAA VVAsA c |s tdtjdg|Dcgc]}tj|tk( c}zt Scc}w)zHDetermine outer offsets from codes using locations of MOVETO codes. z5Empty list passed to outer_offsets_from_list_of_codesrr)r(rr8 count_nonzeror r )r*rs r outer_offsets_from_list_of_codesr`sM PQQ 99aS=Y%B,,Uf_=YY' ))Ys#A c |s tdtjdg|Dcgc]}t|dz c}ztScc}w)z4Determine outer offsets from a list of offsets. z7Empty list passed to outer_offsets_from_list_of_offsetsrrrr[)r6rs r"outer_offsets_from_list_of_offsetsrbsB RSS 99aS/JwCLNJJR^ __JsA ct|tjtj|dddfd}t |dk(r(|tj dt |gt fStj||d}|tjt |z}tjt |dzt }d|d<||ddt ||d<||fS)z^Remove NaN from a points array, also return the offsets corresponding to the NaN removed. NrrrrJrr) rrrXisnanrrQr deletearangeempty)r! nan_offsetsrs r remove_nanrisf**RXXfQTl34Q7K ;1rxxCK 0 EEE6;Q7ryy[!122 #%88C ,l#S #" &k wrct|t|t|dkDrtj||ddS|gS)z]Split a code array at locations specified by an offset array into a list of code arrays. rJrr)rrrrsplit)rrs rsplit_codes_by_offsetsrls>Uw 7|axxwq}--wrct|t|t|dkDrtj||ddS|gS)z_Split a point array at locations specified by an offset array into a list of point arrays. rJrr)rrrrrk)r!rs rsplit_points_by_offsetsrns> fw 7|axx" ..xrcFt|tjtj|dddfd}t |dk(r|gStj dg|t |gf}t |dd|ddDcgc] \}}||dz|c}}Scc}}w)z>Split a points array at NaNs into a list of point arrays. Nrrr)rrrXrdrr)zip)r!rhses rsplit_points_at_nanrssf**RXXfQTl34Q7K ;1xnnrdK#f+%GH *-k#2.> AB*PQ$!Qqs1 QQQsB)rcpy.OffsetArrayreturn cpy.CodeArray)rrtr!cpy.PointArrayrurv)r!rwrurv)r*list[cpy.CodeArray]rurv)r-zlist[cpy.CodeArray | None]ruzcpy.CodeArray | None)r6zlist[cpy.OffsetArray]rurt)r=zlist[cpy.OffsetArray | None]ruzcpy.OffsetArray | None)r@list[cpy.PointArray]rurw)rDzlist[cpy.PointArray | None]ruzcpy.PointArray | None)r!rwrrtrurw)rrvrurt)r@ryrurt)r*rxrurt)r!rwruz&tuple[cpy.PointArray, cpy.OffsetArray])rrvrrtrurx)r!rwrrtrury)r!rwrury)* __future__r itertoolsrtypingrnumpyrcontourpy.typecheckrrrcontourpy.typesr r r r r rcontourpy._contourpy _contourpycpyrr$r&r+r.r;r>rArErHrGrUrYr]r`rbrirlrnrsr2rrrs" WW\\&   ( ;  9  = 7   7   - UVW)`$        Rr