From c27ee89f265f73fa77ec043e4f2a52c5c3859965 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Tue, 22 Oct 2019 10:49:44 +0200
Subject: Add basic Generator class

---
 boltzgen.py | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

(limited to 'boltzgen.py')

diff --git a/boltzgen.py b/boltzgen.py
index 6d410c3..27a34fa 100644
--- a/boltzgen.py
+++ b/boltzgen.py
@@ -9,16 +9,12 @@ argparser.add_argument('language', help = 'Target language (currently either "op
 args = argparser.parse_args()
 
 lbm = LBM(D2Q9)
-geometry = Geometry(32,32)
+generator = Generator(
+    descriptor = D2Q9,
+    moments    = lbm.moments(),
+    collision  = lbm.bgk(f_eq = lbm.equilibrium(), tau = 0.6))
 
-src = source(
-    args.language,
-    D2Q9,
-    lbm.moments(),
-    lbm.bgk(f_eq = lbm.equilibrium(), tau = 0.6),
-    "",
-    'float',
-    geometry
-)
+geometry = Geometry(32,32)
 
+src = generator.kernel(args.language, 'float', geometry)
 print(src)
-- 
cgit v1.2.3