U
    “±Ëh
  ã                   @   sl   d dl Z d dlmZ eƒ ZG dd„ dƒZdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ ZdS )é    N)Úimport_numpyc                   @   sN   e Zd ZdgZeddd„ƒZeddd„ƒZdd„ Zd	d
„ Zdd„ Z	dd„ Z
dS )ÚTrtTableÚ_tabr   c                 C   s.   t j t jj||¡}tƒ }| ||| ¡ |S ©N)ÚflatbuffersÚencodeZGetÚpackerZuoffsetr   ÚInit)ÚclsÚbufÚoffsetÚnÚx© r   úY/tmp/pip-unpacked-wheel-socb9apf/onnxruntime/quantization/CalTableFlatBuffers/TrtTable.pyÚ	GetRootAs   s    zTrtTable.GetRootAsc                 C   s   |   ||¡S )z6This method is deprecated. Please switch to GetRootAs.)r   )r
   r   r   r   r   r   ÚGetRootAsTrtTable   s    zTrtTable.GetRootAsTrtTablec                 C   s   t j ||¡| _d S r   )r   ÚtableZTabler   )Úselfr   Úposr   r   r   r	      s    zTrtTable.Initc                 C   sv   t jj | j d¡¡}|dkrr| j |¡}|t jj |¡d 7 }| j |¡}ddlm	} |ƒ }| 
| jj|¡ |S d S )Né   r   )ÚKeyValue)r   Únumber_typesÚUOffsetTFlagsÚpy_typer   ÚOffsetZVectorZIndirectZ5onnxruntime.quantization.CalTableFlatBuffers.KeyValuer   r	   ZBytes)r   ÚjÚor   r   Úobjr   r   r   ÚDict   s    zTrtTable.Dictc                 C   s.   t jj | j d¡¡}|dkr*| j |¡S dS ©Nr   r   )r   r   r   r   r   r   Z	VectorLen©r   r   r   r   r   Ú
DictLength-   s    zTrtTable.DictLengthc                 C   s   t jj | j d¡¡}|dkS r    )r   r   r   r   r   r   r!   r   r   r   Ú
DictIsNone4   s    zTrtTable.DictIsNoneN)r   )r   )Ú__name__Ú
__module__Ú__qualname__Ú	__slots__Úclassmethodr   r   r	   r   r"   r#   r   r   r   r   r      s   r   c                 C   s   |   d¡ d S )Né   )ZStartObject©Úbuilderr   r   r   ÚStart9   s    r,   c                 C   s   t | ƒS ©z2This method is deprecated. Please switch to Start.)r,   r*   r   r   r   ÚTrtTableStart=   s    r.   c                 C   s   |   dtjj |¡d¡ d S )Nr   )ZPrependUOffsetTRelativeSlotr   r   r   r   ©r+   Údictr   r   r   ÚAddDictB   s    r1   c                 C   s
   t | |ƒS )z4This method is deprecated. Please switch to AddDict.)r1   r/   r   r   r   ÚTrtTableAddDictF   s    r2   c                 C   s   |   d|d¡S )Nr   )ZStartVector©r+   ZnumElemsr   r   r   ÚStartDictVectorK   s    r4   c                 C   s
   t | |ƒS r-   )r4   r3   r   r   r   ÚTrtTableStartDictVectorO   s    r5   c                 C   s   |   ¡ S r   )Z	EndObjectr*   r   r   r   ÚEndT   s    r6   c                 C   s   t | ƒS )z0This method is deprecated. Please switch to End.)r6   r*   r   r   r   ÚTrtTableEndX   s    r7   )r   Zflatbuffers.compatr   Únpr   r,   r.   r1   r2   r4   r5   r6   r7   r   r   r   r   Ú<module>   s   .