ML Equalizer works now.

Not yet perfectly integrated into all the routines

Functions/beautifyBERplot.m

@@ -1,43 +1,97 @@
 function beautifyBERplot(options)
+% BEAUTIFYBERPLOT  Enhances BER-style plots for publication-quality figures.
+% Supports automatic smoothing and trend-line overlay.
+%
+% Usage examples:
+%   beautifyBERplot;                                 % default
+%   beautifyBERplot("polyfit",1);                    % add polynomial fit
+%   beautifyBERplot("polyfit",1,"fitmethod","pchip") % piecewise cubic fit
+%
+% Supported fitmethod options: 'polyfit', 'smoothingspline', 'loess', 'pchip'
+
 arguments
-    options.logscale = 1
+    options.logscale (1,1) logical = 1
+    options.polyfit (1,1) logical = 0
+    options.polyorder (1,1) double = 2
+    options.fitmethod (1,1) string = "polyfit"  % choose fit type
 end
-    % BEAUTIFYBERPLOT Enhances a BER plot for publication-quality figures.
 
-    % Set line properties for all current plot lines
-    lines = findall(gca, 'Type', 'Line');
-    markers = {'o', 's', 'd', '^', 'v', '>', '<', 'p', 'h'}; % Define marker styles
-    num_markers = length(markers);
-    
+% --- find all line objects in current axes
+lines = findall(gca, 'Type', 'Line');
+markers = {'o', 's', 'd', '^', 'v', '>', '<', 'p', 'h'}; 
+num_markers = length(markers);
+
+% --- style all lines consistently
+for i = 1:length(lines)
+    lines(i).LineWidth = 1.1;
+    lines(i).LineStyle = '-';
+    if string(lines(i).Marker) == "none"
+        lines(i).Marker = markers{mod(i-1, num_markers) + 1};
+    end
+    lines(i).MarkerSize = 4;
+    lines(i).MarkerFaceColor = lines(i).Color;
+end
+
+% --- optional smoothing/fitting overlay
+if options.polyfit
+    hold on
     for i = 1:length(lines)
-        lines(i).LineWidth = 1.3;  % Thicker line width
-        lines(i).LineStyle = '-';  % Solid lines for simplicity
-        if string(lines(i).Marker) == "none"
-            lines(i).Marker = markers{mod(i-1, num_markers) + 1}; % Assign markers cyclically
+        x = lines(i).XData;
+        y = lines(i).YData;
+        valid = isfinite(x) & isfinite(y);
+        if sum(valid) < options.polyorder + 1
+            continue;
         end
-        lines(i).MarkerSize = 7;   % Marker size
-        lines(i).MarkerFaceColor = lines(i).Color; % Use line color for marker face
-        lines(i).MarkerEdgeColor = 'white';
+
+        xf = linspace(min(x(valid)), max(x(valid)), 200);
+
+        % ----- choose fitting method -----
+        switch lower(options.fitmethod)
+            case "polyfit"
+                p = polyfit(x(valid), y(valid), options.polyorder);
+                yf = polyval(p, xf);
+
+            case "smoothingspline"
+                try
+                    f = fit(x(valid)', y(valid)', 'smoothingspline');
+                    yf = feval(f, xf);
+                catch
+                    yf = interp1(x(valid), y(valid), xf, 'pchip');
+                end
+
+            case "loess"
+                yf = smooth(x(valid), y(valid), 0.2, 'loess');
+                yf = interp1(x(valid), yf, xf, 'linear', 'extrap');
+
+            case "pchip"
+                yf = interp1(x(valid), y(valid), xf, 'pchip');
+
+            otherwise
+                warning('Unknown fitmethod "%s". Using polyfit.', options.fitmethod);
+                p = polyfit(x(valid), y(valid), options.polyorder);
+                yf = polyval(p, xf);
+        end
+
+        % --- lightened color for fit overlay
+        lightcol = lines(i).Color + 0.4 * (1 - lines(i).Color);
+        lightcol(lightcol > 1) = 1;
+
+        plot(xf, yf, '-', 'Color', lightcol, ...
+            'LineWidth', 0.7, 'Marker', 'none', ...
+            'HandleVisibility','off');
     end
-
-    % Change all text interpreters to LaTeX
-    set(findall(gca, '-property', 'Interpreter'), 'Interpreter', 'latex');
-
-    % Set figure background to white
-    set(gcf, 'Color', 'w');
-
-    
-    % Set logarithmic scale for y-axis, but only if it makes sense.
-    % If this is not always desired, you could condition this on the presence of lines or data.
-    if options.logscale
-        set(gca, 'YScale', 'log');
-    end
-    
-    % Customize grid and box appearance
-    set(gca, 'Box', 'on', 'LineWidth', 0.8);  % Thicker border
-    grid on;
-    % grid minor;
-
-    % Adjust font size and style for better readability
-    set(gca, 'FontSize', 10, 'FontName', 'Times New Roman');
+    hold off
+end
+
+% --- axis scaling and cosmetics
+if options.logscale
+    set(gca, 'YScale', 'log');
+end
+
+set(findall(gca, '-property', 'Interpreter'), 'Interpreter', 'latex');
+set(gcf, 'Color', 'w');
+set(gca, 'Box', 'on', 'LineWidth', 0.8);
+grid on;
+set(gca, 'FontSize', 10, 'FontName', 'Times New Roman');
+
 end