3
���h� � @ sH d dl Z d dlZddlmZ ddlmZ ddlmZ G dd� de�Z dS ) � N� )�Variable)� OrderedDict� )�AbstractWritableDataStorec sL e Zd ZdZd� fdd� Zdd� Zdd� Zdd
d �Zd d
� Zdd� Z � Z
S )�InMemoryDataStorez�
Stores dimensions, variables and attributes in ordered dictionaries, making
this store fast compared to stores which save to disk.
This store exists purely for internal testing purposes.
Nc s< |d krt � n|| _|d kr"t � n|| _tt| �j|� d S )N)r �
_variables� _attributes�superr �__init__)�self� variables�
attributes�writer)� __class__� �6/tmp/pip-build-5_djhm0z/xray/xarray/backends/memory.pyr s zInMemoryDataStore.__init__c C s | j S )N)r )r r r r � get_attrs s zInMemoryDataStore.get_attrsc C s | j S )N)r )r r r r �
get_variables s z InMemoryDataStore.get_variablesFc C s8 t |jtj|�|j�}|jj|j� || j|< ||jfS )N) r Zdims�npZ
empty_like�attrs�update�encodingr �data)r �k�vZcheck_encodingZnew_varr r r �prepare_variable s
z"InMemoryDataStore.prepare_variablec C s t j|�| j|< d S )N)�copy�deepcopyr )r r r r r r �
set_attribute% s z InMemoryDataStore.set_attributec C s d S )Nr )r �d�lr r r �
set_dimension) s z InMemoryDataStore.set_dimension)NNN)F) �__name__�
__module__� __qualname__�__doc__r r r r r r" �
__classcell__r r )r r r s
r )
r Znumpyr Z
core.variabler Z
core.pycompatr �commonr r r r r r �<module> s
|