first try at getting lensed CMB maps in

.travis.yml

@@ -42,7 +42,7 @@ env:
 
         # List other runtime dependencies for the package that are available as
         # pip packages here.
-        - PIP_DEPENDENCIES=''
+        - PIP_DEPENDENCIES='git+https://github.com/bthorne93/PySM_public.git'
 
         # Conda packages for affiliated packages are hosted in channel
         # "astropy" while builds for astropy LTS with recent numpy versions
@@ -72,8 +72,8 @@ matrix:
 
     include:
         # Try MacOS X
-        - os: osx
-          env: SETUP_CMD='test'
+        #- os: osx
+        #  env: SETUP_CMD='test'
 
         # Do a coverage test.
         - os: linux
@@ -132,6 +132,7 @@ install:
 
     - git clone --depth 1 git://github.com/astropy/ci-helpers.git
     - source ci-helpers/travis/setup_conda.sh
+    - pip install https://github.com/simonsobs/so_pysm_models/archive/master.zip
 
     # As described above, using ci-helpers, you should be able to set up an
     # environment with dependencies installed using conda and pip, but in some

mapsims/cmb/__init__.py

@@ -0,0 +1,96 @@
+import healpy as hp, numpy as np
+import so_pysm_models as spm
+
+
+def get_cmb_sky(
+    iteration_num,
+    nside=None,  # set this if healpix is desired
+    shape=None,  # set shape and wcs if CAR maps are desired
+    wcs=None,
+    lensed=True,
+    aberrated=False,
+    modulation=False,
+    has_polarization=True,
+    cmb_set=0,  # We allow for more than one CMB map per lensing map
+    cmb_dir=None,
+    nu=None,
+    input_units="uK_RJ",
+    pixel_indices=None,
+):
+    """
+    Return a CMB map from stored alm's.  This can be in HEALPix format
+    (if nside is specified) or rectangular pixel format (if wcs and shape are
+    specified).  The lensed alm's are pre-stored.
+    If rectangular, it returns a stack of enmaps of shape (nfreqs, ncomp, ny, nx). 
+    If HEALPix, it will return a numpy array of shape (nfreqs, ncomp, npix) 
+
+    Parameters
+    ----------
+    iteration_num : int
+        integer specifying which sim iteration to use
+    nside : int
+        nside of healpix map to project alms to. If None, uses 
+        rectangular pixel geometry specified through shape and wcs.
+    shape : tuple of ints
+        shape of ndmap array (see pixell.enmap)
+    wcs : str
+        World Coordinate System for geometry of map (see pixell.enmap)
+    lensed : bool
+        Whether to load lensed or unlensed sims
+    aberrated : bool
+        Whether to load aberrated or unaberrated sims
+    pol : bool
+        If True, return ncomp=3 I,Q,U components, else just ncomp=1 I
+    cmb_set : int
+        Integer specifying which set of sims to use
+    cmb_dir : path
+        Override the default lensed alm directory path
+    nfreqs : int
+        Number of copies of the CMB sky to provide. When modulation
+        is implemented, this argument will be changed to a frequency bandpass
+        specification that applies frequency-dependent modulation to the sims.
+
+    Returns
+    -------
+    output
+        (nfreqs,ncomp,npix) HEALPix numpy array if nside is specified, else
+        returns (nfreqs,ncomp,Ny,Nx) rectangular pixel ndmap.
+
+    """
+    ncomp = 3 if has_polarization else 1
+    filename = _get_cmb_map_string(cmb_dir, iteration_num, cmb_set, lensed, aberrated)
+    # The hdu = (1, 2,3) means get all of T, E, B
+    # Note the alm's are stored as complex32, so upgrade this for processing
+
+    sgen = spm.PrecomputedAlms(
+        filename,
+        target_nside=nside,
+        target_shape=shape,
+        target_wcs=wcs,
+        input_units=input_units,
+        has_polarization=has_polarization,
+        pixel_indices=pixel_indices,
+    )
+
+    return sgen.signal(nu=nu, modulation=modulation)
+
+
+def _get_default_cmb_directory():
+    # FIXME: remove hard-coding to use preferred directory path system
+    return "/global/project/projectdirs/sobs/v4_sims/mbs/cmb"
+
+
+def _get_cmb_map_string(cmb_dir, iteration_num, cmb_set, lensed, aberrated):
+    # Implements the CMB lensed alms file naming convention
+    # Ideally the same function should be used when saving sims
+    if cmb_dir is None:
+        cmb_dir = _get_default_cmb_directory()
+    lstring = "Lensed" if lensed else "Unlensed"
+    abstring = "Abberated" if aberrated else "Unabberated"
+    cmb_map_type = "%s%sCMB" % (lstring, abstring)
+    filename = cmb_dir + "/fullsky%s_alm_set%02d_%05d.fits" % (
+        cmb_map_type,
+        cmb_set,
+        iteration_num,
+    )
+    return filename