From 8f90cb21daef0e81cfcd6545129a598bf3dc1f99 Mon Sep 17 00:00:00 2001
From: Michele Guerini Rocco <m.guerinirocco2@campus.unimib.it>
Date: Tue, 27 Sep 2022 18:30:43 +0200
Subject: [PATCH] gray_jetto1beam: activate some debug units by defaults

---
 src/gray_jetto1beam.f90 | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/gray_jetto1beam.f90 b/src/gray_jetto1beam.f90
index 8c4b706..55f725d 100644
--- a/src/gray_jetto1beam.f90
+++ b/src/gray_jetto1beam.f90
@@ -2,6 +2,7 @@ subroutine gray_jetto1beam(ijetto, mr, mz, r, z, psin, psia, rax, zax, &
     nbnd, rbnd, zbnd, nrho, psrad, fpol, te, dne, zeff, qpsi, ibeam,   &
     p0mw, alphain, betain, dpdv, jcd, pabs, icd, error)
   use const_and_precisions, only: wp_
+  use units,                only: set_active_units, close_units
   use gray_params,          only: gray_parameters, gray_data, gray_results
   use gray_core,            only: gray_main
 
@@ -29,6 +30,16 @@ subroutine gray_jetto1beam(ijetto, mr, mz, r, z, psin, psia, rax, zax, &
   first_call: if (firstcall) then
     firstcall = .false.
 
+    ! Activate some debug units
+    set_units: block
+      use units
+      call set_active_units([         &
+        ucenr, usumm, uprj0, uprj0+1, &
+        uwbm, udisp, ubres, ucnt,     &
+        uoutr, umaps, uprfin, uflx,   &
+        upec])
+    end block set_units
+
     ! Read parameters from external file
     init_params: block
       use gray_params, only : read_gray_params, set_globals
@@ -133,4 +144,6 @@ subroutine gray_jetto1beam(ijetto, mr, mz, r, z, psin, psia, rax, zax, &
   dpdv = res%dpdv
   jcd  = res%jcd
 
+  call close_units
+
 end subroutine gray_jetto1beam