From 7de2e0484f29c61c7bf72246a82321fd400f27d1 Mon Sep 17 00:00:00 2001
From: Antoine <antoine.hoffmann@epfl.ch>
Date: Thu, 3 Aug 2023 13:10:04 +0200
Subject: [PATCH] adapted for ion and electron fluxes

---
 matlab/plot/plot_radial_transport_and_spacetime.m | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/matlab/plot/plot_radial_transport_and_spacetime.m b/matlab/plot/plot_radial_transport_and_spacetime.m
index 4e2617cd..8fdcbc3d 100644
--- a/matlab/plot/plot_radial_transport_and_spacetime.m
+++ b/matlab/plot/plot_radial_transport_and_spacetime.m
@@ -27,12 +27,18 @@ mvm = @(x) movmean(x,OPTIONS.NMVA);
     FIGURE.fig = figure; FIGURE.FIGNAME = ['ZF_transport_drphi','_',DATA.params_string]; %set(gcf, 'Position',  [500, 1000, 1000, 600])
     FIGURE.ax1 = subplot(2,1,1,'parent',FIGURE.fig);
     for ia = 1:DATA.inputs.Na
+        if ia == 1
+            a = 'i'; color = 'r';
+        elseif ia == 2
+            a = 'e'; color = 'b';
+        end
+
         plot(mvm(DATA.Ts0D),mvm(DATA.PGAMMA_RI(ia,:)*SCALE),'--',...
-            'color',clr_(max(1,(DATA.grids.Np-1)/2+(ia-1)),:),...
-            'DisplayName',['$\Gamma_x$ ',DATA.paramshort]); hold on;
+            'color',color,...
+            'DisplayName',['$\Gamma_{x,',a,'}$ ',DATA.paramshort]); hold on;
         plot(mvm(DATA.Ts0D),mvm(DATA.HFLUX_X(ia,:)*SCALE),'-',...
-            'color',clr_(max(1,(DATA.grids.Np-1)/2+(ia-1)),:),...
-            'DisplayName',['$Q_x$ ',DATA.paramshort]); hold on;
+            'color',color,...
+            'DisplayName',['$Q_{x,',a,'}$ ',DATA.paramshort]); hold on;
         ylabel('Transport')  
         if(~isnan(Qx_infty_avg))
         plot(DATA.Ts0D(its0D:ite0D),ones(ite0D-its0D+1,1)*Qx_avg, '-k',...
-- 
GitLab