audio = audioread('ontwikkelen.wav');

%% linear simple
if 1
    %calculate Yk
    [ xhat,y1, y2, y3, y4, y5, y1hat,y2hat,y3hat,y4hat,y5hat ] = simple_sbc( audio, 1, 0, ones(1,5), ones(1,5)*4);
    Yls(1) = prctile(y1,99.9);
    Yls(2) = prctile(y2,99.9);
    Yls(3) = prctile(y3,99.9);
    Yls(4) = prctile(y4,99.9);
    Yls(5) = prctile(y5,99.9);
    [ xhat,y1, y2, y3, y4, y5, y1hat,y2hat,y3hat,y4hat,y5hat ] = simple_sbc( audio, 1, 0, Yls, ones(1,5)*4);
    E(1) = sum((y1-y1hat).^2);
    E(2) = sum((y2-y2hat).^2);
    E(3) = sum((y3-y3hat).^2);
    E(4) = sum((y4-y4hat).^2);
    E(5) = sum((y5-y5hat).^2);

    %calculate Bk
    Boptls = zeros(1,5);
    minEBls = +inf;
    for i1 = [2:8]
        for i2 = [2:8]
            for i3 = [2:8]
                for i4 = [2:8]
                    for i5 = [2:8]
                        B = [i1, i2, i3, i4, i5];
                        Bsbc = sum(B)/5;
                        if (Bsbc == 4)
                            EB = sum(E.*4.^(4-B));
                            if EB < minEBls
                                minEBls = EB
                                Boptls = B
                            end
                        end
                    end
                end
            end
        end
    end
end


%% log simple
if 1
    %calculate Yk
    [ xhat,y1, y2, y3, y4, y5, y1hat,y2hat,y3hat,y4hat,y5hat ] = simple_sbc( audio, 2, 0, ones(1,5), ones(1,5)*4);
    Yos(1) = prctile(y1,99.9);
    Yos(2) = prctile(y2,99.9);
    Yos(3) = prctile(y3,99.9);
    Yos(4) = prctile(y4,99.9);
    Yos(5) = prctile(y5,99.9);
    [ xhat,y1, y2, y3, y4, y5, y1hat,y2hat,y3hat,y4hat,y5hat ] = simple_sbc( audio, 2, 0, Yos, ones(1,5)*4);
    E(1) = sum((y1-y1hat).^2);
    E(2) = sum((y2-y2hat).^2);
    E(3) = sum((y3-y3hat).^2);
    E(4) = sum((y4-y4hat).^2);
    E(5) = sum((y5-y5hat).^2);

    %calculate Bk
    Boptos = zeros(1,5);
    minEBos = +inf;
    for i1 = [2:8]
        for i2 = [2:8]
            for i3 = [2:8]
                for i4 = [2:8]
                    for i5 = [2:8]
                        B = [i1, i2, i3, i4, i5];
                        Bsbc = sum(B)/5;
                        if (Bsbc == 4)
                            EB = sum(E.*4.^(4-B));
                            if EB < minEBos
                                minEBos = EB
                                Boptos = B
                            end
                        end
                    end
                end
            end
        end
    end
end




%% linear complex
if 1
    %calculate Yk
    [ xhat,y1, y2, y3, y4, y5, y1hat,y2hat,y3hat,y4hat,y5hat ] = complex_sbc( audio, 1, 0, ones(1,5), ones(1,5)*4);
    Ylc(1) = prctile(y1,99.9);
    Ylc(2) = prctile(y2,99.9);
    Ylc(3) = prctile(y3,99.9);
    Ylc(4) = prctile(y4,99.9);
    Ylc(5) = prctile(y5,99.9);
    [ xhat,y1, y2, y3, y4, y5, y1hat,y2hat,y3hat,y4hat,y5hat ] = complex_sbc( audio, 1, 0, Ylc, ones(1,5)*4);
    E(1) = sum((y1-y1hat).^2);
    E(2) = sum((y2-y2hat).^2);
    E(3) = sum((y3-y3hat).^2);
    E(4) = sum((y4-y4hat).^2);
    E(5) = sum((y5-y5hat).^2);


    %calculate Bk
    Boptlc = zeros(1,5);
    minEBlc = +inf;
    for i1 = [2:8]
        for i2 = [2:8]
            for i3 = [2:8]
                for i4 = [2:8]
                    for i5 = [2:8]
                        B = [i1, i2, i3, i4, i5];
                        Bsbc = sum(B)/5;
                        if (Bsbc == 4)
                            EB = sum(E.*4.^(4-B));
                            if EB < minEBlc
                                minEBlc = EB
                                Boptlc = B
                            end
                        end
                    end
                end
            end
        end
    end
end


%% log complex
if 1
    %calculate Yk
    [ xhat,y1, y2, y3, y4, y5, y1hat,y2hat,y3hat,y4hat,y5hat ] = complex_sbc( audio, 2, 0, ones(1,5), ones(1,5)*4);
    Yoc(1) = prctile(y1,99.9);
    Yoc(2) = prctile(y2,99.9);
    Yoc(3) = prctile(y3,99.9);
    Yoc(4) = prctile(y4,99.9);
    Yoc(5) = prctile(y5,99.9);
    [ xhat,y1, y2, y3, y4, y5, y1hat,y2hat,y3hat,y4hat,y5hat ] = complex_sbc( audio, 2, 0, Yos, ones(1,5)*4);
    E(1) = sum((y1-y1hat).^2);
    E(2) = sum((y2-y2hat).^2);
    E(3) = sum((y3-y3hat).^2);
    E(4) = sum((y4-y4hat).^2);
    E(5) = sum((y5-y5hat).^2);


    %calculate Bk
    Boptoc = zeros(1,5);
    minEBoc = +inf;
    for i1 = [2:8]
        for i2 = [2:8]
            for i3 = [2:8]
                for i4 = [2:8]
                    for i5 = [2:8]
                        B = [i1, i2, i3, i4, i5];
                        Bsbc = sum(B)/5;
                        if (Bsbc == 4)
                            EB = sum(E.*4.^(4-B));
                            if EB < minEBoc
                                minEBoc = EB
                                Boptoc = B
                            end
                        end
                    end
                end
            end
        end
    end
end