From b4afecd0f9565bac8cf2e82651b9e2a0b4ba9a87 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Sun, 16 Jun 2019 14:28:53 +0200
Subject: Replace some explicit dimension branching

---
 implosion.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'implosion.py')

diff --git a/implosion.py b/implosion.py
index 2ba831f..63c1950 100644
--- a/implosion.py
+++ b/implosion.py
@@ -17,9 +17,9 @@ def generate_moment_plots(lattice, moments):
     for i, m in enumerate(moments):
         print("Generating plot %d of %d." % (i+1, len(moments)))
 
-        velocity = numpy.ndarray(shape=tuple(reversed(lattice.geometry.inner_span())))
+        velocity = numpy.ndarray(shape=tuple(reversed(lattice.geometry.inner_size())))
         for x, y in lattice.geometry.inner_cells():
-            velocity[y-1,x-1] = numpy.sqrt(m[1,lattice.idx(x,y)]**2 + m[2,lattice.idx(x,y)]**2)
+            velocity[y-1,x-1] = numpy.sqrt(m[1,lattice.gid(x,y)]**2 + m[2,lattice.gid(x,y)]**2)
 
         plt.figure(figsize=(10, 10))
         plt.imshow(velocity, origin='lower', cmap=plt.get_cmap('seismic'))
-- 
cgit v1.2.3