%%%%%%%【基于峰值變換小波系數分析主程序】%%%%%%%%%%
clear all
disp('Running PTWT....')
%%%%%%%%%%%%%%%【讀入圖像數據】%%%%%%%%%%%%%%%%%%%
load woman;  %  下載圖像
X=im2double(X);
f=X;         %  原始圖像
%%%%%%%%%%%%%%%【二維小波分解】%%%%%%%%%%%%%%%%%%%
[sa,sb]=size(f);
[Ain]=hvanalysis(f,1,1,sa,sb,-1);
[Ain]=hvanalysis(Ain,1,1,sa,sb,1);
%%%%%%%%%%%%%%%%【峰值變換】%%%%%%%%%%%%%%%%%%%%
Ain=func_PeakTransform(Ain);           %調用func_PeakTransform函數，此函數包括前向峰值變換和后向峰值變換
%%%%%%%%%%%%%%%%%【編譯碼】%%%%%%%%%%%%%%%%%%%%%
%%%%%%%%%%%%%%%%%【編碼】%%%%%%%%%%%%%%%%%%%%%%%
rate=input('請輸入rate=');  
disp('Running.......');
Enc=SPIHT_Enc(Ain,1,rate);
%%%%%%%%%%%%%%%%%【譯碼】%%%%%%%%%%%%%%%%%%%%%%%%
Dec= func_SPIHT_Dec(Enc);
%%%%%%%%%%%%%%%【重構源圖像】%%%%%%%%%%%%%%%%%%%
[Ain]=hvsynthesis(Dec,1,1,sa,sb,1);
[A]=hvsynthesis(Ain,1,1,sa,sb,-1);
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
construct_pic=A;  %  重建全部圖像
%%%%%%%%%%%%%%%%%%【計算PSNR】%%%%%%%%%%%%%%%%
mse=sum(sum((construct_pic-f).^2))/sa/sb;
psnr=10*log10(255*255/mse)
%%%%%%%%%%%%%%%%%%%%%%【結果顯示】%%%%%%%%%%%
% figure;colormap(map);subplot(2,1,1);
% image(f);                           %  源圖像顯示
% title('original pic');
% subplot(2,1,2);
% image(abs(construct_pic));          %  重構源圖像顯示
% title('reconstructed pic');
disp('%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%');