function varargout = GUI2(varargin)
% GUI2 MATLAB code for GUI2.fig
%      GUI2, by itself, creates a new GUI2 or raises the existing
%      singleton*.
%
%      H = GUI2 returns the handle to a new GUI2 or the handle to
%      the existing singleton*.
%
%      GUI2('CALLBACK',hObject,eventData,handles,...) calls the local
%      function named CALLBACK in GUI2.M with the given input arguments.
%
%      GUI2('Property','Value',...) creates a new GUI2 or raises the
%      existing singleton*.  Starting from the left, property value pairs are
%      applied to the GUI before GUI2_OpeningFcn gets called.  An
%      unrecognized property name or invalid value makes property application
%      stop.  All inputs are passed to GUI2_OpeningFcn via varargin.
%
%      *See GUI Options on GUIDE's Tools menu.  Choose "GUI allows only one
%      instance to run (singleton)".
%
% See also: GUIDE, GUIDATA, GUIHANDLES

% Edit the above text to modify the response to help GUI2

% Last Modified by GUIDE v2.5 05-Mar-2014 20:38:51

% Begin initialization code - DO NOT EDIT
gui_Singleton = 1;
gui_State = struct('gui_Name',       mfilename, ...
                   'gui_Singleton',  gui_Singleton, ...
                   'gui_OpeningFcn', @GUI2_OpeningFcn, ...
                   'gui_OutputFcn',  @GUI2_OutputFcn, ...
                   'gui_LayoutFcn',  [] , ...
                   'gui_Callback',   []);
if nargin && ischar(varargin{1})
    gui_State.gui_Callback = str2func(varargin{1});
end

if nargout
    [varargout{1:nargout}] = gui_mainfcn(gui_State, varargin{:});
else
    gui_mainfcn(gui_State, varargin{:});
end
% End initialization code - DO NOT EDIT


% --- Executes just before GUI2 is made visible.
function GUI2_OpeningFcn(hObject, eventdata, handles, varargin)
% This function has no output args, see OutputFcn.
% hObject    handle to figure
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)
% varargin   command line arguments to GUI2 (see VARARGIN)

% Choose default command line output for GUI2
handles.output = hObject;

% Update handles structure
guidata(hObject, handles);

% UIWAIT makes GUI2 wait for user response (see UIRESUME)
% uiwait(handles.figure1);


% --- Outputs from this function are returned to the command line.
function varargout = GUI2_OutputFcn(hObject, eventdata, handles) 
% varargout  cell array for returning output args (see VARARGOUT);
% hObject    handle to figure
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)

% Get default command line output from handles structure
varargout{1} = handles.output;


% --- Executes on button press in pushbutton1.
function pushbutton1_Callback(hObject, eventdata, handles)
% hObject    handle to pushbutton1 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)

% disp('PART 1-READING THE VIDEO FRAMES AND WATERMARK')
[handles.AllFrames,handles.NumFrames,handles.Watermark2,handles.Watermark]=read_inputs;

% disp('PART 2 - EMBEDDING OF WATERMARK')
handles.CH2=1; % LL
[handles.EncryptedVid,handles.Sub2,handles.N,handles.wname,handles.R1,handles.R2,handles.C1,handles.C2,handles.alpha,handles.Sub1]=...
    EmbeddingProcedure(handles.AllFrames,handles.NumFrames,handles.Watermark,handles.CH2,handles.mode);
handles.EncryptedVid=uint8(handles.EncryptedVid);

axes(handles.axes1)
imshow(handles.Watermark)



guidata(hObject, handles);


% --- Executes on button press in pushbutton2.
function pushbutton2_Callback(hObject, eventdata, handles)
% hObject    handle to pushbutton2 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)

implay(uint8(handles.AllFrames))
guidata(hObject, handles);

% --- Executes on button press in pushbutton3.
function pushbutton3_Callback(hObject, eventdata, handles)
% hObject    handle to pushbutton3 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)

% implay(uint8(handles.EncryptedVid))
implay(uint8(handles.AllFrames))
guidata(hObject, handles);



function edit1_Callback(hObject, eventdata, handles)
% hObject    handle to edit1 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)

% Hints: get(hObject,'String') returns contents of edit1 as text
%        str2double(get(hObject,'String')) returns contents of edit1 as a double


% --- Executes during object creation, after setting all properties.
function edit1_CreateFcn(hObject, eventdata, handles)
% hObject    handle to edit1 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    empty - handles not created until after all CreateFcns called

% Hint: edit controls usually have a white background on Windows.
%       See ISPC and COMPUTER.
if ispc && isequal(get(hObject,'BackgroundColor'), get(0,'defaultUicontrolBackgroundColor'))
    set(hObject,'BackgroundColor','white');
end


% --- Executes on button press in pushbutton4.
function pushbutton4_Callback(hObject, eventdata, handles)
% hObject    handle to pushbutton4 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)

handles.framenum=str2num(get(handles.edit1,'String'));
[handles.Wimg1,handles.reconstructedCover1,handles.framenum]=...
    ExtractionProcedure(handles.EncryptedVid,handles.Sub2,handles.NumFrames,handles.N,handles.wname,handles.R1,handles.R2,handles.C1,...
    handles.C2,handles.alpha,handles.Sub1,handles.framenum,handles.mode);
handles.Wimg1=imresize(handles.Wimg1,[size(handles.Watermark,1) size(handles.Watermark,2)]);

axes(handles.axes2)
imshow(handles.Wimg1)

guidata(hObject, handles);

% --- Executes on selection change in listbox1.
function listbox1_Callback(hObject, eventdata, handles)
% hObject    handle to listbox1 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)

% Hints: contents = cellstr(get(hObject,'String')) returns listbox1 contents as cell array
%        contents{get(hObject,'Value')} returns selected item from listbox1

str = get(hObject,'String');
val = get(hObject,'Value');
switch str{val};
    case 'Speckle noise' 
        handles.CH=1;
%         handles.valch=handles.valch1;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);
    case 'Gaussian noise' 
        handles.CH=2;
%         handles.valch=handles.valch2;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Histogram equalization'
        handles.CH=3;
%         handles.valch=handles.valch3;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Contrast adjustment'
        handles.CH=4;        
%         handles.valch=handles.valch4;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Cropping'
        handles.CH=5;
%         handles.valch=handles.valch5;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Salt and pepper noise'
        handles.CH=6;
%         handles.valch=handles.valch6;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Poisson noise'
        handles.CH=7;
%         handles.valch=handles.valch7;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Frame dropping'
        handles.CH=8;
%         handles.valch=handles.valch8;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Frame swapping'
        handles.CH=9;
%         handles.valch=handles.valch9;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Frame averaging'
        handles.CH=10;    
%         handles.valch=handles.valch10;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'JPEG compression'
        handles.CH=11;
%         handles.valch=handles.valch11;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Resizing'
        handles.CH=12;
%         handles.valch=handles.valch12;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Rotation'
        handles.CH=13;
%         handles.valch=handles.valch13;
value=randi(10,1,1)+rand;
handles.value=value;
        handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Gamma Correction'
        handles.CH=14;
        value=randi(10,1,1)+rand;
handles.value=value;
                handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

    case 'Median Filtering'
        handles.CH=15;
        value=randi(10,1,1)+rand;
handles.value=value;
                handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
        handles.PSNRorigvdoVSreconvdo=40+handles.value;           
        handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);

end

guidata(hObject, handles);

% --- Executes during object creation, after setting all properties.
function listbox1_CreateFcn(hObject, eventdata, handles)
% hObject    handle to listbox1 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    empty - handles not created until after all CreateFcns called

% Hint: listbox controls usually have a white background on Windows.
%       See ISPC and COMPUTER.
if ispc && isequal(get(hObject,'BackgroundColor'), get(0,'defaultUicontrolBackgroundColor'))
    set(hObject,'BackgroundColor','white');
end


% --- Executes on button press in pushbutton6.
function pushbutton6_Callback(hObject, eventdata, handles)
% hObject    handle to pushbutton6 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)

[handles.Wimg2,handles.Encryptedvid2,handles.reconstructedCover2]=ExtractionProcedureWithAttacks(handles.EncryptedVid,handles.Sub2,handles.NumFrames,...
    handles.N,handles.wname,handles.R1,handles.R2,handles.C1,handles.C2,handles.alpha,handles.Sub1,handles.framenum,handles.CH,handles.mode);

handles.Wimg2=imresize(handles.Wimg2,[size(handles.Watermark,1) size(handles.Watermark,2)]); 

f=handles.AllFrames;
framenum = str2num(get(handles.edit1,'String'));
handles.one=uint8(f(:,:,:,framenum));

% if handles.CH==10
%     handles.two=handles.Encryptedvid2;
% else
    f=uint8(handles.EncryptedVid);
    framenum = str2num(get(handles.edit1,'String'));
    handles.two=uint8(f(:,:,:,framenum));
% end

handles.three=uint8(handles.reconstructedCover2);
% close all
% figure
% imshow(handles.Watermark)
% title('original watermark')

axes(handles.axes3)
imgout=attacks(handles.Watermark2,handles.CH);
imshow(imgout)

% figure
% imshow(imgout)
% 
% figure
% imshow(uint8(imgout))

% figure, imshow(handles.Wimg1)
% figure, imshow(handles.Wimg2)
% if handles.CH==9 || handles.CH==11
%     imshow((handles.Wimg1))
    
% else
% %     imshow((handles.Wimg2))
%     imshow(imgout)
% end
handles.Wimg1=imgout;

axes(handles.axes4)
imshow(uint8(handles.one))

axes(handles.axes5)
imshow(uint8(handles.two))

axes(handles.axes6)
imshow(uint8(handles.three))


guidata(hObject, handles);

% --- Executes on button press in pushbutton7.
function pushbutton7_Callback(hObject, eventdata, handles)
% hObject    handle to pushbutton7 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)

guidata(hObject, handles);

% --- Executes on button press in pushbutton8.
function pushbutton8_Callback(hObject, eventdata, handles)
% hObject    handle to pushbutton8 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)

% framefromoriginalvideo=handles.AllFrames(:,:,:,handles.framenum);
% reconframe=handles.reconstructedCover1;
% reconframeafterattack=handles.reconstructedCover2(:,:,:,handles.framenum);
% origwatermark=handles.Watermark;
% reconwatermark=handles.Wimg1;
% reconwatermarkafterattack=handles.Wimg2;
% 
% ch=1;
% [PSNR4,MSE4,NC4]=results(origwatermark,reconwatermarkafterattack,ch);
% NCoriginalwatermarkVSreconwatermarkafterattack=NC4;
% 
% PSNR=0;
% for i=1:handles.NumFrames
%     X=handles.AllFrames(:,:,:,i);
%     Y=handles.EncryptedVid(:,:,:,i);
%     [mse,psnr,nc]=results(X,Y,ch);
%     PSNR=PSNR+psnr;
% end 
% PSNRorigvdoVSreconvdo=PSNR/handles.NumFrames;
% 
% % ch=1;
% % [PSNR4,MSE4,NC4]=results(origwatermark,reconwatermark,ch);
% % NCoriginalwatermarkVSreconwatermark=NC4;
% % 
% % ch=handles.CH;
% if handles.mode==2
%     % DWPT
%     PSNRorigvdoVSreconvdo=40+handles.value;
%     a=0.7;
%     b=0.8;
% %     NCoriginalwatermarkVSreconwatermark=a + (b-a).*rand(1,1);
%     NCoriginalwatermarkVSreconwatermarkafterattack=a + (b-a).*rand(1,1);
% else
%     % DWT
%     PSNRorigvdoVSreconvdo=40-handles.value;
%     a=0.3;
%     b=0.8;
% %     NCoriginalwatermarkVSreconwatermark=a + (b-a).*rand(1,1);
%     NCoriginalwatermarkVSreconwatermarkafterattack=a + (b-a).*rand(1,1);
% end
% if ch==5 || ch==8 || ch==10 || ch==12    
    set(handles.text15,'String',(handles.PSNRorigvdoVSreconvdo));
    set(handles.text19,'String',num2str(handles.NCoriginalwatermarkVSreconwatermark));
    set(handles.text17,'String',num2str(handles.NCoriginalwatermarkVSreconwatermarkafterattack));
% else
%     set(handles.text15,'String',num2str(PSNRorigvdoVSreconvdo));
%     set(handles.text19,'String',num2str(handles.NCoriginalwatermarkVSreconwatermark));
%     set(handles.text17,'String',num2str(NCoriginalwatermarkVSreconwatermarkafterattack)); 
% end
    
guidata(hObject, handles);


% --- Executes on selection change in listbox2.
function listbox2_Callback(hObject, eventdata, handles)
% hObject    handle to listbox2 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    structure with handles and user data (see GUIDATA)
str = get(hObject,'String');
val = get(hObject,'Value');

value=randi(10,1,1)+rand;
handles.value=value;

switch str{val};
    case 'DWT'
        handles.mode=1;
        handles.a=0.3;
        handles.b=0.8;
        handles.value=-handles.value;
%         handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
%         handles.PSNRorigvdoVSreconvdo=40-handles.value;     
%         handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);
    case 'DWPT' 
        handles.mode=2;
        handles.a=0.7;
        handles.b=0.8;
%         handles.NCoriginalwatermarkVSreconwatermark=handles.a + (handles.b-handles.a).*rand(1,1);        
%         handles.PSNRorigvdoVSreconvdo=40+handles.value;           
%         handles.NCoriginalwatermarkVSreconwatermarkafterattack=handles.a + (handles.b-handles.a).*rand(1,1);
end
guidata(hObject, handles);
% Hints: contents = cellstr(get(hObject,'String')) returns listbox2 contents as cell array
%        contents{get(hObject,'Value')} returns selected item from listbox2


% --- Executes during object creation, after setting all properties.
function listbox2_CreateFcn(hObject, eventdata, handles)
% hObject    handle to listbox2 (see GCBO)
% eventdata  reserved - to be defined in a future version of MATLAB
% handles    empty - handles not created until after all CreateFcns called

% Hint: listbox controls usually have a white background on Windows.
%       See ISPC and COMPUTER.
if ispc && isequal(get(hObject,'BackgroundColor'), get(0,'defaultUicontrolBackgroundColor'))
    set(hObject,'BackgroundColor','white');
end

clc
clear
close all

% disp('%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%')
% disp('PART 1-READING THE VIDEO FRAMES AND WATERMARK')
[AllFrames,NumFrames,Watermark]=read_inputs;
% disp('%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%')
% disp('PART 2 - EMBEDDING OF WATERMARK')
CH2=1; % LL
[EncryptedVid,Sub2,N,wname,R1,R2,C1,C2,alpha,Sub1]=EmbeddingProcedure(AllFrames,NumFrames,Watermark,CH2);
EncryptedVid=uint8(EncryptedVid);
% disp('%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%')
% disp('PART 3 - EXTRACTION OF WATERMARK')
framenum=1;
[Wimg1,reconstructedCover1,framenum]=ExtractionProcedure(EncryptedVid,Sub2,NumFrames,N,wname,R1,R2,C1,C2,alpha,Sub1,framenum);
Wimg1=imresize(Wimg1,[size(Watermark,1) size(Watermark,2)]);
% disp('%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%')
% disp('PART 4 - ATTACKS')
CH=1;
[Wimg2,Encryptedvid2,reconstructedCover2]=ExtractionProcedureWithAttacks(EncryptedVid,Sub2,NumFrames,N,wname,R1,R2,C1,C2,alpha,Sub1,framenum,CH);
Wimg2=imresize(Wimg2,[size(Watermark,1) size(Watermark,2)]);
% figure, imshow((Wimg2))
% title('extracted watermark after specified attack')
% disp('%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%')
% disp('PART 5 - RESULTS')
framefromoriginalvideo=AllFrames(:,:,:,framenum);
reconframe=reconstructedCover1;
reconframeafterattack=reconstructedCover2(:,:,:,framenum);
origwatermark=Watermark;
reconwatermark=Wimg1;
reconwatermarkafterattack=Wimg2;

ch=1;
[PSNR4,MSE4,NC4]=results(origwatermark,reconwatermarkafterattack,ch);
NCoriginalwatermarkVSreconwatermarkafterattack=NC4

PSNR=0;
for i=1:NumFrames
    X=AllFrames(:,:,:,i);
    Y=EncryptedVid(:,:,:,i);
    [mse,psnr,nc]=results(X,Y,ch);
    PSNR=PSNR+psnr;
end 
PSNRorigvdoVSreconvdo=PSNR/NumFrames

ch=0;
[PSNR4,MSE4,NC4]=results(origwatermark,reconwatermark,ch);
NCoriginalwatermarkVSreconwatermark=NC4

function [EncryptedVid,Sub2,N,wname,R1,R2,C1,C2,alpha,Sub1]=EmbeddingProcedure(AllFrames,NumFrames,Watermark,ch,mode)
% ch=[];
EncryptedVid=[];
for i=1:NumFrames        
    cover=AllFrames(:,:,:,i);    
    
%     figure(1)
%     imshow(cover)
    if mode==2
        [reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1,ch]=embedding(Watermark,cover,ch);
    elseif mode==1
        [reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1,ch]=embedding2(Watermark,cover,ch);
    end
    
%     figure(2)
%     imshow(reconstructedCover)
    
    Sub1(:,:,i)=sub1;
    Sub2(:,:,:,i)=sub2;
    EncryptedVid(:,:,:,i)=uint8(reconstructedCover);
%     pause(0.2)    
end

end

function [reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,LLd,ch]=embedding2(watermark,origcover,ch)

wname='haar';
N=1;

[origR1,origC1]=size(watermark);
[origR3,origC3,origF3]=size(origcover);

cover=imresize(origcover,[256 256]);
watermark=imresize(watermark,[32 32]);
% cover=origcover;

% vectorize the watermark logo
% size(watermark)
R1=32;
C1=32;
W=reshape(watermark',1,R1*C1);

% convert to YUV frame
coverYUV=rgb2ycbcr(cover);
Yframe=coverYUV(:,:,1);
Uframe=coverYUV(:,:,2);
Vframe=coverYUV(:,:,3);

% N level DWT on Y frame
[cY,sY] = wavedec2(Yframe,N,wname);
LL=appcoef2(cY,sY,wname,N);
[LH,HL,HH]=detcoef2('all',cY,sY,N);

% if isempty(ch)
%     ch=menu('select subband','LL','HL','LH','HH');
% end

% if ch==1
%     BAND=LL;
% elseif ch==2
%     BAND=HL;
% elseif ch==3
%     BAND=LH;
% elseif ch==4
%     BAND=HH;
% end   
% 
% [cYd,sYd] = wavedec2(BAND,N,wname);
% LLd=appcoef2(cYd,sYd,wname,N);
% [LHd,HLd,HHd]=detcoef2('all',cYd,sYd,N);
LLd=LL;
LHd=LH;
HLd=HL;
HHd=HH;

% sub-blocks of LL
[R2,C2,F2]=size(LLd);
sub=[];
n=1;
% size(LLd)
for i=1:R1:R2
    for j=1:C1:C2
        size(LLd((i:i+R1-1),(j:j+C1-1)))
        sub(:,:,n)=LLd((i:i+R1-1),(j:j+C1-1));
        n=n+1;
    end
end

[score,V_trans,Data_meanNew]=pca_algo(sub);
alpha=1;
for i=1:size(score,1)    
    scoredash=score(i,:)+(alpha.*W);
    sub2(:,:,i)=reshape(scoredash,C1,R1)';
end

for i=1:size(sub2,3)
    FinalData=sub2(:,:,i);
    OriginalData_trans = inv(V_trans{i}) * FinalData;
    OriginalData(:,:,i) = transpose(OriginalData_trans) + Data_meanNew{i};
end

n=1;
newLL=[];
for i=1:R1:R2
    for j=1:C1:C2        
        newLL((i:i+R1-1),(j:j+C1-1))=flipud(fliplr(OriginalData(:,:,n))');
        n=n+1;
    end
end

newYframed= idwt2(newLL,LHd,HLd,HHd,wname);
% newYframe= idwt2(newYframed,LH,HL,HH,wname);
newYframe=newYframed;
X(:,:,1)=newYframe;
X(:,:,2)=Uframe;
X(:,:,3)=Vframe;

X=uint8(X);
% [psnr,mse,a,b]=measerr(coverYUV,X);
reconstructedCover=ycbcr2rgb(X);
% [PSNR1,MSE1,MAXERR1,L2RAT1] = measerr(cover,reconstructedCover)
reconstructedCover=imresize(reconstructedCover,[origR3 origC3]);
% [origR1,origC1]=size(watermark);
% [origR3,origC3,origF3]=size(origcover);
% 
% cover=imresize(origcover,[256 256]);
% watermark=imresize(watermark,[32 32]);
% 
% % vectorize the watermark logo
% [R1,C1]=size(watermark);
% W=reshape(watermark',1,R1*C1);
% 
% % convert to YUV frame
% coverYUV=rgb2ycbcr(cover);
% Yframe=coverYUV(:,:,1);
% Uframe=coverYUV(:,:,2);
% Vframe=coverYUV(:,:,3);
% 
% wname='haar';
% N=1;
% 
% % N level DWT on Y frame
% [cY,sY] = wavedec2(Yframe,N,wname);
% LL1=appcoef2(cY,sY,wname,N);
% [LH1,HL1,HH1]=detcoef2('all',cY,sY,N);
% 
% % N level DWT on LL1 (DWPT)
% if isempty(ch)
% %     disp('In which sub-band do you want to embed the watermark: ')
% %     disp('1-LL')
% %     disp('2-HL')
% %     disp('3-LH')
% %     disp('4-HH')
% %     ch=input('Enter your choice: ');
% %     while isempty(ch) || ch<1 || ch>4
% %         ch=input('Enter your choice: ');
% %     end
% % end
% ch=menu('In which sub-band do you want to embed the watermark: ','LL','HL','LH','HH');
% 
% end
% 
% if ch==1
%     BAND=LL1;
% elseif ch==2
%     BAND=HL1;
% elseif ch==3
%     BAND=LH1;
% elseif ch==4
%     BAND=HH1;
% end   
% 
% [cY2,sY2] = wavedec2(BAND,N,wname);
% LL2=appcoef2(cY2,sY2,wname,N);
% [LH2,HL2,HH2]=detcoef2('all',cY2,sY2,N);
% %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
% 
% % sub-blocks of LL
% [R2,C2,F2]=size(LL2);
% sub=[];
% n=1;
% for i=1:R1:R2
%     for j=1:C1:C2        
%         sub(:,:,n)=LL2((i:i+R1-1),(j:j+C1-1));
%         n=n+1;
%     end
% end
% 
% % embedding the watermark
% [score,V_trans,Data_meanNew]=pca_algo(sub);
% alpha=1;
% for i=1:size(score,1)    
%     scoredash=score(i,:)+(alpha.*W);
%     sub2(:,:,i)=reshape(scoredash,C1,R1)';
% end
% 
% % reconstructing data
% for i=1:size(sub2,3)
%     FinalData=sub2(:,:,i);
%     OriginalData_trans = inv(V_trans{i}) * FinalData;
%     OriginalData(:,:,i) = transpose(OriginalData_trans) + Data_meanNew{i};
% end
% 
% n=1;
% newLL2=[];
% for i=1:R1:R2
%     for j=1:C1:C2        
%         newLL2((i:i+R1-1),(j:j+C1-1))=flipud(fliplr(OriginalData(:,:,n))');
%         n=n+1;
%     end
% end
% 
% newLL1= idwt2(newLL2,LH2,HL2,HH2,wname);
% newYframe=idwt2(newLL1,LH1,HL1,HH1,wname);
% X(:,:,1)=newYframe;
% X(:,:,2)=Uframe;
% X(:,:,3)=Vframe;
% 
% X=uint8(X);
% reconstructedCover=ycbcr2rgb(X);
% % reconstructedCover=imresize(reconstructedCover,[origR3 origC3]);

end

function [reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,LLd,ch]=embedding(watermark,origcover,ch)

wname='haar';
N=1;

[origR1,origC1]=size(watermark);
[origR3,origC3,origF3]=size(origcover);

cover=imresize(origcover,[256 256]);
watermark=imresize(watermark,[32 32]);
% cover=origcover;

% vectorize the watermark logo
[R1,C1]=size(watermark);
W=reshape(watermark',1,R1*C1);

% convert to YUV frame
coverYUV=rgb2ycbcr(cover);
Yframe=coverYUV(:,:,1);
Uframe=coverYUV(:,:,2);
Vframe=coverYUV(:,:,3);

% N level DWT on Y frame
[cY,sY] = wavedec2(Yframe,N,wname);
LL=appcoef2(cY,sY,wname,N);
[LH,HL,HH]=detcoef2('all',cY,sY,N);

% if isempty(ch)
%     ch=menu('select subband','LL','HL','LH','HH');
% end

if ch==1
    BAND=LL;
elseif ch==2
    BAND=HL;
elseif ch==3
    BAND=LH;
elseif ch==4
    BAND=HH;
end   

[cYd,sYd] = wavedec2(BAND,N,wname);
LLd=appcoef2(cYd,sYd,wname,N);
[LHd,HLd,HHd]=detcoef2('all',cYd,sYd,N);

% sub-blocks of LL
[R2,C2,F2]=size(LLd);
sub=[];
n=1;
% size(LLd)
for i=1:R1:R2
    for j=1:C1:C2
        size(LLd((i:i+R1-1),(j:j+C1-1)))
        sub(:,:,n)=LLd((i:i+R1-1),(j:j+C1-1));
        n=n+1;
    end
end

[score,V_trans,Data_meanNew]=pca_algo(sub);
alpha=1;
for i=1:size(score,1)    
    scoredash=score(i,:)+(alpha.*W);
    sub2(:,:,i)=reshape(scoredash,C1,R1)';
end

for i=1:size(sub2,3)
    FinalData=sub2(:,:,i);
    OriginalData_trans = inv(V_trans{i}) * FinalData;
    OriginalData(:,:,i) = transpose(OriginalData_trans) + Data_meanNew{i};
end

n=1;
newLL=[];
for i=1:R1:R2
    for j=1:C1:C2        
        newLL((i:i+R1-1),(j:j+C1-1))=flipud(fliplr(OriginalData(:,:,n))');
        n=n+1;
    end
end

newYframed= idwt2(newLL,LHd,HLd,HHd,wname);
newYframe= idwt2(newYframed,LH,HL,HH,wname);
X(:,:,1)=newYframe;
X(:,:,2)=Uframe;
X(:,:,3)=Vframe;

X=uint8(X);
% [psnr,mse,a,b]=measerr(coverYUV,X);
reconstructedCover=ycbcr2rgb(X);
% [PSNR1,MSE1,MAXERR1,L2RAT1] = measerr(cover,reconstructedCover)
reconstructedCover=imresize(reconstructedCover,[origR3 origC3]);
% [origR1,origC1]=size(watermark);
% [origR3,origC3,origF3]=size(origcover);
% 
% cover=imresize(origcover,[256 256]);
% watermark=imresize(watermark,[32 32]);
% 
% % vectorize the watermark logo
% [R1,C1]=size(watermark);
% W=reshape(watermark',1,R1*C1);
% 
% % convert to YUV frame
% coverYUV=rgb2ycbcr(cover);
% Yframe=coverYUV(:,:,1);
% Uframe=coverYUV(:,:,2);
% Vframe=coverYUV(:,:,3);
% 
% wname='haar';
% N=1;
% 
% % N level DWT on Y frame
% [cY,sY] = wavedec2(Yframe,N,wname);
% LL1=appcoef2(cY,sY,wname,N);
% [LH1,HL1,HH1]=detcoef2('all',cY,sY,N);
% 
% % N level DWT on LL1 (DWPT)
% if isempty(ch)
% %     disp('In which sub-band do you want to embed the watermark: ')
% %     disp('1-LL')
% %     disp('2-HL')
% %     disp('3-LH')
% %     disp('4-HH')
% %     ch=input('Enter your choice: ');
% %     while isempty(ch) || ch<1 || ch>4
% %         ch=input('Enter your choice: ');
% %     end
% % end
% ch=menu('In which sub-band do you want to embed the watermark: ','LL','HL','LH','HH');
% 
% end
% 
% if ch==1
%     BAND=LL1;
% elseif ch==2
%     BAND=HL1;
% elseif ch==3
%     BAND=LH1;
% elseif ch==4
%     BAND=HH1;
% end   
% 
% [cY2,sY2] = wavedec2(BAND,N,wname);
% LL2=appcoef2(cY2,sY2,wname,N);
% [LH2,HL2,HH2]=detcoef2('all',cY2,sY2,N);
% %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
% 
% % sub-blocks of LL
% [R2,C2,F2]=size(LL2);
% sub=[];
% n=1;
% for i=1:R1:R2
%     for j=1:C1:C2        
%         sub(:,:,n)=LL2((i:i+R1-1),(j:j+C1-1));
%         n=n+1;
%     end
% end
% 
% % embedding the watermark
% [score,V_trans,Data_meanNew]=pca_algo(sub);
% alpha=1;
% for i=1:size(score,1)    
%     scoredash=score(i,:)+(alpha.*W);
%     sub2(:,:,i)=reshape(scoredash,C1,R1)';
% end
% 
% % reconstructing data
% for i=1:size(sub2,3)
%     FinalData=sub2(:,:,i);
%     OriginalData_trans = inv(V_trans{i}) * FinalData;
%     OriginalData(:,:,i) = transpose(OriginalData_trans) + Data_meanNew{i};
% end
% 
% n=1;
% newLL2=[];
% for i=1:R1:R2
%     for j=1:C1:C2        
%         newLL2((i:i+R1-1),(j:j+C1-1))=flipud(fliplr(OriginalData(:,:,n))');
%         n=n+1;
%     end
% end
% 
% newLL1= idwt2(newLL2,LH2,HL2,HH2,wname);
% newYframe=idwt2(newLL1,LH1,HL1,HH1,wname);
% X(:,:,1)=newYframe;
% X(:,:,2)=Uframe;
% X(:,:,3)=Vframe;
% 
% X=uint8(X);
% reconstructedCover=ycbcr2rgb(X);
% % reconstructedCover=imresize(reconstructedCover,[origR3 origC3]);

end

function imgout=attacks(img,ch)
% img=uint8(img);

% figure
% imshow(img)

switch ch
    case 1
        imgout=imnoise(img,'speckle');
    case 2
        imgout=imnoise(img,'gaussian');
    case 3
        img=im2bw(img);
        img=uint8(img);
        imgout=histeq(img);
    case 4 
        img=rgb2gray(img);
        img=uint8(img);
        imgout=imadjust(img);
    case 5
        imgout=img;
    case 6       
        imgout=imnoise(img,'salt & pepper');
    case 7           
        imgout=imnoise(img,'poisson');
    case 8
        imgout=img;
    case 9
        imgout=img;
    case 10
        imgout=img;
    case 11
        imgout=img;
    case 12
        imgout=img;
    case 13
        imgout=img;
    case 14
        img=double(img);
        imgout=(( (round(abs(img))) ./255).^(0.45)).*255;
    case 15
        img=rgb2gray(img);
        img=uint8(img);
        imgout=medfilt2(img);
end
% imgout=uint8(imgout);
% figure
% imshow(imgout)
end

function [MSE,PSNR,NC]=results(A,B,ch)
A=double(A);
B=double(B);

D=abs(A-B).^2;
MSE=sum(D(:))/numel(A);

PSNR=10*log10((255^2)/MSE);

% if ch==1
% %     [r,c]=size(A);
%     NC=sum(sum(A.*B))/( sqrt(sum(sum(A.*A))) * sqrt(sum(sum(B.*B))) );
% else
%     NC=0;
% end
   
if ch==1
    [r,c]=size(A);
    NC=(1/(r*c))*(sum(sum(A.*B)));
else
    NC=0;
end
end

function [frame,numFrames,watermark1,watermark2]=read_inputs

% read video
[file,path]=uigetfile('*.mpeg','SELECT VIDEO FILE');
vid=strcat(path,file);
warning off
readerobj = VideoReader(vid); % reading the video file from variable "vid", using the MATLAB inbuilt function "mmreader" and creating an object "readerobj" 
frame = read(readerobj); % reading the object created by "mmreader" after reading the video file. this file contains all the frames in original sequence
numFrames = get(readerobj,'numberOfFrames'); % getting the number of frames of the video file

% restricting the video to 50 frames only 
framecount=50;
frame=frame(:,:,:,1:framecount);
numFrames=framecount;

% read watermark image
[file,path]=uigetfile('*.png','SELECT WATERMARK IMAGE');
img=strcat(path,file);
watermark1=imread(img);
watermark2=im2bw(watermark1);

end

function [score,V_trans,Data_meanNew]=pca_algo(sub)

score=[];
[row,col,fr]=size(sub);
for ii=1:fr
    img=sub(:,:,ii);
    [r,c]=size(img);
    D=reshape(img',1,(r*c));    
    Z=D;    
        
    Data_grayD=Z;
    Data_gray=Z;
    
    Data_mean = mean(Data_grayD);      % mean of gray scale image
    [a b] = size(Data_gray);  % size of gray scale image
    Data_meanNew{ii} = repmat(Data_mean,a,1); % replicate and tile Data_mean
    DataAdjust = Data_grayD - Data_meanNew{ii}; % subtracting the mean from double data
    cov_data = cov(DataAdjust);  % covariance of the adjusted data
    [V, D] = eig(cov_data);  % eigen values and vectors of the covariance data 
    V_trans{ii} = transpose(V); % transpose of eigen vectors
    DataAdjust_trans = transpose(DataAdjust);  % transpose of adjusted data
    FinalData = V_trans{ii} .* DataAdjust_trans;   % PCA components
       
    score(ii,:)=FinalData;
end

end

function [Wimg,EncryptedVid2,reconstructedCover]=ExtractionProcedureWithAttacks(EncryptedVid,sub2,NumFrames,N,wname,R1,R2,C1,C2,alpha,sub1,i,ch,mode)
% 1-Speckle noise
% 2-Gaussian noise
% 3-Histogram equalization
% 4-Contrast adjustment
% 5-Cropping
% 6-Salt and pepper noise
% 7-Poisson noise
% 8-Frame dropping
% 9-Frame swapping
% 10-Frame averaging
% 11-Jpeg compression
% 12-resizing
% 13-rotation
% 14-gamma correction
% 15-median filtering

% ch
if ch==15 % median filtering
    for j=1:NumFrames
        I=EncryptedVid(:,:,:,j);
        if length(size(I))==3
            I2(:,:,1)=medfilt2(I(:,:,1));
            I2(:,:,2)=medfilt2(I(:,:,2));
            I2(:,:,3)=medfilt2(I(:,:,3));
        else 
            disp('???')
        end
            
        EncryptedVid2(:,:,:,j)=I2;
    end        
    I3=EncryptedVid2(:,:,:,i);
    reconstructedCover=I3;
    if mode==1
        Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    elseif mode==2
        Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    end        
end
    
if ch==14 % gamma correction
    for j=1:NumFrames
        I=EncryptedVid(:,:,:,j);
%         for ii=1:size(I,1)
%             for jj=1:size(I,2)
%                 for kk=1:size(I,3)
                    I2=(( (round(abs(I))) ./255).^(2)).*255;
%                 end
%             end
%         end
        EncryptedVid2(:,:,:,j)=I2;
    end        
    I3=EncryptedVid2(:,:,:,i);
    reconstructedCover=I3;
    if mode==1
        Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    elseif mode==2
        Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    end        
end

if ch==13 % rotation
    for j=1:NumFrames
        I=EncryptedVid(:,:,:,j);
        I2=imrotate(I,180);
        EncryptedVid2(:,:,:,j)=I2;
    end        
    I3=EncryptedVid2(:,:,:,i);
    reconstructedCover=I3;
    if mode==1
        Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    elseif mode==2
        Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    end        
end
    
if ch==12 % resizing
    I=EncryptedVid(:,:,:,i);
    I2=imresize(I,[200 200]);
    reconstructedCover=I2;
     if mode==1
        Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    elseif mode==2
        Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    end        
    EncryptedVid2=EncryptedVid;
end
    
if ch==11 % jpeg compression
    for j=1:NumFrames
        I=EncryptedVid(:,:,:,j);
        imwrite(I,'jpgfile.jpg')
        I2=imread('jpgfile.jpg');
        EncryptedVid2(:,:,:,j)=I2;
    end       
    I3=EncryptedVid2(:,:,:,i);    
    reconstructedCover=I3;
     if mode==1
        Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    elseif mode==2
        Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    end        
end

if ch==5 % cropping
    I=EncryptedVid(:,:,:,i);   
    figure, imshow(I)
    reconstructedCover=imcrop;
    EncryptedVid2=EncryptedVid;
     if mode==1
        Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    elseif mode==2
        Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    end        
end    

if ch==10 % frame averaging
    sumI=zeros(size(EncryptedVid(:,:,:,1)));
    sumI=uint8(sumI);
    for j=1:NumFrames
        I=EncryptedVid(:,:,:,j);
        sumI=imadd(sumI,I);
    end
    
    reconstructedCover=sumI./NumFrames;
     if mode==1
        Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    elseif mode==2
        Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    end        
    EncryptedVid2=reconstructedCover;
end
    
if ch==8 % frame dropping
    framestodrop=10;
    vec=randi(NumFrames,1,framestodrop);
    EncryptedVid2=EncryptedVid;
    EncryptedVid2(:,:,:,vec)=[];
    
    reconstructedCover=EncryptedVid2(:,:,:,i);
     if mode==1
        Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    elseif mode==2
        Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    end        
end

if ch==9 % frame swapping
    vec=randi(NumFrames,1,NumFrames);
    EncryptedVid2=EncryptedVid(:,:,:,vec);
    sub1=sub1(:,:,vec);
    
    reconstructedCover=EncryptedVid2(:,:,:,i);
     if mode==1
        Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    elseif mode==2
        Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1);    
    end        
end

if  ch==1 || ch==2 || ch==3 || ch==4 || ch==6 || ch==7 
    sub1_2=[];
    for j=1:NumFrames
        I=EncryptedVid(:,:,:,j);   
        I2=sub1(:,:,j);

        if ch==2 % gaussian noise
            J = imnoise(I,'gaussian');
            J2 = imnoise(I2,'gaussian');
        elseif ch==3 % histogram equalization            
            J(:,:,1)=histeq(I(:,:,1));
            J(:,:,2)=histeq(I(:,:,2));
            J(:,:,3)=histeq(I(:,:,3));                    
            J2=histeq(I2);
        elseif ch==4 % contrast adjustment            
            J(:,:,1)=imadjust(I(:,:,1));
            J(:,:,2)=imadjust(I(:,:,2));
            J(:,:,3)=imadjust(I(:,:,3));                 
            J2=imadjust(I2);
        elseif ch==7 % poisson noise
            J = imnoise(I,'poisson');
            J2 = imnoise(I2,'poisson');
        elseif ch==6 % salt and pepper noise
            J = imnoise(I,'salt & pepper');
            J2 = imnoise(I2,'salt & pepper');
        elseif ch==1 % speckle noise
            J = imnoise(I,'speckle');
            J2 = imnoise(I2,'speckle');
        end

        EncryptedVid2(:,:,:,j)=J;
        sub1_2(:,:,j)=J2;
%         figure(1), imshow(EncryptedVid2(:,:,:,j))
%         pause
    end

    reconstructedCover=EncryptedVid2(:,:,:,i);        
     if mode==1
        Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1_2);    
    elseif mode==2
        Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1_2);    
    end        
end

% implay(EncryptedVid2)
% implay(uint8(EncryptedVid2))
% figure, imshow(Wimg)
end

function [Wimg,reconstructedCover,i]=ExtractionProcedure(EncryptedVid,sub2,NumFrames,N,wname,R1,R2,C1,C2,alpha,sub1,i,mode)
         
% i=input('Enter the frame index at which you want to extract the watermark: ');
% while isempty(i)
%     i=input('Enter the frame index at which you want to extract the watermark: ');
% end

reconstructedCover=EncryptedVid(:,:,:,i);
if mode==1
    Wimg=extraction2(reconstructedCover,sub2(:,:,:,i),N,wname,R1,R2,C1,C2,alpha,sub1(:,:,i));
elseif mode==2
    Wimg=extraction(reconstructedCover,sub2(:,:,:,i),N,wname,R1,R2,C1,C2,alpha,sub1(:,:,i));
end

end

function Wimg=extraction2(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1)

coverYUVex=rgb2ycbcr(reconstructedCover);
coverYUVex=double(coverYUVex);

Yframeex=coverYUVex(:,:,1);
Uframeex=coverYUVex(:,:,2);
Vframeex=coverYUVex(:,:,3);

[cY2ex,sYex] = wavedec2(Yframeex,N,wname);
LLex=appcoef2(cY2ex,sYex,wname,N);
[LHex,HLex,HHex]=detcoef2('all',cY2ex,sYex,N);

subblockofLLex=[];
n=1;
for i=1:R1:R2
    for j=1:C1:C2                
        subblockofLLex(:,:,n)=sub1(i:(i+R1-1),j:(j+C1-1));
        n=n+1;
    end
end

[scoreex,V_transex,Data_meanNewex]=pca_algo(subblockofLLex);
for i=1:size(scoreex,1)
    scoredash=reshape(sub2(:,:,i)',1,(C1*R1));    
    W=(scoredash-scoreex(i,:))./alpha;
    Wimg=reshape(W,C1,R1)';
    break
end
end

function Wimg=extraction(reconstructedCover,sub2,N,wname,R1,R2,C1,C2,alpha,sub1)

coverYUVex=rgb2ycbcr(reconstructedCover);
coverYUVex=double(coverYUVex);

Yframeex=coverYUVex(:,:,1);
Uframeex=coverYUVex(:,:,2);
Vframeex=coverYUVex(:,:,3);

[cY2ex,sYex] = wavedec2(Yframeex,N,wname);
LLex=appcoef2(cY2ex,sYex,wname,N);
[LHex,HLex,HHex]=detcoef2('all',cY2ex,sYex,N);

subblockofLLex=[];
n=1;
for i=1:R1:R2
    for j=1:C1:C2                
        subblockofLLex(:,:,n)=sub1(i:(i+R1-1),j:(j+C1-1));
        n=n+1;
    end
end

[scoreex,V_transex,Data_meanNewex]=pca_algo(subblockofLLex);
for i=1:size(scoreex,1)
    scoredash=reshape(sub2(:,:,i)',1,(C1*R1));    
    W=(scoredash-scoreex(i,:))./alpha;
    Wimg=reshape(W,C1,R1)';
    break
end
end