o
    ¢Äi 
  ã                   @   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   úo/home/kim/smarthome/.venv/lib/python3.10/site-packages/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   ÚtableÚTabler   )Úselfr   Úposr   r   r   r	      s   zTrtTable.Initc                 C   sv   t jj | j d¡¡}|dkr9| 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	   Ú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   )Ú__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   ó   r.   c                 C   ó   t | ƒS ©z2This method is deprecated. Please switch to Start.)r.   r,   r   r   r   ÚTrtTableStart=   ó   r2   c                 C   s   |   dtjj |¡d¡ d S )Nr   )ZPrependUOffsetTRelativeSlotr   r   r   r   ©r-   Údictr   r   r   ÚAddDictB   s   r6   c                 C   ó
   t | |ƒS )z4This method is deprecated. Please switch to AddDict.)r6   r4   r   r   r   ÚTrtTableAddDictF   ó   
r8   c                 C   s   |   d|d¡S )Nr   )ZStartVector©r-   ZnumElemsr   r   r   ÚStartDictVectorK   r/   r;   c                 C   r7   r1   )r;   r:   r   r   r   ÚTrtTableStartDictVectorO   r9   r<   c                 C   s   |   ¡ S r   )Z	EndObjectr,   r   r   r   ÚEndT   s   r=   c                 C   r0   )z0This method is deprecated. Please switch to End.)r=   r,   r   r   r   ÚTrtTableEndX   r3   r>   )r   Zflatbuffers.compatr   Únpr   r.   r2   r6   r8   r;   r<   r=   r>   r   r   r   r   Ú<module>   s   .