o ff[@sddlmZddlmZmZddlZddlmZm Z ddl m Z ddl mZmZddlmZmZddlmZmZer@ddlmZd:ddZd;ddZdddZd?ddZd@d!d"ZdAd(d)ZdBd+d,Z dCd.d/Z!dDd1d2Z"dEd4d5Z#dFd8d9Z$dS)G) annotations) TYPE_CHECKINGcastN)FillTypeLineType) as_fill_type as_line_type) check_filled check_lines)MOVETO offset_dtypefilledcpy.FillReturn_OuterCode fill_type_torreturncpy.FillReturncCsB|tjkr|S|tjkr|ddd|dDfSt|ddkr0t|d}t|d}nd}d}|tjkr?|g|gfS|tjkrU|g|durNdgfSt |gfS|tj krq|dur`dnt |d}|g|g|gf}|S|tj kr|durdgdgdgf}|St |}t |d}|g|g|gf}|Std|)NrcSg|]}t|qSarroffsets_from_codes.0codesrr1lib/python3.10/site-packages/contourpy/convert.py z2_convert_filled_from_OuterCode..Invalid FillType )r OuterCode OuterOffsetlenr concat_points concat_codesChunkCombinedCodeChunkCombinedOffsetrChunkCombinedCodeOffsetoffsets_from_lengthsChunkCombinedOffsetOffsetZ outer_offsets_from_list_of_codes ValueError)r rpointsr outer_offsetsret1ret2offsetsrrr_convert_filled_from_OuterCodes4     "   r/cpy.FillReturn_OuterOffsetc CsV|tjkrdd|dD}|d|fS|tjkr|St|ddkr2t|d}t|d}nd}d}|tjkrL|g|durEdgfSt|gfS|tj krW|g|gfS|tj kr|durjdgdgdgf}|St|}t |d}|g|g|gf}|S|tj kr|durdgdgdgf}|St |d}|g|g|gf}|Std|)NcSrr)rcodes_from_offsets)rr.rrrr:rz4_convert_filled_from_OuterOffset..rrr)rrr r!rr"Zconcat_offsetsr$r1r%r&r'r(Z"outer_offsets_from_list_of_offsetsr)) r rseparate_codesr*r.r,rr+r-rrr _convert_filled_from_OuterOffset5s<    "     r3 cpy.FillReturn_ChunkCombinedCodecCsN|tjkr|S|tjkrdd|dD}|d|fStdtjd|d)NcS"g|] }|dur dnt|qSNrrrrrrd"z:_convert_filled_from_ChunkCombinedCode..rrConversion from  to  not supported)rr$r%r))r rrrrr&_convert_filled_from_ChunkCombinedCode]s   r;"cpy.FillReturn_ChunkCombinedOffsetcCs|tjkr1g}t|D]\}}|dur|dq tr!|dus!J|t||q |d|fS|tjkr8|Stdtjd|d)Nrr8r9r:) rr$zipappendrrcodes_from_offsets_and_pointsr%r))r r chunk_codesr*r.rrr(_convert_filled_from_ChunkCombinedOffsetks     rA&cpy.FillReturn_ChunkCombinedCodeOffsetcs|tjkr9g}g}t|D]'\}}}|dur4tr$|dusJ|dus$J|t||7}|t||7}q ||fS|tjkryg}g}t|D].\}}}|durttr]|dusWJ|dus]J|t||7}t||}|dd|D7}qF||fS|tjkr|d|df}|S|tj krdd|dD} |d| f} | S|tj kr|S|tj krg} g} t|ddD]8\}}|dur| d| dqtr|dusJt |tjfdd|Dtd}| | |q|d| | f} | Std|) NcSrrrrrrrrrz@_convert_filled_from_ChunkCombinedCodeOffset..rrcSr5r6rrrrrrcs"g|] }t|kddqS)r)npnonzero)rZoor.rrrr7)Zdtyper)rrr=rrsplit_points_by_offsetssplit_codes_by_offsetsr r$r%r&r(r>rrDarrayr r))r rseparate_pointsr2r*rr+separate_offsetsr,Z all_offsetsr- chunk_offsetschunk_outer_offsetsret3rrFr,_convert_filled_from_ChunkCombinedCodeOffsetsn                   rO(cpy.FillReturn_ChunkCombinedOffsetOffsetc sx|tjkrCg}g}t|D]1\}}|dur>tr$dusJ|dus$Jt|}|}|t||7}|t||7}q ||fS|tjkrg}g}t|D]C\}}|durtrgdusaJ|dusgJt |dkr|fddt|dd|ddD7}n| |t||7}qP||fS|tj krg}t|D]&\}}|dur| dqtrdusJ|dusJ| t|q|d|f} | S|tj kr|d|dfS|tj kr-g}g} t|D]6\}}|dur| d| dqtrdus J|dusJ| t|| |q|d|| f} | S|tjkr5|Std|)Ncs(g|]\}}||d|qS)rr)rserFrrrs(zB_convert_filled_from_ChunkCombinedOffsetOffset..rrr)rrr=rrr?rGrHr r!r>r$r%r&r(r)) r rrJr2r*r+rrKr@r,rMr-rrFr._convert_filled_from_ChunkCombinedOffsetOffsetsx                     rUfill_type_fromFillType | strcCst|}t|}t|||tjkrtrttj|}t||S|tj kr1tr,ttj |}t ||S|tj krCtr>ttj |}t||S|tjkrUtrPttj|}t||S|tjkrgtrbttj|}t||S|tjkrytrtttj|}t||Std|)aReturn the specified filled contours converted to a different :class:`~contourpy.FillType`. Args: filled (sequence of arrays): Filled contour polygons to convert. fill_type_from (FillType or str): :class:`~contourpy.FillType` to convert from as enum or string equivalent. fill_type_to (FillType or str): :class:`~contourpy.FillType` to convert to as enum or string equivalent. Return: Converted filled contour polygons. When converting non-chunked fill types (``FillType.OuterCode`` or ``FillType.OuterOffset``) to chunked ones, all polygons are placed in the first chunk. When converting in the other direction, all chunk information is discarded. Converting a fill type that is not aware of the relationship between outer boundaries and contained holes (``FillType.ChunkCombinedCode`` or) ``FillType.ChunkCombinedOffset``) to one that is will raise a ``ValueError``. .. versionadded:: 1.2.0 r)rr rrrrcpyZFillReturn_OuterCoder/r ZFillReturn_OuterOffsetr3r$ZFillReturn_ChunkCombinedCoder;r%ZFillReturn_ChunkCombinedOffsetrAr&Z"FillReturn_ChunkCombinedCodeOffsetrOr(Z$FillReturn_ChunkCombinedOffsetOffsetrUr))r rVrrrrconvert_filleds8                   rYlinescpy.LineReturn_Separate line_type_torcpy.LineReturnc Cs|tjkr|S|tjkrdd|D}||fS|tjkr>|s&dgdgf}|St|}t|}t||}|g|gf}|S|tjkr[|sMdgdgf}|St|gt|gf}|S|tj krq|shdgf}|St |gf}|St d|)NcSrrrZcodes_from_pointsrlinerrrr@rz0_convert_lines_from_Separate..Invalid LineType ) rSeparate SeparateCoder$rr"r'r?r%ChunkCombinedNanconcat_points_with_nanr)) rZr\r2r,r*r.rr-rNrrr_convert_lines_from_Separate9s6          rfcpy.LineReturn_SeparateCodecCs|tjkr |dS|tjkr|S|tjkr3|ds!dgdgf}|St|dgt|dgf}|S|tjkrV|dsDdgdgf}|St|dgt|dgf}|S|tj krp|dsedgf}|St |dgf}|St d|)Nrrra) rrbrcr$rr"r#r%r'rdrer))rZr\r,r-rNrrr _convert_lines_from_SeparateCode[s.         rh cpy.LineReturn_ChunkCombinedCodec CsN|tjtjfvrRg}t|D]1\}}|dur?tr|dusJt|tkd}t|dkr:|t ||dd7}q| |q|tjkrG|Sdd|D}||fS|tj krY|S|tj krmdd|dD}|d|fS|tj krg}t|D]$\}}|dur| dqxtr|dusJt|} | t|| qx|fStd|)NrrcSrrr^r_rrrrrz9_convert_lines_from_ChunkCombinedCode..cSr5r6rrrrrrrCra)rrbrcr=rrDrEr r!splitr>r$r%rdrrinsert_nan_at_offsetsr)) rZr\separate_linesr*rZsplit_atr2rL points_nanr.rrr%_convert_lines_from_ChunkCombinedCodezsB           rn"cpy.LineReturn_ChunkCombinedOffsetcCsL|tjtjfvr9g}t|D]\}}|dur&tr|dusJ|t||7}q|tjkr.|Sdd|D}||fS|tjkrjg}t|D]\}}|durR|dqDtrZ|dusZJ|t ||qD|d|fS|tj krq|S|tj krg}t|D]\}}|dur|dq|tr|dusJ|t ||q||fSt d|)NcSrrr^r_rrrrrz;_convert_lines_from_ChunkCombinedOffset..rra)rrbrcr=rrrGr$r>r?r%rdrkr))rZr\rlr*r.r2r@rmrrr'_convert_lines_from_ChunkCombinedOffsetsB          rpcpy.LineReturn_ChunkCombinedNanc CsP|tjtjfvr.g}|dD] }|dur|t|7}q|tjkr#|Sdd|D}||fS|tjkrfg}g}|dD]&}|durL|d|dq;t|\}}|||t||q;||fS|tj krg}g}|dD]"}|dur|d|dqst|\}}||||qs||fS|tj kr|St d|)NrcSrrr^)rr*rrrrrz8_convert_lines_from_ChunkCombinedNan..ra) rrbrcrZsplit_points_at_nanr$r>Z remove_nanr?r%rdr)) rZr\rlr*r2Z chunk_pointsr@r.rLrrr$_convert_lines_from_ChunkCombinedNansF              rrline_type_fromLineType | strcCst|}t|}t|||tjkrtrttj|}t||S|tj kr1tr,ttj |}t ||S|tj krCtr>ttj |}t||S|tjkrUtrPttj|}t||S|tjkrgtrbttj|}t||Std|)aGReturn the specified contour lines converted to a different :class:`~contourpy.LineType`. Args: lines (sequence of arrays): Contour lines to convert. line_type_from (LineType or str): :class:`~contourpy.LineType` to convert from as enum or string equivalent. line_type_to (LineType or str): :class:`~contourpy.LineType` to convert to as enum or string equivalent. Return: Converted contour lines. When converting non-chunked line types (``LineType.Separate`` or ``LineType.SeparateCode``) to chunked ones (``LineType.ChunkCombinedCode``, ``LineType.ChunkCombinedOffset`` or ``LineType.ChunkCombinedNan``), all lines are placed in the first chunk. When converting in the other direction, all chunk information is discarded. .. versionadded:: 1.2.0 ra)rr rrbrrrXZLineReturn_SeparaterfrcZLineReturn_SeparateCoderhr$ZLineReturn_ChunkCombinedCodernr%ZLineReturn_ChunkCombinedOffsetrprdZLineReturn_ChunkCombinedNanrrr))rZrsr\rrr convert_liness0                ru)r rrrrr)r r0rrrr)r r4rrrr)r r<rrrr)r rBrrrr)r rPrrrr)r rrVrWrrWrr)rZr[r\rrr])rZrgr\rrr])rZrir\rrr])rZror\rrr])rZrqr\rrr])rZr]rsrtr\rtrr])%Z __future__rtypingrrZnumpyrDZcontourpy._contourpyrrZcontourpy.arrayrIrZcontourpy.enum_utilrrZcontourpy.typecheckr r Zcontourpy.typesr r Z _contourpyrXr/r3r;rArOrUrYrfrhrnrprrrurrrrs.     $ (   < C : "  ) *,