Skip to content
Open
Changes from 5 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion examples/seismic/plotting.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,10 @@ def plot_perturbation(model, model1, colorbar=True):
domain_size = 1.e-3 * np.array(model.domain_size)
extent = [model.origin[0], model.origin[0] + domain_size[0],
model.origin[1] + domain_size[1], model.origin[1]]
dv = np.transpose(model.vp.data) - np.transpose(model1.vp.data)
if model.nbl != model1.nbl:
raise ValueError(f"{model} and {model1} have different values for nbl")
slices = tuple(slice(model.nbl, -model.nbl) for _ in range(2))
dv = np.transpose(model.vp.data[slices]) - np.transpose(model1.vp.data[slices])

plot = plt.imshow(dv, animated=True, cmap=cm.jet,
vmin=min(dv.reshape(-1)), vmax=max(dv.reshape(-1)),
Expand Down