clear;
close all;
taille_ecran = get(0,'ScreenSize');
L = taille_ecran(3);
H = taille_ecran(4);

% Choix du jeu de données :
load Donnees/vase_brillant_bis;
% load Donnees/visage;

[nb_lignes,nb_colonnes] = size(masque);
[n,p] = size(I);

% Affichage des images :
figure('Name','Images d''origine','Position',[0.5*L,0.66*H,0.5*L,0.4*H]);
colormap gray;
n_c = min(4,ceil(sqrt(n)));
n_l = min(2,ceil(n/n_c));
for i = 1:n_c*n_l
	img = reshape(I(i,:),nb_lignes,nb_colonnes);
	subplot(n_l,n_c,i);
	imagesc(img);
	hold on;
	axis image;
	axis off;
	title(['$\mathbf{s}_{' num2str(i,'%2d') '}$'],'Interpreter','Latex','FontSize',20);
end
drawnow;

% Correction des images :
I_bar = correction(I,masque);

% Affichage des images corrigées :
figure('Name','Images corrigees','Position',[0.5*L,0,0.5*L,0.4*H]);
colormap gray;
for i = 1:n_c*n_l
	img = reshape(I_bar(i,:),nb_lignes,nb_colonnes);
	subplot(n_l,n_c,i);
	imagesc(img);
	hold on;
	axis image;
	axis off;
	title(['$\mathbf{s}_{' num2str(i,'%2d') '}$'],'Interpreter','Latex','FontSize',20);
end