diff options
Diffstat (limited to 'implosion.py')
-rw-r--r-- | implosion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/implosion.py b/implosion.py index 370ef92..245f2bb 100644 --- a/implosion.py +++ b/implosion.py @@ -19,7 +19,7 @@ def generate_moment_plots(lattice, moments): 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.gid(x,y)]**2 + m[2,lattice.gid(x,y)]**2) + velocity[y-1,x-1] = numpy.sqrt(m[1,lattice.memory.gid(x,y)]**2 + m[2,lattice.memory.gid(x,y)]**2) plt.figure(figsize=(10, 10)) plt.imshow(velocity, origin='lower', cmap=plt.get_cmap('seismic')) |