�
n1�V�1 � @ sj d d l Z d d l Td d l Td d l Td d � Z d d � Z d d � Z d d
� Z d d � Z d S)
� N)�*c C s. d d l } d d d � } | | j j _ d S)Nr Fc S s� | r1 d } d d d g } d d � | D� } n d } d d g } | r[ t t | � � n g } t j g | } t | � d k r� | St j � �� t j d
d � x� | D]� \ } } d d � | | D� } d
d � | | D� } | rt | d t j � rt j
| � | | <n | r3t j | � | | <n | ret | d t j � ret j | � | | <q� | r� t j
| � | | <q� q� WWd QXt | � S)z�
Computes the maximal extent in 2D and 3D space from
list of 4-tuples or 6-tuples. If zrange is enabled
all extents are converted to 6-tuples to comput
x-, y- and z-limits.
� r � � � � � c S sP g | ]F } t | � d k r$ | n% | d | d d | d | d d f � q S)r r r Nr r )�len)�.0�e� r �../code/patch_holoviews.py�
<listcomp> s z8patch_core_util.<locals>.max_extents.<locals>.<listcomp>�ignorez All-NaN (slice|axis) encounteredc S s" g | ] } | d k r | � q S)Nr )r
�vr r r
r " s c S s" g | ] } | d k r | � q S)Nr )r
r r r r
r # s N)r r )r r )r r )r r )r r )�list�zip�np�NaNr �warnings�catch_warnings�filterwarnings�
isinstance�
datetime64�min�nanmin�max�nanmax�tuple) �extents�zrange�num�inds�arr�lidx�uidx�lower�upperr r r
� max_extents s2
z$patch_core_util.<locals>.max_extents)�holoviews.core.util�core�utilr( )� holoviewsr( r r r
�patch_core_util s #r- c s@ d d l � d � f d d � } d d l } | | j j _ d S)Nr Tc s� | r | \ } } n � j | � � j | � } } | | | | } � j j | d � j � j | � � �} | | � } | r� | d 9} d d � | D� S| Sd S)zg
Maps an array of values to RGB hex strings, given
a color range and colormap.
�mask�� c S s6 g | ], } d j d d � | d d � D� � � q S)z#{0:02x}{1:02x}{2:02x}c s s | ] } t | � Vq d S)N)�int)r
r r r r
� <genexpr>A s zEpatch_plotting_util.<locals>.map_colors.<locals>.<listcomp>.<genexpr>Nr �����)�format)r
�cr r r
r A s z;patch_plotting_util.<locals>.map_colors.<locals>.<listcomp>N)r r �ma�array� logical_not�isfinite)r# �crange�cmap�hex�cmin�cmax)r r r
�
map_colors3 s *
z'patch_plotting_util.<locals>.map_colors)�numpy�holoviews.plotting.util�plottingr+ r> )r> r, r )r r
�patch_plotting_util1 s rB c s. d d l } d d l j � d d l m � � | j j j _ � | j j j _ d d � } | | j j j j _ d d � } | | j j j j _
d d d � } | | j j j j _ d � � f d d
� } | | j j j j _
d d d
� } | | j j j j _
d d d
� } | | j j j j _
d S)
Nr )r> c [ s@ | j d } | j d j d � | j | j � | j d | j d | j � | j | _ | j d k r� | j
j j d � | j
j
j d � n | j d k r� | j j j d � | j j
j d � n | j d k r| j j j d � | j j
j d � n | j r| j � n | j | j � t t | � j | | � S) z�
Extends the ElementPlot _finalize_axis method to set appropriate
labels, and axes options for 3D Plots.
�axis�figF�elev�azimNg � )�handles� set_frameon�grid� show_grid� view_init� elevation�azimuth�distance�dist�xaxis�w_xaxis�line�set_lw�label�set_text�yaxis�w_yaxis�zaxis�w_zaxis� disable_axes� set_axis_off�set_axis_bgcolor�bgcolor�super�Plot3D�_finalize_axis)�self�key�kwargsrC r r r
ra S s$
z)patch_mpl_chart3d.<locals>._finalize_axisc _ s t t | � j | | � d S)N)r_ r` � update_frame)rb �argsrd r r r
re o s z'patch_mpl_chart3d.<locals>.update_framec s� | j d } | j j � | j | j | j d | � } t � | � } | j d } � f d d � t d � D� \ } } } | j | j } � j | j
� } | r d | k r � j | j
� } | | d <d | k r | | j }
| j
d |
d
d |
d � q n � j | j � r4| j � | � | d <n | j | | | d
| j | �} | j d j | � | | j d <| j | d | �S)NrC r c 3 s | ] } � j | � Vq d S)N)�dimension_values)r
�i)�pointsr r
r1 z s z=patch_mpl_chart3d.<locals>.initialize_plot.<locals>.<genexpr>r r: r4 �clim�vminr �vmax�s�zorder�artist�rangesr2 r2 )rH �hmap�last�compute_ranges�keys�
match_spec�range�style� cyclic_index�
get_dimension� color_indexrg �name�update�
size_index�
_compute_size�scatterrn �add_collectionra ) rb rp rC rc �xs�ys�zsrw �cdim�cs�clims� scatterplotr )ri r
�initialize_plott s(
(
$!
z*patch_mpl_chart3d.<locals>.initialize_plotc s� | j d } t � f d d � � j � D� � | _ � j | j � } | j | j } | r� d | k r� � j | j � } d | k r� | d n
| | j } � j
| d � }
� | | |
d � | _ n � j | j � r� | j
| j � | � � n d S)Nro c 3 s | ] } � | Vq d S)Nr )r
�d)ri r r
r1 � s z<patch_mpl_chart3d.<locals>.update_handles.<locals>.<genexpr>r: rj F)rH r �
dimensions�
_offsets3dry rz rw rx rg r{ �get_cmap� _facecolor3dr} � set_sizesr~ ) rb rC ri rc rp ro r� rw r� rj r: )�cmr> )ri r
�update_handles� s
(#z)patch_mpl_chart3d.<locals>.update_handlesc S s� d | j k r0 | j d j j | j d � n | j } | j \ } } | j | | � \ } } }
} } }
t j | | | | t | � � | | | | t | � � f \ } } | j | j
} | j d k r� | j d j | | | | � | j d <n� | j d k rG|
| d <|
| d <| j d j
| | | | � | j d <n8 | j d k r| j d j | | | | � | j d <n d S)Nro rC � wireframe�surfacerk rl �contour)rH � collections�remove�data�shape� get_extentsr �mgrid�floatrw rx � plot_type�plot_wireframe� plot_surface� contour3D)rb rC �elementrc rp �mat�rn�cn�l�b�zmin�r�t�zmaxr4 �
style_optsr r r
r� � s ! $G)
)c s� d | j k r0 | j d j j | j d � n | j | j } � j d d � } | | d } � f d d � | D� \ } } }
| j | | |
d | d d
| d | �} | | j d <d S) Nro rC rU Tr c s g | ] } � j | � � q Sr )rg )r
r� )r� r r
r � s z=patch_mpl_chart3d.<locals>.update_handles.<locals>.<listcomp>rl r rk r )rH r� r� rw rx r� � plot_trisurf) rb rC r� rc rp r� �dims�vrange�x�y�zro r )r� r
r� � s !" )� holoviews.plotting.mpl.chart3d�
matplotlib.cmr� r@ r> rA �mpl�chart3dr` ra re �
Scatter3DPlotr� r� � SurfacePlot�TrisurfacePlot)r, ra re r� r� r )r� r> r
�patch_mpl_chart3dJ s"
r� c
C sp d d l } d d d d d d d d d d � } | | j j j j _ d d d d � } | | j j j j _ d S)Nr c
sf � j | � }
|
� _ � j d } � j rA | j � j � n � j r_ t � j j � � n g } � j d k r�| d k r�� j d k r� d n � j | � } t
� f d d � � g | D� � }
|
d k r.|
r.� j |
| | | | � \ } } } � j r| | } } n � j
| |
| | � |
j d � |
j d � } } d \ } } | d k rjn: | j r| j } n% | j | j k r�| j | j } n | r�| j j t | � � n | d k r�n: | j r�| j } n% | j | j k r | j | j } n | r.| j j t | � � q.n � j d k r�| r�| j � } | ri| j � j � n | j � j � j � | j � j � j � n | r�� j r�| j | � j d � � n | r�� j r�| j | � j d � � n | r� j r| j! | � j d � � n � j" | � � j# | � � j$ rX� j% | |
| | | � n � j& r�| d k r�d � j k r�� j d j' | � q�| j( | � j d � � � j d <q�n � j) | � � j* r�� j+ r�� j, | � n xa � j- D]V } y | � |
� Wq�t. k
rK} z � j/ d
| | f � WYd d } ~ Xq�Xq�Wt0 t1 � � j2 | � S) a*
Applies all the axis settings before the axis or figure is returned.
Only plots with zorder 0 get to apply their settings.
When the number of the frame is supplied as n, this method looks
up and computes the appropriate title, axis labels and axis bounds.
rC r Nc 3 s6 | ], } t | j t � r | j j � j k Vq d S)N)r rq �HoloMap�type� _suppressed)r
�sp)rb r r
r1 � s z<patch_mpl_element.<locals>._finalize_axis.<locals>.<genexpr>r �xlabel�ylabel�titlez+Plotting hook %r could not be applied:
%s)NN)3�
_get_frame�
current_framerH r^ r] �subplotsr �valuesrn �
_format_title�any� _axis_labels� invert_axes�_finalize_limitsry � value_formatr� �type_formattersrQ �set_major_formatter�wrap_formatterrW �
get_legend� set_visible� show_legend� get_xaxisrJ rK � get_yaxis�
set_xlabel� _fontsize�
set_ylabelrY �
set_zlabel�
_apply_aspect�_subplot_label� apply_ticks�_finalize_ticks�
show_titlerV � set_title�_finalize_axes�overlaid�drawn�_finalize_artist�finalize_hooks� Exception�warningr_ � ElementPlotra )rb rc r� rp �xticks�yticks�zticksr� r� �zlabelr� rC r� �suppress�xdim�ydim�xformat�yformat�legend�hookr r )rb r
ra � s|
$$&$
.z)patch_mpl_element.<locals>._finalize_axisc S s~ | j d } | d k r+ | j | � } n | | _ | | _ t | j t � rd | } | j � } n | j } | j � } | j | | | � } x� | j j � D]� \ } } | j
| d � } t | j t � r
t | | | | | � }
|
d k r
| j |
� \ } } q
n | j
| | | � q� Wt | j t � rK| rKt d � � n | j rg| j | | � n | j | d | �d S)NrC zhSome Elements returned by the dynamic callback were not initialized correctly and could not be rendered.rp )rH r� r� � current_keyr rq �
DynamicMap�itemsrs r� �get�dynamic_update�popre r� r� �_adjust_legendra ) rb rc rp r� rC � range_objr� �k�subplot�el�idx�_r r r
re s.
z'patch_mpl_element.<locals>.update_frame)� holoviews.plotting.mpl.elementrA r� r� r� ra � OverlayPlotre )r, ra re r r r
�patch_mpl_element� s S"r� c C s6 t j d � �! t � t � t � t � Wd QXd S)N�error)�param�
logging_levelr- rB r� r� r r r r
� patch_all@ s
r ) r� r) r� r� r- rB r� r� r r r r r
�<module> s
* x ~