From 0a64777ddc063d9b3f4abd68383e4ef998977bb7 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Fri, 30 Aug 2019 17:00:26 +0200
Subject: Adapt examples to new material interface

---
 ldc_2d_gl_interop.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

(limited to 'ldc_2d_gl_interop.py')

diff --git a/ldc_2d_gl_interop.py b/ldc_2d_gl_interop.py
index e5ea866..ec94a75 100644
--- a/ldc_2d_gl_interop.py
+++ b/ldc_2d_gl_interop.py
@@ -19,13 +19,13 @@ updates_per_frame = 200
 lid_speed = 0.1
 relaxation_time = 0.515
 
-def cavity(geometry, x, y):
-    if x == 1 or y == 1 or x == geometry.size_x-2:
-        return 2
-    elif y == geometry.size_y-2:
-        return 3
-    else:
-        return 1
+def get_cavity_material_map(geometry):
+    return [
+        (lambda x, y: x > 0 and x < geometry.size_x-1 and y > 0 and y < geometry.size_y-1,  1), # bulk fluid
+        (lambda x, y: x == 1 or y == 1 or x == geometry.size_x-2,                           2), # left, right, bottom walls
+        (lambda x, y: y == geometry.size_y-2,                                               3), # lid
+        (lambda x, y: x == 0 or x == geometry.size_x-1 or y == 0 or y == geometry.size_y-1, 0)  # ghost cells
+    ]
 
 boundary = Template("""
     if ( m == 2 ) {
@@ -126,7 +126,9 @@ lattice = Lattice(
     opengl       = True
 )
 
-lattice.setup_geometry(cavity)
+lattice.apply_material_map(
+    get_cavity_material_map(lattice.geometry))
+lattice.sync_material()
 
 projection = get_projection()
 
-- 
cgit v1.2.3