From 53cb229998c65f6bb07cd7437741c2ebe5bce8d9 Mon Sep 17 00:00:00 2001 From: Ivan Markin <git@unkaktus.art> Date: Wed, 28 Feb 2024 17:33:49 +0100 Subject: [PATCH] wave.readtxt: Correctly set Psi4 data loaded from CoRe format --- watpy/wave/wave.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/watpy/wave/wave.py b/watpy/wave/wave.py index 62f8da8..81c2687 100644 --- a/watpy/wave/wave.py +++ b/watpy/wave/wave.py @@ -466,7 +466,7 @@ class wave(object): self.h = np.array(re) + 1j *np.array(im) rp4, ip4 = np.loadtxt(os.path.join(self.path,fname.replace('Rh', 'Rpsi4')), unpack=True, usecols=[1,2]) - self.p4 = np.array(re) + 1j *np.array(im) + self.p4 = np.array(rp4) + 1j *np.array(ip4) def write_to_txt(self, var, path): """ -- GitLab