%% Jonathan Wrobel
%  B^2 calculation

%%

function probe_B_squared(shot)
close all
% shot = 113122;
probe_names = build_hitsi_probe_name('ALL','ALL',['P','T'],'ALL');

mdsconnect('landau.hit');
mdsopen('hitsi',shot);

dead_probes = mdsvalue('\dead_probes');
i_tor   = mdsvalue('\i_tor');
i_tor_t = mdsvalue('dim_of(\i_tor)');

B_sum = [];
for n=1:length(probe_names)
    probe_name = probe_names(n,:);
    if isempty(nonzeros(strcmpi(probe_name, dead_probes)))
        probe = mdsvalue(['\B_' probe_name]);
        if isempty(B_sum)
            B_sum = probe.^2;
            B_sum_t = mdsvalue(['dim_of(\B_' probe_name ')']);
        else
            B_sum = B_sum + probe.^2;
        end
        
    end % dead_probe check
end % probe_name loop

figure(1)
    ax(1) = subplot(2,1,1);
        plot(i_tor_t, i_tor), xlabel('time(s)'), ylabel('I_t_o_r')
        title(['Sum of B^2 compared to I_ _t_o_r    Shot: ' num2str(shot)])
    ax(2) = subplot(2,1,2);
        plot(B_sum_t, B_sum), xlabel('time(s)'), ylabel('B^2 summed')
    linkaxes(ax, 'x')