#!/usr/bin/env python3
from scipy.misc import imread, imsave
from numpy import zeros, ones, arange

im = imread('noise1-rawresult-scaled.png')
import noise1_data
data = noise1_data.data.flatten()

values = zeros(256)
all_masks = zeros((1024, 1024))
for idx in range(256):
  px, py = idx%16, idx//16; mask_w = 40; mask = zeros((1024, 1024));
  mask[64*py + (64-mask_w)//2 : 64*py + (64-mask_w)//2 + mask_w,
       64*px + (64-mask_w)//2 : 64*px + (64-mask_w)//2 + mask_w] = 1/(mask_w**2)
  values[idx] = (im*mask).sum()
  all_masks[mask>0] = 1.0

imsave('noise1-rawresult-scaled-mask.png', all_masks)
print(values)

from pylab import *
border = zeros((16, 16))
border[:,0] = 1
border[:,-1] = 1
border[0,:] = 1
border[-1,:] = 1
border = border.flatten()
plot(data[border == 0], values[border == 0], '*', label='not border')
plot(data[border == 1], values[border == 1], '.', label='border')
legend()
xlabel('brightness (input)')
ylabel('brightness (photo)')

show()