aboutsummaryrefslogtreecommitdiff
path: root/boltzgen.py
diff options
context:
space:
mode:
authorAdrian Kummerlaender2019-10-26 13:30:17 +0200
committerAdrian Kummerlaender2019-10-26 13:30:17 +0200
commit4a56c8448653ecc4fea9cee89e70b32dd5aadbc2 (patch)
tree0e083d9a33f3c2b2ce51d4af2c08a0849f3891e5 /boltzgen.py
parentafbb5520cd2140c108adffd6170b5a83faf329f2 (diff)
downloadboltzgen-4a56c8448653ecc4fea9cee89e70b32dd5aadbc2.tar
boltzgen-4a56c8448653ecc4fea9cee89e70b32dd5aadbc2.tar.gz
boltzgen-4a56c8448653ecc4fea9cee89e70b32dd5aadbc2.tar.bz2
boltzgen-4a56c8448653ecc4fea9cee89e70b32dd5aadbc2.tar.lz
boltzgen-4a56c8448653ecc4fea9cee89e70b32dd5aadbc2.tar.xz
boltzgen-4a56c8448653ecc4fea9cee89e70b32dd5aadbc2.tar.zst
boltzgen-4a56c8448653ecc4fea9cee89e70b32dd5aadbc2.zip
Add geometry argument
Diffstat (limited to 'boltzgen.py')
-rw-r--r--boltzgen.py3
1 files changed, 2 insertions, 1 deletions
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)