template/particles.mako


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

__kernel void update_particles(__global float*  moments,
                               __global int*    material,
                               __global float4* particles,
                               __global float4* init_particles,
                               float aging)
{
  const unsigned int pid = get_global_id(0);

  float4 particle = particles[pid];

% if descriptor.d == 2:
  const unsigned int gid = floor(particle.y)*${memory.size_x} + floor(particle.x);
% elif descriptor.d == 3:
  const unsigned int gid = floor(particle.z)*${memory.size_x*memory.size_y} + floor(particle.y)*${memory.size_x} + floor(particle.x);
% endif

  if (material[gid] == 1 && particle.w < 1.0) {
    particle.x += moments[${1*memory.volume}+gid];
    particle.y += moments[${2*memory.volume}+gid];
% if descriptor.d == 3:
    particle.z += moments[${3*memory.volume}+gid];
% endif
    particle.w += min(particle.x, particle.y) * aging;
  } else {
    particle.xyz = init_particles[pid].xyz;
    particle.w   = particle.w-1.0;
  }

  particles[pid] = particle;
}