diff options
Diffstat (limited to 'AsmParser.py')
| -rw-r--r-- | AsmParser.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/AsmParser.py b/AsmParser.py index 4ab25ed..24755f6 100644 --- a/AsmParser.py +++ b/AsmParser.py @@ -29,7 +29,7 @@ class AsmParser(object): self.defined_names = parent.defined_names else: self.parent = None - self.data = [] + self.data = {} self.code = [] # defined labels for the assembly files self.labels = [] @@ -99,7 +99,7 @@ class AsmParser(object): if what == "DS": # DS initializes names with zero, let's convert it to # DW to make it compatible with the Parser - self.addData(name, [0 for x in range(0, int(data))]) + self.addData(name, ["0"] * int(data)) else: self.addData(name, data.split(",")) line = "" @@ -166,4 +166,7 @@ class AsmParser(object): if self.parent: raise RuntimeError("You can only get the lines for @CODE from the" " root node") - return [elm for elm in self.code] + return [str(elm) for elm in self.code] + def getDataDefinitions(self): + """Returns the data section as a dictionary""" + return self.data
\ No newline at end of file |