
    `i                     T    S SK r S SKJr  \" 5       r " S S\5      rS rS rS rS r	g)	    N)import_numpyc                   B    \ rS rSrS/r\S 5       rS rS rS r	S r
Srg	)
TrtTable
   _tabc                     [         R                  R                  [         R                  R                  X5      n[        5       nUR                  XU-   5        U$ N)flatbuffersencodeGetpackeruoffsetr   Init)clsbufoffsetnxs        z/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxruntime_tools/quantization/CalTableFlatBuffers/TrtTable.pyGetRootAsTrtTableTrtTable.GetRootAsTrtTable   sB    "";#5#5#=#=sKJ	sJ    c                 L    [         R                  R                  X5      U l        g r	   )r
   tableTabler   )selfr   poss      r   r   TrtTable.Init   s    %%++C5	r   c                    [         R                  R                  R                  U R                  R                  S5      5      nUS:w  a  U R                  R                  U5      nU[         R                  R                  R                  U5      S-  -  nU R                  R                  U5      nSSKJ	n  U" 5       nUR                  U R                  R                  U5        U$ g )N   r   )KeyValue)r
   number_typesUOffsetTFlagspy_typer   OffsetVectorIndirectCalTableFlatBuffers.KeyValuer!   r   Bytes)r   jor   r!   objs         r   DictTrtTable.Dict   s    $$22::499;K;KA;NO6		  #A))77??BQFFA		""1%A=*CHHTYY__a(Jr   c                     [         R                  R                  R                  U R                  R                  S5      5      nUS:w  a  U R                  R                  U5      $ gNr    r   )r
   r"   r#   r$   r   r%   	VectorLenr   r+   s     r   
DictLengthTrtTable.DictLength&   sL    $$22::499;K;KA;NO699&&q))r   c                     [         R                  R                  R                  U R                  R                  S5      5      nUS:H  $ r0   )r
   r"   r#   r$   r   r%   r2   s     r   
DictIsNoneTrtTable.DictIsNone-   s6    $$22::499;K;KA;NOAvr   )r   N)__name__
__module____qualname____firstlineno__	__slots__classmethodr   r   r-   r3   r6   __static_attributes__ r   r   r   r   
   s/    I 6
r   r   c                 &    U R                  S5        g )N   )StartObjectbuilders    r   TrtTableStartrE   2   s    r   c                 x    U R                  S[        R                  R                  R	                  U5      S5        g )Nr   )PrependUOffsetTRelativeSlotr
   r"   r#   r$   )rD   dicts     r   TrtTableAddDictrI   6   s-    '';+C+C+Q+Q+Y+YZ^+_abcr   c                 (    U R                  SUS5      $ )Nr    )StartVector)rD   numElemss     r   TrtTableStartDictVectorrM   :   s    q(A..r   c                 "    U R                  5       $ r	   )	EndObjectrC   s    r   TrtTableEndrP   >   s    r   )
r
   flatbuffers.compatr   npobjectr   rE   rI   rM   rP   r?   r   r   <module>rT      s6   
  +^%v %Pd/r   