folderpath = 'C:\Users\sioe\Nextcloud\Dokumente\02_Ablage_Office\Lab_Data_24\baudrate_sweep\';
experiment_name = 'PAM4_10km_ffe_';

ffe_only            = 0;
postfilter_approach = 0;
db_channel_approach = 1;
db_coding_approach  = 0;

db_precode = db_coding_approach || db_channel_approach;

%%% SIR Sweep for MPI Experiment %%%
params = struct;
params.fsym = [56,68,80,92].*1e9;
params.fsym = [92].*1e9;

wh = DataStorage(params);

wh.addStorage("ber");
wh.addStorage("pd_in");
wh.addStorage("rop");
wh.addStorage("m");
wh.addStorage("signals");

precomp_path = "C:\Users\sioe\Documents\MATLAB\imdd_simulation\projects\standard_system\";
precomp_fn = "lab_mpi_setup_2";
precomp_mode = 2; %0=do nothing ; 1= measure; 2=precomp active
precomp_amp_max = 2;

M           = 4;
pn_key      = 2;
usemrds     = 0;
fdac        = 92e9;
awg_vpp     = 0.15;
fadc        = 160e9;
rrcalpha    = 0.05;
v_bias      = 2.25;
pd_in_set   = 4;
rop_atten   = 0;

disp(['Start Measurement of ',num2str(prod(wh.dim)),' loops...'])

for fsym = wh.parameter.fsym.values

        loop_name = ['_fsym_',num2str(fsym)];

        %%%%% SET Voltages %%%%%%
        dcs = DC_supply("active",[1,1],"voltage",[v_bias, 9]);
        dcs.set("voltage",[v_bias, 9]);

        %%%%% SET Attenuator %%%%%%
        voa = OptAtten("active",[1,2,1,1],"value",[rop_atten,pd_in_set,0,0],"wavelength",[1310,1310,1310,1310]);
        voa.set('active',[1,2,1,1],'value',[rop_atten,pd_in_set,0,0]);
        % voa.readvals();

        %%%%% Construct AWG and Scope Modules %%%%%%
        SCP = ScopeKeysight("model","DSAZ634A",'autoscale',1,"fadc",'GSa_160',"channel",[1,0],"recordLen",2000000,"removeDC",1);
        AWG = AwgKeysight("model","M8196A","fdac",fdac,"scaletodac",[1,1,1,1],"skews",[0,0,0,0],"voltages",[0,0,0,awg_vpp]);
        A2S = Awg2Scope(AWG,SCP,[0,0,0,1]);


        %%%%% Symbol Generation %%%%%%
        Pform       = Pulseformer("fsym",fsym,"fdac",4*fsym,"pulse","rrc","pulselength",16,"rrcalpha",rrcalpha);

        [Digi_sig,Symbols,Bits] = PAMsource("fsym",fsym,"M",M,"order",18,"useprbs",1,...
            "fs_out",fdac,"applyclipping",0,"clipfactor",1.7,...
            "applypulseform",0,"pulseformer",Pform,"randkey",pn_key,...
            "db_precode",db_precode,...
            "mrds_code",usemrds,"mrds_blocklength",512,"db_encode",db_coding_approach).process();

        %%%%% Precompensation Routine %%%%%%
        if precomp_mode == 1 % measure channel
            freqresp = ChannelFreqResp("Nacq",1024,"Navg",64,"Ncp",63,'f_ref',Digi_sig.fs);
            Digi_sig = freqresp.buildOFDM();
        elseif precomp_mode == 2 % apply precomp
            freqresp = ChannelFreqResp("Nacq",1024,"Navg",64,"Ncp",63,'f_ref',Digi_sig.fs);
            Digi_sig = freqresp.precomp(Digi_sig,'maxampdb',precomp_amp_max,'loadPath',precomp_path,'fileName',precomp_fn);
        end

        %%%%% Resample to DAC rate %%%%%%
        Digi_sig = Digi_sig.resample("fs_out",AWG.fdac);


        %%%%% Plot and Save Routine 1 %%%%%%%%%%%%%%%%%%%%%%%%%
        Digi_sig.spectrum("displayname","Normal Tx","fignum",10);

        save([folderpath,[experiment_name,'bits'],loop_name],"Bits");
        save([folderpath,[experiment_name,'symbols'],loop_name],"Symbols");
        %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

        %%%%% AWG --> Scope %%%%%%
        [~,~,~,Scpe_sig] = A2S.process("signal4",Digi_sig);

        %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
        Scpe_sig.spectrum("displayname","Scope PSD","fignum",20);

        Scpe_sig.plot("displayname","Scope raw signal","fignum",25);
        %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

       
        %%%%%% Sample to 2x fsym %%%%%%
        Scpe_sig = Scpe_sig.resample("fs_in",160e9,"fs_out",2*fsym);

        %%%%% Precompensation Routine %%%%%%
        if precomp_mode == 1
            freqresp.estimate(Scpe_sig,"save",true,"savePath",precomp_path,"fileName",precomp_fn);
            freqresp.plot();
        end

        %%%%%% Sync Rx signal with reference %%%%%%
        [Scpe_sig,S] = Scpe_sig.tsynch("reference",Symbols,"fs_ref",fsym);

        %%%%%% SNR CHEAT - Avg. the measured signal occurences %%%%%%
        average_signals = 0;
        if average_signals
            scope_mean = zeros(size(S{1}.signal));
            for n=1:numel(S)
                scope_mean = scope_mean + S{n}.signal;
            end
            scope_mean = scope_mean ./ n;
            Scpe_sig.signal = scope_mean;
        end

        %%%%% Plot and Save Routine 2 %%%%%%%%%%%%%%%%%%%%%%%%%
        save([folderpath,experiment_name,'rx_signal',loop_name],"S");
        Scpe_sig.eye(fsym,M,"fignum",40,"displayname",' after Scope');

        voa.readvals();
        rop = voa.power_state(1);
        pd_in = voa.power_state(2);


        %%%%% EQUALIZE %%%%%%
        Eq = FFE("epochs_tr",5,"epochs_dd",5,"len_tr",4096*2,"mu_dd",1e-4,"mu_tr",0,"order",50,"sps",2,"decide",0);
        Eq = VNLE("epochs_tr",5,"epochs_dd",5,"len_tr",4096*2,"mu_dd",[0.0004 0.0005 0.0006],"mu_tr",0,"order",[50,7,7],"sps",2,"decide",1);
        Eq = EQ("Ne",[50,7,7],"Nb",[0,0,0],"training_length",4096*2,"training_loops",5,"dd_loops",5,"K",2,"DCmu",0.05,"DDmu",[0.0004 0.0004 0.0004 0.0004 ],"DFEmu",0.005,"FFEmu",0,"plotfinal",0,"ideal_dfe",1);

        if ffe_only %%%%%%%%%%%%%%%%%%%%%%%%%%%

            [EQ_sig] = Eq.process(Scpe_sig,Symbols);

            EQ_sig.plot("fignum",50,"displayname",'After EQ');


            Rx_bits = PAMmapper(M,0).demap(EQ_sig);
            [~,errors_bm,ber,errors] = calc_ber(Rx_bits.signal,Bits.signal,"skip_front",100,"skip_end",150,"returnErrorLocation",1);

            disp(['FFE: ',sprintf('%.1E',ber),'| ROP: ',num2str(rop),' dB | PD_in: ',num2str(pd_in),' dBm']);

            if 1
                figure(53);
                constellation = unique(Symbols.signal);
                received = NaN(numel(constellation),length(Symbols));
                for lvl = 1:numel(constellation)
                    received(lvl,Symbols.signal==constellation(lvl)) = EQ_sig.signal(Symbols.signal==constellation(lvl));
                    hold on
                    histogram(received(lvl,:),1000,"EdgeAlpha",0);
                end
            end

        elseif postfilter_approach %%%%%%%%%%%%%%%%%%%%%%%%%%%

            [EQ_sig] = Eq.process(Scpe_sig,Symbols);

            EQ_sig.plot("fignum",50,"displayname",'After EQ','clear',1);

            Noi = EQ_sig-Symbols;

            Rx_bits = PAMmapper(M,0).demap(EQ_sig);
            [~,errors_bm,ber_ffe_only,errors] = calc_ber(Rx_bits.signal,Bits.signal,"skip_front",100,"skip_end",150,"returnErrorLocation",1);

            nc = 2;
            burg_coeff = arburg(Noi.signal,nc);

            EQ_sig = EQ_sig.filter(burg_coeff,1);

            if 0
                Noi.spectrum('displayname','Noise PSD','fignum',123)
                [h,w] = freqz(1,burg_coeff,length(Noi),"whole",Noi.fs);
                h = h/max(abs(h));
                hold on
                w_ = (w - Noi.fs/2);
                plot(w_.*1e-9,20*log10(fftshift(h)),'DisplayName',['', num2str(nc), ' coefficients for burg alg.']);
            end

            EQ_sig = MLSE("DIR",burg_coeff,"duobinary_output",0,"M",M,"trellis_states",PAMmapper(M,0).levels).process(EQ_sig);

            Rx_bits = PAMmapper(M,0).demap(EQ_sig);
            [~,errors_bm,ber,errors] = calc_ber(Rx_bits.signal,Bits.signal,"skip_front",100,"skip_end",150,"returnErrorLocation",1);

            disp(['FFE: ',sprintf('%.1E',ber_ffe_only),' -> PF -> MLSE: ',sprintf('%.1E',ber),' | SIR: ',num2str(sir),' dB | PD_in: ',num2str(pd_in),' dBm']);

        elseif db_channel_approach %%%%%%%%%%%%%%%%%%%%%%%%%%%

            [EQ_sig, Noi] = Eq.process(Scpe_sig,Duobinary().encode(Symbols));

            EQ_sig.plot("fignum",50,"displayname",'After EQ','clear',1);

            EQ_sig = MLSE("DIR",[1,1],"duobinary_output",1,"M",M,"trellis_states",PAMmapper(M,0).levels).process(EQ_sig);
            EQ_sig = Duobinary().decode(EQ_sig);

            Rx_bits = PAMmapper(M,0).demap(EQ_sig);
            [~,errors_bm,ber,errors] = calc_ber(Rx_bits.signal,Bits.signal,"skip_front",100,"skip_end",150,"returnErrorLocation",1);

            disp([' DB Precode -> Channel -> FFE -> Decode/ Mod  ',sprintf('%.1E',ber),' | dB | PD_in: ',num2str(pd_in),' dBm']);

        elseif db_coding_approach %%%%%%%%%%%%%%%%%%%%%%%%%%%

            [EQ_sig, Noi] = Eq.process(Scpe_sig,Symbols);
            EQ_sig = MLSE("DIR",[1,1],"duobinary_output",1,"M",M,"trellis_states",PAMmapper(M,0).levels).process(EQ_sig);
            EQ_sig = Duobinary().decode(EQ_sig);

            Rx_bits = PAMmapper(M,0).demap(EQ_sig);
            [~,errors_bm,ber,errors] = calc_ber(Rx_bits.signal,Bits.signal,"skip_front",100,"skip_end",150,"returnErrorLocation",1);

            EQ_sig.plot("fignum",50,"displayname",'After EQ');

            disp([' DB Precode -> DB Code -> Channel -> FFE -> Decode/ Mod  ',sprintf('%.1E',ber),' | SIR: ',num2str(sir),' dB | PD_in: ',num2str(pd_in),' dBm']);

        end

        wh.addValueToStorage(ber,'ber',fsym);
        wh.addValueToStorage(rop,'rop',fsym);
        wh.addValueToStorage(pd_in,'pd_in',fsym);
        wh.addValueToStorage(Rx_bits,'signals',fsym);
        wh.addValueToStorage(M,'m',fsym);

        showCurrentMeasurement('BER', ber, 'ROP', rop, 'PD in', pd_in, 'PAM',M, 'Vbias', v_bias, 'AWG Vpp', Awg_vpp);
        autoArrangeFigures(3,3,2);
end

wh.save([folderpath,experiment_name,'_wh']);

cols = linspecer(8);

fsym_vals = wh.parameter.fsym.values;

bers = wh.getStoValue('ber',fsym_vals);
rop_measured = wh.getStoValue('rop',fsym_vals);
pd_in_measured = wh.getStoValue('pd_in',fsym_vals);

figure(90);
hold on; % Retain the plot so new points can be added without complete redraw

% Plot the data and get the line handle
hLine = plot(fsym_vals.*1e-9, bers, "LineWidth", 0.5, "LineStyle", "-", "Marker", ".", "MarkerSize", 15, "DisplayName", experiment_name);

% Store pd_in_measured in the ZData property
hLine.ZData = pd_in_measured;

% Customize the data tips
% Set labels for existing data tip rows
hLine.DataTipTemplate.DataTipRows(1).Label = 'Fsym';
hLine.DataTipTemplate.DataTipRows(2).Label = 'BER';
hLine.DataTipTemplate.DataTipRows(2).Format = '%.2e'; % Format BER as "3e-4"


% Add a new data tip row for PDin
pdinRow = dataTipTextRow('PDin', 'ZData');
hLine.DataTipTemplate.DataTipRows(3) = pdinRow;

% Continue with the rest of your plot settings
yline(3.8e-3, 'DisplayName', 'HD-FEC', 'LineStyle', '--', 'HandleVisibility', 'off');
xlabel('Symbol Rate');
ylabel('Bit Error Rate (BER)');
title('Bit Error Rate vs. ROP');
set(gca, 'yscale', 'log');
set(gca, 'Box', 'on');
grid on;
grid minor;
legend('Interpreter', 'none');

autoArrangeFigures(3,3,2)