make the preprocessor a little faster

This reduced the preprocessing time from 8s to 2.7s.

1 files changed, 8 insertions, 5 deletions

diff --git a/preprocessor.py b/preprocessor.py
index 6b659443b..abb672822 100644
--- a/preprocessor.py
+++ b/preprocessor.py
@@ -393,6 +393,10 @@ def extract_token(asm):
     token = asm.split(" ")[0].replace("\t", "").replace("\n", "")
     return token
 
+def make_macro_table():
+    return dict([(macro.macro_name, macro) for macro in macros])
+macro_table = make_macro_table()
+
 def macro_test(asm):
     """ Returns a matching macro, or None/False.
     """
@@ -401,11 +405,10 @@ def macro_test(asm):
     token = extract_token(asm)
 
     # check against all names
-    for macro in macros:
-        if macro.macro_name == token:
-            return macro, token
-
-    return None, None
+    try:
+        return (macro_table[token], token)
+    except:
+        return (None, None)
 
 def macro_translator(macro, token, line):
     """ Converts a line with a macro into a rgbasm-compatible line.