%注意嵌入與提取源碼需放在同一個(gè)文件中,或者提取時(shí)不要用clear
%源碼由會員robinher提供,感謝robinher對論壇的支持.
clear
%定義常量
size=256;
block=8;
blockno=size/block;
LENGTH=size*size/64;  
Alpha1=0.02; 
Alpha2=0.1; 
T1=3; 
I=zeros(size,size); 
D=zeros(size,size); 
BW=zeros(size,size); 
block_dct1=zeros(block,block);
%產(chǎn)生高斯水印，并顯示水印信息；
randn('seed',10);mark=randn(1,LENGTH);
subplot(3,2,1);plot(mark);title('水印'); 
%顯示原圖
subplot(3,2,2);I=imread('lena.bmp');   
imshow(I);title('原始圖像');
%顯示prewitt為算子的邊緣圖
BW=edge(I,'prewitt');subplot(3,2,3);imshow(BW);
title('顯示邊緣圖像');
%嵌入水印
k=1;
for m=1:blockno
  for n=1:blockno
    x=(m-1)*block+1;     y=(n-1)*block+1;
    block_dct1=I(x:x+block-1,y:y+block-1);
    block_dct1=dct2(block_dct1);
  BW_8_8=BW(x:x+block-1,y:y+block-1);
    if m<=1|n<=1
      T=0;
    else
      T=sum(BW_8_8);   T=sum(T);
    end
    if T>T1
      Alpha=Alpha2;
else
      Alpha=Alpha1;
end
block_dct1(1,1)=block_dct1(1,1)*(1+Alpha*mark(k));  
block_dct1=idct2(block_dct1);
D(x:x+block-1,y:y+block-1)=block_dct1; 
    k=k+1;
  end
end
%顯示嵌入水印后的圖像
subplot(3,2,4);imshow(D,[]);title('嵌入水印后的圖像');
%保存該圖像
D=uint8(D);
imwrite(D,'marked.bmp');
%將帶有水印的圖像marked.bmp以jpg格式壓縮后，產(chǎn)生marked.jpg。
      
%從marked.jpg中可以提取正確的水印，證明dct方法對該壓縮方式的魯棒性。
    %提起水印
    D=imread('marked.bmp');
    D=double(D);
I=imread('lena.bmp');
I=double(I);
array2=zeros(1,LENGTH);
K=1;
l=1;
for m=1:blockno
  for n=1:blockno
      x=(m-1)*block+1;   y=(n-1)*block+1;%算出每格圖像的坐標(biāo)（x,y）,block=8,8*8的圖像小格
      block_dct1=I(x:x+block-1,y:y+block-1);%取原始圖像小格中的像素點(diǎn)到block_dct1矩陣中。
      block_dct2=D(x:x+block-1,y:y+block-1);
Block_dct1=dct2(block_dct1);%對二維數(shù)組進(jìn)行離散余弦變換。dct是有損壓縮如jpeg使用的技術(shù)。Dct是可逆的運(yùn)算
Block_dct2=dct2(block_dct2);
      BW_8_8=BW(x:x+block-1,y:y+block-1);%得到邊界矩陣。
      if m<=1|n<=1
          T=0;
      else
          T=sum(BW_8_8);   T=sum(T);
      end
      if T>T1
        Alpha=Alpha2;
        %block_dct1(1,1)=block_dct1(1,1)*(1+Alpha*mark(k));
        if l<=LENGTH
            tmp=(Block_dct2(1,1)/Block_dct1(1,1)-1);
            tmp=tmp/Alpha;
            array2(l)=double(tmp);
        end
      else
          Alpha= Alpha1;
          %block_dct1(1,1)=block_dct1(1,1)*(1+Alpha*mark(k));
          if l<LENGTH
          tmp=(Block_dct2(1,1)/Block_dct1(1,1)-1);
            tmp=tmp/Alpha; 
           array2(l)=double(tmp);
          end
    end
    l=l+1;
  end
end
subplot(3,2,5);
plot(array2);
title('提取水印');