コード例 #1
0
ファイル: makesky.py プロジェクト: tanderson23/cora
def _21cm(fstate, nside, pol, filename, eor, oversample):
    """Generate a Gaussian simulation of the unresolved 21cm background."""

    from cora.signal import corr21cm

    # Read in arguments.
    if eor:
        cr = corr21cm.EoR21cm()
    else:
        cr = corr21cm.Corr21cm()

    cr.nside = nside
    cr.frequencies = fstate.frequencies
    cr.oversample = oversample if oversample is not None else 3

    # Generate signal realisation and save.
    sg_map = cr.getpolsky() if pol == "full" else cr.getsky()

    # Save map
    write_map(filename, sg_map, cr.frequencies, fstate.freq_width,
              pol != "none")
コード例 #2
0
def _21cm(ctx, eor, oversample):
    """Generate a Gaussian simulation of the unresolved 21cm background.
    """

    from cora.signal import corr21cm

    # Read in arguments.
    if eor:
        cr = corr21cm.EoR21cm()
    else:
        cr = corr21cm.Corr21cm()

    cr.nside = ctx.obj.nside
    cr.frequencies = ctx.obj.freq
    cr.oversample = oversample if oversample is not None else 3

    # Generate signal realisation and save.
    sg_map = cr.getpolsky() if ctx.obj.full_pol else cr.getsky()

    # Save map
    write_map(ctx.obj.filename, sg_map, cr.frequencies, ctx.obj.freq_width,
              ctx.obj.include_pol)
コード例 #3
0
def im21cm_model(lmax, frequencies, npol, cr=None, temponly=False):

    nfreq = frequencies.size

    if not cr:
        global _cr
        if not _cr:
            if not _reionisation:
                _cr = corr21cm.Corr21cm()
            else:
                _cr = corr21cm.EoR21cm()
        cr = _cr

    #cr._freq_window = np.abs(cr.cosmology.comoving_distance(frequencies[0]) - cr.cosmology.comoving_distance(frequencies[1]))

    cv_t = skysim.clarray(cr.angular_powerspectrum, lmax, frequencies)

    if temponly:
        return cv_t
    else:
        cv_sg = np.zeros((npol, npol, lmax + 1, nfreq, nfreq))
        cv_sg[0, 0] = cv_t
        return cv_sg