__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