%%% SIR Sweep for MPI Experiment %%%
params = struct;
params.i_atten = 10;

wh = DataStorage(params);

wh.addStorage("ber");
wh.addStorage("sir");
wh.addStorage("pd_in");

playandrecord = 0;

precomp_mode = 2; %0=do nothing ; 1= measure; 2=precomp active
M           = 4;
pn_key      = 2;
usemrds     = 0;
fdac        = 92e9;
fsym        = 92e9;
fadc        = 160e9;
rrcalpha    = 0.05;
v_bias      = 2.27;
i_atten     = 40;
pd_in_desired = 7;

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

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

voa = OptAtten("active",[0,2,1,1],"value",[0,pd_in_desired,0,i_atten],"wavelength",[1310,1310,1310,1310]);
voa.set('active',[0,2,1,1],'value',[0,pd_in_desired,0,i_atten]);
voa.readvals();
%%%%% SET Volatges %%%%%%


if 1
    precomp_path = "C:\Users\sioe\Documents\MATLAB\imdd_simulation\projects\standard_system\";
    precomp_fn = "lab_mpi_setup_2";
else
    precomp_path = "C:\Users\sioe\Documents\MATLAB\model-collection\sioe_models\Labor_2024\Lab_PAM4\";
    precomp_fn = "precomp_bla__loop1_1";
end


%%%%% 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",0,...
    "mrds_code",usemrds,"mrds_blocklength",512).process();

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',3,'loadPath',precomp_path,'fileName',precomp_fn);
end

%%%%% AWG %%%%%%
AWG = AwgKeysight("model","M8196A","fdac",fdac,"scaletodac",[1,1,1,1],"skews",[0,0,0,0],"voltages",[0,0,0,0.62]);
AWG.upload("signal4",Digi_sig);

for i_atten = wh.parameter.i_atten.values

    %%%%% SET ATTENUATOR %%%%%%
    voa.set('active',[0,2,1,1],'value',[0,7,0,i_atten]);
    
    %%%%% Scope %%%%%%
    SCP = ScopeKeysight("model","DSAZ634A",'autoscale',1,"fadc",'GSa_160',"channel",[1],"recordLen",2000000,"removeDC",1);
    Scpe_sig = SCP.read();
    Scpe_sig = Electricalsignal(Scpe_sig{1},"fs",fadc);
    
    % Scpe_sig.spectrum("displayname",'Rx Signal','fignum',10);
    
    %%%%%% Sample to 2x fsym %%%%%%
    Scpe_sig = Scpe_sig.resample("fs_in",160e9,"fs_out",2*92e9);
    
    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);

    Scpe_sig.eye(fsym,M);
    
    %%%%% EQUALIZE %%%%%%
    
    Eq_ffe_only     = FFE("epochs_tr",5,"epochs_dd",5,"len_tr",4096*2,"mu_dd",1e-4,"mu_tr",0,"order",25,"sps",2,"decide",0);
    ber_ffe_only    = runEQ(Eq_ffe_only,Scpe_sig,Symbols,Bits,M);

    Eq_move_it     = 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);
    ber_move_it    = runEQ(Eq_move_it,Scpe_sig,Symbols,Bits,M);
    
    Eq_nonlin     = EQ("Ne",[50,7,7],"Nb",[0,0,0],"training_length",4096*2,"training_loops",5,"dd_loops",5,"K",2,"DCmu",0.0,"DDmu",[0.0004 0.0004 0.0004 0.0004 ],"DFEmu",0.005,"FFEmu",0,"plotfinal",0,"ideal_dfe",1);
    %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",0);
    ber_nonlin    = runEQ(Eq_nonlin,Scpe_sig,Symbols,Bits,M);

    Eq_adaptive_decision  = FFE_adaptive_decision("epochs_tr",5,"epochs_dd",5,"len_tr",4096*2,"mu_dd",1e-4,"mu_tr",0,"order",25,"sps",2,"decide",0,"buffer_length",80);
    ber_adaptive_decision = runEQ(Eq_adaptive_decision,Scpe_sig,Symbols,Bits,M);

    FFE_FFDCAVG("epochs_tr",5,"epochs_dd",5,"len_tr",4096*2,"mu_dd",1e-4,"mu_tr",0,"order",25,"sps",2,"decide",0,"mu_buff",128);
    ber_ff_dcavg = runEQ(FFE_FFDCAVG,Scpe_sig,Symbols,Bits,M);

    Eq_feedback_removal = FFE_DCremoval("epochs_tr",5,"epochs_dd",5,"len_tr",4096*2,"mu_dd",1e-4,"mu_tr",0,"order",25,"sps",2,"decide",0,"mu_dc",0.05,"dc_buffer_len",1);
    ber_feedback_removal = runEQ(Eq_feedback_removal,Scpe_sig,Symbols,Bits,M);



    sir = voa.power_state(3)-voa.power_state(4);
    pd_in = voa.power_state(2);

    disp(['BER: ',sprintf('%.1E',ber_eq),' | SIR: ',num2str(sir),' dB | PD_in: ',num2str(pd_in),' dBm']);

    wh.addValueToStorage(ber_eq,'ber',i_atten);  
    wh.addValueToStorage(sir,'sir',i_atten);
    wh.addValueToStorage(pd_in,'pd_in',i_atten);

end



cols = linspecer(8);

sir_vals = wh.parameter.i_atten.values;
bers = wh.getStoValue('ber',sir_vals);

figure(44);
a = gca;
hold on; % Retain the plot so new points can be added without complete redraw

plot(sir_vals,bers,"LineWidth",0.5,"LineStyle","-","Marker",".","MarkerSize",15,"DisplayName","FFE only");
yline(3.8e-3,'DisplayName','HD-FEC','LineStyle','--','HandleVisibility','off');
xlabel('Received Optical Power (dBm)');
ylabel('Bit Error Rate (BER)');
title('Bit Error Rate vs. ROP');
set(gca,'yscale','log');
set(gca,'Box','on');
grid on;
grid minor
legend