aboutsummaryrefslogtreecommitdiff
path: root/boltzgen.py
diff options
context:
space:
mode:
authorAdrian Kummerlaender2019-10-21 22:33:13 +0200
committerAdrian Kummerlaender2019-10-21 22:33:13 +0200
commit8fb7705428a6524a436a5dd8806ad8076b7f4e08 (patch)
treeb5256587d5e5d4fc4d94799544ca0b9547ae1647 /boltzgen.py
parent8f88aea2c963ebe63db83480fc9a4b48451220af (diff)
downloadboltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar
boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar.gz
boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar.bz2
boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar.lz
boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar.xz
boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar.zst
boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.zip
Add basic argument parsing
Diffstat (limited to 'boltzgen.py')
-rw-r--r--boltzgen.py24
1 files changed, 24 insertions, 0 deletions
diff --git a/boltzgen.py b/boltzgen.py
new file mode 100644
index 0000000..6d410c3
--- /dev/null
+++ b/boltzgen.py
@@ -0,0 +1,24 @@
+import argparse
+import string
+
+from boltzgen import *
+
+argparser = argparse.ArgumentParser(description='Generate LBM kernels in various languages using a symbolic description.')
+argparser.add_argument('language', help = 'Target language (currently either "opencl" or "cpp")')
+
+args = argparser.parse_args()
+
+lbm = LBM(D2Q9)
+geometry = Geometry(32,32)
+
+src = source(
+ args.language,
+ D2Q9,
+ lbm.moments(),
+ lbm.bgk(f_eq = lbm.equilibrium(), tau = 0.6),
+ "",
+ 'float',
+ geometry
+)
+
+print(src)