# encode.py
from PIL import Image
import sys

base_image = Image.open('./image.png')
loaded_base_image = base_image.load()
qr_image = Image.open('./code.png')
loaded_qr_image = qr_image.load()


if base_image.size < qr_image.size:
	print("QR Code 이미지의 크기는 Base 이미지의 크기보다 작아야 합니다.")
	sys.exit()


print('0000', base_image.size)
# 0000 (360, 672)
print('1111', qr_image.size)
# 1111 (200, 200)
print('2222', loaded_qr_image[0, 1], loaded_qr_image[0, 1][0])
# 2222 (255, 255, 255, 255) 255
print('3333', loaded_base_image[0, 1], sum(loaded_base_image[0, 1]))
# 3333 (226, 231, 227, 255), 939


# qr code image width/height 만큼 이중 for문을 돌음. x, y 좌표 한픽셀씩 돌아가겠다는 의미
for x in range(qr_image.size[0]):
	for y in range(qr_image.size[1]):
		base_image_pixel = loaded_base_image[x, y] # base image의 한 픽셀
		change = False

		# qr code image	는 흑/백만 있는 것으로 합니다. (이 말은 색이 포함된 qr code는 사용 X)
		# loaded_qr_image[x, y]는 한 픽셀의 컬러값. 0번째 요소는 RGB의 R을 나타냄
		if loaded_qr_image[x, y][0]:
			# R 값이 True이면 넘어옴. 흑/백이므로 0 or 255인데, 0일땐 False
			# base_image_pixel의 sum은 R, G, B 색상 모두를 더한 값
			# base_image_pixel의 sum % 2는 0, 1이 나올 수 있고,
			# 이는 흑 백을 나누기 위해 체크를 함
			# if loaded_qr_image[x, y][0]: => 현재 QR이 흰색일 때의 케이스
			if sum(base_image_pixel) % 2:
				# RGB 합산의 mod 2는 0일때 흑, 1일때 백 임을 나타냄
				# mod 연산 후 1일때 True
				# QR이 백 인데 Base 이미지 같은 위치 픽셀이 mod 연산 시 1(255, 백)인 케이스 => 넘어감
				continue
			else:
				# QR이 백 인데 Base 이미지 같은 위치 픽셀이 mod 연산 시 0(0, 흑)인 케이스 => 변경
				change = True
		else:
			# :else => 현재 QR이 검은색일 때의 케이스
			if sum(base_image_pixel) % 2:
				# QR이 흑 인데 Base 이미지 같은 위치 픽셀이 mod 연산 시 1(255, 백)인 케이스 => 변경
				change = True
			else:
				# QR이 흑 인데 Base 이미지 같은 위치 픽셀이 mod 연산 시 0(0, 흑)인 케이스 => 넘어감
				continue
		if change:
			# change가 True일 때, R 값을 1만 낮춤
			# 1만 변경해주면 색상에 크게 영향이 가지 않음
			# 추후 decoding 시, mod 연산으로 흑백 픽셀만 찍히게 작업한다면?
			# QR 코드 영역은 제대로 나오게 됨
			loaded_base_image[x, y] = (base_image_pixel[0] - 1, base_image_pixel[1], base_image_pixel[2])

base_image.save('encoded_image.png')

# decode.py
from PIL import Image

encode_base_image = Image.open('encoded_image.png')
loaded_encode_base_image = encode_base_image.load()

# encode image width/height 만큼 이중 for문을 돌음. x, y 좌표 한픽셀씩 돌아가겠다는 의미
for x in range(encode_base_image.size[0]):
	for y in range(encode_base_image.size[1]):
		if sum(loaded_encode_base_image[x, y]) % 2:
			# encode때 했던 mod 연산을 동일하게 처리
			# mod 연산 시 1 => 흰색으로 변환
			loaded_encode_base_image[x, y] = (255, 255, 255)
		else:
			# mod 연산 시 0 => 검은색으로 변환
			loaded_encode_base_image[x, y] = (0, 0, 0)

encode_base_image.save('decoded_image.png')