# -*- coding: utf-8 -*-
'''
將一張圖片填充為正方形后切為9張圖
'''
from PIL import Image
import sys
#將圖片填充為正方形
def fill_image(image):
  width, height = image.size
  #選取長(zhǎng)和寬中較大值作為新圖片的
  new_image_length = width if width > height else height
  #生成新圖片[白底]
  new_image = Image.new(image.mode, (new_image_length, new_image_length), color='white')
  #將之前的圖粘貼在新圖上，居中
  if width > height:#原圖寬大于高，則填充圖片的豎直維度
    #(x,y)二元組表示粘貼上圖相對(duì)下圖的起始位置
    new_image.paste(image, (0, int((new_image_length - height) / 2)))
  else:
    new_image.paste(image, (int((new_image_length - width) / 2),0))
  return new_image
#切圖
def cut_image(image):
  width, height = image.size
  item_width = int(width / 3)
  box_list = []
  # (left, upper, right, lower)
  for i in range(0,3):#兩重循環(huán)，生成9張圖片基于原圖的位置
    for j in range(0,3):
      #print((i*item_width,j*item_width,(i+1)*item_width,(j+1)*item_width))
      box = (j*item_width,i*item_width,(j+1)*item_width,(i+1)*item_width)
      box_list.append(box)

  image_list = [image.crop(box) for box in box_list]
  return image_list
#保存
def save_images(image_list):
  index = 1
  for image in image_list:
    image.save('./python'+str(index) + '.png', 'PNG')
    index += 1

if __name__ == '__main__':
  file_path = "python.jpeg"
  image = Image.open(file_path)
  #image.show()
  image = fill_image(image)
  image_list = cut_image(image)
  save_images(image_list)