diff options
| -rw-r--r-- | lid_driven_cavity/opencl_gl_interop/SSS.py | 18 | ||||
| l--------- | lid_driven_cavity/opencl_gl_interop/lattice/SSS.py | 1 | ||||
| -rw-r--r-- | lid_driven_cavity/opencl_gl_interop/ldc_2d.py | 13 |
3 files changed, 29 insertions, 3 deletions
diff --git a/lid_driven_cavity/opencl_gl_interop/SSS.py b/lid_driven_cavity/opencl_gl_interop/SSS.py new file mode 100644 index 0000000..a8f5a37 --- /dev/null +++ b/lid_driven_cavity/opencl_gl_interop/SSS.py @@ -0,0 +1,18 @@ +import pyopencl as cl + +from common import MomentsTextureBase + +from lattice.SSS import Lattice + +class MomentsTexture(MomentsTextureBase): + pass + + def collect(self): + cl.enqueue_acquire_gl_objects(self.lattice.queue, [self.cl_gl_moments]) + + self.lattice.program.collect_moments_to_texture( + self.lattice.queue, + self.lattice.geometry.size(), + self.lattice.layout, + self.lattice.memory.cl_control, + self.cl_gl_moments) diff --git a/lid_driven_cavity/opencl_gl_interop/lattice/SSS.py b/lid_driven_cavity/opencl_gl_interop/lattice/SSS.py new file mode 120000 index 0000000..b01ba59 --- /dev/null +++ b/lid_driven_cavity/opencl_gl_interop/lattice/SSS.py @@ -0,0 +1 @@ +../../opencl/SSS.py
\ No newline at end of file diff --git a/lid_driven_cavity/opencl_gl_interop/ldc_2d.py b/lid_driven_cavity/opencl_gl_interop/ldc_2d.py index ac8119b..6373a51 100644 --- a/lid_driven_cavity/opencl_gl_interop/ldc_2d.py +++ b/lid_driven_cavity/opencl_gl_interop/ldc_2d.py @@ -18,14 +18,18 @@ from pyrr import matrix44 geometry = Geometry(512, 512) +precision = 'single' +streaming = 'AA' + functions = ['collide_and_stream', 'equilibrilize', 'collect_moments', 'momenta_boundary'] extras = ['cell_list_dispatch', 'opencl_gl_interop'] -precision = 'single' -streaming = 'AA' +if streaming == 'SSS': + functions = functions + ['update_sss_control_structure'] import AA import AB +import SSS def glut_window(fullscreen = False): glutInit(sys.argv) @@ -73,7 +77,10 @@ wall_cells = CellList(lattice.context, lattice.queue, lattice.float_type, lid_cells = CellList(lattice.context, lattice.queue, lattice.float_type, [ gid(x,y) for x, y in geometry.inner_cells() if y == geometry.size_y-2 ]) -if streaming == 'AB': +if streaming == 'SSS': + lattice.schedule('equilibrilize', ghost_cells) + +if streaming in ['AB', 'SSS']: lattice.schedule('collide_and_stream', bulk_cells) lattice.schedule('velocity_momenta_boundary', wall_cells, numpy.array([0.0, 0.0], dtype=lattice.float_type[0])) lattice.schedule('velocity_momenta_boundary', lid_cells, numpy.array([0.1, 0.0], dtype=lattice.float_type[0])) |