earth2studio.models.dx.DerivedTCWV#

class earth2studio.models.dx.DerivedTCWV(levels=[1000, 850, 700, 500, 300, 200, 100])[source]#

Calculates the Total Column Water Vapor (TCWV) from specific humidity at pressure levels and surface pressure. The calculation is based on the vertical integration of specific humidity using the trapezoidal rule:

TCWV = (1/g) * ∫(p_top to p_surface) q dp

Discretized as: TCWV ≈ (1/g) * Σ [(q_i + q_{i+1})/2 * (p_i - p_{i+1})]

Note

The integration includes the layer between the surface pressure and the lowest pressure level, assuming the surface specific humidity equals the lowest level specific humidity.

Parameters:

levels (list[int], optional) – Pressure levels (hPa) to use for the integration. They will be sorted internally from highest to lowest pressure. Default is [1000, 850, 700, 500, 300, 200, 100].

__call__(x, coords)[source]#

Forward pass of diagnostic

Parameters:

  • x (Tensor)

  • coords (OrderedDict[str, ndarray])

Return type:

tuple[Tensor, OrderedDict[str, ndarray]]