o
    \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ÚencodeÚGetÚpackerÚuoffsetr   ÚInit)ÚclsÚbufÚoffsetÚnÚx© r   úx/home/air/segue/gemini/backup/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   ÚOffsetÚVectorÚIndirectÚ5onnxruntime.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   Ú	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é   )ÚStartObject©Úbuilderr   r   r   ÚStart9   ó   r5   c                 C   ó   t | ƒS ©z2This method is deprecated. Please switch to Start.)r5   r3   r   r   r   ÚTrtTableStart=   ó   r9   c                 C   s   |   dtjj |¡d¡ d S )Nr   )ÚPrependUOffsetTRelativeSlotr   r   r   r   ©r4   Údictr   r   r   ÚAddDictB   s   r>   c                 C   ó
   t | |ƒS )z4This method is deprecated. Please switch to AddDict.)r>   r<   r   r   r   ÚTrtTableAddDictF   ó   
r@   c                 C   s   |   d|d¡S )Nr   )ÚStartVector©r4   ÚnumElemsr   r   r   ÚStartDictVectorK   r6   rE   c                 C   r?   r8   )rE   rC   r   r   r   ÚTrtTableStartDictVectorO   rA   rF   c                 C   s   |   ¡ S r   )Ú	EndObjectr3   r   r   r   ÚEndT   s   rH   c                 C   r7   )z0This method is deprecated. Please switch to End.)rH   r3   r   r   r   ÚTrtTableEndX   r:   rI   )r   Úflatbuffers.compatr   Únpr   r5   r9   r>   r@   rE   rF   rH   rI   r   r   r   r   Ú<module>   s   .