From 08fa3efba9992c0c2496eefb27bcdd2b10737c0b Mon Sep 17 00:00:00 2001
From: Antoine Hoffmann <antoine.hoffmann@epfl.ch>
Date: Mon, 10 Oct 2022 11:12:39 +0200
Subject: [PATCH] use of general indices variable instead of numbers

---
 src/diagnose.F90 | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/diagnose.F90 b/src/diagnose.F90
index c76c7958..bfd9e9bc 100644
--- a/src/diagnose.F90
+++ b/src/diagnose.F90
@@ -413,12 +413,12 @@ SUBROUTINE diagnose_3d
 
   IF (write_Na00) THEN
     IF(KIN_E)THEN
-    IF (ips_e .EQ. 1) &
-      Ne00(ikys:ikye,ikxs:ikxe,izs:ize) = moments_e(ips_e,1,ikys:ikye,ikxs:ikxe,izs:ize,updatetlevel)
+    IF (CONTAINS_ip0_e) &
+      Ne00(ikys:ikye,ikxs:ikxe,izs:ize) = moments_e(ip0_e,ij0_e,ikys:ikye,ikxs:ikxe,izs:ize,updatetlevel)
     CALL write_field3d_kykxz(Ne00(ikys:ikye,ikxs:ikxe,izs:ize), 'Ne00')
     ENDIF
-    IF (ips_i .EQ. 1) &
-      Ni00(ikys:ikye,ikxs:ikxe,izs:ize) = moments_i(ips_i,1,ikys:ikye,ikxs:ikxe,izs:ize,updatetlevel)
+    IF (CONTAINS_ip0_i) &
+      Ni00(ikys:ikye,ikxs:ikxe,izs:ize) = moments_i(ip0_i,ij0_i,ikys:ikye,ikxs:ikxe,izs:ize,updatetlevel)
     CALL write_field3d_kykxz(Ni00(ikys:ikye,ikxs:ikxe,izs:ize), 'Ni00')
 
     ! CALL compute_Napjz_spectrum
-- 
GitLab