| Server IP : 93.86.61.54 / Your IP : 216.73.216.156 Web Server : Apache/2.4.62 (Ubuntu) System : Linux rasin.ddns.net 6.8.0-124-generic #124~22.04.1-Ubuntu SMP PREEMPT_DYNAMIC Tue May 26 21:05:19 UTC x86_64 User : www-data ( 33) PHP Version : 8.4.22 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /usr/lib/python3/dist-packages/mpl_toolkits/mplot3d/__pycache__/ |
Upload File : |
o
�:�a&{ � @ s� d Z ddlZddlZddlmZmZmZmZ m
Z
m Z m
Z ddlmZmZmZmZ ddlmZ ddlmZ ddlmZ d d
� Zd d � Zd
d� Z G dd� de j �Z d?dd�Z G dd� de
j �Z!d?dd�Z"d?dd�Z#d?dd�Z$d?d d �Z%d?d d �Z&G d d!� d!e�Z'd?d"d#�Z(G d$d%� d%e�Z)G d&d'� d'e)�Z*d(d)� Z+d?d*d+�Z,d?d,d-�Z-G d.d/� d/e�Z.G d0d1� d1e�Z/d@d3d4�Z0G d5d6� d6e�Z1d?d7d8�Z2d9d:� Z3d;d<� Z4d=d>� Z5dS )Azx
Module containing 3D artist code and functions to convert 2D
artists into 3D versions which can be added to an Axes3D.
� N)�_api�artist�cbook�colors�lines�text�path)�LineCollection�PolyCollection�PatchCollection�PathCollection)� Normalize)�Patch� )�proj3dc C � | d d } | dkr| d } | S )z?Return the given angle normalized to -180 < *a* <= 180 degrees.ih � � ��ar r �</usr/lib/python3/dist-packages/mpl_toolkits/mplot3d/art3d.py� _norm_angle � r c C r )z=Return the given angle normalized to -90 < *a* <= 90 degrees.r �Z r r r r r �_norm_text_angle r r c C sp | dkr t �d�S | dkrt �d�S | dkrt �d�S | du r$t �d�S t �| �r4t| �d kr4t �| �S td
��) a�
Return a direction vector.
Parameters
----------
zdir : {'x', 'y', 'z', None, 3-tuple}
The direction. Possible values are:
- 'x': equivalent to (1, 0, 0)
- 'y': equivalent to (0, 1, 0)
- 'z': equivalent to (0, 0, 1)
- *None*: equivalent to (0, 0, 0)
- an iterable (x, y, z) is converted to a NumPy array, if not already
Returns
-------
x, y, z : array-like
The direction vector.
�x)r r r �y)r r r �z)r r r N)r r r � z2'x', 'y', 'z', None or vector of length 3 expected)�np�array�iterable�len�
ValueError��zdirr r r �get_dir_vector( s
r& c s\ e Zd ZdZddd�Zdd� Zd� fd
d � Zd d
� Zddd�Ze j
dd� �Z dd� Z � Z
S )�Text3Da�
Text object with 3D position and direction.
Parameters
----------
x, y, z
The position of the text.
text : str
The text string to display.
zdir : {'x', 'y', 'z', None, 3-tuple}
The direction of the text. See `.get_dir_vector` for a description of
the values.
Other Parameters
----------------
**kwargs
All other parameters are passed on to `~matplotlib.text.Text`.
r � r c K s* t jj| |||fi |�� | �||� d S �N)�mtext�Text�__init__�set_3d_properties)�selfr r r r r% �kwargsr r r r, ^ s zText3D.__init__c C s | j | j| jfS )z*Return the (x, y, z) position of the text.)�_x�_y�_z�r. r r r �get_position_3db s zText3D.get_position_3dNc s< t � �|dd� � | �|d � |dur t|�| _dS dS )a?
Set the (*x*, *y*, *z*) position of the text.
Parameters
----------
xyz : (float, float, float)
The position in 3D space.
zdir : {'x', 'y', 'z', None, 3-tuple}
The direction of the text. If unspecified, the zdir will not be
changed.
N� )�super� set_position�set_zr& �_dir_vec)r. �xyzr% �� __class__r r �set_position_3df s
�zText3D.set_position_3dc C � || _ d| _dS )zl
Set the *z* position of the text.
Parameters
----------
z : float
TN)r2 �stale)r. r r r r r8 w s
z Text3D.set_zc C s || _ t|�| _d| _d S �NT)r2 r& r9 r? )r. r r% r r r r- � s
zText3D.set_3d_propertiesc C s� t �| j| j| jf�}t�||| j g| jj �}|d d |d d }|d d |d d }t
� t
� ||��}t
j| |d d |d d t|�d�� tj�| |� W d � n1 s^w Y d| _d S )Nr r )r0 r1 � _rotationF)r r r0 r1 r2 r �proj_trans_pointsr9 �axes�M�math�degrees�atan2r � _setattr_cmr r* r+ �drawr? )r. �renderer�
position3d�proj�dx�dy�angler r r rI � s �� �
z Text3D.drawc C s d S r) r �r. rJ r r r �
get_tightbbox� s zText3D.get_tightbbox)r r r r( r r) �r r )�__name__�
__module__� __qualname__�__doc__r, r4 r= r8 r- r �allow_rasterizationrI rQ �
__classcell__r r r; r r' J s
r' r c C � t | _| �||� dS )z"Convert a Text to a Text3D object.N)r' r< r- )�objr r% r r r �
text_2d_to_3d� s r[ c sL e Zd ZdZ� fdd�Zddd�Zdd � Zd
d � Zej � fd d
��Z
� Z S )�Line3Dz
3D line object.
c s, t � jg g g|�R i |�� |||f| _dS )zU
Keyword arguments are passed onto :func:`~matplotlib.lines.Line2D`.
N)r6 r, �_verts3d)r. �xs�ys�zs�argsr/ r; r r r, � s zLine3D.__init__r r c C s: | � � }| �� }t�|t|��}t||||�| _d| _d S r@ )� get_xdata� get_ydatar � broadcast_tor" � juggle_axesr] r? )r. r` r% r^ r_ r r r r- � s
zLine3D.set_3d_propertiesc G s( t |�dkr |d | _n|| _d| _dS )ag
Set the x, y and z data
Parameters
----------
x : array-like
The x-data to be plotted.
y : array-like
The y-data to be plotted.
z : array-like
The z-data to be plotted.
Notes
-----
Accepts x, y, z arguments or a single array-like (x, y, z)
r r TN)r" r] r? )r. ra r r r � set_data_3d� s
zLine3D.set_data_3dc C � | j S )z�
Get the current data
Returns
-------
verts3d : length-3 tuple or array-like
The current data as a tuple or array-like.
)r] r3 r r r � get_data_3d� s zLine3D.get_data_3dc sH | j \}}}t�|||| jj�\}}}| �||� t� �|� d| _d S )NF) r] r �proj_transformrC rD �set_datar6 rI r? )r. rJ �xs3d�ys3d�zs3dr^ r_ r` r; r r rI � s
z Line3D.drawrR ) rS rT rU rV r, r- rf rh r rW rI rX r r r; r r\ � s
r\ c C rY )zConvert a 2D line to 3D.N)r\ r< r- )�liner` r% r r r �
line_2d_to_3d� s ro c sH t �|t| ��}| jddd�}dd� t||�D �}� fdd�|D �}|S )z Convert a path to a 3D segment.F��simplify�curvesc S s" g | ]
\\\}}}}|||f�qS r r ��.0r r �coder r r r �
<listcomp>� � " z'_path_to_3d_segment.<locals>.<listcomp>c � g | ] \}}}t |||� ��qS r �re �rt r r r r$ r r rv � � )r rd r" �
iter_segments�zip)r r` r% �pathsegs�seg�seg3dr r$ r �_path_to_3d_segment� s
r� c sP t �|�st �|t| ��}n t|�t| �krtd��� fdd�t| |�D �}|S )z6Convert paths from a collection object to 3D segments.z-Number of z-coordinates does not match paths.c � g | ]
\}}t ||� ��qS r )r� �rt r �pathzr$ r r rv � � �z)_paths_to_3d_segments.<locals>.<listcomp>)r r! rd r" r# r} )�pathsr` r% �segsr r$ r �_paths_to_3d_segments� s
�r� c sj t �|t| ��}| jddd�}dd� t||�D �}|r+t|� \}}� fdd�|D �}ng }g }|t|�fS )z/Convert a path to a 3D segment with path codes.Frp c S s& g | ]\\\}}}}|||f|f�qS r r rs r r r rv s & z2_path_to_3d_segment_with_codes.<locals>.<listcomp>c rx r ry rz r$ r r rv r{ )r rd r" r| r} �list)r r` r% r~ � seg_codesr �codesr� r r$ r � _path_to_3d_segment_with_codes� s r� c sT t �|t| ��}� fdd�t| |�D �}|r t|� \}}ng g }}t|�t|�fS )zP
Convert paths from a collection object to 3D segments with path codes.
c r� r )r� r� r$ r r rv r� z4_paths_to_3d_segments_with_codes.<locals>.<listcomp>)r rd r" r} r� )r� r` r% �segments_codes�segmentsr� r r$ r � _paths_to_3d_segments_with_codes s
�
r� c sb e Zd ZdZdd� Z� fdd�Ze�dd�dd d
��Ze j
ejdd d
d �d� fdd� ��Z � Z S )�Line3DCollectionz#
A collection of 3D lines.
c C r> �z&Set the position to use for z-sorting.TN��
_sort_zposr? �r. �valr r r �
set_sort_zpos �
z Line3DCollection.set_sort_zposc s || _ t� �g � dS )z"
Set 3D segments.
N)� _segments3dr6 � set_segments)r. r� r; r r r� % s z Line3DCollection.set_segments�3.4rJ Nc sT � fdd�� j D �}dd� |D �}t�� |� d}|D ] \}}}t|t|��}q|S )zB
Project the points according to renderer matrix.
c s g | ]
}t �|� jj��qS r )r rB rC rD )rt �pointsr3 r r rv 1 s �z5Line3DCollection.do_3d_projection.<locals>.<listcomp>c S s g | ] \}}}t �||g��qS r )r � column_stack)rt r^ r_ r` r r r rv 3 r{ g e��A)r� r r� �min)r. rJ �xyslist� segments_2d�minzr^ r_ r` r r3 r �do_3d_projection, s
� z!Line3DCollection.do_3d_projection�project)� alternativeFc s |r| � � t� �|� d S r) )r� r6 rI )r. rJ r� r; r r rI < s zLine3DCollection.drawr) )F)
rS rT rU rV r� r� r �delete_parameterr� r rW rI rX r r r; r r� s
�r� c C s$ t | �� ||�}t| _| �|� dS )z6Convert a LineCollection to a Line3DCollection object.N)r� � get_pathsr� r< r� )�colr` r% �
segments3dr r r �line_collection_2d_to_3dE s r� c sP e Zd ZdZddd�� fdd�
Zddd �Zd
d � Ze�d d
�ddd��Z � Z
S )�Patch3Dz
3D patch object.
r r �r` r% c s" t � j|i |�� | �||� d S r) )r6 r, r- )r. r` r% ra r/ r; r r r, Q s zPatch3D.__init__r c s. t �|t|��}� fdd�t||�D �| _d S )Nc s" g | ]
\\}}}t |||� ��qS r ry rz r$ r r rv W s
�z-Patch3D.set_3d_properties.<locals>.<listcomp>)r rd r" r} �
_segment3d)r. �vertsr` r% r r$ r r- U s
�zPatch3D.set_3d_propertiesc C rg r) )�_path2dr3 r r r �get_pathZ � zPatch3D.get_pathr� rJ Nc
C sN | j }t|� \}}}t�|||| jj�\}}}} t�t� ||g��| _
t |�S r) ) r� r} r �proj_transform_cliprC rD �mpath�Pathr r� r� r� �
r. rJ �sr^ r_ r` �vxs�vys�vzs�visr r r r� ] s
�zPatch3D.do_3d_projectionrR r) ) rS rT rU rV r, r- r� r r� r� rX r r r; r r� L s
r� c @ s@ e Zd ZdZddd�dd�Zddd �Ze�d
d �dd
d��Zd S )� PathPatch3Dz
3D PathPatch object.
r r r� c K s$ t j| fi |�� | �|||� d S r) )r r, r- )r. r r` r% r/ r r r r, l s zPathPatch3D.__init__r c C s t j| |j||d� |j| _d S )Nr� )r� r- �verticesr� �_code3d)r. r r` r% r r r r- q s z PathPatch3D.set_3d_propertiesr� rJ Nc
C sR | j }t|� \}}}t�|||| jj�\}}}} t�t� ||g�| j
�| _ t |�S r) )
r� r} r r� rC rD r� r� r r� r� r� r� r� r r r r� u s
�z PathPatch3D.do_3d_projectionrR r) ) rS rT rU rV r, r- r r� r� r r r r r� g s
r� c C s4 | � � }| �� }|�|�}t|�r|d S t�g �S )z2Return a list of vertices for the path of a patch.r )�get_patch_transformr� � to_polygonsr" r r )�patch�transr �polygonsr r r �_get_patch_verts s
r� c C s t | �}t| _| �|||� dS )z$Convert a Patch to a Patch3D object.N)r� r� r< r- )r� r r% r� r r r �patch_2d_to_3d� s r� c C s2 | � � }| �� }|�|�}t| _| �|||� dS )z,Convert a PathPatch to a PathPatch3D object.N)r� r� �transform_pathr� r< r- )� pathpatchr r% r r� r� r r r �pathpatch_2d_to_3d� s
r� c s� e Zd ZdZdddd�� fdd�
Zdd � Zd
d � Zd d
� Zdd� Ze �
dd�d� fdd� �Z dd� Z � fdd�Z
� fdd�Z� ZS ) �Patch3DCollectionz%
A collection of 3D patches.
r r T�r` r% �
depthshadec s( || _ t� j|i |�� | �||� dS )a�
Create a collection of flat 3D patches with its normal vector
pointed in *zdir* direction, and located at *zs* on the *zdir*
axis. 'zs' can be a scalar or an array-like of the same length as
the number of patches in the collection.
Constructor arguments are the same as for
:class:`~matplotlib.collections.PatchCollection`. In addition,
keywords *zs=0* and *zdir='z'* are available.
Also, the keyword argument *depthshade* is available to
indicate whether or not to shade the patches in order to
give the appearance of depth (default is *True*).
This is typically desired in scatter plots.
N)� _depthshader6 r, r- �r. r` r% r� ra r/ r; r r r, � s zPatch3DCollection.__init__c C rg r) �r� r3 r r r �get_depthshade� r� z Patch3DCollection.get_depthshadec C r> �z�
Set whether depth shading is performed on collection members.
Parameters
----------
depthshade : bool
Whether to shade the patches in order to give the appearance of
depth.
TN�r� r? �r. r� r r r �set_depthshade� �
z Patch3DCollection.set_depthshadec C r> r� r� r� r r r r� � r� z Patch3DCollection.set_sort_zposc C s` | � � | �� }t|�dkr|j\}}ng }g }t||t�|�|�| _td�| _ d | _
d| _ d S �Nr ���T) �update_scalarmappable� get_offsetsr" �Tre r �
atleast_1d�
_offsets3d�slice�_z_markers_idx�_vzsr? �r. r` r% �offsetsr^ r_ r r r r- � s
z#Patch3DCollection.set_3d_propertiesr� rJ Nc s\ | j \}}}t�|||| jj�\}}}}|| _t� �t� ||g�� |j
dkr+t |�S tj S �Nr )
r� r r� rC rD r� r6 � set_offsetsr r� �sizer� �nan) r. rJ r^ r_ r` r� r� r� r� r; r r r� � s
�
z"Patch3DCollection.do_3d_projectionc C �D | j d ur| jrt|| j �n|}t|�dkr|| j }t�|| j�S �Nr �r� r� �_zalphar" r� �mcolors�
to_rgba_array�_alpha�r. � color_arrayr r r �"_maybe_depth_shade_and_sort_colors� � ��
z4Patch3DCollection._maybe_depth_shade_and_sort_colorsc � | � t� �� �S r) �r� r6 �
get_facecolorr3 r; r r r� � � z Patch3DCollection.get_facecolorc �& t �| jd�r | �� S | �t� �� �S �N�face�r �
_str_equal� _edgecolorsr� r� r6 �
get_edgecolorr3 r; r r r� � � z Patch3DCollection.get_edgecolorr) )rS rT rU rV r, r� r� r� r- r r� r� r� r� r� rX r r r; r r� � s
r� c s� e Zd ZdZdddd�� fdd�
Z� fdd �Zd
d � Zd d
� Zd"� fdd� Z� fdd�Z dd� Z
dd� Z e �
dd�d#dd��Zd d � Z� fd d �Z� fd d!�Z� ZS )$�Path3DCollectionz#
A collection of 3D paths.
r r Tr� c s. || _ d| _t� j|i |�� | �||� dS )a�
Create a collection of flat 3D paths with its normal vector
pointed in *zdir* direction, and located at *zs* on the *zdir*
axis. 'zs' can be a scalar or an array-like of the same length as
the number of paths in the collection.
Constructor arguments are the same as for
:class:`~matplotlib.collections.PathCollection`. In addition,
keywords *zs=0* and *zdir='z'* are available.
Also, the keyword argument *depthshade* is available to
indicate whether or not to shade the patches in order to
give the appearance of depth (default is *True*).
This is typically desired in scatter plots.
FN)r� �_in_drawr6 r, r- r� r; r r r, � s zPath3DCollection.__init__c s@ t j| dd�� t� �|� W d � d S 1 sw Y d S )NT)r� )r rH r6 rI rP r; r r rI s "�zPath3DCollection.drawc C r> r� r� r� r r r r� r� z Path3DCollection.set_sort_zposc C s| | � � | �� }t|�dkr|j\}}ng }g }t||t�|�|�| _| j| _ | j
| _ | j\}}}t d�| _
d | _d| _d S r� )r� r� r" r� re r r� r� �_sizes�_sizes3d� _linewidths�
_linewidths3dr� r� r� r? r� r r r r- s
z"Path3DCollection.set_3d_properties� R@c s" t � �||� | js|| _d S d S r) )r6 � set_sizesr� r� )r. �sizes�dpir; r r r ; s
�zPath3DCollection.set_sizesc s t � �|� | js|| _d S d S r) )r6 �
set_linewidthr� � _linewidth3d)r. �lwr; r r r @ s
�z Path3DCollection.set_linewidthc C rg r) r� r3 r r r r� E r� z Path3DCollection.get_depthshadec C r> r� r� r� r r r r� H r� z Path3DCollection.set_depthshader� rJ Nc
C s� | j \}}}t�|||| jj�\}}}}t�|�d d d� } | _|| _t | j
�dkr1| j
| | _ t | j �dkr>| j | | _
|| }|| }|| }t�| t�||f�� |jr]t�|�S tjS )Nr� r )r� r r� rC rD r �argsortr� r� r" r� r� r� r� r r� r� r� r� r� )
r. rJ r^ r_ r` r� r� r� r� �
z_markers_idxr r r r� U s
� z!Path3DCollection.do_3d_projectionc C r� r� r� r� r r r r� t r� z3Path3DCollection._maybe_depth_shade_and_sort_colorsc r� r) r� r3 r; r r r� ~ r� z Path3DCollection.get_facecolorc r� r� r� r3 r; r r r� � r� z Path3DCollection.get_edgecolor)r� r) )rS rT rU rV r, rI r� r- r r r� r� r r� r� r� r� r� rX r r r; r r� � s
r� Tc C s> t | t�r t| _nt | t�rt| _|| _d| _| �||� dS )a
Convert a :class:`~matplotlib.collections.PatchCollection` into a
:class:`Patch3DCollection` object
(or a :class:`~matplotlib.collections.PathCollection` into a
:class:`Path3DCollection` object).
Parameters
----------
za
The location or locations to place the patches in the collection along
the *zdir* axis. Default: 0.
zdir
The axis in which to place the patches. Default: "z".
depthshade
Whether to shade the patches to give a sense of depth. Default: *True*.
FN) �
isinstancer r� r< r r� r� r� r- )r� r` r% r� r r r �patch_collection_2d_to_3d� s
r c s� e Zd ZdZdd�� fdd�
Zejejejd�Z dd� Z
d d
� Z d#� fd d
� Z dd� Z
dd� Zdd� Ze�dd�d$dd��Z� fdd�Z� fdd �Zd d � Zd d � Zd!d"� Z� ZS )%�Poly3DCollectiona�
A collection of 3D polygons.
.. note::
**Filling of 3D polygons**
There is no simple definition of the enclosed surface of a 3D polygon
unless the polygon is planar.
In practice, Matplotlib fills the 2D projection of the polygon. This
gives a correct filling appearance only for planar polygons. For all
other polygons, you'll find orientations in which the edges of the
polygon intersect in the projection. This will lead to an incorrect
visualization of the 3D area.
If you need filled areas, it is recommended to create them via
`~mpl_toolkits.mplot3d.axes3d.Axes3D.plot_trisurf`, which creates a
triangulation and thus generates consistent surfaces.
�average)�zsortc sh t � j|g|�R i |�� t|tj�r |jdkr td��n
tdd� |D ��r*td��| �|� d| _ dS )aS
Parameters
----------
verts : list of (N, 3) array-like
Each element describes a polygon as a sequence of ``N_i`` points
``(x, y, z)``.
zsort : {'average', 'min', 'max'}, default: 'average'
The calculation method for the z-order.
See `~.Poly3DCollection.set_zsort` for details.
*args, **kwargs
All other parameters are forwarded to `.PolyCollection`.
Notes
-----
Note that this class does a bit of magic with the _facecolors
and _edgecolors properties.
r z)verts must be a list of (N, 3) array-likec s s" � | ] }t t�|��d kV qdS )r5 N)r" r �shape)rt �vertr r r � <genexpr>� s � z,Poly3DCollection.__init__.<locals>.<genexpr>N)
r6 r, r r �ndarray�ndimr# �any� set_zsort�_codes3d)r. r� r ra r/ r; r r r, � s
�
zPoly3DCollection.__init__)r r� �maxc C s | j | | _d| _d| _dS )a
Set the calculation method for the z-order.
Parameters
----------
zsort : {'average', 'min', 'max'}
The function applied on the z-coordinates of the vertices in the
viewer's coordinate system, to determine the z-order.
NT)�_zsort_functions�
_zsortfuncr� r? )r. r r r r r � s
zPoly3DCollection.set_zsortc C s� t |�rt�|�j\}}}ng g g }}}t�t |��}t�||||g�| _dgt�dd� |D ���}g tt |dd� |dd� ��| _
dS )z Optimize points for projection.r c S s g | ]}t |��qS r )r" )rt �segmentr r r rv � s z/Poly3DCollection.get_vector.<locals>.<listcomp>Nr� r ) r" r � row_stackr� �onesr �_vec�cumsum�mapr� �
_segslices)r. r� r^ r_ r` r �indicesr r r �
get_vector� s &zPoly3DCollection.get_vectorTc s" | � |� t� �g d� || _dS )zSet 3D vertices.FN)r r6 � set_verts�_closed)r. r� �closedr; r r r! � s
zPoly3DCollection.set_vertsc C s | j |dd� || _dS )z Set 3D vertices with path codes.F)r# N)r! r )r. r� r� r r r �set_verts_and_codes� s
z$Poly3DCollection.set_verts_and_codesc C sF | � � d | _| �d� t�| �| _t�| �| _t�| �| _ d| _
d S )Nr T) r� r� r r
r� � _facecolor3dr� � _edgecolor3d� get_alpha�_alpha3dr? r3 r r r r- s
z"Poly3DCollection.set_3d_propertiesc C r> r� r� r� r r r r� r� z Poly3DCollection.set_sort_zposr� rJ Nc s� � j dur� �� � jr� j� _� jr� j� _t� � j
� j j �\������fdd�� j
D �}� j}� j}t|�t|�krF|jt|�dd�}t|�t|�kr`t|�dkrW|}n |jt|�dd�}|r�t� fdd�tt|||��D �dd � d
d �}t|� \}}� _� _}ng }t�d �� _t�d �� _g }� jdur�� fd
d�|D �} t�� || � nt�� |� j� t� j�t|�kr�� j� _� j dur�t� dgdg� j gdgg�}
t� |
� j j �} | d d S �j dkr�t� ��S tj S )z<
Perform the 3D projection for this object.
Nc s" g | ]
}� | �| �| f�qS r r )rt �sl)�txs�tys�tzsr r rv ( rw z5Poly3DCollection.do_3d_projection.<locals>.<listcomp>r )�axisc 3 s> � | ]\}\\}}}}}� � |�t�||g�|||fV qd S r) )r r r� )rt �idxr^ r_ r` �fc�ecr3 r r r 8 s � &�z4Poly3DCollection.do_3d_projection.<locals>.<genexpr>c S s | d S r� r )r r r r �<lambda>; s z3Poly3DCollection.do_3d_projection.<locals>.<lambda>T)�key�reverse�r � c s g | ]}� j | �qS r )r )rt r. r3 r r rv F s r r5 )!�_Ar� �_face_is_mapped� _facecolorsr% �_edge_is_mappedr� r& r �_proj_transform_vecr rC rD r r" �repeat�sorted� enumerater} �
_facecolors2d�
_edgecolors2dr �emptyr r
r$ r! r" r� r r� r� r� ) r. rJ �xyzlist�cface�cedge�
z_segments_2d�_r� �idxsr� �zvec�ztransr )r. r* r+ r, r r� sT
���
z!Poly3DCollection.do_3d_projectionc � t � �|� t�| �| _d S r) )r6 �
set_facecolorr
r� r% �r. r r; r r rJ [ � z Poly3DCollection.set_facecolorc rI r) )r6 �
set_edgecolorr
r� r& rK r; r r rM ` rL z Poly3DCollection.set_edgecolorc
C sx t j�| |� z t�| j| j�| _W n ttt fy Y nw z t�| j
| j�| _ W n ttt fy6 Y nw d| _ d S r@ )
r �Artist� set_alphar� r� r% r� �AttributeError� TypeError�
IndexErrorr& r� r? )r. �alphar r r rO e s
��
��
zPoly3DCollection.set_alphac C rg r) )r> r3 r r r r� t r� z Poly3DCollection.get_facecolorc C rg r) )r? r3 r r r r� w r� z Poly3DCollection.get_edgecolor)Tr) )rS rT rU rV r, r r r� r r r r r! r$ r- r� r r� r� rJ rM rO r� r� rX r r r; r r
� s( �
E r
c C s2 t | �� ||�\}}t| _| �||� | �� dS )z6Convert a PolyCollection to a Poly3DCollection object.N)r� r� r
r< r$ r- )r� r` r% � segments_3dr� r r r �poly_collection_2d_to_3d{ s
� rU c C sH |dkr || |fS |dkr| ||fS |d dkr t | |||�S | ||fS )z�
Reorder coordinates so that 2D xs, ys can be plotted in the plane
orthogonal to zdir. zdir is normally x, y or z. However, if zdir
starts with a '-' it is interpreted as a compensation for rotate_axes.
r r r �-)� rotate_axes�r^ r_ r` r% r r r re � s
re c C sR |dkr ||| fS |dkr|| |fS |dkr|| |fS |dkr$||| fS | ||fS )z�
Reorder coordinates so that the axes are rotated with zdir along
the original z axis. Prepending the axis with a '-' does the
inverse transform, so zdir can be x, -x, y, -y, z or -z
r z-xr z-yr rX r r r rW � s
rW c C s� t | �dks t |�dkrt�d�S tt|�t|��}d||�d }t�t�| �t |�df�}t� |dd�dd�f |dd�df | g�S )z7Modify the alphas of the color list according to depth.r r4 r gffffff�?r5 Nr )
r" r �zerosr
r� r rd r� r� r� )r r` �norm�sats�rgbar r r r� � s
.r� rR )r r T)6rV rE �numpyr �
matplotlibr r r r r� r r r* r r� �matplotlib.collectionsr r
r r �matplotlib.colorsr
�matplotlib.patchesr r( r r r r& r+ r' r[ �Line2Dr\ ro r� r� r� r� r� r� r� r� r� r� r� r� r� r r
rU re rW r� r r r r �<module> sH $ "
P
>
*
`
W