From f2fd544e20a54aece3bb1699e2b4c6a5ff0d4078 Mon Sep 17 00:00:00 2001
From: Bryan Bishop <kanzure@gmail.com>
Date: Thu, 10 Jan 2013 15:15:59 -0600
Subject: [PATCH] make the preprocessor a little faster

This reduced the preprocessing time from 8s to 2.7s.
---
 preprocessor.py | 13 ++++++++-----
 1 file 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.