In [1]:
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))

In [2]:
wrapper = BMIWrapper(engine="xbeach", configfile='/Users/baart_f/models/xbeach/Zwin_T01/params.txt')
wrapper.initialize()

In [3]:
zs0 = wrapper.get_var('zs').copy()
uu0 = wrapper.get_var('uu').copy()
vv0 = wrapper.get_var('vv').copy()

In [5]:
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



In [ ]: