Relativistic kinetic energy from Smilei output

ekin.py

def ekin(p_r, m_r = 1, to_MeV = True):
    """ compute relativistic kinetic energy from Smilei momentum and mass
    Parameters
    ----------
    p_r : float
        particle momentum in reduced units.
    m_r : float, optional
        particle mass in reduced units, default value is hydrogen mass.
    to_MeV : bool, optional
        computes energy in MeV if true, otherwise in E_r (reduced energy units).
    Returns
    -------
    ekin
        relativistic kinetic energy, float.
    """
    if to_MeV:
        return (gamma(p_r, m_r) - 1) * m_r * 0.51122
    else:
        return (gamma(p_r, m_r) - 1) * m_r