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