silas/imdd_silas
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

MPI FFE with A1 and A2 algorithms

Browse Source
This commit is contained in:
Silas Oettinghaus
2026-07-10 09:14:34 +02:00
parent aa210e5352
commit a6cf742121
11 changed files with 863 additions and 229 deletions
Download Patch File Download Diff File Expand all files Collapse all files

191
Tests/04_DSP/Equalizer/FFE_test.m
View File

@@ -12,8 +12,19 @@ classdef FFE_test < IMDDTestCase
testCase.verifyEqual(ffe.dd_mode, 1);
testCase.verifyEqual(ffe.mu_dd, 1e-5);
testCase.verifyEqual(ffe.epochs_dd, 5);
testCase.verifyFalse(logical(ffe.optimize_dc_params));
testCase.verifyEqual(ffe.dc_optimization_delay_weight, 1e-3);
testCase.verifyFalse(logical(ffe.optimize_dc_tracking_params));
testCase.verifyEqual(ffe.dc_tracking_optimization_delay_weight, 1e-3);
testCase.verifyEqual(ffe.dc_tracking_power_exponent, 2);
testCase.verifyEqual(ffe.dc_avg_bufferlength_a1, 0);
testCase.verifyEqual(ffe.dc_smoothing_a1, 0);
testCase.verifyEqual(ffe.dc_level_avg_bufferlength_a2, 0);
testCase.verifyEqual(ffe.dc_smoothing_a2, 0);
testCase.verifyEqual(ffe.dc_level_weights_a2, 0);
testCase.verifyEqual(ffe.dc_tracking_buffer_len, 1);
testCase.verifyEqual(ffe.ffe_update_buffer_len, 1);
testCase.verifyFalse(logical(ffe.optimize_a2_level_weights));
testCase.verifyEqual(ffe.a2_level_weight_optimization_max_evals, 30);
testCase.verifyEqual(ffe.a2_level_weight_max, 1);
testCase.verifyFalse(ffe.decide);
testCase.verifyEqual(ffe.e, zeros(ffe.order, 1));
testCase.verifyEqual(ffe.error, 0);
@@ -71,6 +82,182 @@ classdef FFE_test < IMDDTestCase
testCase.verifyNotEqual(y.signal, zeros(size(x.signal)));
testCase.verifyEqual(length(y.signal), length(x.signal));
end
function dcTrackingBufferDelaysDcUpdateUntilBlockBoundary(testCase)
x = zeros(4, 1);
d = ones(4, 1);
ffe = FFE( ...
"sps", 1, ...
"order", 1, ...
"dc_tracking_mu", 0.1, ...
"dc_tracking_buffer_len", 2, ...
"dd_mode", 0, ...
"adaption_technique", adaption_method.lms);
ffe.constellation = unique(d);
ffe.equalize(x, d, 0, 1, numel(x), true, false);
testCase.verifyEqual(ffe.e_dc, 0.19, "AbsTol", 1e-12);
end
function ffeBufferDelaysTapUpdateUntilBlockBoundary(testCase)
x = ones(4, 1);
d = ones(4, 1);
ffe = FFE( ...
"sps", 1, ...
"order", 1, ...
"dc_tracking_mu", 0, ...
"ffe_update_buffer_len", 2, ...
"dd_mode", 0, ...
"adaption_technique", adaption_method.lms);
ffe.constellation = unique(d);
ffe.equalize(x, d, 0.1, 1, numel(x), true, false);
testCase.verifyEqual(ffe.e, 0.19, "AbsTol", 1e-12);
end
function dcAvgA1BufferlengthOneLeavesInputUnchanged(testCase)
x = (1:4).';
d = zeros(4, 1);
ffe = FFE( ...
"sps", 1, ...
"order", 1, ...
"dc_avg_bufferlength_a1", 1, ...
"save_debug", true, ...
"dd_mode", 0, ...
"adaption_technique", adaption_method.lms);
ffe.e = 1;
[y,~] = ffe.equalize(x, d, 0, 1, numel(x), true, false);
testCase.verifyEqual(y, x, "AbsTol", 1e-12);
testCase.verifyEqual(ffe.debug_struct.dc_avg_offset, zeros(1,4), "AbsTol", 1e-12);
end
function dcAvgA1SubtractsCausalMovingAverageByDefault(testCase)
x = (1:4).';
d = zeros(4, 1);
expected_offset = [0, 0, 0, 2];
expected_y = [1; 2; 3; 2];
ffe = FFE( ...
"sps", 1, ...
"order", 1, ...
"dc_avg_bufferlength_a1", 3, ...
"save_debug", true, ...
"dd_mode", 0, ...
"adaption_technique", adaption_method.lms);
ffe.e = 1;
[y,~] = ffe.equalize(x, d, 0, 1, numel(x), true, false);
testCase.verifyEqual(y, expected_y, "AbsTol", 1e-12);
testCase.verifyEqual(ffe.debug_struct.dc_avg_offset, expected_offset, "AbsTol", 1e-12);
end
function dcAvgA1SmoothingIsReservedForOfflineVariants(testCase)
x = (1:4).';
d = zeros(4, 1);
expected_offset = [0, 0, 0, 2];
expected_y = [1; 2; 3; 2];
ffe = FFE( ...
"sps", 1, ...
"order", 1, ...
"dc_avg_bufferlength_a1", 3, ...
"dc_smoothing_a1", 0.5, ...
"save_debug", true, ...
"dd_mode", 0, ...
"adaption_technique", adaption_method.lms);
ffe.e = 1;
[y,~] = ffe.equalize(x, d, 0, 1, numel(x), true, false);
testCase.verifyEqual(y, expected_y, "AbsTol", 1e-12);
testCase.verifyEqual(ffe.debug_struct.dc_avg_offset, expected_offset, "AbsTol", 1e-12);
end
function dcAvgA1WarnsWhenCombinedWithDcTracking(testCase)
testCase.verifyWarning(@() FFE( ...
"dc_avg_bufferlength_a1", 3, ...
"dc_tracking_mu", 0.1), "FFE:DCAvgWithDcTracking");
end
function dcLevelAvgA2SubtractsScalarWeightedResidual(testCase)
x = (1:4).';
d = zeros(4, 1);
expected_y = [1; 2; 1.5; 2.5];
ffe = FFE( ...
"sps", 1, ...
"order", 1, ...
"dc_level_avg_bufferlength_a2", 2, ...
"dc_level_weights_a2", 1, ...
"save_debug", true, ...
"dd_mode", 0, ...
"adaption_technique", adaption_method.lms);
ffe.e = 1;
ffe.constellation = 0;
[y,~] = ffe.equalize(x, d, 0, 1, numel(x), true, false);
testCase.verifyEqual(y, expected_y, "AbsTol", 1e-12);
testCase.verifyEqual(ffe.debug_struct.dc_level_mpi_est, [0, 0, 1.5, 1.5], "AbsTol", 1e-12);
testCase.verifyEqual(ffe.debug_struct.dc_level_weight, [0, 0, 1, 1], "AbsTol", 1e-12);
testCase.verifyEqual(ffe.debug_struct.dc_level_valid_count, [0, 0, 2, 2]);
end
function dcLevelAvgA2UsesVectorWeightsInConstellationOrder(testCase)
x = [1; 3; 1; 4];
d = [0; 2; 0; 2];
ffe = FFE( ...
"sps", 1, ...
"order", 1, ...
"dc_level_avg_bufferlength_a2", 2, ...
"dc_level_weights_a2", [0, 1], ...
"dd_mode", 0, ...
"adaption_technique", adaption_method.lms);
ffe.e = 1;
ffe.constellation = [0; 2];
[y,~] = ffe.equalize(x, d, 0, 1, numel(x), true, false);
testCase.verifyEqual(y, [1; 3; 1; 3.5], "AbsTol", 1e-12);
end
function dcLevelAvgA2WarnsWhenCombinedWithDcTracking(testCase)
testCase.verifyWarning(@() FFE( ...
"dc_level_avg_bufferlength_a2", 2, ...
"dc_level_weights_a2", 1, ...
"dc_tracking_mu", 0.1), "FFE:DCLevelAvgWithOtherDcSuppression");
end
function a2InitialWeightGuessFollowsLevelVarianceSlope(testCase)
x = [ ...
0.98; 1.02; 1.00; 1.01; ...
1.90; 2.10; 2.00; 2.08; ...
2.50; 3.50; 3.00; 3.40];
d = [ ...
ones(4,1); ...
2*ones(4,1); ...
3*ones(4,1)];
ffe = FFE("sps", 1, "a2_level_weight_max", 1);
ffe.constellation = [1; 2; 3];
[weights,stats] = ffe.a2LevelWeightInitialGuess(x,d);
testCase.verifySize(weights, [3, 1]);
testCase.verifyGreaterThan(weights(3), weights(2));
testCase.verifyGreaterThan(weights(2), weights(1));
testCase.verifyEqual(max(weights), 0.7, "AbsTol", 1e-12);
testCase.verifyGreaterThan(stats.variance_slope, 0);
end
end
end