fig_subway9_map.m

clear all;


figure('pos', [10 500 1000 600] * 3/4);
fontsize = 13;
axisfontsize = 10;
lettersize = 14;

ii = 2;
jj = 1:4;

%modelfile = 'model_all_data_20samples_MAP_2alpha.mat'; % <--- preprint
%modelfile = 'model_all_data_samples=40_MAP_alpha=2.0000.mat'; % <-- sample_c
%modelfile = 'model_Exp_1_thru_4_samples=10000_alpha=1.0000_last.mat'; % sample_c
modelfile = 'model_Exp_1_thru_4_samples=10000_alpha=1.0000_eps=0.6000_last.mat';

% A: graph
%
subplot(3,6,1);

%load('model_all_data_10samples_MAP_5alpha.mat');
load(modelfile);

H = pl(ii).H{jj(1)}(1,1);
D = pl(ii).D{jj(1)}(1,1);
[h, xs, ys] = plot_subway9_graph(H, D);
%labelnode(h, 1:D.G.N, 1:D.G.N);
for i = 1:D.G.N
    text(h.XData(i) , h.YData(i) + 0.01, num2str(i), 'FontSize', 10, 'HorizontalAlignment', 'center', 'VerticalAlignment', 'middle');
end
set(gca, 'xlim', [-1 3]);
xlabel('bad');



subplot(3,6,2);

c = [1 2 2 2 2 3 3 3 1];
[h, xs, ys] = plot_subway9_graph(H, D, c);
%labelnode(h, 1:D.G.N, 1:D.G.N);
for i = 1:D.G.N
    text(h.XData(i) , h.YData(i) + 0.01, num2str(i), 'FontSize', 10, 'HorizontalAlignment', 'center', 'VerticalAlignment', 'middle');
end
set(gca, 'xlim', [-1 3]);
xlabel('good');

title('Experimental Design', 'fontsize', fontsize);


h = subplot(3,6,3);
pos = get(h, 'position');
pos(1) = pos(1) * 1.0;
pos(2) = pos(2) * 1.0;
pos(3) = pos(3) * 1.0;
pos(4) = pos(4) * 1.0;
subplot(3,6, 3, 'position', pos);
PICpng = imread('subway9_map_crop.png');
[rows columns numberOfColorChannels] = size(PICpng);
imshow(PICpng, 'InitialMagnification', 'fit');  




h = subplot(3,2,2);
pos = get(h, 'position');
pos(1) = pos(1) * 0.90;
pos(2) = pos(2) * 0.98;
pos(3) = pos(3) * 1.3;
pos(4) = pos(4) * 1.3;
subplot(3,2, 2, 'position', pos);
PICpng = imread('subway9_map_trials.png');
[rows columns numberOfColorChannels] = size(PICpng);
imshow(PICpng, 'InitialMagnification', 'fit');  




% B: Data
%


subplot(3,2,3);

%load('analyze_all_data.mat') % preprint;
load('analyze_Exp_1_thru_5.mat');

i = ii;
m = pl(i).m ./ pl(i).n;
ci = pl(i).ci ./ pl(i).n;
for j = 1:length(pl(i).m)
    se(j) = std([ones(1,pl(i).m(j)) zeros(1,pl(i).n(j) - pl(i).m(j))]) / sqrt(pl(i).n(j));
end

hold on;
col = [];
for j = jj
    h = bar(j, m(j));
    if pl(i).fake(j)
        set(h, 'facealpha', 0.5);
    end
    if isempty(col)
        col = get(h, 'facecolor');
    else
        set(h, 'facecolor', col);
    end

    %text(j, 1, sprintf('p = %.3f\nN = %d', pl(i).p(j), pl(i).n(j)));
    %text(j - 0.3, 0.9 - j * 0.1, sprintf('p = %.3f', pl(i).p(j)));
end
errorbar(m(jj), se(jj), 'linestyle', 'none', 'color', 'black');
%errorbar(m, ci, 'linestyle', 'none', 'color', 'black');
line([0 length(jj)+1], [0.5 0.5], 'linestyle', '--', 'color', [0.6 0.6 0.6]);
%h = fill([0 2 2 0], [0.5 - ci(j) 0.5 - ci(j) 0.5 + ci(j) 0.5 + ci(j)], [0.4 0.4 0.4]);
%set(h, 'facealpha', 0.5, 'edgecolor', 'none');
set(gca, 'xlim', [0 length(jj)+1]);
set(gca, 'ylim', [0 1]);
set(gca, 'ytick', [0 0.5 1]);
set(gca, 'xtick', [1 2 3 4]);
%xticklabels({'P(fewer boundaries)'});
xticklabels({'bad', 'control 1', 'control 2', 'good'});
%ylabel('fraction of participants');
ylabel('P(action 6 \rightarrow 5)');

hold off;



title('Data', 'fontsize', fontsize);


% C: Model

load(modelfile);

subplot(3,2,4);

i = ii;
m = pl(i).m ./ pl(i).n;
ci = pl(i).ci ./ pl(i).n;
for j = 1:length(pl(i).m)
    se(j) = std([ones(1,pl(i).m(j)) zeros(1,pl(i).n(j) - pl(i).m(j))]) / sqrt(pl(i).n(j));
end

hold on;
col = [];
for j = jj
    h = bar(j, m(j));
    if pl(i).fake(j)
        set(h, 'facealpha', 0.5);
    end
    if isempty(col)
        col = get(h, 'facecolor');
    else
        set(h, 'facecolor', col);
    end

    %text(j, 1, sprintf('p = %.3f\nN = %d', pl(i).p(j), pl(i).n(j)));
    %text(j - 0.3, 0.9 - j * 0.1, sprintf('p = %.3f', pl(i).p(j)));
end
errorbar(m(jj), se(jj), 'linestyle', 'none', 'color', 'black');
%errorbar(m, ci, 'linestyle', 'none', 'color', 'black');
line([0 length(jj)+1], [0.5 0.5], 'linestyle', '--', 'color', [0.6 0.6 0.6]);
%h = fill([0 2 2 0], [0.5 - ci(j) 0.5 - ci(j) 0.5 + ci(j) 0.5 + ci(j)], [0.4 0.4 0.4]);
%set(h, 'facealpha', 0.5, 'edgecolor', 'none');
set(gca, 'xlim', [0 length(jj)+1]);
set(gca, 'ylim', [0 1]);
set(gca, 'ytick', [0 0.5 1]);
set(gca, 'xtick', [1 2 3 4]);
%xticklabels({'P(fewer boundaries)'});
xticklabels({'bad', 'control 1', 'control 2', 'good'});
%ylabel('fraction of participants');



title('Model', 'fontsize', fontsize);


% D: Hierarchies

%{
j = [1 1 1 2 2 2 3 3 3 2 2 2];
for s = 1:12
    subplot(6,6, 6*4 + s);

    H = pl(i).H{j(s)}(s);
    D = pl(i).D{j(s)}(s);
    h = plot_H(H, D);
    set(h, 'XData', xs);
    set(h, 'YData', ys);

    labelnode(h, 1:D.G.N, '');
    set(gca, 'xtick', []);
    set(gca, 'ytick', []);
    set(gca, 'xlim', [-2 4]);
    h.MarkerSize = 6;

    if s == 3
        %title(['                            ', 'Example hierarchies'], 'fontsize', fontsize);
    end
    if s == 1
        ylabel('bad');
    end
    if s == 4
        ylabel('control 1');
    end
    if s == 7
        ylabel('control 2');
    end
    if s == 10
        ylabel('good');
    end
end
%}



ax1 = axes('Position',[0 0 1 1],'Visible','off');
axes(ax1);
text(0.09, 0.96, 'A', 'FontSize', lettersize, 'FontWeight', 'bold');
text(0.09, 0.67, 'B', 'FontSize', lettersize, 'FontWeight', 'bold');
text(0.53, 0.67, 'C', 'FontSize', lettersize, 'FontWeight', 'bold');
%text(0.09, 0.35, 'D', 'FontSize', lettersize, 'FontWeight', 'bold');



% save figure
h = gcf;
%set(h, 'PaperPositionMode', 'auto');
set(h, 'PaperOrientation', 'landscape');
print('figures/subway9_map.pdf', '-dpdf');