Commit ce79980c authored by Philipp Arras's avatar Philipp Arras
Browse files

Add checks

parent b39f819c
Pipeline #63190 failed with stages
in 4 minutes and 14 seconds
......@@ -53,11 +53,16 @@ def _normal(mean, sig, key):
sig*ducktape(DomainTuple.scalar_domain(), None, key))
def _log_k_lengths(pspace):
return np.log(pspace.k_lengths[1:])
class _SlopeRemover(EndomorphicOperator):
def __init__(self, domain, logkl):
self._domain = makeDomain(domain)
assert len(self._domain) == 1
assert isinstance(self._domain[0], PowerSpace)
self._sc = logkl/float(logkl[-1])
self._capability = self.TIMES | self.ADJOINT_TIMES
def apply(self, x, mode):
......@@ -72,13 +77,11 @@ class _SlopeRemover(EndomorphicOperator):
return from_global_data(self._tgt(mode), res)
def _log_k_lengths(pspace):
return np.log(pspace.k_lengths[1:])
class _TwoLogIntegrations(LinearOperator):
def __init__(self, target):
self._target = makeDomain(target)
assert len(self._target) == 1
assert isinstance(self._target[0], PowerSpace)
self._domain = makeDomain(
UnstructuredDomain((2, self.target.shape[0] - 2)))
self._capability = self.TIMES | self.ADJOINT_TIMES
......@@ -112,6 +115,8 @@ class _TwoLogIntegrations(LinearOperator):
class _Normalization(Operator):
def __init__(self, domain):
self._domain = self._target = makeDomain(domain)
assert len(self._domain) == 1
assert isinstance(domain[0], PowerSpace)
hspace = self._domain[0].harmonic_partner
pd = PowerDistributor(hspace, power_space=self._domain[0])
cst = pd.adjoint(full(pd.target, 1.)).to_global_data_rw()
......@@ -131,6 +136,7 @@ class _Normalization(Operator):
class _SpecialSum(EndomorphicOperator):
def __init__(self, domain):
self._domain = makeDomain(domain)
assert len(self._domain) == 1
self._capability = self.TIMES | self.ADJOINT_TIMES
def apply(self, x, mode):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment