From 4a56c8448653ecc4fea9cee89e70b32dd5aadbc2 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Sat, 26 Oct 2019 13:30:17 +0200
Subject: Add geometry argument

---
 boltzgen.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'boltzgen.py')

diff --git a/boltzgen.py b/boltzgen.py
index 6f8f20c..95cccd9 100644
--- a/boltzgen.py
+++ b/boltzgen.py
@@ -11,6 +11,7 @@ argparser.add_argument('language', help = 'Target language (currently either "cl
 argparser.add_argument('--lattice',   dest = 'lattice',   required = True, help = 'Lattice type (D2Q9, D3Q7, D3Q19, D3Q27)')
 argparser.add_argument('--layout',    dest = 'layout',    required = True, help = 'Memory layout ("AOS" or "SOA")')
 argparser.add_argument('--precision', dest = 'precision', required = True, help = 'Floating precision ("single" or "double")')
+argparser.add_argument('--geometry',  dest = 'geometry',  required = True, help = 'Size of the block geometry ("x:y(:z)")')
 
 args = argparser.parse_args()
 
@@ -22,7 +23,7 @@ generator = Generator(
     moments    = lbm.moments(),
     collision  = lbm.bgk(f_eq = lbm.equilibrium(), tau = 0.6))
 
-geometry = Geometry(1024,1024)
+geometry = Geometry.parse(args.geometry)
 
 src = generator.kernel(args.language, args.precision, args.layout, geometry)
 print(src)
-- 
cgit v1.2.3