A significant part of inertia-gravity waves consists of processes that are unresolved in the GCMs. In most of the relations presented for parameterization of inertia–gravity waves generated by convection, latent heating is explicitly present. In this research, at firstthe baroclinic waves have been simulated using the WRF model. Then a new equation is introduced for calculating the vertical velocity in a way that diabatic heating is explicitly involved in the forcing terms. The vertical velocity equation is solved in both compressible and incompressible forms and by using the diabatic vertical velocity latent heating is determined. The results showed that solving the vertical velocity equation in incompressible form provides optimal solution for determining latent heating and cooling. In addition, the maximum heating is in the upper levels jet exit region and the maximum cooling is in the jet entrance region