U
    /e                     @  s   d dl mZ d dlZd dlmZmZmZmZ eej	dd Z
eej	dd Zeejdd	 Zeejd
d Zeejdd Zeejdd ZdS )    )annotationsN)dask_deserializedask_serializedeserialize	serializec                 C  s   d|   ig fS )Nfilename)filepath)Zds r	   @/tmp/pip-unpacked-wheel-g426oqom/distributed/protocol/netcdf4.pyserialize_netcdf4_dataset   s    r   c                 C  s   t j| d ddS )Nr   r)mode)netCDF4Dataset)headerframesr	   r	   r
   deserialize_netcdf4_dataset   s    r   c                 C  s:   t |  \}}|d |d< |d |d< | j|d< |g fS )Ntypeparent-typetype-serializedparent-type-serializedname)r   groupr   )xr   _r	   r	   r
   serialize_netcdf4_variable   s
    
r   c                 C  s0   | d | d< | d | d< t | |}|j| d  S )Nr   r   r   r   r   )r   	variables)r   r   parentr	   r	   r
   deserialize_netcdf4_variable!   s    
r   c                 C  s0   | }|j r|j }qt|\}}| j|d< |g fS Npath)r   r   r    )gr   r   r   r	   r	   r
   serialize_netcdf4_group)   s    
r"   c                 C  s   t | |}|| d  S r   )r   )r   r   filer	   r	   r
   deserialize_netcdf4_group3   s    
r$   )
__future__r   r   Zdistributed.protocol.serializer   r   r   r   registerr   r   r   Variabler   r   Groupr"   r$   r	   r	   r	   r
   <module>   s   









	
