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
CollideSweepDoublePrecisionThermalizedAVX.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 CollideSweepDoublePrecisionThermalizedAVX.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:
58 BlockDataID pdfsID_, double kT,
59 double omega_bulk,
60 double omega_even, double omega_odd,
61 double omega_shear, uint32_t seed,
62 uint32_t time_step)
63 : forceID(forceID_), pdfsID(pdfsID_), kT_(kT), omega_bulk_(omega_bulk),
64 omega_even_(omega_even), omega_odd_(omega_odd),
65 omega_shear_(omega_shear), seed_(seed), time_step_(time_step),
66 block_offset_0_(uint32_t(0)), block_offset_1_(uint32_t(0)),
67 block_offset_2_(uint32_t(0)), configured_(false) {}
68
69 void run(IBlock *block);
70
71 void runOnCellInterval(const shared_ptr<StructuredBlockStorage> &blocks,
72 const CellInterval &globalCellInterval,
73 cell_idx_t ghostLayers, IBlock *block);
74
75 void operator()(IBlock *block) { run(block); }
76
77 static std::function<void(IBlock *)> getSweep(
78 const shared_ptr<CollideSweepDoublePrecisionThermalizedAVX> &kernel) {
79 return [kernel](IBlock *b) { kernel->run(b); };
80 }
81
82 static std::function<void(IBlock *)> getSweepOnCellInterval(
83 const shared_ptr<CollideSweepDoublePrecisionThermalizedAVX> &kernel,
84 const shared_ptr<StructuredBlockStorage> &blocks,
85 const CellInterval &globalCellInterval, cell_idx_t ghostLayers = 1) {
86 return [kernel, blocks, globalCellInterval, ghostLayers](IBlock *b) {
87 kernel->runOnCellInterval(blocks, globalCellInterval, ghostLayers, b);
88 };
89 }
90
91 std::function<void(IBlock *)> getSweep() {
92 return [this](IBlock *b) { this->run(b); };
93 }
94
95 std::function<void(IBlock *)>
96 getSweepOnCellInterval(const shared_ptr<StructuredBlockStorage> &blocks,
97 const CellInterval &globalCellInterval,
98 cell_idx_t ghostLayers = 1) {
99 return [this, blocks, globalCellInterval, ghostLayers](IBlock *b) {
100 this->runOnCellInterval(blocks, globalCellInterval, ghostLayers, b);
101 };
102 }
103
104 void configure(const shared_ptr<StructuredBlockStorage> &blocks,
105 IBlock *block) {
106 Cell BlockCellBB = blocks->getBlockCellBB(*block).min();
107 block_offset_0_ = uint32_t(BlockCellBB[0]);
108 block_offset_1_ = uint32_t(BlockCellBB[1]);
109 block_offset_2_ = uint32_t(BlockCellBB[2]);
110 configured_ = true;
111 }
112
113 inline uint32_t getBlock_offset_0() const { return block_offset_0_; }
114 inline uint32_t getBlock_offset_1() const { return block_offset_1_; }
115 inline uint32_t getBlock_offset_2() const { return block_offset_2_; }
116 inline double getKt() const { return kT_; }
117 inline double getOmega_bulk() const { return omega_bulk_; }
118 inline double getOmega_even() const { return omega_even_; }
119 inline double getOmega_odd() const { return omega_odd_; }
120 inline double getOmega_shear() const { return omega_shear_; }
121 inline uint32_t getSeed() const { return seed_; }
122 inline uint32_t getTime_step() const { return time_step_; }
123 inline void setBlock_offset_0(const uint32_t value) {
124 block_offset_0_ = value;
125 }
126 inline void setBlock_offset_1(const uint32_t value) {
127 block_offset_1_ = value;
128 }
129 inline void setBlock_offset_2(const uint32_t value) {
130 block_offset_2_ = value;
131 }
132 inline void setKt(const double value) { kT_ = value; }
133 inline void setOmega_bulk(const double value) { omega_bulk_ = value; }
134 inline void setOmega_even(const double value) { omega_even_ = value; }
135 inline void setOmega_odd(const double value) { omega_odd_ = value; }
136 inline void setOmega_shear(const double value) { omega_shear_ = value; }
137 inline void setSeed(const uint32_t value) { seed_ = value; }
138 inline void setTime_step(const uint32_t value) { time_step_ = value; }
139
140private:
141 BlockDataID forceID;
142 BlockDataID pdfsID;
143 uint32_t block_offset_0_;
144 uint32_t block_offset_1_;
145 uint32_t block_offset_2_;
146 double kT_;
147 double omega_bulk_;
148 double omega_even_;
149 double omega_odd_;
150 double omega_shear_;
151 uint32_t seed_;
152 uint32_t time_step_;
153
154 bool configured_;
155};
156
157} // namespace pystencils
158} // namespace walberla
159
160#if (defined WALBERLA_CXX_COMPILER_IS_GNU) || \
161 (defined WALBERLA_CXX_COMPILER_IS_CLANG)
162#pragma GCC diagnostic pop
163#endif
Definition Cell.hpp:96
static std::function< void(IBlock *)> getSweep(const shared_ptr< CollideSweepDoublePrecisionThermalizedAVX > &kernel)
CollideSweepDoublePrecisionThermalizedAVX(BlockDataID forceID_, BlockDataID pdfsID_, double kT, double omega_bulk, double omega_even, double omega_odd, double omega_shear, uint32_t seed, uint32_t time_step)
void runOnCellInterval(const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers, IBlock *block)
std::function< void(IBlock *)> getSweepOnCellInterval(const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers=1)
void configure(const shared_ptr< StructuredBlockStorage > &blocks, IBlock *block)
static std::function< void(IBlock *)> getSweepOnCellInterval(const shared_ptr< CollideSweepDoublePrecisionThermalizedAVX > &kernel, 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