From 99827b3bc2d4d28fbd2956a1a56c419a2cd360a9 Mon Sep 17 00:00:00 2001
From: Antoine Cyril David Hoffmann <ahoffman@spcpc606.epfl.ch>
Date: Mon, 22 Jun 2020 14:46:47 +0200
Subject: [PATCH] write first ion moment Ni00 as 2D output

---
 src/diagnose.F90 | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/diagnose.F90 b/src/diagnose.F90
index 7e8c419f..61f8bbd3 100644
--- a/src/diagnose.F90
+++ b/src/diagnose.F90
@@ -65,16 +65,23 @@ SUBROUTINE diagnose(kstep)
      IF (cstep==0) THEN 
       iframe5d=0
      END IF
-     CALL attach(fidres,"/data/var5d/" , "frames", iframe3d)
+     CALL attach(fidres,"/data/var5d/" , "frames", iframe5d)
 
      !  File group
      CALL creatg(fidres, "/files", "files")
      CALL attach(fidres, "/files",  "jobnum", jobnum)
 
-     !  var2d group (electro. pot.)
+     !  var2d group (electro. pot., Ni00 moment)
      rank = 0
      CALL creatd(fidres, rank, dims,  "/data/var2d/time",     "Time t*c_s/R")
      CALL creatd(fidres, rank, dims, "/data/var2d/cstep", "iteration number")
+
+     IF (write_Ni00) THEN
+      CALL creatg(fidres, "/data/var2d/Ni00", "Ni00")
+      CALL putarr(fidres, "/data/var2d/Ni00/coordkr", krarray(ikrs:ikre), "kr*rho_s0",ionode=0)     
+      CALL putarr(fidres, "/data/var2d/Ni00/coordkz", kzarray(ikzs:ikze), "kz*rho_s0",ionode=0)     
+     END IF
+
      IF (write_phi) THEN
       CALL creatg(fidres, "/data/var2d/phi", "phi")
       CALL putarr(fidres, "/data/var2d/phi/coordkr", krarray(ikrs:ikre), "kr*rho_s0",ionode=0)     
@@ -231,6 +238,9 @@ SUBROUTINE diagnose_2d
      CALL write_field2d(phi(:,:), 'phi')
   END IF
 
+  IF (write_Ni00) THEN
+     CALL write_field2d(moments_i(1,1,:,:,updatetlevel), 'Ni00')
+  END IF
 CONTAINS
 
   SUBROUTINE write_field2d(field, text)
-- 
GitLab