gl_space.py 3.62 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
# 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 GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
Theo Steininger's avatar
Theo Steininger committed
13
#
Martin Reinecke's avatar
Martin Reinecke committed
14
# Copyright(C) 2013-2018 Max-Planck-Society
Theo Steininger's avatar
Theo Steininger committed
15 16 17
#
# NIFTy is being developed at the Max-Planck-Institut fuer Astrophysik
# and financially supported by the Studienstiftung des deutschen Volkes.
18

csongor's avatar
csongor committed
19 20
from __future__ import division
import numpy as np
Martin Reinecke's avatar
Martin Reinecke committed
21
from .structured_domain import StructuredDomain
Theo Steininger's avatar
Theo Steininger committed
22

Martin Reinecke's avatar
Martin Reinecke committed
23

Martin Reinecke's avatar
Martin Reinecke committed
24
class GLSpace(StructuredDomain):
Martin Reinecke's avatar
Martin Reinecke committed
25 26
    """NIFTy subclass for Gauss-Legendre pixelizations of the two-sphere.

Martin Reinecke's avatar
Martin Reinecke committed
27 28
    Its harmonic partner domain is the
    :class:`~nifty4.domains.lm_space.LMSpace`.
csongor's avatar
csongor committed
29

Martin Reinecke's avatar
Martin Reinecke committed
30 31 32 33 34
    Parameters
    ----------
    nlat : int
        Number of latitudinal bins (or rings) that are used for this
        pixelization.
Martin Reinecke's avatar
Martin Reinecke committed
35
    nlon : int, optional
Martin Reinecke's avatar
Martin Reinecke committed
36 37
        Number of longitudinal bins that are used for this pixelization.
        Default value is 2*nlat + 1.
csongor's avatar
csongor committed
38 39
    """

Martin Reinecke's avatar
Martin Reinecke committed
40 41
    _needed_for_hash = ["_nlat", "_nlon"]

Martin Reinecke's avatar
Martin Reinecke committed
42 43
    def __init__(self, nlat, nlon=None):
        super(GLSpace, self).__init__()
csongor's avatar
csongor committed
44

Martin Reinecke's avatar
Martin Reinecke committed
45 46 47 48 49 50 51 52 53
        self._nlat = int(nlat)
        if self._nlat < 1:
            raise ValueError("nlat must be a positive number.")
        if nlon is None:
            self._nlon = 2*self._nlat - 1
        else:
            self._nlon = int(nlon)
            if self._nlon < 1:
                raise ValueError("nlon must be a positive number.")
54
        self._dvol = None
csongor's avatar
csongor committed
55

56 57 58
    def __repr__(self):
        return ("GLSpace(nlat=%r, nlon=%r)" % (self.nlat, self.nlon))

59 60 61
    @property
    def harmonic(self):
        return False
csongor's avatar
csongor committed
62 63 64

    @property
    def shape(self):
65
        return (np.int((self.nlat * self.nlon)),)
csongor's avatar
csongor committed
66

67
    @property
Martin Reinecke's avatar
Martin Reinecke committed
68
    def size(self):
69
        return np.int((self.nlat * self.nlon))
70

Martin Reinecke's avatar
Martin Reinecke committed
71
    @property
72
    def scalar_dvol(self):
73 74
        return None

Martin Reinecke's avatar
Martin Reinecke committed
75 76
    # MR FIXME: this is potentially wasteful, since the return array is
    #           blown up by a factor of self.nlon
Martin Reinecke's avatar
Martin Reinecke committed
77
    @property
78
    def dvol(self):
79
        from pyHealpix import GL_weights
80 81 82
        if self._dvol is None:
            self._dvol = GL_weights(self.nlat, self.nlon)
        return np.repeat(self._dvol, self.nlon)
83

Martin Reinecke's avatar
Martin Reinecke committed
84
    @property
Martin Reinecke's avatar
Martin Reinecke committed
85 86 87
    def total_volume(self):
        return 4*np.pi

88 89
    @property
    def nlat(self):
Martin Reinecke's avatar
Martin Reinecke committed
90
        """int : number of rings in this domain"""
91 92 93 94
        return self._nlat

    @property
    def nlon(self):
Martin Reinecke's avatar
Martin Reinecke committed
95
        """int : pixels per ring in this domain"""
96 97
        return self._nlon

Martin Reinecke's avatar
Martin Reinecke committed
98
    def get_default_codomain(self):
Martin Reinecke's avatar
Martin Reinecke committed
99 100 101 102 103 104 105
        """Returns a :class:`LMSpace` object, which is capable of storing an
        accurate representation of data residing on `self` (if this data is
        band-limited).

        Returns
        -------
        LMSpace
Martin Reinecke's avatar
Martin Reinecke committed
106
            The partner domain
Martin Reinecke's avatar
Martin Reinecke committed
107
        """
Martin Reinecke's avatar
Martin Reinecke committed
108
        from .. import LMSpace
109
        return LMSpace(lmax=self._nlat-1, mmax=self._nlon//2)
Martin Reinecke's avatar
Martin Reinecke committed
110 111

    def check_codomain(self, codomain):
Martin Reinecke's avatar
Martin Reinecke committed
112 113 114 115 116 117 118 119
        """Raises `TypeError` if `codomain` is not a matching partner domain
        for `self`.

        Notes
        -----
        This function only checks whether `codomain` is of type
        :class:`LMSpace`.
        """
Martin Reinecke's avatar
Martin Reinecke committed
120 121 122
        from .. import LMSpace
        if not isinstance(codomain, LMSpace):
            raise TypeError("codomain must be a LMSpace.")