%Name:		Chris Shoemaker 
%Course:	EER-280 - Digital Watermarking 
%Project: 	Embeding of CDMA watermark into H1,V1,D1 componants of a 1-scale DWT 
%           Watermark Embeding 
 
clear all; 
close all; 
clc 
 
% save start time 
start_time=cputime; 
 
k=2;            % set the gain factor for embeding 
 
% read in the cover object 
file_name='lena.jpg'; 
cover_object=double(imread(file_name)); 
 
% determine size of watermarked image 
Mc=size(cover_object,1);	%Height 
Nc=size(cover_object,2);	%Width 
 
% read in the message image and reshape it into a vector 
file_name='dmg1.tif'; 
message=double(imread(file_name)); 
Mm=size(message,1);	                        %Height 
Nm=size(message,2);	                        %Width 
message_vector=round(reshape(message,Mm*Nm,1)./2); 
 
% read in key for PN generator 
% file_name='_key.bmp'; 
% key=double(imread(file_name))./2; 
 
% reset MATLAB's PN generator to state "key" 
rand('state',16); 
 
%三層harr小波變換 
[cA1,cH1,cV1,cD1] = dwt2(cover_object,'haar'); 
% [cA2,cH2,cV2,cD2] = dwt2(cA1,'haar'); 
% [cA3,cH3,cV3,cD3] = dwt2(cA2,'haar'); 
 
% [r1,c1] = size(cA3); 
% pn_sequence_h1=round(2*(rand(r1,c1)-0.5)); 
% cA3=cA3+20*pn_sequence_h1; 
% add pn sequences to H1 and V1 componants when message = 0  
%  pn_sequence_h=round(2*(rand(Mc/2,Nc/2)-0.5)); 
%  pn_sequence_v=round(2*(rand(Mc/2,Nc/2)-0.5)); 
for (kk=1:length(message_vector))   
    pn_sequence_h=round(2*(rand(Mc/2,Nc/2)-0.5)); 
    pn_sequence_v=round(2*(rand(Mc/2,Nc/2)-0.5)); 
    if (message(kk) == 0) 
        cH1=cH1+k*pn_sequence_h; 
        cV1=cV1+k*pn_sequence_v; 
    end 
end 
 
% perform IDWT 
% cA2 = idwt2(cA3,cH3,cV3,cD3,'haar',[Mc/4,Nc/4]); 
% cA1 = idwt2(cA2,cH2,cV2,cD2,'haar',[Mc/2,Nc/2]); 
watermarked_image = idwt2(cA1,cH1,cV1,cD1,'haar',[Mc,Nc]);  
 
% convert back to uint8 
watermarked_image_uint8=uint8(watermarked_image); 
 
% write watermarked Image to file 
imwrite(watermarked_image_uint8,'dwt_watermarked.bmp','bmp'); 
 
% display processing time 
elapsed_time=cputime-start_time, 
 
% calculate the PSNR 
psnr=psnr(cover_object,watermarked_image_uint8,Mc,Nc), 
 
% display watermarked image 
figure(1) 
imshow(watermarked_image_uint8,[]) 
title('Watermarked Image') 
 
% figure(2) 
% subplot(2,2,1); 
% imshow(cA1,[]); 
% subplot(2,2,2);imshow(cA2,[]); 
% subplot(2,2,3);imshow(cA3,[]); 
% subplot(2,2,4);imshow(cD1,[]); 
%  
% figure(3)