blob: 3b95a31d66fe2c8367e85fce017de26737324fb2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
|
void equilibrilize(${float_type}* f_next,
${float_type}* f_prev,
std::size_t gid)
{
${float_type}* preshifted_f_next = f_next + ${layout.cell_preshift('gid')};
${float_type}* preshifted_f_prev = f_prev + ${layout.cell_preshift('gid')};
% for i, w_i in enumerate(descriptor.w):
preshifted_f_next[${layout.pop_offset(i)}] = ${w_i.evalf()};
preshifted_f_prev[${layout.pop_offset(i)}] = ${w_i.evalf()};
% endfor
}
|