mirror of
https://github.com/google/flatbuffers.git
synced 2026-06-02 04:04:19 +00:00
Flatbuffers Python Object API (#5616)
* Flatbuffers Python Object API Implement the logic to generate the Python object API that can unpack the data from a buf class into an object class, and pack the data of an object class to a buf class. * Fix the build issues Remove unused parameters and replace auto in the for-loop statement with std::string to make it compatible with VS2010. * Fix the build issues. * Add support for Array type Added logic to handle Array type in Python Object API. Updated the generated code accordingly. * Fix the old style casting from int to char * Fixed another conversion from int to char * Fixed the import for typing Importing typing may cause errors when a machine do not have the moduel typing installed. This PR fixes the issue by guarding "import typing" with the "try/except" statement. * Fix issue of iterating the vector of import list * Update the generated examples using generate_code.sh * Fix the import order for typing The import list was stored in unordered_set, so that each generated codes may have different import order. Therefore, it failed in the consistency test where two generated copies need to have exactly the same apperance. * Optimize unpack using numpy Use numpy to unpack vector whenever it is possible to improve unpack performance. Also, added codegen command for Python specificly in generate_code.sh, because --no-includes cannot be turn on for Python. * Fix the import order * Update generate_code.bat for windows accordingly * Replace error message with pass Avoid printing error message for every Python2 users about typing. Replace it with pass.
This commit is contained in:
committed by
Wouter van Oortmerssen
parent
75823cc275
commit
65f8703572
@@ -3,6 +3,8 @@
|
||||
# namespace: Example
|
||||
|
||||
import flatbuffers
|
||||
from flatbuffers.compat import import_numpy
|
||||
np = import_numpy()
|
||||
|
||||
class ArrayTable(object):
|
||||
__slots__ = ['_tab']
|
||||
@@ -27,7 +29,7 @@ class ArrayTable(object):
|
||||
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
||||
if o != 0:
|
||||
x = o + self._tab.Pos
|
||||
from .ArrayStruct import ArrayStruct
|
||||
from MyGame.Example.ArrayStruct import ArrayStruct
|
||||
obj = ArrayStruct()
|
||||
obj.Init(self._tab.Bytes, x)
|
||||
return obj
|
||||
@@ -36,3 +38,43 @@ class ArrayTable(object):
|
||||
def ArrayTableStart(builder): builder.StartObject(1)
|
||||
def ArrayTableAddA(builder, a): builder.PrependStructSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(a), 0)
|
||||
def ArrayTableEnd(builder): return builder.EndObject()
|
||||
|
||||
import MyGame.Example.ArrayStruct
|
||||
try:
|
||||
from typing import Optional
|
||||
except:
|
||||
pass
|
||||
|
||||
class ArrayTableT(object):
|
||||
|
||||
# ArrayTableT
|
||||
def __init__(self):
|
||||
self.a = None # type: Optional[MyGame.Example.ArrayStruct.ArrayStructT]
|
||||
|
||||
@classmethod
|
||||
def InitFromBuf(cls, buf, pos):
|
||||
arrayTable = ArrayTable()
|
||||
arrayTable.Init(buf, pos)
|
||||
return cls.InitFromObj(arrayTable)
|
||||
|
||||
@classmethod
|
||||
def InitFromObj(cls, arrayTable):
|
||||
x = ArrayTableT()
|
||||
x._UnPack(arrayTable)
|
||||
return x
|
||||
|
||||
# ArrayTableT
|
||||
def _UnPack(self, arrayTable):
|
||||
if arrayTable is None:
|
||||
return
|
||||
if arrayTable.A() is not None:
|
||||
self.a = MyGame.Example.ArrayStruct.ArrayStructT.InitFromObj(arrayTable.A())
|
||||
|
||||
# ArrayTableT
|
||||
def Pack(self, builder):
|
||||
ArrayTableStart(builder)
|
||||
if self.a is not None:
|
||||
a = self.a.Pack(builder)
|
||||
ArrayTableAddA(builder, a)
|
||||
arrayTable = ArrayTableEnd(builder)
|
||||
return arrayTable
|
||||
|
||||
Reference in New Issue
Block a user