Further tidy up

Author: EdCaunt

devitoboundary/__init__.py

@@ -1,4 +1,3 @@
-from .structures import *
 from .geometry import *
 from .distance import *
 from .stencils import *

devitoboundary/topography.py

@@ -261,10 +261,6 @@ def subs(self, spec):
                     switchconfig(log_level='ERROR')(op_weights.apply)()
                     # DIY Progress Bar
                     print('■', end='', flush=True)
-            for i in range(5):
-                plt.imshow(w_x.data[:, 100, :, i])
-                plt.colorbar()
-                plt.show()
             print("\nWeight calculation complete.")
 
             weights.append(Coefficient(deriv,

examples/seismic_topography_example.py

@@ -1,4 +1,5 @@
 import matplotlib.pyplot as plt
+import numpy as np
 
 from devito import Grid, TimeFunction, Eq, solve, Operator, ConditionalDimension
 from devitoboundary import ImmersedBoundary
@@ -78,6 +79,9 @@
 # And run
 op.apply(dt=dt)
 
+outfile = 'data/seismic_topography_wavefield.npy'
+np.save(outfile, usave.data)
+"""
 plot_extent = [0, grid.extent[0],
                origin[2], grid.extent[2] + origin[2]]
 for i in range(usave.data.shape[0] - 1):
@@ -91,3 +95,4 @@
     plt.savefig("images/image-%s" % str(i))
     # plt.show()
     plt.close()
+"""