shithub: pokecrystal

Info  •  Files  •  Log  •  Branches

make the preprocessor a little faster

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

--- a/preprocessor.py

+++ b/preprocessor.py

@@ -393,6 +393,10 @@

     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 @@

     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.