3
���h�# � @ s2 d dl m Z mZ d dlZd dlZd dlZddlmZ ddl m
Z
m Z m Z m
Z
dd� Zd7d d
�Zd d � Zd
d� Zdd� Zd8dd�Zd9dd�Zdd� Zdd� Zd:dd �Zd d � Zd d � Zd!d"� Zd;d$d%�Zd<d&d'�Z d(Z d)d*� Z d=d+d,�Z ej e d-ed.�Z!ej e d/ed.�Z"ej e d0e d.�Z#d1d2� Z$d3d4� Z%d5d6� Z&dS )>� )�datetime� timedeltaN� )�OPTIONS)� iteritems� unicode_type�
bytes_type�dask_array_typec C s@ t | �}t|�|kr(|d|d � d S |d|t|� S dS )z�Given an object `x`, call `str(x)` and format the returned string so
that it is numchars long, padding with trailing spaces or truncating with
ellipses as necessary
N� z...� )�str�len)�x�numchars�s� r �6/tmp/pip-build-5_djhm0z/xray/xarray/core/formatting.py� pretty_print s r � c C s8 |d krt |�}dd| }||jdd� | j� D �� S )N�
r c s s | ]
}|V qd S )Nr )�.0r r r r � <genexpr> s z wrap_indent.<locals>.<genexpr>)r
�join�
splitlines)�text�start�length�indentr r r � wrap_indent s r c C s� d| k ot j| �kn s"t�t j| d d d� �}t j||k�}tt jt|�t jd|f | ��}dt | �d | t
|�f t
d �f| }|S )Nr r ���)r ) �np�prod�AssertionErrorZcumprod�argmax�int�ceil�floatZr_r
�slice)�shape� n_desiredZ cum_itemsZn_steps�stop�indexerr r r � _get_indexer_at_least_n_items s "$ r, c C sP |dk rt d��| jdkr g S || jk r<t| j|�}| | } tj| �jd|� S )z-Returns the first n_desired items of an arrayr z must request at least one itemr N)�
ValueError�sizer, r( r �asarrayZflat)r r) r+ r r r �
first_n_items( s
r0 c C sP t tj| ��}y|j� \}}W n tk
r2 |S X |dkr@|S d||f S dS )zECast given object to a Timestamp and return a nicely formatted stringz00:00:00z%sT%sN)r �pdZ Timestamp�splitr- )�tZ datetime_strZdate_str�time_strr r r �format_timestamp; s r5 c C sZ t tj| ��}y|jd�\}}W n tk
r4 |S X |dkrF|d S |dkrR|S |S dS )zECast given object to a Timestamp and return a nicely formatted stringz days �datez days�timeN)r r1 Z Timedeltar2 r- )r3 �timedelta_formatZ
timedelta_strZdays_strr4 r r r �format_timedeltaJ s r9 Tc C sx t | tjtf�rt| �S t | tjtf�r4t| |d�S t | tt f�rR|rNt
| �S | S t | t tj f�rldj | �S t
| �S dS )z3Returns a succinct summary of an object as a string)r8 z{0:.4}N)�
isinstancer Z
datetime64r r5 � timedelta64r r9 r r �reprr& �formatr )r r8 Z
quote_stringsr r r � format_item[ s
r> c s� t j| �} d� t j| jt j�r�t j| dd�} | tj| � jd�jd�}| |k}|t jdd�k}t j|�j � rtd� nt j|�j � r�d� � fd d
�| D �}|S ) zBReturns a succinct summaries of all items in a sequence as stringsr ztimedelta64[ns])�dtypeztimedelta64[D]r �nsr7 r6 c s g | ]}t |� ��qS r )r> )r �xi)r8 r r �
<listcomp>y s z format_items.<locals>.<listcomp>)
r r/ Z
issubdtyper? r; r1 ZisnullZastypeZ logical_not�all)r Zday_partZ time_neededZ
day_needed� formattedr )r8 r � format_itemsi s
rE c C s� t ttj|d ��d�}t| |�}t|�}tjdd� |D ��d }|| jk sX||kj� r�d}t tj |t
|� |k�d�}|d|� }nd}dj |�| }|S ) z�Return a formatted string for as many items in the flattened version of
items_ndarray that will fit within max_width characters
g @r c S s g | ]}t |�d �qS )r )r
)r r r r r rB � s z%format_array_flat.<locals>.<listcomp>z ...Nr r ) �maxr$ r r% r0 rE Zcumsumr. �anyr# r
r ) Z
items_ndarray� max_widthZmax_possibly_relevantZrelevant_itemsZ pprint_itemsZcum_lenZ end_padding�countZ
pprint_strr r r �format_array_flat} s
rJ r c
C sx |d krt d }td|| f |�}|jr>ddjtt|j�� nd}|| d|j }|rlt||t|� �} nd} || S )NZ
display_widthz %s %s z(%s) z, r z%s z...) r r �dimsr �mapr r? rJ r
)
�name�var� col_width� show_values�markerrH Z first_colZdims_strZ front_strZ
values_strr r r �_summarize_var_or_coord� s rR c C s* | j jd�}|r&|jd�r&| j r&dS dS )zhHelper function to identify if array is positively identifiable as
coming from a remote source.
�source�httpFT)�encoding�get�
startswith�
_in_memory)rN rS r r r � _not_remote� s rY c C s t |�}t| |||�S )N)rY rR )rM rN rO rP r r r �
summarize_var� s rZ c C s2 | |j k}|pt|�}|r dnd}t| ||||�S )N�*r )rK rY rR )rM rN rO Zis_indexrP rQ r r r �summarize_coord� s
r\ �� c C s, t | �}t|�|kr(|d |d � d }|S )Nr
z...)r r
)�obj�maxlenr r r r �_maybe_truncate� s r` c C s d| t |�f S )Nz
%s: %s)r` )�key�valuerO r r r �summarize_attr� s rc z *empty*c C s, | rt dd� | D ��nd}t |d�d }|S )Nc s s | ]}t t|��V qd S )N)r
r )r �kr r r r � s z'_calculate_col_width.<locals>.<genexpr>r � � )rF )�mappingZmax_name_lengthrO r r r �_calculate_col_width� s rh c sP � d krt | �� d| g}| r<|� �fdd�| j� D �7 }n
|tg7 }dj|�S )Nz%s:c s g | ]\}}�||� ��qS r r )r rd �v)rO �
summarizerr r rB � s z!_mapping_repr.<locals>.<listcomp>r )rh �items�
EMPTY_REPRr )rg �titlerj rO �summaryr )rO rj r �
_mapping_repr� s
ro Z Coordinates)rm rj zData variablesZ
Attributesc C s< g }x,| j � D ] \}}|jtt|�d| �� qW dj|�S )Nz%s: r )rk �appendr r< r )�indexesrn rd ri r r r � indexes_repr� s rr c C s� t | d�r | jd k r d| j }nd}djdd� t| j| j�D ��}dt| �j||f g}tt | d| �j
t �r~|j t
| j�� n8| js�| jd k r�|j t
| j�� n|j d
| j| jf � t | d �r�| jr�|j t
| j�� | jr�|j t| j�� d j|�S )
NrM z%r r z, c s s | ]\}}d ||f V qdS )z%s: %sNr )r rd ri r r r r � s z array_repr.<locals>.<genexpr>z<xarray.%s %s(%s)>�variableg j�@z[%s values with dtype=%s]�coordsr )�hasattrrM r �ziprK r( �type�__name__r: �getattr�_datar rp r< �datarX r. �valuesr? rt �attrs�
attrs_repr)ZarrZname_strZ dim_summaryrn r r r �
array_repr� s$
r c C s� dt | �j g}t| �}td|�}dd� t| j�D �}|jd|dj|�f � |jt| j |d�� |jt
| j |d�� | j r�|jt
| j �� dj|�S ) Nz <xarray.%s>z Dimensions:c S s g | ]\}}d ||f �qS )z%s: %sr )r rd ri r r r rB s z dataset_repr.<locals>.<listcomp>z%s(%s)z, )rO r )rw rx rh r r rK rp r � coords_reprrt � vars_reprZ data_varsr} r~ )Zdsrn rO Z
dims_startZall_dim_stringsr r r � dataset_repr s
r� )r N)N)NT)Tr N)r] )N)N)'r r � functoolsZnumpyr Zpandasr1 �optionsr Zpycompatr r r r r r r, r0 r5 r9 r> rE rJ rR rY rZ r\ r` rc rl rh ro �partialr� r� r~ rr r r� r r r r �<module> sB
|