Add an option in dynamics to calculate omega in non-hydrostatic runs similar to hydrostatic method

Add an option in dynamics to calculate omega in non-hydrostatic runs similar to hydrostatic method #850