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                   @   sF   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S )KeyValue_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/KeyValue.py	GetRootAs   s   zKeyValue.GetRootAsc                 C   s   |  ||S )z6This method is deprecated. Please switch to GetRootAs.)r   )r   r   r   r   r   r   GetRootAsKeyValue   s   zKeyValue.GetRootAsKeyValuec                 C   s   t j||| _d S r   )r   tableTabler   )selfr   posr   r   r   r      s   zKeyValue.Initc                 C   6   t jj| jd}|dkr| j|| jj S d S )N   r   r   number_typesUOffsetTFlagspy_typer   OffsetStringPosr   or   r   r   Key      zKeyValue.Keyc                 C   r   )N   r   r   r"   r   r   r   Value&   r%   zKeyValue.ValueN)r   )
__name__
__module____qualname__	__slots__classmethodr   r   r   r$   r'   r   r   r   r   r      s    r   c                 C   s   |  d d S )N   )StartObjectbuilderr   r   r   Start-   s   r1   c                 C      t | S )z2This method is deprecated. Please switch to Start.)r1   r/   r   r   r   KeyValueStart1      r3   c                 C   s   |  dtjj|d d S )Nr   PrependUOffsetTRelativeSlotr   r   r   r   r0   keyr   r   r   AddKey6      r9   c                 C   
   t | |S )z3This method is deprecated. Please switch to AddKey.)r9   r7   r   r   r   KeyValueAddKey:      
r<   c                 C   s   |  dtjj|d d S )N   r   r5   r0   valuer   r   r   AddValue?   r:   rA   c                 C   r;   )z5This method is deprecated. Please switch to AddValue.)rA   r?   r   r   r   KeyValueAddValueC   r=   rB   c                 C   s   |   S r   )	EndObjectr/   r   r   r   EndH   s   rD   c                 C   r2   )z0This method is deprecated. Please switch to End.)rD   r/   r   r   r   KeyValueEndL   r4   rE   )r   flatbuffers.compatr   npr   r1   r3   r9   r<   rA   rB   rD   rE   r   r   r   r   <module>   s   "