o Ьf4#у@s<ddlmZddlmZddlmZddlZddlm Z m Z mZddlm Z mZmZmZmZmZer8ddlmZdFddДZdGddДZdHddДZdIddДZdJddДZdKddДZdLd#d$ДZdMd'd(ДZdNd,d-ДZdNd.d/ДZdMd0d1ДZ dOd2d3ДZ!dPd5d6ДZ"dQd7d8ДZ#dRd9d:ДZ$dKd;d<ДZ%dSd>d?ДZ&dTd@dAДZ'dUdBdCДZ(dVdDdEДZ)dS)Wщ)┌annotations)┌chain)┌ TYPE_CHECKINGN)┌check_code_array┌check_offset_array┌check_point_array)┌ CLOSEPOLY┌LINETO┌MOVETO┌ code_dtype┌offset_dtype┌point_dtype┌offsets·cpy.OffsetArray┌return· cpy.CodeArraycCsHt|Г|d}tj|ttdН}t||ddЕ<t||ddЕd<|S)zSDetermine codes from offsets, assuming they all correspond to closed polygons. щ йZdtypeNщ)r┌np┌fullr rr r)r┌n┌codesйr·/lib/python3.10/site-packages/contourpy/array.py┌codes_from_offsetssr┌points·cpy.PointArraycCsvt|Гt|Гtjt|ГttdН}t||ddЕ<|ddЕd}tj||ddЕ||kddН}t |||<|S)zЦDetermine 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. rNrrйZaxis) rrrr┌lenr rr ┌allr)rrrZend_offsets┌closedrrr┌codes_from_offsets_and_pointss"r"cCsJt|Гt|Г}tj|ttdН}t|d<tа|d|dkбr#t|d<|S)zЕDetermine 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 rr r r)rrrrrr┌codes_from_points/sr#┌ list_of_codes·list[cpy.CodeArray]cCє|stdГВtj|tdНS)zAConcatenate a list of codes arrays into a single code array. z!Empty list passed to concat_codesr)┌ ValueErrorr┌concatenaterйr$rrr┌concat_codes=єr*┌list_of_codes_or_none·list[cpy.CodeArray | None]·cpy.CodeArray | NonecCєddД|DГ}|r t|ГSdS)zQConcatenate a list of codes arrays or None into a single code array or None. cSєg|]}|dur|СqSйNrй┌.0rrrr┌ Iєz(concat_codes_or_none..N)r*)r,r$rrr┌concat_codes_or_noneFsr6┌list_of_offsets·list[cpy.OffsetArray]csbИstdГВtИГ}tjddДИDГtdНЙtjИdgЗЗfddДt|dГDГвRtdН}|S) zEConcatenate a list of offsets arrays into a single offset array. z#Empty list passed to concat_offsetscSsg|]}|dСqS)rrйr3rrrrr4Wєz"concat_offsets..rrc3s,Б|]}И|dddЕИ|VqdS)rNr)r3┌iйZ cumulativer7rr┌ YsА*z!concat_offsets..r)r'rr┌cumsumrr(┌range)r7r┌retrr<r┌concat_offsetsPs&■rA┌list_of_offsets_or_none·list[cpy.OffsetArray | None]·cpy.OffsetArray | NonecCr/)zUConcatenate a list of offsets arrays or None into a single offset array or None. cSr0r1rr9rrrr4dr5z*concat_offsets_or_none..N)rA)rBr7rrr┌concat_offsets_or_none_єrE┌list_of_points·list[cpy.PointArray]cCr&)zBConcatenate a list of point arrays into a single point array. z"Empty list passed to concat_pointsr)r'rr(r йrGrrr┌ concat_pointskr+rJ┌list_of_points_or_none·list[cpy.PointArray | None]·cpy.PointArray | NonecCr/)zRConcatenate a list of point arrays or None into a single point array or None. cSr0r1rйr3rrrrr4yr5z)concat_points_or_none..N)rJйrKrGrrr┌concat_points_or_nonetrFrPcCr/)zvConcatenate a list of points or None into a single point array or None, with NaNs used to separate each line. cSr0r1rrNrrrr4Жr5z2concat_points_or_none_with_nan..N)┌concat_points_with_nanrOrrr┌concat_points_or_none_with_nanАsrRcsf|stdГВt|Гdkr|dStjdtjtdНЙ|dgttЗfddД|ddЕDГОГв}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)rщrc3sБ|]}И|fVqdSr1r)r3┌xй┌ nan_spacerrrr=ШsАz)concat_points_with_nan..N) r'rrr┌nanr ┌listrrJrIrrUrrQНs rQcCsXt|Гt|Гt|Гdkr|StjtjtjgtdН}tj||ddЕаtj б|ddНS)zNInsert NaNs into a point array at locations specified by an offset array. rSrrrrr) rrrr┌arrayrWr ┌insert┌astypeZint64)rrrVrrr┌insert_nan_at_offsetsЬs"r\rcCs,t|Гtаtа|tkбdt|ГбаtбS)zBDetermine offsets from codes using locations of MOVETO codes. r)rr┌append┌nonzeror rr[r)rrrr┌offsets_from_codesкs$r_cCє*|stdГВtjdgddД|DГtdНS)z4Determine offsets from lengths of point arrays. z)Empty list passed to offsets_from_lengthsrcSsg|]}t|ГСqSrйr)r3┌linerrrr4╕r:z(offsets_from_lengths..rйr'rr>rrIrrr┌offsets_from_lengths▓єrdcCr`)zHDetermine outer offsets from codes using locations of MOVETO codes. z5Empty list passed to outer_offsets_from_list_of_codesrcSsg|] }tа|tkбСqSr)rZ count_nonzeror r2rrrr4┴sz4outer_offsets_from_list_of_codes..rrcr)rrr┌ outer_offsets_from_list_of_codes╗s rfcCr`)z4Determine outer offsets from a list of offsets. z7Empty list passed to outer_offsets_from_list_of_offsetsrcSsg|]}t|ГdСqSйrrar9rrrr4╦r5z6outer_offsets_from_list_of_offsets..rrc)r7rrr┌"outer_offsets_from_list_of_offsets┼rerh·&tuple[cpy.PointArray, cpy.OffsetArray]cCsоt|Гtаtа|ddЕdfббd}t|Гdkr'|tjdt|ГgtdНfStj||ddН}|tаt|Гб8}tj t|ГdtdН}d|d<||ddЕ<t|Г|d<||fS)z^Remove NaN from a points array, also return the offsets corresponding to the NaN removed. NrrrrSrr) rrr^┌isnanrrYr┌deleteZarange┌empty)r┌nan_offsetsrrrr┌ remove_nan╬s rncCє6t|Гt|Гt|Гdkrtа||ddЕбS|gS)z]Split a code array at locations specified by an offset array into a list of code arrays. rSrr)rrrr┌split)rrrrr┌split_codes_by_offsetsрs rqcCro)z_Split a point array at locations specified by an offset array into a list of point arrays. rSrr)rrrrrp)rrrrr┌split_points_by_offsetsьs rrcsztИГtаtаИddЕdfббd}t|ГdkrИgStаdg|tИГgfб}ЗfddДt|ddЕ|ddЕГDГS)z>Split a points array at NaNs into a list of point arrays. Nrrcs g|]\}}И|d|ЕСqSrgr)r3┌s┌eйrrrr4s z'split_points_at_nan..r)rrr^rjrr(┌zip)rrmrrur┌split_points_at_nan√s (rw)rrrr)rrrrrr)rrrr)r$r%rr)r,r-rr.)r7r8rr)rBrCrrD)rGrHrr)rKrLrrM)rrrrrr)rrrr)rGrHrr)r$r%rr)rrrri)rrrrrr%)rrrrrrH)rrrrH)*Z __future__r┌ itertoolsr┌typingrZnumpyrZcontourpy.typecheckrrrZcontourpy.typesrr r rrr Zcontourpy._contourpyZ _contourpyZcpyrr"r#r*r6rArErJrPrRrQr\r_rdrfrhrnrqrrrwrrrr┌s8