diff options
author | Adrian Kummerlaender | 2019-10-21 18:42:24 +0200 |
---|---|---|
committer | Adrian Kummerlaender | 2019-10-21 18:48:38 +0200 |
commit | 82a44e0d64afb8818ea98d68dc08108885d503c2 (patch) | |
tree | 6e8f08acd83b2886cd296ed3831acc83e309906c /boltzgen/geometry.py | |
download | boltzgen-82a44e0d64afb8818ea98d68dc08108885d503c2.tar boltzgen-82a44e0d64afb8818ea98d68dc08108885d503c2.tar.gz boltzgen-82a44e0d64afb8818ea98d68dc08108885d503c2.tar.bz2 boltzgen-82a44e0d64afb8818ea98d68dc08108885d503c2.tar.lz boltzgen-82a44e0d64afb8818ea98d68dc08108885d503c2.tar.xz boltzgen-82a44e0d64afb8818ea98d68dc08108885d503c2.tar.zst boltzgen-82a44e0d64afb8818ea98d68dc08108885d503c2.zip |
Pull in basics from symlbm_playground
It's time to extract the generator-part of my GPU LBM playground and turn it
into a nice reusable library. The goal is to produce a framework that can be
used to generate collision and streaming programs from symbolic descriptions.
i.e. it should be possible to select a LB model, the desired boundary
conditions as well as a data structure / streaming model and use this
information to automatically generate matching OpenCL / CUDA / C++
programs.
Diffstat (limited to 'boltzgen/geometry.py')
-rw-r--r-- | boltzgen/geometry.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/boltzgen/geometry.py b/boltzgen/geometry.py new file mode 100644 index 0000000..4b61982 --- /dev/null +++ b/boltzgen/geometry.py @@ -0,0 +1,24 @@ +from utility.ndindex import ndindex + +class Geometry: + def __init__(self, size_x, size_y, size_z = 1): + self.size_x = size_x + self.size_y = size_y + self.size_z = size_z + self.volume = size_x * size_y * size_z + + def inner_cells(self): + for idx in numpy.ndindex(self.inner_size()): + yield tuple(map(lambda i: i + 1, idx)) + + def size(self): + if self.size_z == 1: + return (self.size_x, self.size_y) + else: + return (self.size_x, self.size_y, self.size_z) + + def inner_size(self): + if self.size_z == 1: + return (self.size_x-2, self.size_y-2) + else: + return (self.size_x-2, self.size_y-2, self.size_z-2) |