def compute_energy_spatial(sim):
    """ Compute energy without energy in shear modes """
    dict_spatial = sim.output.spatial_means.load()
    E = dict_spatial['E'] - dict_spatial['E_shear']
    t = dict_spatial['t']
    PK_tot = dict_spatial['PK_tot']
    PA_tot = dict_spatial['PA_tot']
    P_tot = PK_tot + PA_tot
    return E, t, P_tot
