Add .imag to Operator

nifty6/linearization.py

@@ -133,6 +133,10 @@ class Linearization(Operator):
     def real(self):
         return self.new(self._val.real, self._jac.real)
 
+    @property
+    def imag(self):
+        return self.new(self._val.imag, self._jac.imag)
+
     def _myadd(self, other, neg):
         if np.isscalar(other) or other.jac is None:
             return self.new(self._val-other if neg else self._val+other,

nifty6/operators/operator.py

@@ -143,6 +143,11 @@ class Operator(metaclass=NiftyMeta):
         from .simple_linear_operators import Realizer
         return Realizer(self.target)(self)
 
+    @property
+    def imag(self):
+        from .simple_linear_operators import Imaginizer
+        return Imaginizer(self.target)(self)
+
     def __neg__(self):
         return self.scale(-1)
 

nifty6/operators/simple_linear_operators.py

@@ -112,6 +112,24 @@ class Realizer(EndomorphicOperator):
         return x.real
 
 
+class Imaginizer(EndomorphicOperator):
+    """Operator returning the imaginary component of its input.
+
+    Parameters
+    ----------
+    domain: Domain, tuple of domains or DomainTuple
+        domain of the input field
+
+    """
+    def __init__(self, domain):
+        self._domain = DomainTuple.make(domain)
+        self._capability = self.TIMES | self.ADJOINT_TIMES
+
+    def apply(self, x, mode):
+        self._check_input(x, mode)
+        return x.imag
+
+
 class FieldAdapter(LinearOperator):
     """Operator for conversion between Fields and MultiFields.