endomorphic_operator.py 2.13 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
# Copyright(C) 2013-2019 Max-Planck-Society
Theo Steininger's avatar
Theo Steininger committed
15
#
16
# NIFTy is being developed at the Max-Planck-Institut fuer Astrophysik.
17
18
19

import numpy as np

Martin Reinecke's avatar
Martin Reinecke committed
20
from .linear_operator import LinearOperator
21
22
23


class EndomorphicOperator(LinearOperator):
24
    """NIFTy class for endomorphic operators.
25

Martin Reinecke's avatar
Martin Reinecke committed
26
27
    The  NIFTy EndomorphicOperator describes linear operators with identical
    domain and target.
28
29
    """

30
31
    @property
    def target(self):
Martin Reinecke's avatar
Martin Reinecke committed
32
33
34
35
        """DomainTuple : returns :attr:`domain`

        Returns `self.domain`, because this is also the target domain
        for endomorphic operators."""
Martin Reinecke's avatar
Martin Reinecke committed
36
        return self._domain
clienhar's avatar
clienhar committed
37

38
    def draw_sample(self, from_inverse=False, dtype=np.float64):
clienhar's avatar
clienhar committed
39
40
41
42
43
        """Generate a zero-mean sample

        Generates a sample from a Gaussian distribution with zero mean and
        covariance given by the operator.

44
45
46
47
48
49
50
        Parameters
        ----------
        from_inverse : bool (default : False)
            if True, the sample is drawn from the inverse of the operator
        dtype : numpy datatype (default : numpy.float64)
            the data type to be used for the sample

clienhar's avatar
clienhar committed
51
52
53
54
55
56
        Returns
        -------
        Field
            A sample from the Gaussian of given covariance.
        """
        raise NotImplementedError
Martin Reinecke's avatar
Martin Reinecke committed
57

58
59
60
61
62
63
    def _dom(self, mode):
        return self._domain

    def _tgt(self, mode):
        return self._domain

Martin Reinecke's avatar
Martin Reinecke committed
64
65
    def _check_input(self, x, mode):
        self._check_mode(mode)
Martin Reinecke's avatar
Martin Reinecke committed
66
        if self.domain != x.domain:
Martin Reinecke's avatar
Martin Reinecke committed
67
            raise ValueError("The operator's and field's domains don't match.")