import brainweb
import matplotlib.pyplot as plt
import numpy as np
from niftypet.nimpa import imscroll
from niftypet.nimpa.prc import prc

brainweb.seed(1337)
vol = brainweb.get_mmr_fromfile(brainweb.get_files()[0])
img = vol['PET'][:, 100:-100, 100:-100]
knl = np.zeros((3, 5), 'f4')
knl[:, 0:5] = [0.15, 0.15, 0.4, 0.15, 0.15]
img = prc.conv_separable(img, knl)

ref = vol['T1'][:, 100:-100, 100:-100]
res = prc.nlm(img, ref, sigma=1, half_width=2)
imscroll([img, ref, res], titles=["PET", "T1", "NLM"])
plt.show()