def mass_column_integral(bk, pk, ps):
"""
Total mass per square meter of atmospheric column.
Explicitly computed by integrating over pressure, rather than implicitly
using surface pressure. Useful for checking if model data conserves mass.
:param dp: Pressure thickness of the model levels.
"""
dp = dp_from_ps(bk, pk, ps)
return dp.sum(dim=PFULL_STR)
def dp(ps, bk, pk, arr):
"""Pressure thickness of hybrid coordinate levels from surface pressure."""
return dp_from_ps(bk, pk, ps, arr[PFULL_STR])