From 4152323248f0b6521c0ea2e6ac0d3f723d6d24fb Mon Sep 17 00:00:00 2001
From: Philipp Arras <parras@mpa-garching.mpg.de>
Date: Tue, 19 Jun 2018 16:08:15 +0200
Subject: [PATCH] Add .real and .imag to MultiField
---
nifty5/multi/multi_field.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/nifty5/multi/multi_field.py b/nifty5/multi/multi_field.py
index cce6c79a1..b878cf84e 100644
--- a/nifty5/multi/multi_field.py
+++ b/nifty5/multi/multi_field.py
@@ -52,6 +52,16 @@ class MultiField(object):
def dtype(self):
return {key: val.dtype for key, val in self._val.items()}
+ @property
+ def real(self):
+ """MultiField : The real part of the multi field"""
+ return MultiField({key: field.real for key, field in self.items()})
+
+ @property
+ def imag(self):
+ """MultiField : The imaginary part of the multi field"""
+ return MultiField({key: field.imag for key, field in self.items()})
+
@staticmethod
def from_random(random_type, domain, dtype=np.float64, **kwargs):
dtype = MultiField.build_dtype(dtype, domain)
--
GitLab