%嵌入水印的函數
%參數說明
%origne_source--原始圖像（載體圖像）
%watermark_source--水印圖像（灰色圖像或二值圖像）
%單位：江南大學信安0602班
%導師：Ms.sun?
%作者：寒星
function?[UV]=in（origne_sourcewatermark_sourcewatermarked_source）
%N=512;
%先處理水印圖像（128x128），如下：
arnold_image=Arnold（watermark_source100）;%讀入水印圖像并且進行arnold變換，10表示進行的次數（可作密鑰）
%arnold_image是uint8的格式
%[cachcvcd]=dwt2（arnold_image‘haar‘）;%進行一級dwt變換

%然后處理載體圖像，如下：
origne_image=imread（origne_source）;%讀入載體圖像
dims=ndims（origne_image）;
if（dims==2）
[image_rolimage_col]=size（origne_image）;
if（image_rol<512）
????if（image_col<512）
????????disp（‘載體圖像的高度和寬度都必須大于512‘）;
????????return;
????end
end

origne_image2=double（origne_image）;

%part_origne_image2=zeros（N）;

%for?i=1:N????????????????????????
%????for?j=1:N
%????????part_origne_image2（ij）=origne_image2（ij）;
%????end
%end
part_origne_image2=origne_image2;
[LL1LH1HL1HH1]=dwt2（part_origne_image2‘haar‘）;%對載體圖像先進行一級dwt變換（320x256）
[LL2LH2HL2HH2]=dwt2（LL1‘haar‘）;%對LL1進行dwt變換（160x128）
[LL3LH3HL3HH3]=dwt2（LL2‘haar‘）;%變成80x64，水印圖像為64x64
%水印嵌入過程，如下：
%嵌入分兩個部分
%a1=0.15;a2=0.15;

%ｃｖ，ｃｄ的存放
arnold_image=double（arnold_image）;
[UWSWVW]=svd（arnold_image）;%對水印作svd分解
%size（SW）
[ULSLVL]=svd（LL3）;%對3級低頻分量作svd分解80x64
SLL=SL;
%size（SL）
[UHSHVH]=svd（HH3）;%對3級高頻分量作svd分解
SHH=SH;
%[ULHSLHVLH]=svd（LH3）;

%[UHLSHLVHL]=svd（HL3）;
U=UW;V=VW;%128x128
g1=0.7;g2=0.7;g3=0.5;
%由于第一個奇異值很大，改變一點，圖像失真很多，所以強度不應很大
[ab]=size（LL3）;
col=min（ab）;

SL（11）=SL（11）+g3*SW（11）;
for?i=2:col
????SL（ii）=SL（ii）+g1*SW（ii）;
end
SH（11）=SH（11）+g3*SW（6565）;
for?i=2:col
????SH（ii）=SH（ii）+g2*SW（i+64i+64）;
end
%欠兩下
%SLH（11）=SLH（11）+g3*SW（11）;
%col=64;
%for?i=2:col
%????SLH（ii）=SLH（ii）+g1*SW（ii）;
%end
%SHL（11）=SHL（11）+g3*SW（i+64i+64）;
%for?i=2:col
%????SHL（ii）=SHL（ii）+g2*SW（i+64i+64）;
%end
LL3=UL*SL*VL‘;
HH3=UH*SH*VH‘;
%LH3=ULH*SLH*VLH‘;
%HL3=UHL*SHL*VHL‘;
LL2=idwt2（LL3LH3HL3HH3‘haar‘）;%小波逆變換
LL1=idwt2（LL2LH2HL2HH2‘haar‘）;
watermarked_image1=idwt2（LL1LH1HL1HH1‘haar‘）;
%for?i=1:N????????????????????????
%????for?j=1:N
%????????origne_image2（ij）=watermarked_image1（ij）;
%????end
%end

watermarked_image=uint8（watermarked_image1）;
imwrite（watermarked_image