__author__ = 'chunk'

import numpy as np
import jpegObj

b = jpegObj.Jpeg("res/test2.jpg")
# c =  b.getCoefBlocks(channel='Y')
# c1,c2,c3 =  b.getCoefBlocks(channel='All')
# print c2


print b.setCoefBlock(np.array([[0] * 8 for i in range(8)]))

c = b.getCoefBlock(channel='Y', loc=(0, 0))
print c

c = b.getCoefBlock(channel='Y', loc=(-1, 1))
print c

b.setCoefMatrix(np.array([[0] * 800 for i in range(600)]),channel='Y')

c = b.getCoefBlock(channel='Y', loc=(-1, 1))
print c

c = b.getCoefBlock(channel='Y', loc=(0, 0))
print c