a
    5ghL                     @  s   d dl mZ d dlmZmZ d dlmZmZ d dlm	Z	m
Z
mZmZ d dlmZmZ d dlmZmZ ertd dlmZ dd	dd
ddZddddddZdd	ddddZddddddZdS )    )annotations)TYPE_CHECKINGcast)FillTypeLineType)concat_codes_or_noneconcat_offsets_or_noneconcat_points_or_noneconcat_points_or_none_with_nan)as_fill_typeas_line_type)check_filledcheck_linesNzcpy.FillReturnzFillType | str)filled	fill_typereturnc                 C  s  t |}|tjtjfv r| S t| | t| d dk r:| S trJttj	| } t
| d }|tjkrtrpttj| } |du rdgdgf}n|gt| d gf}|S |tjkrtrttj| } |du rdgdgf}n|gt| d gf}|S |tjkrHtrttj| } |du r dgdgdgf}n$t| d }|gt| d g|gf}|S |tjkrtrfttj| } |du rdgdgdgf}n$t| d }|gt| d g|gf}|S td| dS )a  Return the specified filled contours with chunked data moved into the first chunk.

    Filled contours that are not chunked (``FillType.OuterCode`` and ``FillType.OuterOffset``) and
    those that are but only contain a single chunk are returned unmodified. Individual polygons are
    unchanged, they are not geometrically combined.

    Args:
        filled (sequence of arrays): Filled contour data, such as returned by
            :meth:`.ContourGenerator.filled`.
        fill_type (FillType or str): Type of :meth:`~.ContourGenerator.filled` as enum or string
            equivalent.

    Return:
        Filled contours in a single chunk.

    .. versionadded:: 1.2.0
    r      N   zInvalid FillType )r   r   	OuterCodeOuterOffsetr   lenr   r   cpyZFillReturn_Chunkr	   ChunkCombinedCodeZFillReturn_ChunkCombinedCoder   ChunkCombinedOffsetZFillReturn_ChunkCombinedOffsetr   ZChunkCombinedCodeOffsetZ"FillReturn_ChunkCombinedCodeOffsetZChunkCombinedOffsetOffsetZ$FillReturn_ChunkCombinedOffsetOffset
ValueError)r   r   pointsret1ret2ret3Zouter_offsetsZret4 r   a/mounts/lovelace/software/anaconda3/envs/metaDMG/lib/python3.9/site-packages/contourpy/dechunk.pydechunk_filled   sP    




r!   zcpy.LineReturnzLineType | str)lines	line_typer   c                 C  s:  t |}|tjtjfv r| S t| | t| d dk r:| S trJttj	| } |tj
krtrdttj| } t| d }|du rdgdgf}n|gt| d gf}|S |tjkrtrttj| } t| d }|du rdgdgf}n|gt| d gf}|S |tjkr(trttj| } t| d }|gf}|S td| dS )a  Return the specified contour lines with chunked data moved into the first chunk.

    Contour lines that are not chunked (``LineType.Separate`` and ``LineType.SeparateCode``) and
    those that are but only contain a single chunk are returned unmodified. Individual lines are
    unchanged, they are not geometrically combined.

    Args:
        lines (sequence of arrays): Contour line data, such as returned by
            :meth:`.ContourGenerator.lines`.
        line_type (LineType or str): Type of :meth:`~.ContourGenerator.lines` as enum or string
            equivalent.

    Return:
        Contour lines in a single chunk.

    .. versionadded:: 1.2.0
    r   r   Nr   zInvalid LineType )r   r   SeparateSeparateCoder   r   r   r   r   ZLineReturn_Chunkr   ZLineReturn_ChunkCombinedCoder	   r   r   ZLineReturn_ChunkCombinedOffsetr   ZChunkCombinedNanZLineReturn_ChunkCombinedNanr
   r   )r"   r#   r   r   r   r   r   r   r    dechunk_linesZ   s>    


r&   zlist[cpy.FillReturn])multi_filledr   r   c                   s.   t    tjtjfv r| S  fdd| D S )a  Return multiple sets of filled contours with chunked data moved into the first chunks.

    Filled contours that are not chunked (``FillType.OuterCode`` and ``FillType.OuterOffset``) and
    those that are but only contain a single chunk are returned unmodified. Individual polygons are
    unchanged, they are not geometrically combined.

    Args:
        multi_filled (nested sequence of arrays): Filled contour data, such as returned by
            :meth:`.ContourGenerator.multi_filled`.
        fill_type (FillType or str): Type of :meth:`~.ContourGenerator.filled` as enum or string
            equivalent.

    Return:
        Multiple sets of filled contours in a single chunk.

    .. versionadded:: 1.3.0
    c                   s   g | ]}t | qS r   )r!   ).0r   r   r   r    
<listcomp>       z(dechunk_multi_filled.<locals>.<listcomp>)r   r   r   r   )r'   r   r   r)   r    dechunk_multi_filled   s    r,   zlist[cpy.LineReturn])multi_linesr#   r   c                   s.   t    tjtjfv r| S  fdd| D S )a  Return multiple sets of contour lines with all chunked data moved into the first chunks.

    Contour lines that are not chunked (``LineType.Separate`` and ``LineType.SeparateCode``) and
    those that are but only contain a single chunk are returned unmodified. Individual lines are
    unchanged, they are not geometrically combined.

    Args:
        multi_lines (nested sequence of arrays): Contour line data, such as returned by
            :meth:`.ContourGenerator.multi_lines`.
        line_type (LineType or str): Type of :meth:`~.ContourGenerator.lines` as enum or string
            equivalent.

    Return:
        Multiple sets of contour lines in a single chunk.

    .. versionadded:: 1.3.0
    c                   s   g | ]}t | qS r   )r&   )r(   r"   r#   r   r    r*      r+   z'dechunk_multi_lines.<locals>.<listcomp>)r   r   r$   r%   )r-   r#   r   r.   r    dechunk_multi_lines   s    r/   )
__future__r   typingr   r   Zcontourpy._contourpyr   r   Zcontourpy.arrayr   r   r	   r
   Zcontourpy.enum_utilr   r   Zcontourpy.typecheckr   r   Z
_contourpyr   r!   r&   r,   r/   r   r   r   r    <module>   s   G<