For a river flowing on a slope, the resistive force is $R = av^2$ and gravitational force, $F = bh$. The flow speed adjusts itself to $av^2=bh \implies v^2=\frac{b}{a}h$. Given the mass conservation law,
$$h_t + (hv)_x = r.$$
If $v^2=\frac{b}{a}h$,
$$v^2=\frac{b}{a}h\rightarrow (v^2)_t =\frac{b}{a} h_t\rightarrow h_t =\frac{a}{b}(v^2)_t$$ $$(hv)_x=(\frac{a}{b}v^3)_x=\frac{a}{b}3v^2v_x$$
Thus, $$h_t+(hv)_x=\frac{a}{b}(v^2)_t+\frac{a}{b}3v^2v_x=r$$ $$(v^2)_t+3v^2v_x=r\frac{b}{a}$$
With $u=3\frac{b}{a}h=3v^2$ write, $$\frac{u_t}{3}+u\frac{(u^{\frac{1}{2}})_x}{3}=r\frac{b}{a}$$ $$\frac{u_t}{3}+u\frac{\frac{1}{2}u_x}{3u^{\frac{1}{2}}}=r\frac{b}{a}$$ $$\frac{u_t}{3}+\frac{u^{\frac{1}{2}}u_x}{6}=r\frac{b}{a}$$
I am trying to get the PDE in the form $u_t+uu_x=f$ (inviscid Burger's equation) but am out of tricks.