An energy compartment model for propagation, non-linear interaction and dissipation of internal gravity waves