gl_space.py 3.63 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
14
15
16
17
#
# Copyright(C) 2013-2017 Max-Planck-Society
#
# 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
    """NIFTy subclass for Gauss-Legendre pixelizations [#]_ of the two-sphere.
csongor's avatar
csongor committed
26

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

    Raises
    ------
    ValueError
        If input `nlat` or `nlon` is invalid.

    See Also
    --------
Martin Reinecke's avatar
Martin Reinecke committed
43
    HPSpace, LMSpace
Martin Reinecke's avatar
Martin Reinecke committed
44
45
46
47
48
49
50
51
52

    References
    ----------
    .. [#] M. Reinecke and D. Sverre Seljebotn, 2013, "Libsharp - spherical
           harmonic transforms revisited";
           `arXiv:1303.4945 <http://www.arxiv.org/abs/1303.4945>`_
    .. [#] K.M. Gorski et al., 2005, "HEALPix: A Framework for
           High-Resolution Discretization and Fast Analysis of Data
           Distributed on the Sphere", *ApJ* 622..759G.
csongor's avatar
csongor committed
53
54
    """

Martin Reinecke's avatar
Martin Reinecke committed
55
56
    def __init__(self, nlat, nlon=None):
        super(GLSpace, self).__init__()
57
        self._needed_for_hash += ["_nlat", "_nlon"]
csongor's avatar
csongor committed
58

Martin Reinecke's avatar
Martin Reinecke committed
59
60
61
62
63
64
65
66
67
        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.")
68
        self._dvol = None
csongor's avatar
csongor committed
69

70
71
72
    def __repr__(self):
        return ("GLSpace(nlat=%r, nlon=%r)" % (self.nlat, self.nlon))

73
74
75
    @property
    def harmonic(self):
        return False
csongor's avatar
csongor committed
76
77
78

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

81
    @property
Martin Reinecke's avatar
Martin Reinecke committed
82
    def size(self):
83
        return np.int((self.nlat * self.nlon))
84

85
    def scalar_dvol(self):
86
87
        return None

Martin Reinecke's avatar
Martin Reinecke committed
88
89
    # MR FIXME: this is potentially wasteful, since the return array is
    #           blown up by a factor of self.nlon
90
    def dvol(self):
91
        from pyHealpix import GL_weights
92
93
94
        if self._dvol is None:
            self._dvol = GL_weights(self.nlat, self.nlon)
        return np.repeat(self._dvol, self.nlon)
95

Martin Reinecke's avatar
Martin Reinecke committed
96
97
98
    def total_volume(self):
        return 4*np.pi

99
100
    @property
    def nlat(self):
Theo Steininger's avatar
Theo Steininger committed
101
102
        """ Number of latitudinal bins (or rings) that are used for this
        pixelization.
103
        """
104
105
106
107
        return self._nlat

    @property
    def nlon(self):
Martin Reinecke's avatar
Martin Reinecke committed
108
        """Number of longitudinal bins that are used for this pixelization."""
109
110
        return self._nlon

Martin Reinecke's avatar
Martin Reinecke committed
111
112
    def get_default_codomain(self):
        from .. import LMSpace
113
        return LMSpace(lmax=self._nlat-1, mmax=self._nlon//2)
Martin Reinecke's avatar
Martin Reinecke committed
114
115
116
117
118

    def check_codomain(self, codomain):
        from .. import LMSpace
        if not isinstance(codomain, LMSpace):
            raise TypeError("codomain must be a LMSpace.")