ó
±éRc           @   sƒ   d  d l  Z d  d l m Z m Z m Z d d e d „ Z d „  Z	 d  d l
 m Z d e f d „  ƒ  YZ d	 e f d
 „  ƒ  YZ d S(   iÿÿÿÿN(   t   make_axes_locatablet   Sizet   locatable_axes_factoryg{®Gáz„?c         C   sC  t  |  ƒ } t j | t j |  ƒ ƒ } t j d d | d t j |  ƒ ƒ } t j d d | d t j |  ƒ ƒ } | j t j |  ƒ | | g ƒ | j | | | | | g ƒ |  j | j d d d d ƒƒ g  } | d k ry t
 |  j ƒ } Wqt k
 rt
 t |  ƒ ƒ } qXn  xì d d d g D]Û }	 | |  j ƒ  |  j d	 t ƒ d
 |  d |  ƒ}
 | j d d d |	 ƒ } |
 j | ƒ x1 |
 j j ƒ  |
 j j ƒ  D] } | j t ƒ q¢Wy. x' |
 j j ƒ  D] } | j j t ƒ qÌWWn t k
 rún X| j |
 ƒ q-W| r?|  j ƒ  } x | D] }
 | j |
 ƒ q%Wn  | S(   s,   
    pad : fraction of the axes height.
    g      ð?g       @g      @i    t   ny1iÿÿÿÿi   i   t   originalt   sharext   shareyt   nxt   nyN(   R    R   t   Fractiont   AxesYt   AxesXt   set_horizontalt   set_verticalt   set_axes_locatort   new_locatort   NoneR   t   _axes_classt   AttributeErrort   typet
   get_figuret   get_positiont   Truet   yaxist   get_ticklabelst   xaxist   set_visiblet   Falset   axist   valuest   major_ticklabelst   appendt   add_axes(   t   axt   padt
   axes_classt   add_allt   dividert   pad_sizet   xsizet   ysizet   ax_rgbR   t   ax1t   locatort   tR   t   fig(    (    sZ   /opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_rgb.pyt   make_rgb_axes   s@    ''#c         K   sÌ   | j  \ } } t j | | d g d d ƒ} | | d  d  … d  d  … d f <t j | ƒ } | | d  d  … d  d  … d f <t j | ƒ }	 | |	 d  d  … d  d  … d f <| | |	 }
 |  j |
 | � } | S(   Ni   t   dtypet   di    i   i   (   t   shapet   npt   zerost
   zeros_liket   imshow(   R!   t   rt   gt   bt   kwargsR   R   t   Rt   Gt   Bt   RGBt   im_rgb(    (    sZ   /opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_rgb.pyt
   imshow_rgb6   s    (   t   Axest   RGBAxesBasec           B   s,   e  Z d  „  Z d „  Z d „  Z d „  Z RS(   c      	   O   s  | j  d d ƒ } | j  d t ƒ } | j  d d  ƒ } | d  k rN |  j } n  | | | Ž  } t | ƒ } t j | t j | ƒ ƒ } t j d d | d t j | ƒ ƒ }	 t j d d | d t j | ƒ ƒ }
 | j	 t j | ƒ | |	 g ƒ | j
 |
 | |
 | |
 g ƒ | j | j d d d	 d
 ƒƒ g  } x� d d d g D]| } | | j ƒ  | j d t ƒ d | d | | �} | j d d d | ƒ } | j | ƒ | j j d t ƒ | j | ƒ qBW| |  _ | \ |  _ |  _ |  _ | r| j ƒ  } | j | ƒ |  j ƒ  n  |  j ƒ  d  S(   NR"   g        R$   R#   g      ð?g       @g      @i    R   iÿÿÿÿi   i   R   R   R   R   R   t
   ticklabels(   t   popR   R   t   _defaultAxesClassR    R   R	   R
   R   R   R   R   R   R   R   R   t   toggleR   R   R=   R:   R;   R<   R    t   add_RGB_to_figuret   _config_axes(   t   selft   klR9   R"   R$   R#   R!   R%   R&   R'   R(   R)   R   R*   R+   R-   (    (    sZ   /opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_rgb.pyt   __init__J   s:    ''	c         C   sU   xN |  j  |  j |  j |  j g D]. } | j j j d ƒ | j j j d ƒ q Wd  S(   Nt   w(	   R=   R:   R;   R<   R   t   linet	   set_colort   major_tickst   set_mec(   RH   R*   (    (    sZ   /opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_rgb.pyRG   |   s    %c         C   sO   |  j  j ƒ  j |  j ƒ |  j  j ƒ  j |  j ƒ |  j  j ƒ  j |  j ƒ d  S(   N(   R=   R   R    R:   R;   R<   (   RH   (    (    sZ   /opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_rgb.pyRF   ˆ   s    c         K   s  | j  \ } } t j | | d g d d ƒ} | | d  d  … d  d  … d f <t j | ƒ } | | d  d  … d  d  … d f <t j | ƒ }	 | |	 d  d  … d  d  … d f <| | |	 }
 |  j j |
 | � } |  j j | | � } |  j j | | � } |  j j |	 | � } | | | | f S(   Ni   R/   R0   i    i   i   (	   R1   R2   R3   R4   R=   R5   R:   R;   R<   (   RH   R6   R7   R8   R9   R   R   R:   R;   R<   R=   R>   t   im_rt   im_gt   im_b(    (    sZ   /opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_rgb.pyR?   �   s    (   t   __name__t
   __module__RJ   RG   RF   R?   (    (    (    sZ   /opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_rgb.pyRA   H   s   	2		t   RGBAxesc           B   s   e  Z e Z RS(    (   RS   RT   R@   RD   (    (    (    sZ   /opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_rgb.pyRU       s   (   t   numpyR2   t   axes_dividerR    R   R   R   R   R.   R?   t   mpl_axesR@   t   objectRA   RU   (    (    (    sZ   /opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_rgb.pyt   <module>   s   2	X