From 1bf577b1c5e606ac2c0553857297ce8c0c04ccb7 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Sat, 26 Oct 2019 15:01:39 +0200
Subject: Support passing additional string arguments to the generator

---
 boltzgen/kernel/generator.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'boltzgen/kernel/generator.py')

diff --git a/boltzgen/kernel/generator.py b/boltzgen/kernel/generator.py
index 434d163..8d0de37 100644
--- a/boltzgen/kernel/generator.py
+++ b/boltzgen/kernel/generator.py
@@ -13,7 +13,7 @@ class Generator:
         self.collision  = collision
         self.boundary   = boundary
 
-    def kernel(self, target, precision, layout, geometry):
+    def kernel(self, target, precision, layout, geometry, extras = []):
         template_path = Path(__file__).parent/('template/basic.' + target + '.mako')
         if not template_path.exists():
             raise Exception("Target '%s' not supported" % target)
@@ -34,9 +34,9 @@ class Generator:
 
             moments_subexpr    = self.moments[0],
             moments_assignment = self.moments[1],
-
             collision_subexpr    = self.collision[0],
             collision_assignment = self.collision[1],
+            ccode = sympy.ccode,
 
             float_type = {
                 'single': 'float',
@@ -49,5 +49,5 @@ class Generator:
                 float_type = precision
             ),
 
-            ccode = sympy.ccode
+            extras = extras
         )
-- 
cgit v1.2.3