summaryrefslogtreecommitdiff
path: root/preprocessor.py
diff options
context:
space:
mode:
authorBryan Bishop <kanzure@gmail.com>2013-08-31 10:13:17 -0500
committerBryan Bishop <kanzure@gmail.com>2013-08-31 10:13:17 -0500
commit5815edf382d586e84db8d08612d5f4d1c2e7ac96 (patch)
tree54c66f23f49e28b4bdc0ff5b831403e7181762d0 /preprocessor.py
parent63c2dc2f1ffb3ec93f8fb6b97b0b909c36ecc0bf (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.py36
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())