#! /usr/bin/env python3
import cv2 as cv
import numpy as np
import scipy.signal as sig
from noise import *
f3 = np.ones((3,3)) / 9
f5 = np.ones((5,5)) / 25
f7 = np.ones((7,7)) / 49
frame = cv.imread("lenna-awgn.png")
awgn = normaliseImage(frame)
frame = cv.imread("lenna-snp.png")
snp = normaliseImage(frame)
snp1 = sig.convolve2d(snp,f7)
snp2 = cv.filter2D(snp,-1,f7)
awgn1 = sig.convolve2d(awgn,f7)
awgn2 = cv.filter2D(awgn,-1,f7)
cv.imwrite("lenna-awgn-blur.png",toIntImage(awgn1))
cv.imwrite("lenna-awgn-filter.png",toIntImage(awgn2))
cv.imwrite("lenna-snp-blur.png",toIntImage(snp1))
cv.imwrite("lenna-snp-filter.png",toIntImage(snp2))