o 2ha@s ddlZddlmZddlmZddlmZddlmZddlmZddlmZm Z dd lm Z dd lm Z m Z m Z e ZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZGdddeZdZGdddeZddZdS)N)compat)encode) number_types)packer)memoryview_type)NumpyRequiredForThisFeature import_numpy) range_func) SOffsetTFlags UOffsetTFlags VOffsetTFlagsc@eZdZdZdS)OffsetArithmeticErrorzError caused by an Offset arithmetic error. Probably caused by bad writing of fields. This is considered an unreachable situation in normal circumstances. N__name__ __module__ __qualname____doc__rrN/var/www/html/chatgem/venv/lib/python3.10/site-packages/flatbuffers/builder.pyr!src@r)IsNotNestedErrorzUError caused by using a Builder to write Object data when not inside an Object. Nrrrrrr+rc@r) IsNestedErrorz_Error caused by using a Builder to begin an Object when an Object is already being built. Nrrrrrr4rrc@r)StructIsNotInlineErrorzeError caused by using a Builder to write a Struct at a location that is not the current Offset. Nrrrrrr=rrc@r)BuilderSizeErrorzUError caused by causing a Builder to exceed the hardcoded limit of 2 gigabytes. NrrrrrrFrrc@r)BuilderNotFinishedErrorz=Error caused by not calling `Finish` before calling `Output`.NrrrrrrOsrc@r)EndVectorLengthMismatchedzdThe number of elements passed to EndVector does not match the number specified in StartVector. NrrrrrrUrrc@s eZdZdZdZ dZdddZdd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zdd#d$Zdd'd(Zdd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zdd7d8Zdd9d:Zdd;d<Zd=d>Z d?d@Z!dAdBZ"dCdDZ#dEdFZ$dGdHZ%dIdJZ&dKdLZ'dMdNZ(dOdPZ)dQdRZ*dSdTZ+dUdVZ,dWdXZ-dYdZZ.d[d\Z/d]d^Z0d_d`Z1dadbZ2dcddZ3dedfZ4dgdhZ5didjZ6dkdlZ7dmdnZ8dodpZ9dqdrZ:dsdtZ;dudvZd{d|Z?d}d~Z@ddZAdS)Buildera|A Builder is used to construct one or more FlatBuffers. Typically, Builder objects will be used from code generated by the `flatc` compiler. A Builder constructs byte buffers in a last-first manner for simplicity and performance during reading. Internally, a Builder is a state machine for creating FlatBuffer objects. It holds the following internal state: - Bytes: an array of bytes. - current_vtable: a list of integers. - vtables: a hash of vtable entries. Attributes: Bytes: The internal `bytearray` for the Builder. finished: A boolean determining if the Builder has been finalized. ) Bytescurrent_vtableheadminalign objectEndvtablesnested forceDefaultsfinishedvectorNumElems sharedStringslcCsnd|kr tjksnd}t|t||_d|_t||_d|_ d|_ i|_ d|_ d|_ i|_d|_dS)z_Initializes a Builder of size `initial_size`. The internal buffer is grown as needed. rz;flatbuffers: Cannot create Builder larger than 2 gigabytes.NrF)rMAX_BUFFER_SIZEr bytearrayr r!r py_typer"r#r$r%r&r'r*r()self initialSizemsgrrr__init__s   zBuilder.__init__returnNcCsLd|_tt|j|_d|_d|_i|_d|_ d|_ i|_ d|_ d|_ dS)NrF)r!r r.lenr r"r#r$r%r&r'r*r)r(r/rrrClears z Builder.ClearcCs|jst|j|dS)aReturn the portion of the buffer that has been used for writing data. This is the typical way to access the FlatBuffer data inside the builder. If you try to access `Builder.Bytes` directly, you would need to manually index it with `Head()`, since the buffer is constructed backwards. It raises BuilderNotFinishedError if the buffer has not been finished with `Finish`. N)r(rr Headr5rrrOutputs zBuilder.OutputcCs0|ddt|D|_||_d|_dS)z=StartObject initializes bookkeeping for writing a new object.cSsg|]}dqS)rr).0_rrr sz'Builder.StartObject..TN)assertNotNestedr r!Offsetr$r&)r/ numfieldsrrr StartObjects  zBuilder.StartObjectc Cs|d|}g}d}t|jD]}|dkr|rqn||}d}||qt|}|j|}|durt|jd}d}d}|dkrkd}|j|}|d8}|dkr\|r[|d7}qAn||}d}| ||dksEt ||j } | t | t|j|t} | tj j9} | t | t t|j|} ttj|j| t ||||j|<n!t t|j|} t | |_ttj|j|t ||d|_|S)aWriteVtable serializes the vtable for the current object, if needed. Before writing out the vtable, this checks pre-existing vtables for equality to this one. If an equal vtable is found, point the object to the existing vtable and return. Because vtable values are sensitive to alignment of object data, not all logically-equal vtables will be deduplicated. A vtable has the following format: * N, where N is the number of fields in the schema for this type. Includes deprecated fields. Thus, a vtable is made of 2 + N elements, each VOffsetT bytes wide. An object has the following format: + rTFNr)PrependSOffsetTRelativer=reversedr!appendtupler%getr4PrependVOffsetTr r.r$r VtableMetadataFieldsN bytewidthr r rWritersoffsetr"r7) r/ objectOffsetvtKeytrimelem vt2Offsetitrailingoff objectSizevBytes objectStartrrr WriteVtablesl         zBuilder.WriteVtablecCs|d|_|S)z>EndObject writes data necessary to finish object construction.F) assertNestedr&rVr5rrr EndObject<szBuilder.EndObjectcCsht|jtjkrd}t|tt|jdtj}|dkrd}t|}|j||t|jd<||_dS)zDoubles the size of the byteslice, and copies the old data towards the end of the new buffer (since we build the buffer backwards). z2flatbuffers: cannot grow buffer beyond 2 gigabytesrrrN)r4r rr,rminr-)r/r1newSizebytes2rrrgrowByteBufferBs zBuilder.growByteBuffercCs|jS)zGet the start of useful data in the underlying byte buffer. Note: unlike other functions, this value is interpreted as from the left. )r"r5rrrr7Tsz Builder.HeadcCstt|j|S)z)Offset relative to the end of the buffer.)r r.r4r r7r5rrrr=_szBuilder.OffsetcCs t|D] }|dtjqdS)z'Pad places zeros at the current offset.rN)r PlacerG Uint8Flags)r/nrPrrrPadcs z Builder.PadcCs||jkr||_t|j||d}||dM}||||krIt|j}||jt|j|}t||_||||ks&||dS)a4Prep prepares to write an element of `size` after `additional_bytes` have been written, e.g. if you write a string, you need to align such the int length field is aligned to SizeInt32, and the string data follows it directly. If all you need to do is align, `additionalBytes` will be 0. rN) r#r4r r7r\r"r r.r`)r/sizeadditionalBytes alignSize oldBufSize updated_headrrrPrephs    z Builder.PrepcCJ|tjjd||ksd}t|||tjj}||dS)z]PrependSOffsetTRelative prepends an SOffsetT, relative to where it will be written. r%flatbuffers: Offset arithmetic error.N)rfrGr rHr=r PlaceSOffsetTr/rRr1off2rrrr@  zBuilder.PrependSOffsetTRelativecCrg)z]Prepends an unsigned offset into vector data, relative to where it will be written. rrhN)rfrGr rHr=r PlaceUOffsetTrjrrrPrependUOffsetTRelativerlzBuilder.PrependUOffsetTRelativecCs@|d|_||_|tjj|||||||S)zStartVector initializes bookkeeping for writing a new vector. A vector has the following format: - - +, where T is the type of elements of this vector. T)r<r&r)rfrG Uint32FlagsrHr=)r/elemSizenumElems alignmentrrr StartVectors zBuilder.StartVectorcCsL|d|_|rtjdtdd||jkrt||jd|_|S)z>EndVector writes data necessary to finish vector construction.FznumElems is deprecated.r) stacklevelN) rWr&warningswarnDeprecationWarningr)rrmr=)r/rqrrr EndVectors  zBuilder.EndVectorutf-8strictcCs0||jvr |j|S||||}||j|<|S)zoCreateSharedString checks if the string is already written to the buffer before calling CreateString. )r* CreateString)r/sencodingerrorsrRrrrCreateSharedStrings   zBuilder.CreateSharedStringcCs|d|_t|tjr|||}n t|tjr|}ntd|t j j t |dt j j |dt j t t |}t |||_||j|||<t ||_|S)z>CreateString writes a null-terminated byte string as a vector.Tz!non-string passed to CreateStringrr)r<r& isinstancer string_typesr binary_types TypeErrorrfrGr rHr4r^r]r.r7r"r r)rx)r/r|r}r~xlrrrr{s    zBuilder.CreateStringcCs|d|_t|tjstd|tjj t |tj j t t |}t | ||_||j| | |<t ||_|S)z"CreateString writes a byte vector.Tz*non-byte vector passed to CreateByteVector)r<r&rrrrrfrGr rHr4r^r.r7r"r r)rx)r/rrrrrCreateByteVectors  zBuilder.CreateByteVectorcCstdurtdt|tjstd|jjdvrtd|jdkr%td||j |j |jj |jj dd kr;|}n|j d d }t|j |j }t|||_|jd d |j|||<|j |_|S)z7CreateNumpyVector writes a numpy array into the buffer.NzNumpy was not found.z-non-numpy-ndarray passed to CreateNumpyVector)brPufz4numpy-ndarray holds elements of unsupported datatyperz4multidimensional-ndarray passed to CreateNumpyVectorrBBBBrT)rGrr rH Int32Flagsrfr#rFILE_IDENTIFIER_LENGTHstructunpackranger]r^rnr4r r7 PrependInt32r()r/ rootTable sizePrefixfile_identifierprepSizerPrarrr__FinishPs&    zBuilder.__FinishcC|j|d|dS)rFr_Builder__Finishr/rrrrrFinishnszBuilder.FinishcCr)z_Finish finalizes a buffer, pointing to the given `rootTable`, with the size prefixed. TrrrrrrFinishSizePrefixedrszBuilder.FinishSizePrefixedcCs||jd|||dS)Nr)rfrHr])r/flagsrRrrrPrependzszBuilder.PrependcCs`|dur t|||durt||||ks|jr,|dur.|||||dSdSdSN)rGrr'rr)r/rordrrr PrependSlot~s   zBuilder.PrependSlotcG|jtjg|RdSr)rrG BoolFlagsr/argsrrrPrependBoolSlotzBuilder.PrependBoolSlotcGrrrrGr^rrrrPrependByteSlotrzBuilder.PrependByteSlotcGrrrrrrrPrependUint8SlotrzBuilder.PrependUint8SlotcGrr)rrG Uint16FlagsrrrrPrependUint16SlotrzBuilder.PrependUint16SlotcGrr)rrGrorrrrPrependUint32SlotrzBuilder.PrependUint32SlotcGrr)rrG Uint64FlagsrrrrPrependUint64SlotrzBuilder.PrependUint64SlotcGrr)rrG Int8FlagsrrrrPrependInt8SlotrzBuilder.PrependInt8SlotcGrr)rrG Int16FlagsrrrrPrependInt16SlotrzBuilder.PrependInt16SlotcGrr)rrGrrrrrPrependInt32SlotrzBuilder.PrependInt32SlotcGrr)rrG Int64FlagsrrrrPrependInt64SlotrzBuilder.PrependInt64SlotcGrr)rrG Float32FlagsrrrrPrependFloat32SlotrzBuilder.PrependFloat32SlotcGrr)rrG Float64FlagsrrrrPrependFloat64SlotrzBuilder.PrependFloat64SlotcCs*||ks|jr||||dSdS)zPrependUOffsetTRelativeSlot prepends an UOffsetT onto the object at vtable slot `o`. If value `x` equals default `d`, then the slot will be set to zero and no other data will be written. N)r'rnr)r/rrrrrrPrependUOffsetTRelativeSlots z#Builder.PrependUOffsetTRelativeSlotcCs2t|tj||kr||||dSdS)zPrependStructSlot prepends a struct onto the object at vtable slot `o`. Structs are stored inline, so nothing additional is being added. In generated code, `d` is always 0. N)rGrr rr)r/vrrrrrPrependStructSlots  zBuilder.PrependStructSlotcC|tj|dS)zTPrepend a `bool` to the Builder buffer. Note: aligns and checks for space. N)rrGrr/rrrr PrependBoolzBuilder.PrependBoolcCr)zTPrepend a `byte` to the Builder buffer. Note: aligns and checks for space. NrrGr^rrrr PrependByterzBuilder.PrependBytecCr)zVPrepend an `uint8` to the Builder buffer. Note: aligns and checks for space. Nrrrrr PrependUint8rzBuilder.PrependUint8cCr)zWPrepend an `uint16` to the Builder buffer. Note: aligns and checks for space. N)rrGrrrrr PrependUint16rzBuilder.PrependUint16cCr)zWPrepend an `uint32` to the Builder buffer. Note: aligns and checks for space. N)rrGrorrrr PrependUint32rzBuilder.PrependUint32cCr)zWPrepend an `uint64` to the Builder buffer. Note: aligns and checks for space. N)rrGrrrrr PrependUint64rzBuilder.PrependUint64cCr)zUPrepend an `int8` to the Builder buffer. Note: aligns and checks for space. N)rrGrrrrr PrependInt8rzBuilder.PrependInt8cCr)zVPrepend an `int16` to the Builder buffer. Note: aligns and checks for space. N)rrGrrrrr PrependInt16rzBuilder.PrependInt16cCr)zVPrepend an `int32` to the Builder buffer. Note: aligns and checks for space. N)rrGrrrrrrrzBuilder.PrependInt32cCr)zVPrepend an `int64` to the Builder buffer. Note: aligns and checks for space. N)rrGrrrrr PrependInt64rzBuilder.PrependInt64cCr)zWPrepend a `float32` to the Builder buffer. Note: aligns and checks for space. N)rrGrrrrrPrependFloat32 rzBuilder.PrependFloat32cCr)zWPrepend a `float64` to the Builder buffer. Note: aligns and checks for space. N)rrGrrrrrPrependFloat64rzBuilder.PrependFloat64cCs ||_dS)zIn order to save space, fields that are set to their default value don't get serialized into the buffer. Forcing defaults provides a way to manually disable this optimization. When set to `True`, will always serialize default values. N)r')r/r'rrr ForceDefaultss zBuilder.ForceDefaultscCs|tj|dSr)rrGr rrrrrE$szBuilder.PrependVOffsetTcCs6t|||j|j|_t|j|j||dS)zkPlace prepends a value specified by `flags` to the Builder, without checking for available space. N) rGrr"rHrrI packer_typer r7)r/rrrrrr]'s z Builder.PlacecC:t|tj|jtjj|_ttj|j | |dS)zWPlaceVOffsetT prepends a VOffsetT to the Builder, without checking for space. N) rGrr r"rHrrIrvoffsetr r7rrrr PlaceVOffsetT1zBuilder.PlaceVOffsetTcCr)zWPlaceSOffsetT prepends a SOffsetT to the Builder, without checking for space. N) rGrr r"rHrrIrrJr r7rrrrri:rzBuilder.PlaceSOffsetTcCr)zWPlaceUOffsetT prepends a UOffsetT to the Builder, without checking for space. N) rGrr r"rHrrIruoffsetr r7rrrrrmCrzBuilder.PlaceUOffsetT)r+)r3Nr)ryrz)Brrrr __slots__r,r2r6r8r?rVrXr\r7r=r`rfr@rnrsrxrr{rrrWr<rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrEr]rrirmrrrrrcs   l    %        rcCst|tjt|tjjt|krdSt|D]$\}}tt j ||tjj}|dkr2|dkr2q||}||kr=dSqdS)z=vtableEqual compares an unwritten vtable to a written vtable.FrT) rGrr r4r rH enumeraterGetrr)arUrrPrNryrrr vtableEqualPsr)rurrrrGrrrr r r r r r RuntimeErrorrrrrrrrrFobjectrrrrrrs2             r