improvements in section

ffcx/codegeneration/definitions.py

@@ -146,6 +146,9 @@ def coefficient(self, mt, tabledata, quadrature_rule, access):
         code += [L.VariableDecl(access, 0.0)]
         code += [L.create_nested_for_loops([ic], body)]
 
+        code = L.Section("Coefficient definition", code)
+        pre_code = L.Section("Coefficient pre definition", pre_code)
+
         return pre_code, code
 
     def _define_coordinate_dofs_lincomb(self, mt, tabledata, quadrature_rule, access):

ffcx/codegeneration/integral_generator.py

@@ -301,6 +301,13 @@ def generate_partition(self, symbol, F, mode, quadrature_rule):
                     # Backend specific modified terminal translation
                     vaccess = self.backend.access.get(mt, tabledata, quadrature_rule)
                     predef, vdef = self.backend.definitions.get(mt, tabledata, quadrature_rule, vaccess)
+
+                    if isinstance(predef, L.Section):
+                        predef = predef.statements
+
+                    if isinstance(vdef, L.Section):
+                        vdef = vdef.statements
+
                     if predef:
                         access = predef[0].symbol.name
                         pre_definitions[str(access)] = predef

ffcx/codegeneration/lnodes.py

@@ -728,16 +728,17 @@ def as_statement(node):
 class Section(LNode):
     """A section of code with a name and a list of statements."""
 
-    def __init__(self, name, statements):
+    def __init__(self, name: str, statements: LExpr, annotations: List[str] = None):
         self.name = name
         self.statements = [as_statement(st) for st in statements]
+        self.annotations = annotations or []
 
     def __eq__(self, other):
         return isinstance(other, type(self)) and self.name == other.name and self.statements == other.statements
 
 
 class StatementList(LNode):
-    """A simple sequence of statements. No new scopes are introduced."""
+    """A simple sequence of statements."""
 
     def __init__(self, statements):
         self.statements = [as_statement(st) for st in statements]