plot_phi_ballooning.m
40660cf3cbf6master

plot_phi_ballooning.m
View Options

			[~,it] = min(abs(Ts3D - time_2_plot));
			phi_real=(real(PHI(:,:,:,it)));
			phi_imag=(imag(PHI(:,:,:,it)));
			% Apply baollooning tranform
			for iky=2
			dims = size(phi_real);
			phib_real = zeros( dims(1)*dims(3) ,1);
			phib_imag= phib_real;
			b_angle = phib_real;

			midpoint = floor((dims(1)*dims(3) )/2)+1;
			for ip =1: dims(1)
			start_ = (ip -1)*dims(3) +1;
			end_ = ip*dims(3);
			phib_real(start_:end_) = phi_real(ip,iky,:);
			phib_imag(start_:end_) = phi_imag(ip,iky, :);
			end

			% Define ballooning angle
			Nkx = numel(kx)-1; coordz = z;
			idx = -Nkx:1:Nkx;
			for ip =1: dims(1)
			for iz=1:dims(3)
			ii = dims(3)*(ip -1) + iz;
			b_angle(ii) =coordz(iz) + 2piidx(ip);
			end
			end

			% normalize real and imaginary parts at chi =0
			[~,idxLFS] = min(abs(b_angle -0));
			phib = phib_real(:) + 1i * phib_imag(:);
			% Normalize to the outermid plane
			phib_norm = phib;% / phib( idxLFS) ;
			phib_real_norm = real( phib_norm);%phib_real(:)/phib_real(idxLFS);
			phib_imag_norm = imag( phib_norm);%phib_imag(:)/ phib_imag(idxLFS);


			figure; hold all;
			plot(b_angle/pi, phib_real_norm,'-b');
			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(['HeLaZ(-) molix(o) benchmark, t=',num2str(Ts3D(it))]);
			end