From bcc087a36ea7d84ac61a2c756458a208f6bc3ff3 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Thu, 25 Jul 2019 20:59:06 +0200
Subject: Dampen channel inflow

---
 simulation.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'simulation.py')

diff --git a/simulation.py b/simulation.py
index b64ff96..700b8b6 100644
--- a/simulation.py
+++ b/simulation.py
@@ -109,6 +109,8 @@ class Lattice:
         self.geometry   = geometry
         self.grid       = Grid(self.geometry, padding)
 
+        self.time = 0
+
         self.float_type = {
             'single': (numpy.float32, 'float'),
             'double': (numpy.float64, 'double'),
@@ -186,14 +188,15 @@ class Lattice:
         self.program = cl.Program(self.context, program_src).build(self.compiler_args)
 
     def evolve(self):
+        self.time += 1
         if self.tick:
             self.tick = False
             self.program.collide_and_stream(
-                self.queue, self.grid.size(), self.layout, self.memory.cl_pop_a, self.memory.cl_pop_b, self.memory.cl_material)
+                self.queue, self.grid.size(), self.layout, self.memory.cl_pop_a, self.memory.cl_pop_b, self.memory.cl_material, numpy.uint32(self.time))
         else:
             self.tick = True
             self.program.collide_and_stream(
-                self.queue, self.grid.size(), self.layout, self.memory.cl_pop_b, self.memory.cl_pop_a, self.memory.cl_material)
+                self.queue, self.grid.size(), self.layout, self.memory.cl_pop_b, self.memory.cl_pop_a, self.memory.cl_material, numpy.uint32(self.time))
 
     def sync(self):
         self.queue.finish()
-- 
cgit v1.2.3