BCJR implementation

WDM code added (Pol Cont., Opt MUX/DEMUX, Opt Atten, DP_Fiber) -> the codebase is not optimized to always work with dp signals!

Functions/EQ_visuals/show2Dconstellation.m

@@ -0,0 +1,85 @@
+function [symbols_for_lvl,avg_for_lvl] = show2Dconstellation(eq_signal,ref_symbols,options)
+arguments
+    eq_signal
+    ref_symbols
+    options.fignum (1,1) double = NaN % Default to NaN if not provided
+    options.displayname (1,:) char = '' % Default to an empty string if not provided
+end
+
+plot_shit = 1;
+
+if isa(eq_signal,'Signal')
+    eq_signal = eq_signal.signal;
+end
+if isa(ref_symbols,'Signal')
+    ref_symbols = ref_symbols.signal;
+end
+
+
+if plot_shit
+    % Determine the figure number to use or create a new figure
+    if isnan(options.fignum)
+        fig = figure; % Create a new figure and get its handle
+    else
+        fig = figure(options.fignum); % Use the specified figure number
+        clf;
+    end
+end
+
+
+rx_symbols = eq_signal; %./ rms(eq_signal);
+correct_symbols = ref_symbols;
+
+col = cbrewer2('Paired',numel(unique(correct_symbols))*2);
+ccnt = -1;
+
+levels = unique(correct_symbols);
+M = numel(levels);
+symbols_for_lvl = NaN(numel(levels),length(correct_symbols));
+start = 1;
+ende = length(correct_symbols);
+
+
+
+
+for l = 1:numel(levels)
+    ccnt = ccnt+2;
+
+    level_amplitude = levels(l);
+
+    symbols_for_lvl(l,correct_symbols==level_amplitude) = rx_symbols(correct_symbols==level_amplitude);
+
+    statistical_mean(l) = mean(symbols_for_lvl(l,:),'omitnan');
+
+end
+
+tx_even = correct_symbols(1:2:end);
+tx_odd = correct_symbols(2:2:end);
+
+rx_even = rx_symbols(1:2:end);
+rx_odd = rx_symbols(2:2:end);
+
+D_even = tx_even - rx_even;
+D_odd  = tx_odd - rx_odd;
+
+D = sqrt(D_even.^2 + D_odd.^2);
+
+[X,Y] = meshgrid(levels, levels);
+[X_,Y_] = meshgrid(statistical_mean, statistical_mean);
+
+hold on;
+scatter(rx_even,rx_odd,5*ones(1,length(D)),D,'.','DisplayName',['Even/Odd PAM-',num2str(M)],'MarkerEdgeColor',col(2,:));
+% colormap(gca,flip(cbrewer2('Spectral',100)))
+colormap(gca,'hsv');
+scatter(X_(:), Y_(:), 2, 'x', 'LineWidth', 10, 'MarkerEdgeColor', col(6,:),'DisplayName','Statistical Rx Levels');
+scatter(X(:), Y(:), 2, 'x', 'LineWidth', 10, 'MarkerEdgeColor', col(4,:),'DisplayName','Tx Levels');
+xlim([floor(min(X_(:)))-1, ceil(max(X_(:)))+1]);
+ylim([floor(min(Y_(:)))-1, ceil(max(Y_(:)))+1]);
+
+yticks((levels(1:end-1) + levels(2:end)) / 2);
+xticks((levels(1:end-1) + levels(2:end)) / 2);
+legend 
+xlabel('even symbols');
+ylabel('odd symbols');
+axis equal; grid on;
+end

Functions/EQ_visuals/showLevelScatter.m

@@ -4,7 +4,7 @@ arguments
     ref_symbols
     options.fignum (1,1) double = NaN % Default to NaN if not provided
     options.displayname (1,:) char = '' % Default to an empty string if not provided
-    options.f_sym = [];
+    options.f_sym =1e6;
 end
 
 plot_shit = 1;
@@ -30,7 +30,7 @@ if plot_shit
 end
 
 
-rx_symbols = eq_signal ./ rms(eq_signal);
+rx_symbols = eq_signal; %./ rms(eq_signal);
 correct_symbols = ref_symbols;
 f_sym = options.f_sym;
 
@@ -47,7 +47,6 @@ for l = 1:numel(levels)
 
     level_amplitude = levels(l);
 
-
     symbols_for_lvl(l,correct_symbols==level_amplitude) = rx_symbols(correct_symbols==level_amplitude);
     std_lvl(l) = std(symbols_for_lvl(l,:),'omitnan');
     xax_in_sec = ((1:length(correct_symbols)) / f_sym) * 1e6;
@@ -68,7 +67,8 @@ for l = 1:numel(levels)
     ccnt = ccnt+2;
     level_amplitude = levels(l);
 
-    movmean = 1/250 .* movsum(rx_symbols(correct_symbols==level_amplitude),[250/2,250/2], 'Endpoints', 'fill');
+    L = 500;
+    movmean = 1/L .* movsum(rx_symbols(correct_symbols==level_amplitude),[L/2,L/2], 'Endpoints', 'fill');
 
     avg_for_lvl(l,correct_symbols==level_amplitude) = movmean;
 
@@ -125,7 +125,7 @@ if 0
 end
 
 if plot_shit
-yline(levels);
+% yline(levels);
 xlabel('Time in $\mu$s');
 ylabel('Normalized Amplitude');
 ylim([-3 3]);