function [sim,PSNR]=dct_total_ss(yuzhi,a,attack_style,attack_strength)
%%% attack_style 用于設置選擇何種攻擊
%%% yuzhi 用于選擇嵌入系數
%%% a 用于控制嵌入強度
%%% sim 返回相關系數
%%% psnr 返回峰值信噪比
%%% attack_strength 用于控制攻擊的強弱
%%%%%%%%%%%%%%%% 讀入原始載體圖像 %%%%%%%%%%%%%%%%%%%%%%%%%%%
[x,map]=imread('lena.bmp');
[row,col]=size(x);
M=row;
N=col;
MN=col*row;
figure(1),imshow(uint8(x));
title('原始圖像','Fontsize',16,'color','blue');
x_source=x;
fun1=@dct2; %%%分割圖像進行二維DCT變換
J1=blkproc(x,[256 256],fun1);
%%%%%%%%%%%%%%%% 產生長度為1000的偽隨機序列作為水印信息 %%%%%%%
mm=normrnd(0,1,[1 1000]);
MNm=1000;
z=mm;
num=1;
num1=1;
num2=1;
t=1;
%%%%%%%%%%%%%%%% 水印信息的嵌入 %%%%%%%%%%%%%%%%%%%%%%%%%%%%
J11=J1;
for i=1:M
    for k=1:N
        if abs(J1(i,k))>yuzhi && t<=MNm && (i+k>2)
            J11(i,k)=J1(i,k)*(1+a*z(t));
            t=t+1;
        end
    end
end
marksize=t-1;
J2=IDCT2(J11); %%% 二維DCT逆變換
imwrite(uint8(round(J2)),'ss_watermarked.bmp');
figure(2),imshow(uint8(round(J2)));
%%%%%%%%%%%%%%%% 計算峰值信噪比PSNR %%%%%%%%%%%%%%%%%%%%%%%%
x_temp1=J2-double(x_source);
figure(3),imshow(uint8(round(100*x_temp1)));
imwrite((uint8(round(100*x_temp1))),'ss_chazhi.bmp');
x_temp2=x_temp1( : );
x_temp3=abs(x_temp2);
x_temp4=x_temp3'*x_temp3;
d_embed=x_temp4/(M*M);
SDR1=255*255/d_embed;
PSNR=10*log10(SDR1);
%%%%%%%%%%%%%%%% 攻擊 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
s=imread('ss_watermarked.bmp');
J2=s;
if attack_style==1
    %%放大兩倍操作(提取前先縮小兩倍)
    xxx1=imresize(J2,2,'bicubic');
    xxx2=imresize(xxx1,1/2,'bicubic');
    yy=double(xxx2);
end
if attack_style==2
    %%放大4倍操作(提取前先縮小4倍)
    xxx1=imresize(J2,4,'bicubic');
    xxx2=imresize(xxx1,1/4,'bicubic');
    yy=double(xxx2);
end
if attack_style==3
    %%縮小1/4操作
    xxx1=imresize(J2,3/4,'bicubic');
    xxx2=imresize(xxx1,4/3,'bicubic');
    yy=double(xxx2);
end
if attack_style==4
    %%
    xxx1=imresize(J2,2/4,'bicubic');
    xxx2=imresize(xxx1,4/2,'bicubic');
    yy=double(xxx2);
end
%%3*3空域低通濾波
if attack_style==5
    B=(1/9)*ones(3,3);
    xxx2=filter2(B,J2);
    yy=double(xxx2);
end
%%4領域平均
if attack_style==6
    B=[0 1 0;1 0 1;0 1 0]*(1/4);
    xxx2=filter2(B,J2);
    yy=double(xxx2);
end
%%8領域平均
if attack_style==7
    B=[1 1 1;1 0 1;1 1 1]*(1/8);
    xxx2=filter2(B,J2);
    yy=double(xxx2);
end
%%窗口中值濾波
if attack_style==8
    xxx2=medfilt2(J2); %默認3*3
    yy=double(xxx2);
end
if attack_style==9
    % a1=input('Please input length of window a1:');
    % b1=input('Please input length of window b1:');
    a1=1;
    b1=3;
    xxx2=medfilt2(J2,[a1 b1]);
    save al al;
    save b1 b2;
    yy=double(xxx2);
end
%% 裁減
if attack_style==10
    for i=128-44:128+45
        for j=128-45:128+44
            J2(i,j)=0;
        end
    end
    yy=double(J2);
end
if attack_style==11
    for i=128-64:128+63
        for j=128-64:128+63
            J2(i,j)=0;
        end
    end
    yy=double(J2);
end
if attack_style==12
    yy=imnoise(uint8(round(J2)),'gaussian',0,attack_strength); %高斯噪聲
end
if attack_style==13
    imwrite(uint8(round(J2)),'jpeg_n.jpg','jpg','Quality',attack_strength);
    [yy,map]=imread('jpeg_n.jpg','jpg');
end
if attack_style>13
    yy=J2;
end
J3=blkproc(yy,[256 256],fun1);
%%%%%%%%%%%%%%%%%%%%% 水印提取 %%%%%%%%%%%%%%%%%
t1=zeros(MNm,1);
t0=zeros(MNm,1);
t=1;
for i=1:M
    for k=1:N
        if abs(J1(i,k))>yuzhi && t<=marksize && (i+k>2)
            mn(t)=(J3(i,k)-J1(i,k))/(a*J1(i,k));
            t=t+1;
        end
    end
end
if marksize<MNm
    mn(marksize+1:MNm)=mm(marksize+1:MNm);
end
%%%%%%%%%%%%%% 計算相關性 %%%%%%%%%%%%%%%%%%%]
sim=(sum(mn(1:MNm).*mm(1:MNm)))/((sum(mn(1:MNm).*mn(1:MNm)))^0.5);