1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
#pragma once
#include "common.hpp"
namespace kel {
namespace lbm {
template<typename T, typename Desc>
saw::error_or<void> setup_initial_conditions(
saw::data<sch::ChunkStruct<T,Desc>>& fields,
saw::data<sch::MacroStruct<T,Desc>>& macros,
saw::data<sch::ParticleSpheroidGroup<T,Desc>>& particles
){
auto& info_f = fields.template get<"info">();
auto& porous_f = macros.template get<"porosity">();
// Set everything as walls
iterator<Desc::D>::apply(
[&](auto& index){
info_f.at(index).set(1u);
},
{},
info_f.get_dims(),
{}
);
// Fluid
iterator<Desc::D>::apply(
[&](auto& index){
info_f.at(index).set(2u);
},
{},
info_f.get_dims(),
{{1u,1u}}
);
// Inflow
iterator<Desc::D>::apply(
[&](auto& index){
info_f.at(index).set(3u);
},
{{0u,0u}},
{{1u,dim_y}},
{{0u,1u}}
);
// Outflow
iterator<Desc::D>::apply(
[&](auto& index){
info_f.at(index).set(4u);
},
{{dim_x-1u,0u}},
{{dim_x, dim_y}},
{{0u,1u}}
);
//
auto& df_f = fields.template get<"dfs_old">();
auto& rho_f = macros.template get<"density">();
auto& vel_f = macros.template get<"velocity">();
auto& por_f = macros.template get<"porosity">();
iterator<Desc::D>::apply(
[&](auto& index){
auto& df = df_f.at(index);
auto& rho = rho_f.at(index);
por_f.at(index).at({}) = {1};
rho.at({}) = {1};
auto& vel = vel_f.at(index);
auto eq = equilibrium<T,Desc>(rho,vel);
df = eq;
},
{},// 0-index
df_f.get_dims()
);
iterator<Desc::D>::apply(
[&](auto& index){
auto& df = df_f.at(index);
auto& rho = rho_f.at(index);
rho.at({}) = {1};
auto& vel = vel_f.at(index);
if(info_f.at(index).get() == 2u){
vel.at({{0u}}) = 0.0;
}
auto eq = equilibrium<T,Desc>(rho,vel);
df = eq;
},
{},// 0-index
df_f.get_dims(),
{{1u,1u}}
);
iterator<Desc::D>::apply(
[&](auto& index){
saw::data<sch::Vector<T,Desc::D>> middle, ind_vec;
middle.at({{0u}}) = dim_x * 0.25;
middle.at({{1u}}) = dim_y * 0.5;
ind_vec.at({{0u}}) = index.at({{0u}}).template cast_to<T>();
ind_vec.at({{1u}}) = index.at({{1u}}).template cast_to<T>();
auto dist = middle - ind_vec;
auto dist_2 = saw::math::dot(dist,dist);
if(dist_2.at({}).get() < dim_y*dim_y*0.01){
porous_f.at(index).at({}) = 0.0;
}
},
{},// 0-index
df_f.get_dims()
);
{
saw::data<sch::Scalar<T>> dense_p;
dense_p.at({}).set(1);
particles = create_spheroid_particle_group<T,Desc::D,2.0f>(dense_p, {{16u}});
}
return saw::make_void();
}
}
}
|
