Height-dependent expression for the energy flux of equatorial waves: exact and approximate versions for unified model diagnosis