diff options
author | Adrian Kummerlaender | 2019-11-02 17:18:32 +0100 |
---|---|---|
committer | Adrian Kummerlaender | 2019-11-02 17:18:32 +0100 |
commit | 24847cbb2567f508a7c30b39c6fb7ba6379d1adc (patch) | |
tree | ae70f1e36e80609d424f7e3b1b489083accfda33 /boltzgen.py | |
parent | e2b00f4ec963060be98939c7b0d12d6c00e50a02 (diff) | |
download | boltzgen-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-x | boltzgen.py | 11 |
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, |