From 2bff37d2d4c0e210fd3e4288bdd0add12b9eb86b Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Wed, 30 Oct 2019 16:08:19 +0100
Subject: Move C++ LDC template from upstream, improve build

---
 ldc_2d/cpp/generate.py | 30 +++++++++++++++++++++++-------
 1 file changed, 23 insertions(+), 7 deletions(-)

(limited to 'ldc_2d/cpp/generate.py')

diff --git a/ldc_2d/cpp/generate.py b/ldc_2d/cpp/generate.py
index a1aa561..167cc1b 100755
--- a/ldc_2d/cpp/generate.py
+++ b/ldc_2d/cpp/generate.py
@@ -1,14 +1,20 @@
 #!/usr/bin/env python
 
+import argparse
+
 from boltzgen import LBM, Generator, Geometry
 from boltzgen.lbm.model import D2Q9
 
-geometry = Geometry(256, 256)
+argparser = argparse.ArgumentParser(
+    description = 'Generate a C++ implementation of a lid driven cavity simulation using LBM')
+argparser.add_argument(
+    '--output', required = False, help = 'Target directory for the generated sources')
+
+args = argparser.parse_args()
 
-functions = ['collide_and_stream', 'equilibrilize', 'collect_moments', 'momenta_boundary', 'example']
-extras    = ['omp_parallel_for', 'moments_vtk']
+geometry = Geometry(128, 128)
 
-precision = 'double'
+functions = ['collide_and_stream', 'equilibrilize', 'collect_moments', 'momenta_boundary']
 
 lbm = LBM(D2Q9)
 generator = Generator(
@@ -16,9 +22,19 @@ generator = Generator(
     moments    = lbm.moments(),
     collision  = lbm.bgk(f_eq = lbm.equilibrium(), tau = 0.52),
     target     = 'cpp',
-    precision  = precision,
+    precision  = 'double',
     index      = 'XYZ',
     layout     = 'AOS')
 
-with open("kernel.h", "w") as kernel:
-    kernel.write(generator.kernel(geometry, functions, extras))
+if args.output is None:
+    args.output = '.'
+
+with open('%s/kernel.h' % args.output, 'w') as kernel:
+    kernel.write(generator.kernel(geometry, functions))
+
+ldc_src = ''
+with open('ldc.cpp.mako', 'r') as template:
+    ldc_src = template.read()
+
+with open('%s/ldc.cpp' % args.output, 'w') as app:
+    app.write(generator.custom(geometry, ldc_src))
-- 
cgit v1.2.3