Merge pull request #62 from slothy-optimizer/macros

Allow spaces instead of commas in macro definitions and invocations
slothy-optimizer · Jul 8, 2024 · c6930b8 · c6930b8
2 parents 5385c76 + 12aa032
commit c6930b8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+**/__pycache__
+venv/
diff --git a/slothy/helper.py b/slothy/helper.py
@@ -789,7 +789,7 @@ def unfold_in(self, source, change_callback=None, inherit_comments=False):
         for arg in self.args:
             arg_regexps.append(rf"\s*(?P<{arg}>[^,]+)\s*")
 
-        macro_regexp_txt += ','.join(arg_regexps)
+        macro_regexp_txt += '(,|\s)'.join(arg_regexps)
         macro_regexp = re.compile(macro_regexp_txt)
 
         output = []
@@ -875,7 +875,7 @@ def extract(source):
                 if cur.tags.get("no-unfold", None) is not None:
                     continue
 
-                current_args = [ a.strip() for a in p.group("args").split(',') ]
+                current_args = [ a.strip() for a in re.split(r'\s|\,', p.group("args")) if a.strip() != ""]
                 current_macro = p.group("name")
                 current_body = []