From 797f6f0cd4c5a35445f97f8d0cf09a5577c7863f Mon Sep 17 00:00:00 2001
From: Nuwan Yapa <yapanuwan@gmail.com>
Date: Wed, 16 Apr 2025 17:00:48 -0400
Subject: [PATCH] Avoid dirty trick if unnecessary

---
 nucleons.jl | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/nucleons.jl b/nucleons.jl
index 095c212..564534a 100644
--- a/nucleons.jl
+++ b/nucleons.jl
@@ -177,8 +177,10 @@ function calculateNucleonDensity(p::Bool, s::system)::Tuple{Vector{Float64}, Vec
 
     r2s = s.r_mesh.r .^ 2
     ρ = ρr2 ./ transpose(r2s)
-    ρ[:, 1] .= ρ[:, 2] # dirty fix for NaN at r=0
-    
+    if !all(isfinite.(ρ[:, 1]))
+        ρ[:, 1] .= ρ[:, 2] # dirty fix for NaN at r=0
+    end
+
     ρ_s = ρ[1, :] - ρ[2, :] # g^2 - f^2
     ρ_v = ρ[1, :] + ρ[2, :] # g^2 + f^2