o h#@s@ddlmZddlmZmZddlmZddlZddl m Z m Z m Z ddl mZmZmZmZmZmZer:ddlmZdFd d ZdGddZdHddZdIddZdJddZdKddZdLd#d$ZdMd'd(ZdNd,d-ZdNd.d/Z dMd0d1Z!dOd2d3Z"dPd5d6Z#dQd7d8Z$dRd9d:Z%dKd;d<Z&dSd>d?Z'dTd@dAZ(dUdBdCZ)dVdDdEZ*dS)W) annotations)chainpairwise) TYPE_CHECKINGN)check_code_arraycheck_offset_arraycheck_point_array) CLOSEPOLYLINETOMOVETO code_dtype offset_dtype point_dtypeoffsetscpy.OffsetArrayreturn cpy.CodeArraycCsHt||d}tj|ttd}t||dd<t||ddd<|S)zSDetermine codes from offsets, assuming they all correspond to closed polygons. dtypeN)rnpfullr r r r )rncodesrc/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/contourpy/array.pycodes_from_offsetss rpointscpy.PointArraycCsvt|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. rNrraxis) rrrrlenr r r allr )rrr end_offsetsclosedrrrcodes_from_offsets_and_pointss" r&cCsJt|t|}tj|ttd}t|d<t|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 )rrrrrrcodes_from_points/sr' list_of_codeslist[cpy.CodeArray]cC|stdtj|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=r.list_of_codes_or_nonelist[cpy.CodeArray | None]cpy.CodeArray | NonecCdd|D}|r t|SdS)zQConcatenate a list of codes arrays or None into a single code array or None. cSg|]}|dur|qSNr.0rrrr Iz(concat_codes_or_none..N)r.)r0r(rrrconcat_codes_or_noneFsr:list_of_offsetslist[cpy.OffsetArray]csbstdt}tjddDtdtjdgfddt|dDRtd}|S) zEConcatenate a list of offsets arrays into a single offset array. z#Empty list passed to concat_offsetscSsg|]}|dqS)rrr7rrrrr8Wz"concat_offsets..rrc3s,|]}|ddd|VqdS)rNr)r7i cumulativer;rr Ys*z!concat_offsets..r)r+r"rcumsumr r,range)r;rretrr@rconcat_offsetsPs&rFlist_of_offsets_or_nonelist[cpy.OffsetArray | None]cpy.OffsetArray | NonecCr3)zUConcatenate a list of offsets arrays or None into a single offset array or None. cSr4r5rr=rrrr8dr9z*concat_offsets_or_none..N)rF)rGr;rrrconcat_offsets_or_none_rJlist_of_pointslist[cpy.PointArray]cCr*)zBConcatenate a list of point arrays into a single point array. z"Empty list passed to concat_pointsr)r+rr,rrLrrr concat_pointskr/rOlist_of_points_or_nonelist[cpy.PointArray | None]cpy.PointArray | NonecCr3)zRConcatenate a list of point arrays or None into a single point array or None. cSr4r5rr7rrrrr8yr9z)concat_points_or_none..N)rOrPrLrrrconcat_points_or_nonetrKrUcCr3)zvConcatenate a list of points or None into a single point array or None, with NaNs used to separate each line. cSr4r5rrSrrrr8r9z2concat_points_or_none_with_nan..N)concat_points_with_nanrTrrrconcat_points_or_none_with_nansrWcsf|stdt|dkr|dStjdtjtd|dgttfdd|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)rrc3s|]}|fVqdSr5r)r7x nan_spacerrrrBsz)concat_points_with_nan..N) r+r"rrnanrlistrrOrNrrZrrVs  rVcCsXt|t|t|dkr|Stjtjtjgtd}tj||ddtj |ddS)zNInsert NaNs into a point array at locations specified by an offset array. rXrrrrr ) rrr"rarrayr\rinsertastypeint64)rrr[rrrinsert_nan_at_offsetss  "rbrcCs,t|tt|tkdt|tS)zBDetermine offsets from codes using locations of MOVETO codes. r)rrappendnonzeror r"r`r )rrrroffsets_from_codess$recC*|stdtjdgdd|DtdS)z4Determine offsets from lengths of point arrays. z)Empty list passed to offsets_from_lengthsrcSsg|]}t|qSrr")r7linerrrr8r>z(offsets_from_lengths..rr+rrCr rNrrroffsets_from_lengthsrjcCrf)zHDetermine outer offsets from codes using locations of MOVETO codes. z5Empty list passed to outer_offsets_from_list_of_codesrcSsg|] }t|tkqSr)r count_nonzeror r6rrrr8sz4outer_offsets_from_list_of_codes..rrir-rrr outer_offsets_from_list_of_codess rmcCrf)z4Determine outer offsets from a list of offsets. z7Empty list passed to outer_offsets_from_list_of_offsetsrcSsg|]}t|dqSrrgr=rrrr8r9z6outer_offsets_from_list_of_offsets..rri)r;rrr"outer_offsets_from_list_of_offsetsrkro&tuple[cpy.PointArray, cpy.OffsetArray]cCst|tt|dddfd}t|dkr'|tjdt|gtdfStj||dd}|tt|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. Nrrr rXrr) rrrdisnanr"r^r deletearangeempty)r nan_offsetsrrrr remove_nans    rvcC6t|t|t|dkrt||ddS|gS)z]Split a code array at locations specified by an offset array into a list of code arrays. rXrr)rrr"rsplit)rrrrrsplit_codes_by_offsetss  rycCrw)z_Split a point array at locations specified by an offset array into a list of point arrays. rXrr)rrr"rrx)rrrrrsplit_points_by_offsetss  rzcshtttdddfd}t|dkrgStdg|tgf}fddt|DS)z>Split a points array at NaNs into a list of point arrays. Nrrcs g|] \}}|d|qSrnr)r7serrrr8s z'split_points_at_nan..)rrrdrqr"r,r)rrurr}rsplit_points_at_nans   r~)rrrr)rrrrrr)rrrr)r(r)rr)r0r1rr2)r;r<rr)rGrHrrI)rLrMrr)rPrQrrR)rrrrrr)rrrr)rLrMrr)r(r)rr)rrrrp)rrrrrr))rrrrrrM)rrrrM)+ __future__r itertoolsrrtypingrnumpyrcontourpy.typecheckrrrcontourpy.typesr r r r r rcontourpy._contourpy _contourpycpyrr&r'r.r:rFrJrOrUrWrVrbrerjrmrorvryrzr~rrrrs8