From 1a41c14b0cd5189c11c1cb7d8a4e01b0406d94c6 Mon Sep 17 00:00:00 2001
From: Antoine Cyril David Hoffmann <ahoffman@spcpc606.epfl.ch>
Date: Mon, 17 May 2021 15:53:08 +0200
Subject: [PATCH] Source flag to remove ei ie collision terms

---
 src/collision_mod.F90 | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/collision_mod.F90 b/src/collision_mod.F90
index 4abc1bcc..2bdeb5ce 100644
--- a/src/collision_mod.F90
+++ b/src/collision_mod.F90
@@ -400,6 +400,8 @@ CONTAINS
 
     CHARACTER(len=256) :: mat_filename, kperp_string, NFLR_string
 
+    LOGICAL     :: CO_AA_ONLY = .false. ! Flag to remove ei ie collision
+
     !! Some terminal info
     IF (CO .EQ. 2) THEN
       IF (my_id .EQ. 0) WRITE(*,*) '=== Load GK Sugama matrix ==='
@@ -446,7 +448,7 @@ CONTAINS
           mat_filename = selfmat_file
         ENDIF
         ! write(*,*) 'loading ', mat_filename
-        
+
         CALL openf(mat_filename,fid1, 'r', 'D', mpicomm=comm_p);
         CALL getatt(fid1,'/Caapj/Ceepj/','Pmaxe',pdime)
         CALL getatt(fid1,'/Caapj/Ceepj/','Jmaxe',jdime)
@@ -638,6 +640,13 @@ CONTAINS
     DEALLOCATE (Ceepj__kp ); DEALLOCATE (CeipjT_kp); DEALLOCATE (CeipjF_kp)
     DEALLOCATE (Ciipj__kp ); DEALLOCATE (CiepjT_kp); DEALLOCATE (CiepjF_kp)
 
+    IF( CO_AA_ONLY ) THEN
+      CeipjF = 0._dp;
+      CeipjT = 0._dp;
+      CiepjF = 0._dp;
+      CiepjT = 0._dp;
+    ENDIF
+
     IF (my_id .EQ. 0) WRITE(*,*) '============DONE==========='
 
   END SUBROUTINE load_COSOlver_mat
-- 
GitLab