From b3fe7ee4980d8c3e68321619b264c2aa94ff20ea Mon Sep 17 00:00:00 2001
From: Marina Kreme <amamarinak@gmail.com>
Date: Sun, 22 Nov 2020 07:51:07 +0100
Subject: [PATCH] refactor
---
matlab/tfgm/tf_fading/solver_tfgm_zero.m | 19 ------------
matlab/tfgm/tf_fading/zero_fill_solver.m | 37 ++++++++++++++++++++++++
2 files changed, 37 insertions(+), 19 deletions(-)
delete mode 100644 matlab/tfgm/tf_fading/solver_tfgm_zero.m
create mode 100644 matlab/tfgm/tf_fading/zero_fill_solver.m
diff --git a/matlab/tfgm/tf_fading/solver_tfgm_zero.m b/matlab/tfgm/tf_fading/solver_tfgm_zero.m
deleted file mode 100644
index 9eef531..0000000
--- a/matlab/tfgm/tf_fading/solver_tfgm_zero.m
+++ /dev/null
@@ -1,19 +0,0 @@
-function x_zero= solver_tfgm_zero(X, mask, idgt)
-
-%% x_zero= solver_tfgm_zero(X, mask, idgt)
-% Thid function reconstruct the signal after filling the masked regions by zeros.
-% Inputs:
-% - X: time-frequency matrix
-% - mask: binary mask
-% - idgt: Inverse of Gabor transform operator
-% Outputs:
-% -x_interp: estimated signal
-%
-%
-% Author: Marina KREME
-
-%%
-X(mask==1)=0;
-x_zero = compute_idgt(X, idgt);
-
-end
\ No newline at end of file
diff --git a/matlab/tfgm/tf_fading/zero_fill_solver.m b/matlab/tfgm/tf_fading/zero_fill_solver.m
new file mode 100644
index 0000000..c962e68
--- /dev/null
+++ b/matlab/tfgm/tf_fading/zero_fill_solver.m
@@ -0,0 +1,37 @@
+function x_est= zero_fill_solver(x, mask, dgt, idgt, dgt_params,...,
+ signal_params, fig_dir)
+
+%% x_est= zero_fill_solver(x, mask, dgt, idgt, dgt_params,...,
+% signal_params, fig_dir)
+% Thid function reconstruct the signal after filling the masked regions by zeros.
+% Inputs:
+% - x (nd array): mix signals
+% - mask: time-frequency mask
+% - idgt,dgt (handle): DGT and IDGT . see utils/get_stft_operators.m
+% - dgt_params (struct) : DGT parameters
+% - signal_params (struct) . :signals parameters
+% - fig_dir : folder where figures are stored
+% Outputs:
+% -x_est (nd array): estimated signal
+%
+%
+% Author: Marina KREME
+
+%%
+x_tf = dgt(x);
+x_tf(mask==1)=0;
+x_est = idgt(x_tf);
+
+%%
+
+figure
+plot_spectrogram(x_tf, dgt_params,signal_params, dgt);
+title('TF matrix filled by zeros in mask')
+saveas(gcf,fullfile(fig_dir,'tf_filled_zeros.pdf'));
+
+figure
+plot_spectrogram(x_est, dgt_params,signal_params, dgt);
+title('Reconstructed signal by zero fill')
+saveas(gcf,fullfile(fig_dir,'zero_fill_est.pdf'));
+
+end
\ No newline at end of file
--
GitLab