diff options
| -rw-r--r-- | pokemontools/wram.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/pokemontools/wram.py b/pokemontools/wram.py index 6d2c2cc..80244e3 100644 --- a/pokemontools/wram.py +++ b/pokemontools/wram.py @@ -112,6 +112,8 @@ class WRAMProcessor(object): self.setup_hram_constants() self.setup_gbhw_constants() + self.reformat_wram_labels() + def read_wram_sections(self): """ Opens the wram file and calls read_bss_sections. @@ -166,3 +168,14 @@ class WRAMProcessor(object): """ self.gbhw_constants = self.read_gbhw_constants() return self.gbhw_constants + + def reformat_wram_labels(self): + """ + Flips the wram_constants dictionary the other way around to access + addresses by label. + """ + self.wram = {} + + for (address, labels) in self.wram_constants.iteritems(): + for label in labels: + self.wram[label] = address |