blob: 608b89ed24fa8fc051c64bf396a128bffc22d840 (
plain)
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
|
#pragma once
namespace kel {
namespace lbm {
/**
* Calculate the macroscopic variables rho and u in Lattice Units.
*/
template<typename Desc>
void compute_rho_u (
const saw::data<sch::DfCell<Desc>>& dfs,
typename saw::native_data_type<sch::T>::type& rho,
std::array<typename saw::native_data_type<sch::T>::type, 2>& vel
)
{
using dfi = df_info<sch::T, Desc>;
rho = 0;
std::fill(vel.begin(), vel.end(), 0);
for(size_t j = 0; j < Desc::Q; ++j){
rho += dfs(j).get();
for(size_t i = 0; i < Desc::D; ++i){
vel[i] += dfi::directions[j][i] * dfs(j).get();
}
}
for(size_t i = 0; i < Desc::D; ++i){
vel[i] /= rho;
}
}
}
}
|
