Energy and momentum equation for non-uniform flow - Engelund, Frank (Series paper, 1981)