| 26,26 → 26,39 |
|---|
| # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| # |
| """ |
| Convert descriptive structure definitions to struct formats |
| Convert descriptive structure definitions to structure object |
| """ |
| |
| import struct |
| |
| def convert(definition): |
| "Convert structure defition to struct format" |
| class Struct: |
| def size(self): |
| return struct.calcsize(self._format_) |
| |
| def pack(self): |
| list = [] |
| for variable in self._list_: |
| list.append(self.__dict__[variable]) |
| |
| return struct.pack(self._format_, *list) |
| |
| def create(definition): |
| "Create structure object" |
| |
| tokens = definition.split(None) |
| |
| # Initial byte order tag |
| struct = { |
| format = { |
| "little:": lambda: "<", |
| "big:": lambda: ">", |
| "network:": lambda: "!" |
| }[tokens[0]]() |
| inst = Struct() |
| list = [] |
| |
| # Member tags |
| |
| comment = False |
| variable = False |
| for token in tokens[1:]: |
| if (comment): |
| if (token == "*/"): |
| 52,13 → 65,23 |
|---|
| comment = False |
| continue |
| |
| if (variable): |
| inst.__dict__[token] = None |
| list.append(token) |
| variable = False |
| continue |
| |
| if (token == "/*"): |
| comment = True |
| elif (token[0:8] == "padding["): |
| size = token[8:].split("]")[0] |
| format += "%dx" % int(size) |
| elif (token[0:5] == "char["): |
| size = token[5:].split("]")[0] |
| struct += ("%d" % int(size)) + "s" |
| format += "%ds" % int(size) |
| variable = True |
| else: |
| struct += { |
| format += { |
| "uint8_t": lambda: "B", |
| "uint16_t": lambda: "H", |
| "uint32_t": lambda: "L", |
| 69,11 → 92,8 |
|---|
| "int32_t": lambda: "l", |
| "int64_t": lambda: "q" |
| }[token]() |
| variable = True |
| |
| return struct |
| |
| def little_string(string): |
| return struct.pack("<" + ("%d" % len(string)) + "s", string) |
| |
| def little_padding(length): |
| return struct.pack("<" + ("%d" % length) + "x") |
| inst.__dict__['_format_'] = format |
| inst.__dict__['_list_'] = list |
| return inst |