def __init__(self, config):
super(LBFreeSurface, self).__init__(config)
self.equilibrium, self.equilibrium_vars = sym.shallow_water_equilibrium(self.grid)
def __init__(self, geo_class, options=[], args=None, defaults=None):
LBMSim.__init__(self, geo_class, options, args, defaults)
self._set_grid('D2Q9')
self._set_model('bgk')
self.equilibrium, self.equilibrium_vars = sym.shallow_water_equilibrium(self.grid)
self.gravity = self.options.gravity