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 )ÚNormalÚ_tabc             C   s.   t j t jj||¡}tƒ }| ||| ¡ |S )N)ÚflatbuffersÚencodeÚGetÚpackerÚuoffsetr   ÚInit)ÚclsÚbufÚoffsetÚnÚx© r   úU/global/project/projectdirs/dasrepo/etalumis/pyprob_master_mar20/pyprob/ppx/Normal.pyÚGetRootAsNormal
   s    zNormal.GetRootAsNormalc             C   s   t j ||¡| _d S )N)r   ÚtableÚTabler   )Úselfr   Úposr   r   r   r	      s    zNormal.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   ÚMean   s    zNormal.Meanc             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   ÚStddev!   s    zNormal.StddevN)	Ú__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   ÚNormalStart+   s    r-   c             C   s   |   dtjj |¡d¡ d S )Nr   )ÚPrependUOffsetTRelativeSlotr   r   r   r   )r,   Úmeanr   r   r   ÚNormalAddMean,   s    r0   c             C   s   |   dtjj |¡d¡ d S )Nr   r   )r.   r   r   r   r   )r,   Ústddevr   r   r   ÚNormalAddStddev-   s    r2   c             C   s   |   ¡ S )N)Ú	EndObject)r,   r   r   r   Ú	NormalEnd.   s    r4   )r   Úobjectr   r-   r0   r2   r4   r   r   r   r   Ú<module>   s
   $