KwantND

clear all;

Q = 10;
T = 50;

I = imread('imgnoise.jpg');

RI = I(:,:,1);
GI = I(:,:,2);
BI = I(:,:,3);

%transforamta
RIt = dct2(RI);
BIt = dct2(BI);
GIt = dct2(GI);
RIf = fwht(double(RI));
BIf = fwht(double(BI));
GIf = fwht(double(GI));
[CR,SR]=wavedec2(RI,2,'haar');
[CG,SG]=wavedec2(GI,2,'haar');
[CB,SB]=wavedec2(BI,2,'haar');


q = @(x) floor(x*(2^Q)+0.5)/(2^Q);

%kwantyzacja
RIt = arrayfun(q, RIt);
BIt = arrayfun(q, BIt);
GIt = arrayfun(q, GIt);

RIf(:) = arrayfun(q, RIt);
BIf(:) = arrayfun(q, BIt);
GIf(:) = arrayfun(q, GIt);

CR(:) = arrayfun(q, RIt);
CG(:) = arrayfun(q, BIt);
CB(:) = arrayfun(q, GIt);




%odwrotana trtanformata
RIti = idct2(RIt);
BIti = idct2(BIt);
GIti = idct2(GIt);
RIfi = ifwht(RIf);
BIfi = ifwht(BIf);
GIfi = ifwht(GIf);
RIfi = ifwht(RIf);
BIfi = ifwht(BIf);
GIfi = ifwht(GIf);
Rwi = waverec2(CR, SR, 'haar');
Bwi = waverec2(CB, SB, 'haar');
Gwi = waverec2(CG, SG, 'haar');

finishedT(:,:,1) = RIti;
finishedT(:,:,2) = GIti;
finishedT(:,:,3) = BIti;
finishedT = uint8(finishedT);

finishedF(:,:,1) = RIfi;
finishedF(:,:,2) = GIfi;
finishedF(:,:,3) = BIfi;
finishedF = uint8(finishedF);

finishedW(:,:,1) = Rwi;
finishedW(:,:,2) = Gwi;
finishedW(:,:,3) = Bwi;
finishedW = uint8(finishedW);

f = figure;
% imshowpair(I,finishedT,'montage');
%or = imread('Leopard.jpg');
% figure, imshow(I-finishedT);

%f = figure;
% imshowpair(I,finishedF,'montage');
%f = figure;
imshowpair(I,finishedW,'montage');