B
    áÉ’\Ý  ã               @   s<   d dl Z G dd„ deƒZdd„ Zdd„ Zdd	„ Zd
d„ ZdS )é    Nc               @   s6   e Zd ZdgZedd„ ƒZdd„ Zdd„ Zdd	„ Zd
S )ÚUniformÚ_tabc             C   s.   t j t jj||¡}tƒ }| ||| ¡ |S )N)ÚflatbuffersÚencodeÚGetÚpackerÚuoffsetr   ÚInit)ÚclsÚbufÚoffsetÚnÚx© r   úV/global/project/projectdirs/dasrepo/etalumis/pyprob_master_mar20/pyprob/ppx/Uniform.pyÚGetRootAsUniform
   s    zUniform.GetRootAsUniformc             C   s   t j ||¡| _d S )N)r   ÚtableÚTabler   )Úselfr   Úposr   r   r   r	      s    zUniform.Initc             C   s\   t jj | j d¡¡}|dkrX| j || jj ¡}ddlm} |ƒ }| 	| jj
|¡ |S d S )Né   r   é   )ÚTensor)r   Únumber_typesÚUOffsetTFlagsÚpy_typer   ÚOffsetÚIndirectÚPosr   r	   ÚBytes)r   Úor   r   Úobjr   r   r   ÚLow   s    zUniform.Lowc             C   s\   t jj | j d¡¡}|dkrX| j || jj ¡}ddlm} |ƒ }| 	| jj
|¡ |S d S )Né   r   r   )r   )r   r   r   r   r   r   r   r   r   r	   r   )r   r    r   r   r!   r   r   r   ÚHigh!   s    zUniform.HighN)	Ú__name__Ú
__module__Ú__qualname__Ú	__slots__Úclassmethodr   r	   r"   r$   r   r   r   r   r      s
   r   c             C   s   |   d¡ d S )Né   )ÚStartObject)Úbuilderr   r   r   ÚUniformStart+   s    r-   c             C   s   |   dtjj |¡d¡ d S )Nr   )ÚPrependUOffsetTRelativeSlotr   r   r   r   )r,   Úlowr   r   r   ÚUniformAddLow,   s    r0   c             C   s   |   dtjj |¡d¡ d S )Nr   r   )r.   r   r   r   r   )r,   Úhighr   r   r   ÚUniformAddHigh-   s    r2   c             C   s   |   ¡ S )N)Ú	EndObject)r,   r   r   r   Ú
UniformEnd.   s    r4   )r   Úobjectr   r-   r0   r2   r4   r   r   r   r   Ú<module>   s
   $