import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm
%matplotlib inline
from bmi.wrapper import BMIWrapper
/Users/baart_f/.virtualenvs/main/lib/python2.7/site-packages/matplotlib/__init__.py:872: UserWarning: axes.color_cycle is deprecated and replaced with axes.prop_cycle; please use the latter. warnings.warn(self.msg_depr % (key, alt_key))
wrapper = BMIWrapper(engine="xbeach", configfile='/Users/baart_f/models/xbeach/Zwin_T01/params.txt')
wrapper.initialize()
zs0 = wrapper.get_var('zs').copy()
uu0 = wrapper.get_var('uu').copy()
vv0 = wrapper.get_var('vv').copy()
from IPython.html.widgets import interact, interactive
@interactive
def plot(waterlevel=(0,10)):
zs = wrapper.get_var('zs')
zs[:] = zs0
uu = wrapper.get_var('uu')
uu[:] = uu0
vv = wrapper.get_var('vv')
vv[:] = vv0
zs[40:50, 40:50] = waterlevel
for i in range(100):
wrapper.update(-1)
uu = wrapper.get_var('uu')
zs = wrapper.get_var('zs')
plt.imshow(uu, cmap=matplotlib.cm.viridis, vmin=-1.0, vmax=1.0)
plt.colorbar()
plot