Homec4science
Diffusion GYACOMO (OBSOLETE: LATEST VERSION ON gitlab.epfl.ch) (shear_version) Browse

gyacomo/matlab/plot/plot_ballooning.m
ea846d5c10f7shear_version

Last Change
Raw File

plot_ballooning.m
View Options

function [FIG] = plot_ballooning(data,options)
FIG.fig = figure; iplot = 1;
[~,it0] = min(abs(data.Ts3D - options.time_2_plot(1)));
[~,it1] = min(abs(data.Ts3D - options.time_2_plot(end)));
[~,ikyarray] = min(abs(data.ky - options.kymodes));
% phi_real=mean(real(data.PHI(:,:,:,it0:it1)),4);
% phi_imag=mean(imag(data.PHI(:,:,:,it0:it1)),4);
phi_real=real(data.PHI(:,:,:,it1));
phi_imag=imag(data.PHI(:,:,:,it1));
% Apply baollooning tranform
for iky=ikyarray
dims = size(phi_real);
Nkx = dims(2);
is = max(1,iky-1);
Npi = (Nkx-1)-2*(is-1);
if(Npi <= 0)
break
elseif(Npi == 1)
ordered_ikx = 1;
else
tmp_ = (Nkx-is+1):-is:(Nkx/2+2);
ordered_ikx = [tmp_(end:-1:1), 1:is:(Nkx/2)];
end
idx=data.kx./data.kx(2);
idx= idx(ordered_ikx);
Nkx = numel(idx);
phib_real = zeros( Nkx*dims(3) ,1);
phib_imag = phib_real;
b_angle = phib_real;
kk_=[];
ss_=[];
ee_=[];
for i_ =1:Nkx
start_ = (i_ -1)*dims(3) +1;
end_ = i_*dims(3);
ikx = ordered_ikx(i_);
phib_real(start_:end_) = phi_real(iky,ikx,:);
phib_imag(start_:end_) = phi_imag(iky,ikx,:);
kk_ = [kk_ data.kx(ikx)];
ss_ = [ss_ start_];
ee_ = [ee_ end_];
end
% Define ballooning angle
coordz = data.z;
for i_ =1: Nkx
for iz=1:dims(3)
ii = dims(3)*(i_ -1) + iz;
b_angle(ii) =coordz(iz) + 2*pi*idx(i_)./is;
end
end
phib = phib_real(:) + 1i * phib_imag(:);
% normalize real and imaginary parts at chi =0
if options.normalized
[~,idxLFS] = min(abs(b_angle -0));
normalization = phib( idxLFS);
else
normalization = 1;
end
phib_norm = phib / normalization ;
phib_real_norm = real( phib_norm);%phib_real(:)/phib_real(idxLFS);
phib_imag_norm = imag( phib_norm);%phib_imag(:)/ phib_imag(idxLFS);
subplot(numel(ikyarray),1,iplot)
plot(b_angle/pi, phib_real_norm,'-b'); hold on;
plot(b_angle/pi, phib_imag_norm ,'-r');
plot(b_angle/pi, sqrt(phib_real_norm .^2 + phib_imag_norm.^2),'-k');
legend('real','imag','norm')
xlabel('$\chi / \pi$')
ylabel('$\phi_B (\chi)$');
title(['$k_y=',sprintf('%1.1f',data.ky(iky)),...
',t_{avg}\in [',sprintf('%1.1f',data.Ts3D(it0)),',',sprintf('%1.1f',data.Ts3D(it1)),']$']);
iplot = iplot + 1;
end
end
c4science · Help