Commit | Line | Data |
---|---|---|
e3f82424 MW |
1 | // -------------------------------------------------------------------------- |
2 | // | |
3 | // Copyright | |
4 | // Markus Wittmann, 2016-2017 | |
5 | // RRZE, University of Erlangen-Nuremberg, Germany | |
6 | // markus.wittmann -at- fau.de or hpc -at- rrze.fau.de | |
7 | // | |
8 | // Viktor Haag, 2016 | |
9 | // LSS, University of Erlangen-Nuremberg, Germany | |
10 | // | |
11 | // This file is part of the Lattice Boltzmann Benchmark Kernels (LbmBenchKernels). | |
12 | // | |
13 | // LbmBenchKernels is free software: you can redistribute it and/or modify | |
14 | // it under the terms of the GNU General Public License as published by | |
15 | // the Free Software Foundation, either version 3 of the License, or | |
16 | // (at your option) any later version. | |
17 | // | |
18 | // LbmBenchKernels is distributed in the hope that it will be useful, | |
19 | // but WITHOUT ANY WARRANTY; without even the implied warranty of | |
20 | // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
21 | // GNU General Public License for more details. | |
22 | // | |
23 | // You should have received a copy of the GNU General Public License | |
24 | // along with LbmBenchKernels. If not, see <http://www.gnu.org/licenses/>. | |
25 | // | |
26 | // -------------------------------------------------------------------------- | |
27 | #ifndef __BENCH_KERNEL_D3Q19_AA_COMMON_H__ | |
28 | #define __BENCH_KERNEL_D3Q19_AA_COMMON_H__ | |
29 | ||
30 | #include "Kernel.h" | |
31 | ||
32 | typedef struct KernelDataAa_ | |
33 | { | |
34 | KernelData kd; | |
35 | int Blk[3]; // Blocking in X, Y, and Z direction, value of 0 disables blocking. | |
36 | int Iteration; // Current iteration number. | |
37 | } KernelDataAa; | |
38 | ||
39 | // Macro for casting KernelData * to KernelDataAa *. | |
40 | #define KDA(_x_) ((KernelDataAa *)(_x_)) | |
41 | ||
42 | // Build a function name extended by the propagation model name and the data layout. | |
43 | // FNANEM(test) will be expanded to test_PushSoA if DATA_LAYOUT_NAME is defined | |
44 | // as SoA and PROP_MODEL is defined as Push. | |
45 | #define FNAME(functionName) JOIN(JOIN(functionName,_),JOIN(PROP_MODEL_NAME,DATA_LAYOUT_NAME)) | |
46 | ||
47 | #ifndef DATA_LAYOUT_NAME | |
48 | #error DATA_LAYOUT_NAME must be defined | |
49 | #endif | |
50 | ||
51 | #ifndef PROP_MODEL_NAME | |
52 | #error PROP_MODEL_NAME must be defined | |
53 | #endif | |
54 | ||
55 | // ----------------------------------------------------------------------- | |
56 | // Index function for accesssing PDF array for different data layouts. | |
57 | ||
58 | #define P_INDEX_5 FNAME(PINDEX5) | |
59 | ||
60 | static inline int FNAME(PINDEX5)(int dims[3], int x, int y, int z, int d) | |
61 | { | |
62 | Assert(dims[0] > 0); | |
63 | Assert(dims[1] > 0); | |
64 | Assert(dims[2] > 0); | |
65 | ||
66 | Assert(x >= 0); | |
67 | Assert(x < dims[0]); | |
68 | Assert(y >= 0); | |
69 | Assert(y < dims[1]); | |
70 | Assert(z >= 0); | |
71 | Assert(z < dims[2]); | |
72 | Assert(d >= 0); | |
73 | #ifdef D3Q19 | |
74 | Assert(d < N_D3Q19); | |
75 | #else | |
76 | #error Not implemented for this discretization. | |
77 | #endif | |
78 | ||
79 | #ifdef DATA_LAYOUT_SOA | |
80 | return d * dims[0] * dims[1] * dims[2] + x * dims[1] * dims[2] + y * dims[2] + z; | |
81 | #elif DATA_LAYOUT_AOS | |
82 | return x * dims[1] * dims[2] * N_D3Q19 + y * dims[2] * N_D3Q19 + z * N_D3Q19 + d; | |
83 | #else | |
84 | #error P_INDEX_5 function no implemented for chosen data layout. | |
85 | #endif | |
86 | } | |
87 | ||
88 | #endif // __BENCH_KERNEL_D3Q19_AA_COMMON_H__ | |
89 |