aboutsummaryrefslogtreecommitdiff
path: root/boltzgen.py
diff options
context:
space:
mode:
authorAdrian Kummerlaender2019-11-02 17:18:32 +0100
committerAdrian Kummerlaender2019-11-02 17:18:32 +0100
commit24847cbb2567f508a7c30b39c6fb7ba6379d1adc (patch)
treeae70f1e36e80609d424f7e3b1b489083accfda33 /boltzgen.py
parente2b00f4ec963060be98939c7b0d12d6c00e50a02 (diff)
downloadboltzgen-24847cbb2567f508a7c30b39c6fb7ba6379d1adc.tar
boltzgen-24847cbb2567f508a7c30b39c6fb7ba6379d1adc.tar.gz
boltzgen-24847cbb2567f508a7c30b39c6fb7ba6379d1adc.tar.bz2
boltzgen-24847cbb2567f508a7c30b39c6fb7ba6379d1adc.tar.lz
boltzgen-24847cbb2567f508a7c30b39c6fb7ba6379d1adc.tar.xz
boltzgen-24847cbb2567f508a7c30b39c6fb7ba6379d1adc.tar.zst
boltzgen-24847cbb2567f508a7c30b39c6fb7ba6379d1adc.zip
Restructure LBM model / lattice distinction
Diffstat (limited to 'boltzgen.py')
-rwxr-xr-xboltzgen.py11
1 files changed, 8 insertions, 3 deletions
diff --git a/boltzgen.py b/boltzgen.py
index 82adafe..4317c68 100755
--- a/boltzgen.py
+++ b/boltzgen.py
@@ -8,7 +8,8 @@ argparser = argparse.ArgumentParser(
argparser.add_argument('target', help = 'Target language (currently either "cl" or "cpp")')
-argparser.add_argument('--lattice', required = True, help = 'Lattice type (D2Q9, D3Q7, D3Q19, D3Q27)')
+argparser.add_argument('--lattice', required = True, help = 'Lattice type ("D2Q9", "D3Q7", "D3Q19", "D3Q27")')
+argparser.add_argument('--model', required = False, help = 'LBM model (currently only "BGK")')
argparser.add_argument('--layout', required = True, help = 'Memory layout ("AOS" or "SOA")')
argparser.add_argument('--index', required = False, help = 'Cell indexing ("XYZ" or "ZYX")')
argparser.add_argument('--precision', required = True, help = 'Floating precision ("single" or "double")')
@@ -21,13 +22,17 @@ argparser.add_argument('--extras', action = 'append', nargs = '+', default
args = argparser.parse_args()
-lattice = eval("lbm.model.%s" % args.lattice)
+if args.model is None:
+ args.model = "BGK"
+
+lattice = eval("lbm.lattice.%s" % args.lattice)
+model = eval("lbm.model.%s" % args.model)
if args.index is None:
args.index = 'XYZ'
generator = Generator(
- model = LBM(lattice, tau = float(args.tau), optimize = not args.disable_cse),
+ model = model(lattice, tau = float(args.tau), optimize = not args.disable_cse),
target = args.target,
precision = args.precision,
index = args.index,