ringsSkyMap.py

Go to the documentation of this file.

#
# LSST Data Management System
# Copyright 2008, 2009, 2010, 2012 LSST Corporation.
#
# This product includes software developed by the
# LSST Project (http://www.lsst.org/).
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the LSST License Statement and
# the GNU General Public License along with this program.  If not,
# see <http://www.lsstcorp.org/LegalNotices/>.
#
 
__all__ = ["RingsSkyMapConfig", "RingsSkyMap"]
 
import struct
import math
import numpy as np
 
from lsst.pex.config import Field
import lsst.geom as geom
import lsst.sphgeom as sphgeom
from .cachingSkyMap import CachingSkyMap
from .tractInfo import ExplicitTractInfo
 
 
class RingsSkyMapConfig(CachingSkyMap.ConfigClass):
    """Configuration for the RingsSkyMap"""
    numRings = Field(dtype=int, doc="Number of rings", check=lambda x: x > 0)
    raStart = Field(dtype=float, default=0.0, doc="Starting center RA for each ring (degrees)",
                    check=lambda x: x >= 0.0 and x < 360.0)
 
 
class RingsSkyMap(CachingSkyMap):
    """Rings sky map pixelization.
 
    We divide the sphere into N rings of Declination, plus the two polar
    caps, which sets the size of the individual tracts.  The rings are
    divided in RA into an integral number of tracts of this size; this
    division is made at the Declination closest to zero so as to ensure
    full overlap.
 
    Rings are numbered in the rings from south to north. The south pole cap is
    ``tract=0``, then the tract at ``raStart`` in the southernmost ring is
    ``tract=1``. Numbering continues (in the positive RA direction) around that
    ring and then continues in the same fashion with the next ring north, and
    so on until all reaching the north pole cap, which is
    ``tract=len(skymap) - 1``.
 
    However, ``version=0`` had a bug in the numbering of the tracts: the first
    and last tracts in the first (southernmost) ring were identical, and the
    first tract in the last (northernmost) ring was missing. When using
    ``version=0``, these tracts remain missing in order to preserve the
    numbering scheme.
 
    Parameters
    ----------
    config : `lsst.skymap.RingsSkyMapConfig`
        The configuration for this SkyMap.
    version : `int`, optional
        Software version of this class, to retain compatibility with old
        verisons. ``version=0`` covers the period from first implementation
        until DM-14809, at which point bugs were identified in the numbering
        of tracts (affecting only tracts at RA=0). ``version=1`` uses the
        post-DM-14809 tract numbering.
    """
    ConfigClass = RingsSkyMapConfig
    _version = (1, 0)  # for pickle
 
    def __init__(self, config, version=1):
        assert version in (0, 1), "Unrecognised version: %s" % (version,)
        # We count rings from south to north
        # Note: pole caps together count for one additional ring when
        # calculating the ring size.
        self._ringSize = math.pi / (config.numRings + 1)  # Size of a ring in Declination (radians)
        self._ringNums = []  # Number of tracts for each ring
        for i in range(config.numRings):
            startDec = self._ringSize*(i + 0.5) - 0.5*math.pi
            stopDec = startDec + self._ringSize
            dec = min(math.fabs(startDec), math.fabs(stopDec))  # Declination for determining division in RA
            self._ringNums.append(int(2*math.pi*math.cos(dec)/self._ringSize) + 1)
        numTracts = sum(self._ringNums) + 2
        super(RingsSkyMap, self).__init__(numTracts, config, version)
        self._raStart = self.configconfig.raStart*geom.degrees
 
    def getRingIndices(self, index):
        """Calculate ring indices given a numerical index of a tract.
 
        The ring indices are the ring number and the tract number within
        the ring.
 
        The ring number is -1 for the south polar cap and increases to the
        north.  The north polar cap has ring number = numRings.  The tract
        number is zero for either of the polar caps.
        """
        if index == 0:  # South polar cap
            return -1, 0
        if index == self._numTracts - 1:  # North polar cap
            return self.configconfig.numRings, 0
        if index < 0 or index >= self._numTracts:
            raise IndexError("Tract index %d is out of range [0, %d]" % (index, len(self) - 1))
        ring = 0  # Ring number
        tractNum = index - 1  # Tract number within ring
        if self._version_version_version == 0:
            # Maintain the off-by-one bug in version=0 (DM-14809).
            # This means that the first tract in the first ring is duplicated
            # and the first tract in the last ring is missing.
            while ring < self.configconfig.numRings and tractNum > self._ringNums[ring]:
                tractNum -= self._ringNums[ring]
                ring += 1
        else:
            while ring < self.configconfig.numRings and tractNum >= self._ringNums[ring]:
                tractNum -= self._ringNums[ring]
                ring += 1
        return ring, tractNum
 
    def generateTract(self, index):
        """Generate TractInfo for the specified tract index."""
        ringNum, tractNum = self.getRingIndices(index)
        if ringNum == -1:  # South polar cap
            ra, dec = 0, -0.5*math.pi
        elif ringNum == self.configconfig.numRings:  # North polar cap
            ra, dec = 0, 0.5*math.pi
        else:
            dec = self._ringSize*(ringNum + 1) - 0.5*math.pi
            ra = ((2*math.pi*tractNum/self._ringNums[ringNum])*geom.radians
                  + self._raStart).wrap().asRadians()
 
        center = geom.SpherePoint(ra, dec, geom.radians)
        wcs = self._wcsFactory.makeWcs(crPixPos=geom.Point2D(0, 0), crValCoord=center)
 
        raMin, raMax, decMin, decMax = self.getRaDecRange(index)
 
        innerBoxCorners = [
            sphgeom.LonLat(lon=sphgeom.NormalizedAngle(raMin), lat=sphgeom.Angle(decMin)),
            sphgeom.LonLat(lon=sphgeom.NormalizedAngle(raMax), lat=sphgeom.Angle(decMax)),
        ]
 
        return ExplicitTractInfo(
            index,
            self._tractBuilder_tractBuilder,
            center,
            0.5*self._ringSize*geom.radians,
            self.configconfig.tractOverlap*geom.degrees,
            wcs,
            innerBoxCorners=innerBoxCorners,
        )
 
    def getRaDecRange(self, index):
        """Get the ra and dec ranges for the inner region of a
        specified tract index.
 
        Parameters
        ----------
        index : `int`
            Tract index number.
 
        Returns
        -------
        raMin, raMax, decMin, decMax : `lsst.geom.Angle`
            RA/Dec boundaries of the inner region.
        """
        ringNum, tractNum = self.getRingIndices(index)
 
        if ringNum == -1:
            # South polar cap.
            decMin = (-np.pi/2.)*geom.radians
            decMax = (-np.pi/2. + (ringNum + 1.5)*self._ringSize)*geom.radians
            raMin = 0.0*geom.radians
            raMax = 2.*np.pi*geom.radians
        elif ringNum == self.configconfig.numRings:
            # North polar cap.
            decMin = (-np.pi/2. + (ringNum + 0.5)*self._ringSize)*geom.radians
            decMax = (np.pi/2.)*geom.radians
            raMin = 0.0*geom.radians
            raMax = 2.*np.pi*geom.radians
        else:
            decMin = (-np.pi/2. + (ringNum + 0.5)*self._ringSize)*geom.radians
            decMax = (-np.pi/2. + (ringNum + 1.5)*self._ringSize)*geom.radians
 
            deltaStart = ((tractNum - 0.5)*2.*np.pi/self._ringNums[ringNum])*geom.radians
            deltaStop = ((tractNum + 0.5)*2.*np.pi/self._ringNums[ringNum])*geom.radians
 
            raMin = (self._raStart + deltaStart).wrap()
            raMax = (self._raStart + deltaStop).wrap()
 
        return raMin, raMax, decMin, decMax
 
    def _decToRingNum(self, dec):
        """Calculate ring number from Declination.
 
        Parameters
        ----------
        dec : `lsst.geom.Angle`
            Declination.
 
        Returns
        -------
        ringNum : `int`
            Ring number: -1 for the south polar cap, and increasing to the
            north, ending with ``numRings`` for the north polar cap.
        """
        firstRingStart = self._ringSize*0.5 - 0.5*math.pi
        if dec < firstRingStart:
            # Southern cap
            return -1
        elif dec > firstRingStart*-1:
            # Northern cap
            return self.configconfig.numRings
        return int((dec.asRadians() - firstRingStart)/self._ringSize)
 
    def _raToTractNum(self, ra, ringNum):
        """Calculate tract number from the Right Ascension.
 
        Parameters
        ----------
        ra : `lsst.geom.Angle`
            Right Ascension.
        ringNum : `int`
            Ring number (from ``_decToRingNum``).
 
        Returns
        -------
        tractNum : `int`
            Tract number within the ring (starts at 0 for the tract at
            ``raStart``).
        """
        if ringNum in (-1, self.configconfig.numRings):
            return 0
        assert ringNum in range(self.configconfig.numRings)
        tractNum = int((ra - self._raStart).wrap().asRadians()
                       / (2*math.pi/self._ringNums[ringNum]) + 0.5)
        return 0 if tractNum == self._ringNums[ringNum] else tractNum  # Allow wraparound
 
    def findTract(self, coord):
        ringNum = self._decToRingNum(coord.getLatitude())
        if ringNum == -1:
            # Southern cap
            return self[0]
        if ringNum == self.configconfig.numRings:
            # Northern cap
            return self[self._numTracts - 1]
        tractNum = self._raToTractNum(coord.getLongitude(), ringNum)
 
        if self._version_version_version == 0 and tractNum == 0 and ringNum != 0:
            # Account for off-by-one error in getRingIndices
            # Note that this means that tract 1 gets duplicated.
            ringNum += 1
 
        index = sum(self._ringNums[:ringNum], tractNum + 1)  # Allow 1 for south pole
        return self[index]
 
    def findTractIdArray(self, ra, dec, degrees=False):
        if degrees:
            _ra = np.deg2rad(ra)
            _dec = np.deg2rad(dec)
        else:
            _ra = np.atleast_1d(ra)
            _dec = np.atleast_1d(dec)
 
        # Set default to -1 to distinguish from polar tracts
        indexes = np.full(_ra.size, -1, dtype=np.int32)
 
        # Do the dec search
        firstRingStart = self._ringSize*0.5 - 0.5*np.pi
        ringNums = np.zeros(len(_dec), dtype=np.int32)
 
        ringNums[_dec < firstRingStart] = -1
        ringNums[_dec > -1*firstRingStart] = self.configconfig.numRings
 
        mid = (_dec >= firstRingStart) & (_dec <= -1*firstRingStart)
        ringNums[mid] = ((_dec[mid] - firstRingStart)/self._ringSize).astype(np.int32)
 
        indexes[ringNums == -1] = 0
        indexes[ringNums == self.configconfig.numRings] = self._numTracts - 1
 
        # We now do the full lookup for all non-polar tracts that have not
        # been set.
        inRange, = np.where(indexes < 0)
 
        # Do the ra search
        _ringNumArray = np.array(self._ringNums)
        _ringCumulative = np.cumsum(np.insert(_ringNumArray, 0, 0))
 
        deltaWrap = (_ra[inRange] - self._raStart.asRadians()) % (2.*np.pi)
        tractNum = (deltaWrap/(2.*np.pi/_ringNumArray[ringNums[inRange]]) + 0.5).astype(np.int32)
        # Allow wraparound
        tractNum[tractNum == _ringNumArray[ringNums[inRange]]] = 0
 
        if self._version_version_version == 0:
            # Account for off-by-one error in getRingIndices
            # Note that this means tract 1 gets duplicated
            offByOne, = np.where((tractNum == 0)
                                 & (ringNums[inRange] != 0))
            ringNums[inRange[offByOne]] += 1
 
        indexes[inRange] = _ringCumulative[ringNums[inRange]] + tractNum + 1
 
        return indexes
 
    def findAllTracts(self, coord):
        """Find all tracts which include the specified coord.
 
        Parameters
        ----------
        coord : `lsst.geom.SpherePoint`
            ICRS sky coordinate to search for.
 
        Returns
        -------
        tractList : `list` of `TractInfo`
            The tracts which include the specified coord.
        """
        ringNum = self._decToRingNum(coord.getLatitude())
 
        tractList = list()
        # ringNum denotes the closest ring to the specified coord
        # I will check adjacent rings which may include the specified coord
        for r in [ringNum - 1, ringNum, ringNum + 1]:
            if r < 0 or r >= self.configconfig.numRings:
                # Poles will be checked explicitly outside this loop
                continue
            tractNum = self._raToTractNum(coord.getLongitude(), r)
            # Adjacent tracts will also be checked.
            for t in [tractNum - 1, tractNum, tractNum + 1]:
                # Wrap over raStart
                if t < 0:
                    t = t + self._ringNums[r]
                elif t > self._ringNums[r] - 1:
                    t = t - self._ringNums[r]
 
                extra = 0
                if self._version_version_version == 0 and t == 0 and r != 0:
                    # Account for off-by-one error in getRingIndices
                    # Note that this means that tract 1 gets duplicated.
                    extra = 1
 
                index = sum(self._ringNums[:r + extra], t + 1)  # Allow 1 for south pole
                tract = self[index]
                if tract.contains(coord):
                    tractList.append(tract)
 
        # Always check tracts at poles
        # Southern cap is 0, Northern cap is the last entry in self
        for entry in [0, len(self)-1]:
            tract = self[entry]
            if tract.contains(coord):
                tractList.append(tract)
 
        return tractList
 
    def findTractPatchList(self, coordList):
        retList = []
        for coord in coordList:
            for tractInfo in self.findAllTracts(coord):
                patchList = tractInfo.findPatchList(coordList)
                if patchList and not (tractInfo, patchList) in retList:
                    retList.append((tractInfo, patchList))
        return retList
 
    def updateSha1(self, sha1):
        """Add subclass-specific state or configuration options to the SHA1."""
        sha1.update(struct.pack("<id", self.configconfig.numRings, self.configconfig.raStart))