compiler: Add List.inline for aesthetics

Author: FabioLuporini

devito/ir/iet/nodes.py

@@ -195,7 +195,7 @@ class List(Node):
 
     _traversable = ['body']
 
-    def __init__(self, header=None, body=None, footer=None):
+    def __init__(self, header=None, body=None, footer=None, inline=False):
         body = as_tuple(body)
         if len(body) == 1 and all(type(i) is List for i in [self, body[0]]):
             # De-nest Lists
@@ -213,6 +213,8 @@ def __init__(self, header=None, body=None, footer=None):
             self.body = as_tuple(body)
             self.footer = as_tuple(footer)
 
+        self.inline = inline
+
     def __repr__(self):
         return "<%s (%d, %d, %d)>" % (self.__class__.__name__, len(self.header),
                                       len(self.body), len(self.footer))

devito/ir/iet/visitors.py

@@ -538,7 +538,11 @@ def visit_Block(self, o):
 
     def visit_List(self, o):
         body = flatten(self._visit(i) for i in self._blankline_logic(o.children))
-        return c.Module(o.header + (c.Collection(body),) + o.footer)
+        if o.inline:
+            body = c.Line(' '.join(str(i) for i in body))
+        else:
+            body = c.Collection(body)
+        return c.Module(o.header + (body,) + o.footer)
 
     def visit_Section(self, o):
         body = flatten(self._visit(i) for i in o.children)

tests/test_iet.py

@@ -488,3 +488,11 @@ class MyArray(Array):
     a = MyArray(name='a', dimensions=dim, scope='shared', dtype=np.uint8)
 
     assert str(Definition(a)) == "extern  unsigned char a[];"
+
+
+def test_list_inline():
+    expr0 = DummyExpr(Symbol(name='a'), 1)
+    expr1 = DummyExpr(Symbol(name='b'), 2)
+
+    lst = List(body=[expr0, expr1], inline=True)
+    assert str(lst) == """a = 1; b = 2;"""