From f233beddfc74d5933f46684adab5298e03c08871 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Sat, 2 Nov 2019 17:29:56 +0100
Subject: Import `sympy.ccode` inside templates instead of as argument

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

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

diff --git a/boltzgen/kernel/generator.py b/boltzgen/kernel/generator.py
index 4fd3c25..8da91ba 100644
--- a/boltzgen/kernel/generator.py
+++ b/boltzgen/kernel/generator.py
@@ -1,5 +1,3 @@
-import sympy
-
 from mako.template import Template
 from pathlib import Path
 
@@ -33,12 +31,8 @@ class Generator:
             geometry   = geometry,
             index      = self.index_impl(geometry),
             layout     = self.layout_impl(self.descriptor, self.index_impl, geometry),
-
-            ccode = sympy.ccode,
-
             float_type = self.float_type,
-
-            extras = extras
+            extras     = extras
         )
 
     def kernel(self, geometry, functions, extras = []):
@@ -47,15 +41,13 @@ class Generator:
 
         return "\n".join(map(lambda f: self.instantiate(f, geometry, extras), functions))
 
-    def custom(self, geometry, source):
+    def custom(self, geometry, source, extras = []):
         return Template(text = source).render(
             descriptor = self.descriptor,
             model      = self.model,
             geometry   = geometry,
             index      = self.index_impl(geometry),
             layout     = self.layout_impl(self.descriptor, self.index_impl, geometry),
-
-            ccode = sympy.ccode,
-
             float_type = self.float_type,
+            extras     = extras
         )
-- 
cgit v1.2.3