geometry_remover.py 2.04 KB
Newer Older
 Martin Reinecke committed Jan 17, 2018 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 ``````# 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 . # # Copyright(C) 2013-2018 Max-Planck-Society # # NIFTy is being developed at the Max-Planck-Institut fuer Astrophysik # and financially supported by the Studienstiftung des deutschen Volkes. `````` Martin Reinecke committed Feb 13, 2018 19 ``````from ..domains.unstructured_domain import UnstructuredDomain `````` Martin Reinecke committed Jan 22, 2018 20 ``````from ..domain_tuple import DomainTuple `````` Martin Reinecke committed Oct 07, 2017 21 ``````from .linear_operator import LinearOperator `````` 22 `````` `````` Theo Steininger committed May 15, 2017 23 `````` `````` Martin Reinecke committed Jan 06, 2018 24 ``````class GeometryRemover(LinearOperator): `````` Martin Reinecke committed Feb 16, 2018 25 `````` """Operator which transforms between a structured and an unstructured `````` Martin Reinecke committed Feb 21, 2018 26 27 28 29 30 31 32 33 34 35 36 37 38 `````` domain. Parameters ---------- domain: Domain, tuple of Domain, or DomainTuple: the full input domain of the operator. Notes ----- The operator will convert every sub-domain of its input domain to an UnstructuredDomain with the same shape. No weighting by volume factors is carried out. """ `````` Martin Reinecke committed Feb 16, 2018 39 `````` `````` Martin Reinecke committed Jan 06, 2018 40 41 `````` def __init__(self, domain): super(GeometryRemover, self).__init__() `````` Martin Reinecke committed Sep 18, 2017 42 `````` self._domain = DomainTuple.make(domain) `````` Martin Reinecke committed Feb 02, 2018 43 `````` target_list = [UnstructuredDomain(dom.shape) for dom in self._domain] `````` Martin Reinecke committed Sep 18, 2017 44 `````` self._target = DomainTuple.make(target_list) `````` Theo Steininger committed Jul 29, 2017 45 `````` `````` 46 47 48 49 50 51 52 53 `````` @property def domain(self): return self._domain @property def target(self): return self._target `````` Martin Reinecke committed Dec 28, 2017 54 55 56 57 58 59 `````` @property def capability(self): return self.TIMES | self.ADJOINT_TIMES def apply(self, x, mode): self._check_input(x, mode) `````` Martin Reinecke committed Jan 06, 2018 60 `````` if mode == self.TIMES: `````` Martin Reinecke committed Feb 21, 2018 61 62 `````` return x.cast_domain(self._target) return x.cast_domain(self._domain)``````