From 52e4fd8e86e226106745bb06cfe147954616fe32 Mon Sep 17 00:00:00 2001
From: Antoine Hoffmann <antoine.hoffmann@epfl.ch>
Date: Wed, 20 Sep 2023 09:53:02 +0200
Subject: [PATCH] improve single mode init
---
src/initial_mod.F90 | 17 ++++++++++-------
1 file changed, 10 insertions(+), 7 deletions(-)
diff --git a/src/initial_mod.F90 b/src/initial_mod.F90
index befe8f65..82863512 100644
--- a/src/initial_mod.F90
+++ b/src/initial_mod.F90
@@ -271,15 +271,18 @@ CONTAINS
USE fields, ONLY: moments
USE prec_const, ONLY: xp
USE parallel, ONLY: my_id
+ USE grid, ONLY: local_nkx, local_nkx_offset, local_nky, local_nky_offset
IMPLICIT NONE
+ INTEGER :: ikx,iky
moments = 0._xp
- IF (my_id .EQ. 0) THEN
- moments(:,:,:,iky_init,ikx_init,:,:) = init_amp
- ! moments(:,:,:,2,2,:,:) = init_amp
- ! moments(:,:,:,4,3,:,:) = init_amp
- ! moments(:,:,:,4,4,:,:) = init_amp
- ! moments(:,:,:,5,5,:,:) = init_amp
- ENDIF
+ DO ikx=1,local_nkx
+ DO iky=1,local_nky
+ IF ( (ikx+local_nkx_offset .EQ. ikx_init) .AND. &
+ (iky+local_nky_offset .EQ. iky_init) ) THEN
+ moments(:,:,:,iky,ikx,:,:) = init_amp
+ ENDIF
+ ENDDO
+ ENDDO
END SUBROUTINE init_single_mode
!******************************************************************************!
--
GitLab