ESPResSo
Extensible Simulation Package for Research on Soft Matter Systems
Loading...
Searching...
No Matches
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.3, lbmpy v1.3.3,
21// lbmpy_walberla/pystencils_walberla from waLBerla commit
22// b0842e1a493ce19ef1bbb8d2cf382fc343970a7f
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#include <set>
34
35#ifdef __GNUC__
36#define RESTRICT __restrict__
37#elif _MSC_VER
38#define RESTRICT __restrict
39#else
40#define RESTRICT
41#endif
42
43#if (defined WALBERLA_CXX_COMPILER_IS_GNU) || \
44 (defined WALBERLA_CXX_COMPILER_IS_CLANG)
45#pragma GCC diagnostic push
46#pragma GCC diagnostic ignored "-Wunused-parameter"
47#pragma GCC diagnostic ignored "-Wreorder"
48#endif
49
50namespace walberla {
51namespace pystencils {
52
54public:
56 BlockDataID pdfsID_, double kT,
57 double omega_bulk,
58 double omega_even, double omega_odd,
59 double omega_shear, uint32_t seed,
60 uint32_t time_step)
61 : forceID(forceID_), pdfsID(pdfsID_), kT_(kT), omega_bulk_(omega_bulk),
62 omega_even_(omega_even), omega_odd_(omega_odd),
63 omega_shear_(omega_shear), seed_(seed), time_step_(time_step),
64 configured_(false){};
65
66 void run(IBlock *block);
67
68 void runOnCellInterval(const shared_ptr<StructuredBlockStorage> &blocks,
69 const CellInterval &globalCellInterval,
70 cell_idx_t ghostLayers, IBlock *block);
71
72 void operator()(IBlock *block) { run(block); }
73
74 static std::function<void(IBlock *)> getSweep(
75 const shared_ptr<CollideSweepDoublePrecisionThermalizedAVX> &kernel) {
76 return [kernel](IBlock *b) { kernel->run(b); };
77 }
78
79 static std::function<void(IBlock *)> getSweepOnCellInterval(
80 const shared_ptr<CollideSweepDoublePrecisionThermalizedAVX> &kernel,
81 const shared_ptr<StructuredBlockStorage> &blocks,
82 const CellInterval &globalCellInterval, cell_idx_t ghostLayers = 1) {
83 return [kernel, blocks, globalCellInterval, ghostLayers](IBlock *b) {
84 kernel->runOnCellInterval(blocks, globalCellInterval, ghostLayers, b);
85 };
86 }
87
88 std::function<void(IBlock *)> getSweep() {
89 return [this](IBlock *b) { this->run(b); };
90 }
91
92 std::function<void(IBlock *)>
93 getSweepOnCellInterval(const shared_ptr<StructuredBlockStorage> &blocks,
94 const CellInterval &globalCellInterval,
95 cell_idx_t ghostLayers = 1) {
96 return [this, blocks, globalCellInterval, ghostLayers](IBlock *b) {
97 this->runOnCellInterval(blocks, globalCellInterval, ghostLayers, b);
98 };
99 }
100
101 void configure(const shared_ptr<StructuredBlockStorage> &blocks,
102 IBlock *block) {
103 Cell BlockCellBB = blocks->getBlockCellBB(*block).min();
104 block_offset_0_ = uint32_t(BlockCellBB[0]);
105 block_offset_1_ = uint32_t(BlockCellBB[1]);
106 block_offset_2_ = uint32_t(BlockCellBB[2]);
107 configured_ = true;
108 }
109
110 BlockDataID forceID;
111 BlockDataID pdfsID;
115 double kT_;
120 uint32_t seed_;
121 uint32_t time_step_;
123};
124
125} // namespace pystencils
126} // namespace walberla
127
128#if (defined WALBERLA_CXX_COMPILER_IS_GNU) || \
129 (defined WALBERLA_CXX_COMPILER_IS_CLANG)
130#pragma GCC diagnostic pop
131#endif
Definition Cell.hpp:97
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