shallow water equations github