a 5gh#@sddlmZddlmZddlmZddlZddlm Z m Z m Z ddl m Z mZmZmZmZmZerpddlmZddd d d Zdd dd ddZd ddddZdddddZdddddZdddddZd d!d"d#d$Zd%d d&d'd(Zd)d*d+d,d-Zd)d*d+d.d/Zd%d d&d0d1Z d dd d2d3d4Z!ddd5d6d7Z"d%dd&d8d9Z#dddd:d;Z$dddddd?d@Z&ddddAdBdCZ'd dd%d2dDdEZ(d d%ddFdGZ)dS)H) annotations)chain) TYPE_CHECKINGN)check_code_arraycheck_offset_arraycheck_point_array) CLOSEPOLYLINETOMOVETO code_dtype offset_dtype point_dtypezcpy.OffsetArrayz cpy.CodeArray)offsetsreturncCsHt||d}tj|ttd}t||dd<t||ddd<|S)zSDetermine codes from offsets, assuming they all correspond to closed polygons. ZdtypeN)rnpfullr r r r)rncodesr_/mounts/lovelace/software/anaconda3/envs/metaDMG/lib/python3.9/site-packages/contourpy/array.pycodes_from_offsetss rzcpy.PointArray)rpointsrcCsvt|t|tjt|ttd}t||dd<|ddd}tj||dd||kdd}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. rNrrZaxis) rrrrlenr r r allr)rrrZ end_offsetsclosedrrrcodes_from_offsets_and_pointss" r)rrcCsJt|t|}tj|ttd}t|d<t|d|dkrFt|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 rr)rrrrrrcodes_from_points/sr zlist[cpy.CodeArray]) list_of_codesrcCs|s tdtj|tdS)zAConcatenate a list of codes arrays into a single code array. z!Empty list passed to concat_codesr) ValueErrorr concatenater r!rrr concat_codes=sr%zlist[cpy.CodeArray | None]zcpy.CodeArray | None)list_of_codes_or_nonercCs"dd|D}|rt|SdSdS)zQConcatenate a list of codes arrays or None into a single code array or None. cSsg|]}|dur|qSNr.0rrrr Iz(concat_codes_or_none..N)r%)r&r!rrrconcat_codes_or_noneFsr,zlist[cpy.OffsetArray])list_of_offsetsrcsbs tdt}tjddDtdtjdgfddt|dDRtd}|S) zEConcatenate a list of offsets arrays into a single offset array. z#Empty list passed to concat_offsetscSsg|] }|dqS)rrr)rrrrr*Wr+z"concat_offsets..rrc3s*|]"}|ddd|VqdS)rNr)r)iZ cumulativer-rr Yr+z!concat_offsets..r)r"rrcumsumr r#range)r-rretrr0rconcat_offsetsPs&r5zlist[cpy.OffsetArray | None]zcpy.OffsetArray | None)list_of_offsets_or_nonercCs"dd|D}|rt|SdSdS)zUConcatenate a list of offsets arrays or None into a single offset array or None. cSsg|]}|dur|qSr'rr.rrrr*dr+z*concat_offsets_or_none..N)r5)r6r-rrrconcat_offsets_or_none_sr7zlist[cpy.PointArray])list_of_pointsrcCs|s tdtj|tdS)zBConcatenate a list of point arrays into a single point array. z"Empty list passed to concat_pointsr)r"rr#r r8rrr concat_pointsksr:zlist[cpy.PointArray | None]zcpy.PointArray | None)list_of_points_or_nonercCs"dd|D}|rt|SdSdS)zRConcatenate a list of point arrays or None into a single point array or None. cSsg|]}|dur|qSr'rr)rrrrr*yr+z)concat_points_or_none..N)r:r;r8rrrconcat_points_or_nonetsr>cCs"dd|D}|rt|SdSdS)zvConcatenate a list of points or None into a single point array or None, with NaNs used to separate each line. cSsg|]}|dur|qSr'rr<rrrr*r+z2concat_points_or_none_with_nan..N)concat_points_with_nanr=rrrconcat_points_or_none_with_nansr@csj|s tdt|dkr |dStjdtjtd|dgttfdd|ddD}t|SdS) 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)rrc3s|]}|fVqdSr'r)r)x nan_spacerrrr1r+z)concat_points_with_nan..N) r"rrrnanr listrr:r9rrCrr?s  r?)rrrcCs\t|t|t|dkr |Stjtjtjgtd}tj||ddtj |ddSdS)zNInsert NaNs into a point array at locations specified by an offset array. rArrrrrN) rrrrarrayrEr insertastypeZint64)rrrDrrrinsert_nan_at_offsetss  rJ)rrcCs,t|tt|tkdt|tS)zBDetermine offsets from codes using locations of MOVETO codes. r)rrappendnonzeror rrIr )rrrroffsets_from_codessrMcCs*|s tdtjdgdd|DtdS)z4Determine offsets from lengths of point arrays. z)Empty list passed to offsets_from_lengthsrcSsg|] }t|qSrr)r)linerrrr*r+z(offsets_from_lengths..rr"rr2r r9rrroffsets_from_lengthssrQcCs*|s tdtjdgdd|DtdS)zHDetermine outer offsets from codes using locations of MOVETO codes. z5Empty list passed to outer_offsets_from_list_of_codesrcSsg|]}t|tkqSr)rZ count_nonzeror r(rrrr*r+z4outer_offsets_from_list_of_codes..rrPr$rrr outer_offsets_from_list_of_codess rRcCs*|s tdtjdgdd|DtdS)z4Determine outer offsets from a list of offsets. z7Empty list passed to outer_offsets_from_list_of_offsetsrcSsg|]}t|dqSrrNr.rrrr*r+z6outer_offsets_from_list_of_offsets..rrP)r-rrr"outer_offsets_from_list_of_offsetssrTz&tuple[cpy.PointArray, cpy.OffsetArray]cCst|tt|dddfd}t|dkrN|tjdt|gtdfStj||dd}|tt|8}tj t|dtd}d|d<||dd<t||d<||fSdS)z^Remove NaN from a points array, also return the offsets corresponding to the NaN removed. NrrrrArr) rrrLisnanrrGr deleteZarangeempty)r nan_offsetsrrrr remove_nans    rY)rrrcCs:t|t|t|dkr0t||ddS|gSdS)z]Split a code array at locations specified by an offset array into a list of code arrays. rArrN)rrrrsplit)rrrrrsplit_codes_by_offsetss  r[cCs:t|t|t|dkr0t||ddS|gSdS)z_Split a point array at locations specified by an offset array into a list of point arrays. rArrN)rrrrrZ)rrrrrsplit_points_by_offsetss  r\cs~tttdddfd}t|dkr:gStdg|tgf}fddt|dd|ddDSdS)z>Split a points array at NaNs into a list of point arrays. Nrrcs g|]\}}|d|qSrSr)r)serrrr*r+z'split_points_at_nan..r)rrrLrUrr#zip)rrXrr_rsplit_points_at_nans   ra)* __future__r itertoolsrtypingrnumpyrZcontourpy.typecheckrrrZcontourpy.typesrr r r r r Zcontourpy._contourpyZ _contourpyZcpyrrr r%r,r5r7r:r>r@r?rJrMrQrRrTrYr[r\rarrrrs6