#159 - disc and full sky maps

glass/observations.py

@@ -23,6 +23,8 @@
 Visibility
 ----------
 
+.. autofunction:: vmap_full
+.. autofunction:: vmap_disc
 .. autofunction:: vmap_galactic_ecliptic
 
 
@@ -38,6 +40,51 @@
 from .core.array import cumtrapz
 
 
+def vmap_full(nside: int) -> np.ndarray:
+
+    '''visibility map for full sky
+    
+    Parameters
+    ----------
+    nside : int
+        The NSIDE parameter of the resulting HEALPix map.
+    
+    Returns
+    -------
+    vis : array_like
+        A HEALPix :term:`visibility map`.
+    
+    '''
+
+    m = np.ones(hp.nside2npix(nside))
+    return m
+
+def vmap_disc(nside: int, coords: Tuple[float, float] = (60, 60),
+                           radius: float = 20) -> np.ndarray:
+
+    '''visibility map for a disc on the full sky
+    
+    Parameters
+    ----------
+    nside : int
+        The NSIDE parameter of the resulting HEALPix map.
+    coords : Tuple[float, float]
+        colatitude and longitude in degress
+    
+
+    Returns
+    -------
+    vis : array_like
+        A HEALPix :term:`visibility map`.
+    
+    '''
+
+    vec = hp.ang2vec(coords[0]/180*np.pi, coords[1]/180*np.pi)
+    ipix_disc = hp.query_disc(nside=nside, vec=vec, radius=np.radians(radius))
+    m = np.zeros(hp.nside2npix(nside))
+    m[ipix_disc] = 1
+    return m
+
 def vmap_galactic_ecliptic(nside: int, galactic: Tuple[float, float] = (30, 90),
                            ecliptic: Tuple[float, float] = (20, 80)
                            ) -> np.ndarray:
@@ -72,7 +119,7 @@ def vmap_galactic_ecliptic(nside: int, galactic: Tuple[float, float] = (30, 90),
         raise TypeError('ecliptic stripe must be a pair of numbers')
 
     m = np.ones(hp.nside2npix(nside))
-    m[hp.query_strip(nside, *galactic)] = 0
+    #m[hp.query_strip(nside, *galactic)] = 0
     m = hp.Rotator(coord='GC').rotate_map_pixel(m)
     m[hp.query_strip(nside, *ecliptic)] = 0
     m = hp.Rotator(coord='CE').rotate_map_pixel(m)