from PIL import Image

def init():
  # 設置每個像素區塊的大小
  block_size = 75
  img = Image.open("a.jpg")
  # 獲取圖片的寬高
  width, height = img.size
  # 獲取像素點對應RGB顏色值，可以改變img_array中的值來改變顏色值
  img_array = img.load()
  # 為了處理最后的區塊，加了一次循環
  max_width = width + block_size
  max_height = height + block_size
  for x in range(block_size - 1, max_width, block_size):
    for y in range(block_size - 1, max_height, block_size):
      # 如果是最后一次循環，則x坐標等于width - 1
      if x == max_width - max_width % block_size - 1:
        x = width - 1
      # 如果是最后一次循環，則x坐標等于height - 1
      if y == max_height - max_height % block_size - 1:
        y = height - 1
      # 改變每個區塊的顏色值
      change_block(x, y, block_size, img_array)
      y += block_size
    x += block_size
  img.save(r'D:\python\pixel_image\awesome_copy.png')
  img.show()

"""
:param x坐標 x: 
:param y坐標 y: 
:param 區塊大小 black_size: 
:param 可操作圖片數組 img_array: 
"""
def change_block(x, y, black_size, img_array):

  color_dist = {}
  block_pos_list = []
  for pos_x in range(-black_size + 1, 1):
    for pos_y in range(-black_size + 1, 1):
      # todo print(x + pos_x,y + pos_y)
      block_pos_list.append([x + pos_x, y + pos_y])
  for pixel in block_pos_list:
    if not str(img_array[pixel[0], pixel[1]]) in color_dist.keys():
      color_dist[str(img_array[pixel[0], pixel[1]])] = 1
    else:
      color_dist[str(img_array[pixel[0], pixel[1]])] += 1
  # key-->value => value-->key
  new_dict = {v: k for k, v in color_dist.items()}
  max_color = new_dict[max(color_dist.values())]
  # 將區塊內所有的顏色值設置為顏色最多的顏色
  for a in block_pos_list:
    img_array[a[0], a[1]] = tuple(list(map(int, max_color[1:len(max_color) - 1].split(","))))


def get_key(dict, value):
  return [k for k, v in dict.items() if v == value]


if __name__ == "__main__":
  init()