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

---
 simulation.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'simulation.py')

diff --git a/simulation.py b/simulation.py
index badd2bb..1f9a6c1 100644
--- a/simulation.py
+++ b/simulation.py
@@ -160,6 +160,10 @@ class Lattice:
 
         self.material = numpy.ndarray(shape=(self.memory.volume, 1), dtype=numpy.int32)
 
+    def apply_material_map(self, material_map):
+        for indicator, material in material_map:
+            self.material[[indicator(*idx) for idx in self.memory.cells()]] = material
+
     def sync_material(self):
         cl.enqueue_copy(self.queue, self.memory.cl_material, self.material).wait();
 
-- 
cgit v1.2.3