From cea91d307d9012cdfb8926892fe495e5ba8e1d10 Mon Sep 17 00:00:00 2001
From: Chichi Lalescu <chichilalescu@gmail.com>
Date: Sat, 18 Nov 2017 09:34:43 -0700
Subject: [PATCH] fix infinity issue
---
bfps/DNS.py | 3 +++
bfps/NavierStokes.py | 3 +++
2 files changed, 6 insertions(+)
diff --git a/bfps/DNS.py b/bfps/DNS.py
index 3207ccce..9d57338d 100644
--- a/bfps/DNS.py
+++ b/bfps/DNS.py
@@ -236,6 +236,9 @@ class DNS(_code):
ii0 = iter0 // self.parameters['niter_stat']
ii1 = iter1 // self.parameters['niter_stat']
self.statistics['kshell'] = data_file['kspace/kshell'].value
+ for kk in [-1, -2]:
+ if (self.statistics['kshell'][kk] == 0):
+ self.statistics['kshell'][kk] = np.nan
self.statistics['kM'] = data_file['kspace/kM'].value
self.statistics['dk'] = data_file['kspace/dk'].value
computation_needed = True
diff --git a/bfps/NavierStokes.py b/bfps/NavierStokes.py
index 59fb907c..b7bf4ff8 100644
--- a/bfps/NavierStokes.py
+++ b/bfps/NavierStokes.py
@@ -598,6 +598,9 @@ class NavierStokes(_fluid_particle_base):
ii0 = iter0 // self.parameters['niter_stat']
ii1 = iter1 // self.parameters['niter_stat']
self.statistics['kshell'] = data_file['kspace/kshell'].value
+ for kk in [-1, -2]:
+ if (self.statistics['kshell'][kk] == 0):
+ self.statistics['kshell'][kk] = np.nan
self.statistics['kM'] = data_file['kspace/kM'].value
self.statistics['dk'] = data_file['kspace/dk'].value
computation_needed = True
--
GitLab