Loading [MathJax]/extensions/TeX/AMSmath.js
ESPResSo
Extensible Simulation Package for Research on Soft Matter Systems
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages Concepts
StreamSweepSinglePrecision.h
Go to the documentation of this file.
1//======================================================================================================================
2//
3// This file is part of waLBerla. waLBerla is free software: you can
4// redistribute it and/or modify it under the terms of the GNU General Public
5// License as published by the Free Software Foundation, either version 3 of
6// the License, or (at your option) any later version.
7//
8// waLBerla is distributed in the hope that it will be useful, but WITHOUT
9// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
11// for more details.
12//
13// You should have received a copy of the GNU General Public License along
14// with waLBerla (see COPYING.txt). If not, see <http://www.gnu.org/licenses/>.
15//
16//! \\file StreamSweepSinglePrecision.h
17//! \\author pystencils
18//======================================================================================================================
19
20// kernel generated with pystencils v1.3.7, lbmpy v1.3.7, sympy v1.12.1,
21// lbmpy_walberla/pystencils_walberla from waLBerla commit
22// f36fa0a68bae59f0b516f6587ea8fa7c24a41141
23
24#pragma once
25#include "core/DataTypes.h"
26#include "core/logging/Logging.h"
27
28#include "domain_decomposition/BlockDataID.h"
29#include "domain_decomposition/IBlock.h"
30#include "domain_decomposition/StructuredBlockStorage.h"
31#include "field/GhostLayerField.h"
32#include "field/SwapableCompare.h"
33
34#include <functional>
35#include <unordered_map>
36
37#ifdef __GNUC__
38#define RESTRICT __restrict__
39#elif _MSC_VER
40#define RESTRICT __restrict
41#else
42#define RESTRICT
43#endif
44
45#if (defined WALBERLA_CXX_COMPILER_IS_GNU) || \
46 (defined WALBERLA_CXX_COMPILER_IS_CLANG)
47#pragma GCC diagnostic push
48#pragma GCC diagnostic ignored "-Wunused-parameter"
49#pragma GCC diagnostic ignored "-Wreorder"
50#endif
51
52namespace walberla {
53namespace pystencils {
54
56public:
57 StreamSweepSinglePrecision(BlockDataID forceID_, BlockDataID pdfsID_,
58 BlockDataID velocityID_)
59 : forceID(forceID_), pdfsID(pdfsID_), velocityID(velocityID_) {}
60
62 for (auto p : cache_pdfs_) {
63 delete p.second;
64 }
65 }
66
67 void run(IBlock *block);
68
69 void runOnCellInterval(const shared_ptr<StructuredBlockStorage> &blocks,
70 const CellInterval &globalCellInterval,
71 cell_idx_t ghostLayers, IBlock *block);
72
73 void operator()(IBlock *block) { run(block); }
74
75 static std::function<void(IBlock *)>
76 getSweep(const shared_ptr<StreamSweepSinglePrecision> &kernel) {
77 return [kernel](IBlock *b) { kernel->run(b); };
78 }
79
80 static std::function<void(IBlock *)>
81 getSweepOnCellInterval(const shared_ptr<StreamSweepSinglePrecision> &kernel,
82 const shared_ptr<StructuredBlockStorage> &blocks,
83 const CellInterval &globalCellInterval,
84 cell_idx_t ghostLayers = 1) {
85 return [kernel, blocks, globalCellInterval, ghostLayers](IBlock *b) {
86 kernel->runOnCellInterval(blocks, globalCellInterval, ghostLayers, b);
87 };
88 }
89
90 std::function<void(IBlock *)> getSweep() {
91 return [this](IBlock *b) { this->run(b); };
92 }
93
94 std::function<void(IBlock *)>
95 getSweepOnCellInterval(const shared_ptr<StructuredBlockStorage> &blocks,
96 const CellInterval &globalCellInterval,
97 cell_idx_t ghostLayers = 1) {
98 return [this, blocks, globalCellInterval, ghostLayers](IBlock *b) {
99 this->runOnCellInterval(blocks, globalCellInterval, ghostLayers, b);
100 };
101 }
102
103 void configure(const shared_ptr<StructuredBlockStorage> & /*blocks*/,
104 IBlock * /*block*/) {}
105
106private:
107 BlockDataID forceID;
108 BlockDataID pdfsID;
109 BlockDataID velocityID;
110 std::unordered_map<IBlock *, field::GhostLayerField<float, 19> *> cache_pdfs_;
111};
112
113} // namespace pystencils
114} // namespace walberla
115
116#if (defined WALBERLA_CXX_COMPILER_IS_GNU) || \
117 (defined WALBERLA_CXX_COMPILER_IS_CLANG)
118#pragma GCC diagnostic pop
119#endif
void runOnCellInterval(const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers, IBlock *block)
static std::function< void(IBlock *)> getSweepOnCellInterval(const shared_ptr< StreamSweepSinglePrecision > &kernel, const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers=1)
void configure(const shared_ptr< StructuredBlockStorage > &, IBlock *)
StreamSweepSinglePrecision(BlockDataID forceID_, BlockDataID pdfsID_, BlockDataID velocityID_)
static std::function< void(IBlock *)> getSweep(const shared_ptr< StreamSweepSinglePrecision > &kernel)
std::function< void(IBlock *)> getSweepOnCellInterval(const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers=1)
static double * block(double *p, std::size_t index, std::size_t size)
Definition elc.cpp:172
\file PackInfoPdfDoublePrecision.cpp \author pystencils