1 files changed, 30 insertions, 0 deletions
diff --git a/template/particles.mako b/template/particles.mako
new file mode 100644
index 0000000..0b9a67e
--- /dev/null
+++ b/template/particles.mako
@@ -0,0 +1,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;
+}