% skripta koja umece odabrani zig u odabranu sliku s odabranim
% koeficijentom

zig = load('sig.txt');
[img,map]=imread('lena.pgm');
[CA,CH,CV,CD] = DWT2(img,'haar');
[CA2,CH2,CV2,CD2] = DWT2(CA,'haar');

alpha = 0.2;

%umetanje u detalje prve razine dekompozicije
CHo = oznaci(CH, alpha, zig);
CVo = oznaci(CV, alpha, zig);
CDo = oznaci(CD, alpha, zig);

%umetanje u detalje druge razine dekompozicije
CH2o = oznaci(CH2, alpha, zig);
CV2o = oznaci(CV2, alpha, zig);
CD2o = oznaci(CD2, alpha, zig);

%inverzna transformacija
imgPom = IDWT2(CA2,CH2o,CV2o,CD2o,'haar');
imgR = IDWT2(imgPom,CHo,CVo,CDo,'haar');

subplot(2,2,1); imshow(img); title('Originalna slika');
subplot(2,2,2); imshow(uint8(imgR)); title('Oznacena slika');

razlika = double(imgR) - double(img);
subplot(2,2,3); imshow(uint8(razlika)); title('Razlika slika');

imwrite(imgR, 'lenaR.pgm');