diff options
Diffstat (limited to 'simulation.py')
| -rw-r--r-- | simulation.py | 4 |
1 files changed, 4 insertions, 0 deletions
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(); |