% Created by 12/12/2017 by Tarmizi Adam.
% This demo is to show ADMM total variation deblurring. The main solver is
% "ADMM_DeblurTV.m". Refer to mentioned file for more details

clc;
clear all;
close all;

Img = imread('Lighthouse256.bmp');

if size(Img,3) > 1
    Img = rgb2gray(Img);
end

K     =   fspecial('average',3); % For denoising
f = imfilter(Img,K,'circular');

f = double(f);

BSNR = 20;
sigma = BSNR2WGNsigma(f, BSNR);

f = f +  sigma * randn(size(Img)); %Add a little noise

%*** ADMM algorithm parameter set up ***

opts.lam = 1.5;
opts.rho = 1.3;
opts.tol = 1e-5;
opts.Nit = 400;

out = ADMM_DeblurTV(f,Img,K,opts);

figure;
imshow(uint8(f));

figure;
imshow(uint8(out.sol))
title(sprintf('ADMM-TV Deblurred (PSNR = %3.3f dB,SSIM = %3.3f) ',...
                       psnr_fun(out.sol,double(Img)),ssim_index(out.sol,double(Img))));