What the author implemented is a specific case of Saint Venant shallow water equations (in case you want to know where it comes form).