o
    
j 
                     @   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   h/home/kuhnn/.local/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Indirect5onnxruntime.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   .