diff options
author | Bryan Bishop <kanzure@gmail.com> | 2013-08-31 10:13:17 -0500 |
---|---|---|
committer | Bryan Bishop <kanzure@gmail.com> | 2013-08-31 10:13:17 -0500 |
commit | 5815edf382d586e84db8d08612d5f4d1c2e7ac96 (patch) | |
tree | 54c66f23f49e28b4bdc0ff5b831403e7181762d0 /preprocessor.py | |
parent | 63c2dc2f1ffb3ec93f8fb6b97b0b909c36ecc0bf (diff) |
refactor preprocessor macros into a function
Remove the "macros" global and instead use a function to construct a
list of macros.
Diffstat (limited to 'preprocessor.py')
-rw-r--r-- | preprocessor.py | 36 |
1 files changed, 22 insertions, 14 deletions
diff --git a/preprocessor.py b/preprocessor.py index b6bdfbe7d..e69089c18 100644 --- a/preprocessor.py +++ b/preprocessor.py @@ -16,20 +16,28 @@ from extras.pokemontools.crystal import ( effect_classes, ) -even_more_macros = [ - Warp, - XYTrigger, - Signpost, - PeopleEvent, - DataByteWordMacro, -] +def load_pokecrystal_macros(): + """ + Construct a list of macros that are needed for pokecrystal preprocessing. + """ + ourmacros = [] + + even_more_macros = [ + Warp, + XYTrigger, + Signpost, + PeopleEvent, + DataByteWordMacro, + ] + + ourmacros += command_classes + ourmacros += even_more_macros + ourmacros += [each[1] for each in text_command_classes] + ourmacros += movement_command_classes + ourmacros += music_classes + ourmacros += effect_classes -macros = command_classes -macros += even_more_macros -macros += [each[1] for each in text_command_classes] -macros += movement_command_classes -macros += music_classes -macros += effect_classes + return ourmacros # show lines before preprocessing in stdout show_original_lines = False @@ -628,4 +636,4 @@ def preprocess(macros, lines=None): # only run against stdin when not included as a module if __name__ == "__main__": - preprocess(macros) + preprocess(load_pokecrystal_macros()) |