0;


set(0,'DefaultAxesFontName','Times')
set(0,'DefaultAxesFontSize',10)
widthCM=13.5; % width of plots in cm.  JFM width is 135mm.



dim = 2;

if dim == 2
    load('s_hatvalues2D_eta40_alphamx13_2000.mat','omega_grid',...
        's_hat_posvecgrid','calcparams')
    load('s_hatWHvalues2D.mat','omega_gridWH','s_hat_gridWH')
    X1 = calcparams.X1;
    X  = 1;
else
    load('s_hatvalues3D_eta40_alphamx8_1200.mat','omega_grid',...
        's_hat_posvecgrid','calcparams')
    load('s_hatWHvalues3D.mat','omega_gridWH','s_hat_gridWH')
    R1 = calcparams.R1;
    R  = 0.5;
end


hold off

if dim == 2

    plot(omega_gridWH*X,s_hat_gridWH(:,1)/X,...
        'DisplayName','$$\hat{s}_0^1/X$$, W-H','Color','r')
    hold on
    plot(omega_gridWH*X,s_hat_gridWH(:,3)/X,...
        'DisplayName','$$\hat{s}_2^1/X$$, W-H','Color','g')
    plot(omega_gridWH*X,s_hat_gridWH(:,5)/X,...
        'DisplayName','$$\hat{s}_4^1/X$$, W-H','Color','b')
    plot(omega_gridWH*X,s_hat_gridWH(:,7)/X,...
        'DisplayName','$$\hat{s}_6^1/X$$, W-H','Color','c')
    plot(omega_gridWH*X,s_hat_gridWH(:,9)/X,...
        'DisplayName','$$\hat{s}_8^1/X$$, W-H','Color','m')
    plot(omega_gridWH*X,s_hat_gridWH(:,11)/X,...
        'DisplayName','$$\hat{s}_{10}^1/X$$, W-H','Color','y')
    plot(omega_gridWH*X,s_hat_gridWH(:,13)/X,...
        'DisplayName','$$\hat{s}_{12}^1/X$$, W-H','Color','k')

    % plot(omega_gridWH*X,s_hat_gridWH(:,2)/X, ...
    %     'DisplayName','$$\hat{s}_1^1/X$$, W-H','Color','r')
    % hold on
    % plot(omega_gridWH*X,s_hat_gridWH(:,4)/X, ...
    %     'DisplayName','$$\hat{s}_3^1/X$$, W-H','Color','g')
    % plot(omega_gridWH*X,s_hat_gridWH(:,6)/X, ...
    %     'DisplayName','$$\hat{s}_5^1/X$$, W-H','Color','b')
    % plot(omega_gridWH*X,s_hat_gridWH(:,8)/X, ...
    %     'DisplayName','$$\hat{s}_7^1/X$$, W-H','Color','c')
    % plot(omega_gridWH*X,s_hat_gridWH(:,10)/X, ...
    %     'DisplayName','$$\hat{s}_9^1/X$$, W-H','Color','m')

    plot(omega_grid*X1,reshape(s_hat_posvecgrid(1,1,:)/X1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_0^1/X$$','Color',...
        'r','LineStyle','-.')
    plot(omega_grid*X1,reshape(s_hat_posvecgrid(3,1,:)/X1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_2^1/X$$','Color',...
        'g','LineStyle','-.')
    plot(omega_grid*X1,reshape(s_hat_posvecgrid(5,1,:)/X1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_4^1/X$$','Color',...
        'b','LineStyle','-.')
    plot(omega_grid*X1,reshape(s_hat_posvecgrid(7,1,:)/X1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_6^1/X$$','Color',...
        'c','LineStyle','-.')
    plot(omega_grid*X1,reshape(s_hat_posvecgrid(9,1,:)/X1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_8^1/X$$','Color',...
        'm','LineStyle','-.')
    plot(omega_grid*X1,reshape(s_hat_posvecgrid(11,1,:)/X1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_{10}^1/X$$','Color',...
        'y','LineStyle','-.')
    plot(omega_grid*X1,reshape(s_hat_posvecgrid(13,1,:)/X1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_{12}^1/X$$','Color',...
        'k','LineStyle','-.')

    % plot(omega_grid*X2,reshape(s_hat_posmatgrid(2,2,1,:)/X2,...
    %     size(omega_grid)),'DisplayName','$$\hat{s}_1^1/X$$','Color',...
    %     'r','LineStyle','--')
    % plot(omega_grid*X2,reshape(s_hat_posmatgrid(4,4,1,:)/X2,...
    %     size(omega_grid)),'DisplayName','$$\hat{s}_3^1/X$$','Color',...
    %     'g','LineStyle','--')
    % plot(omega_grid*X2,reshape(s_hat_posmatgrid(6,6,1,:)/X2,...
    %     size(omega_grid)),'DisplayName','$$\hat{s}_5^1/X$$',...
    %     'Color','b','LineStyle','--')
    % plot(omega_grid*X2,reshape(s_hat_posmatgrid(8,8,1,:)/X2,...
    %     size(omega_grid)),'DisplayName','$$\hat{s}_7^1/X$$','Color',...
    %     'c','LineStyle','--')
    % plot(omega_grid*X2,reshape(s_hat_posmatgrid(10,10,1,:)/X2,...
    %     size(omega_grid)),'DisplayName','$$\hat{s}_9^1/X$$','Color',...
    %     'm','LineStyle','--')



    xlabel('$$\omega X_1$$')

elseif dim == 3
    plot(omega_gridWH*R,s_hat_gridWH(:,1)/R,...
        'DisplayName','$$\hat{s}_{(0,0,0)}^1/R$$, W-H','Color','r')
    hold on
    plot(omega_gridWH*R,s_hat_gridWH(:,4)/R,...
        'DisplayName','$$\hat{s}_{(0,1,0)}^1/R$$, W-H','Color','g')
    plot(omega_gridWH*R,s_hat_gridWH(:,10)/R,...
        'DisplayName','$$\hat{s}_{(0,2,0)}^1/R$$, W-H','Color','b')
    plot(omega_gridWH*R,s_hat_gridWH(:,15)/R,...
        'DisplayName','$$\hat{s}_{(0,3,0)}^1/R$$, W-H','Color','c')
    plot(omega_gridWH*R,s_hat_gridWH(:,21)/R,...
        'DisplayName','$$\hat{s}_{(0,4,0)}^1/R$$, W-H','Color','m')
    plot(omega_gridWH*R,s_hat_gridWH(:,27)/R,...
        'DisplayName','$$\hat{s}_{(0,5,0)}^1/R$$, W-H','Color','y')
    plot(omega_gridWH*R,s_hat_gridWH(:,33)/R,...
        'DisplayName','$$\hat{s}_{(0,6,0)}^1/R$$, W-H','Color','k')

    % plot(omega_gridWH*R,s_hat_gridWH(:,2)/R,...
    %     'DisplayName','$$\hat{s}_{(1,1,0)}^1/R$$, W-H','Color','r')
    % hold on
    % plot(omega_gridWH*R,s_hat_gridWH(:,7)/R,...
    %     'DisplayName','$$\hat{s}_{(1,2,0)}^1/R$$, W-H','Color','g')
    % plot(omega_gridWH*R,s_hat_gridWH(:,12)/R,...
    %     'DisplayName','$$\hat{s}_{(1,3,0)}^1/R$$, W-H','Color','b')
    % plot(omega_gridWH*R,s_hat_gridWH(:,18)/R,...
    %     'DisplayName','$$\hat{s}_{(1,4,0)}^1/R$$, W-H','Color','c')
    % plot(omega_gridWH*R,s_hat_gridWH(:,24)/R,...
    %     'DisplayName','$$\hat{s}_{(1,5,0)}^1/R$$, W-H','Color','m')
    % plot(omega_gridWH*R,s_hat_gridWH(:,30)/R,...
    %     'DisplayName','$$\hat{s}_{(1,6,0)}^1/R$$, W-H','Color','y')

    % plot(omega_gridWH*R,s_hat_gridWH(:,3)/R,...
    %     'DisplayName','$$\hat{s}_{(2,1,0)}^1/R$$, W-H','Color','r')
    % hold on
    % plot(omega_gridWH*R,s_hat_gridWH(:,9)/R,...
    %     'DisplayName','$$\hat{s}_{(2,2,0)}^1/R$$, W-H','Color','g')
    % plot(omega_gridWH*R,s_hat_gridWH(:,14)/R,...
    %     'DisplayName','$$\hat{s}_{(2,3,0)}^1/R$$, W-H','Color','b')
    % plot(omega_gridWH*R,s_hat_gridWH(:,20)/R,...
    %     'DisplayName','$$\hat{s}_{(2,4,0)}^1/R$$, W-H','Color','c')
    % plot(omega_gridWH*R,s_hat_gridWH(:,26)/R,...
    %     'DisplayName','$$\hat{s}_{(2,5,0)}^1/R$$, W-H','Color','m')
    % plot(omega_gridWH*R,s_hat_gridWH(:,32)/R,...
    %     'DisplayName','$$\hat{s}_{(2,6,0)}^1/R$$, W-H','Color','y')


    % plot(omega_gridWH*R,s_hat_gridWH(:,8)/R,...
    %     'DisplayName','$$\hat{s}_{(5,1,0)}^1/R$$, W-H','Color','r')
    % hold on
    % plot(omega_gridWH*R,s_hat_gridWH(:,16)/R,...
    %     'DisplayName','$$\hat{s}_{(5,2,0)}^1/R$$, W-H','Color','g')
    % plot(omega_gridWH*R,s_hat_gridWH(:,22)/R,...
    %     'DisplayName','$$\hat{s}_{(5,3,0)}^1/R$$, W-H','Color','b')
    % plot(omega_gridWH*R,s_hat_gridWH(:,28)/R,...
    %     'DisplayName','$$\hat{s}_{(5,4,0)}^1/R$$, W-H','Color','c')

    plot(omega_grid*R1,reshape(s_hat_posvecgrid(1,1,:)/R1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_{(0,0,0)}^1/R$$',...
        'Color','r','LineStyle','-.')
    plot(omega_grid*R1,reshape(s_hat_posvecgrid(2,1,:)/R1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_{(0,1,0)}^1/R$$',...
        'Color','g','LineStyle','-.')
    plot(omega_grid*R1,reshape(s_hat_posvecgrid(3,1,:)/R1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_{(0,2,0)}^1/R$$',...
        'Color','b','LineStyle','-.')
    plot(omega_grid*R1,reshape(s_hat_posvecgrid(4,1,:)/R1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_{(0,3,0)}^1/R$$',...
        'Color','c','LineStyle','-.')
    plot(omega_grid*R1,reshape(s_hat_posvecgrid(5,1,:)/R1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_{(0,4,0)}^1/R$$',...
        'Color','m','LineStyle','-.')
    plot(omega_grid*R1,reshape(s_hat_posvecgrid(6,1,:)/R1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_{(0,5,0)}^1/R$$',...
        'Color','y','LineStyle','-.')
    plot(omega_grid*R1,reshape(s_hat_posvecgrid(7,1,:)/R1,...
        size(omega_grid)),'DisplayName','$$\hat{s}_{(0,6,0)}^1/R$$',...
        'Color','k','LineStyle','-.')
    xlabel('$$\omega R_1$$')

end


legend

exportgraphics(gcf,'endcorrectionplot.pdf','ContentType','vector')
% exportgraphics(gcf,'endcorrectionplot.jpg','Resolution',500)
