import os
import time
from os import listdir
from PIL import Image
from loguru import logger
from PIL import Image
 
 
def image_synthesis(mother_img, son_img, save_img, size_data, coefficient=2.5, coordinate=None):
 """
 mother_img="C:/Users/Administrator/Desktop/QRCode/b.jpg",
 son_img="C:/Users/Administrator/Desktop/QRCode/y.png",
 save_img="C:/Users/Administrator/Desktop/QRCode/newimg.png",
 coordinate=None#如果為None表示直接將子圖在母圖中居中也可以直接賦值坐標
 # coordinate=(50,50)
 :param mother_img: 母圖
 :param son_img: 子圖
 :param save_img: 保存圖片名
 :param size_data: 母圖的高
 :param coefficient: 子圖相對于母圖高度壓縮系數(shù)
 :param coordinate: 子圖在母圖的坐標 （50， 100）- （距離Y軸水平距離， 距離X軸垂直距離）
 :return:
 """
 # 將圖片賦值,方便后面的代碼調(diào)用
 M_Img = Image.open(mother_img)
 S_Img = Image.open(son_img)
 
 # 給圖片指定色彩顯示格式
 M_Img = M_Img.convert("RGBA")  # CMYK/RGBA 轉(zhuǎn)換顏色格式（CMYK用于打印機的色彩，RGBA用于顯示器的色彩）
 
 # 獲取圖片的尺寸
 M_Img_w, M_Img_h = M_Img.size  # 獲取被放圖片的大?。笀D）
 logger.info(f"母圖尺寸：{M_Img.size}")
 S_Img_w, S_Img_h = S_Img.size  # 獲取小圖的大?。ㄗ訄D）
 logger.info(f"子圖尺寸：{S_Img.size}")
 
 son_resize_h = size_data / coefficient
 factor = son_resize_h / S_Img_h if son_resize_h > S_Img_h else S_Img_h / son_resize_h  # 子圖縮小的倍數(shù)1代表不變，2就代表原來的一半
 logger.info(f"子圖重置比例: {factor}")
 size_w = int(S_Img_w / factor)
 size_h = int(S_Img_h / factor)
 
 # 防止子圖尺寸大于母圖
 if S_Img_w > size_w:
  logger.info(f"防止子圖尺寸大于母圖")
  S_Img_w = size_w
 if S_Img_h > size_h:
  logger.info(f"防止子圖尺寸大于母圖")
  S_Img_h = size_h
 
 # 重新設(shè)置子圖的尺寸
 icon = S_Img.resize((S_Img_w, S_Img_h), Image.ANTIALIAS)
 logger.info(f"重置后子圖尺寸：{(S_Img_w, S_Img_h)}")
 
 try:
  if not coordinate or coordinate == "":
w = int((M_Img_w - S_Img_w) / 2)
h = int((M_Img_h - S_Img_h))
coordinate = (w, h)
# 粘貼子圖到母圖的指定坐標（當前水平居中，垂直靠下）
M_Img.paste(icon, coordinate, mask=None)
  else:
logger.info("已經(jīng)指定坐標")
# 粘貼子圖到母圖的指定坐標（指定坐標）
M_Img.paste(icon, coordinate, mask=None)
 except:
  logger.info("坐標指定出錯 ")
 # 保存圖片
 M_Img.save(save_img)
 return save_img
 
 
def image_stitching(origin_img_path, result_img_path, output_img_path, size_data):
 # 獲取當前文件夾中所有JPG圖像
 # im_list = [Image.open(fn) for fn in listdir() if fn.endswith('.jpg')]
 
 origin_data = Image.open(origin_img_path)
 result_data = Image.open(result_img_path)
 
 M_Img_w, M_Img_h = origin_data.size  # 獲取被放圖片的大小
 logger.info(f"待拼接圖片的原尺寸: {(M_Img_w, M_Img_h)}")
 
 # 圖片轉(zhuǎn)化尺寸(注：此業(yè)務(wù)中，origin和result均為尺寸比例相同的圖片（寬高比相同的圖片）)
 factor = M_Img_h / size_data if size_data > M_Img_h else size_data / M_Img_h  # 子圖縮小的倍數(shù)1代表不變，2就代表原來的一半
 size_w = int(M_Img_w / factor)
 logger.info(f"待拼接圖片重置尺寸: {(size_w, size_data)}")
 
 origin_img = origin_data.resize((size_w, size_data), Image.BILINEAR)
 result_img = result_data.resize((size_w, size_data), Image.BILINEAR)
 
 image_list = [origin_img, result_img]
 
 # 單幅圖像尺寸
 width, height = image_list[0].size
 logger.info(f"--- width = {width}, height = {height}")
 
 # 創(chuàng)建空白長圖
 result = Image.new(image_list[0].mode, (width * len(image_list), height))
 
 # # 拼接圖片
 for i, im in enumerate(image_list):
  result.paste(im, box=(i * width, 0))
 
 # 保存圖片
 result.save(output_img_path)
 return stitching_img_path
 
 
if __name__ == '__main__':
 """圖片拼接與堆疊合成腳本"""
 
 # root_path = './1000x966'
 root_path = './500x841'
 # root_path = './1000x667'
 
 size_data = 1280  # 原圖重制尺寸值 TODO 實現(xiàn)圖片重制大小的時候按比例進行寬高的縮放
 origin_img_path = os.path.join(root_path, 'origin_image.png')
 result_img_path = os.path.join(root_path, 'result_image.png')
 face_img_path = os.path.join(root_path, 'face_image.png')
 stitching_img_path = os.path.join(root_path, 'stitching_.png')
 
 # 兩圖左右拼接
 last_img_path = image_stitching(origin_img_path, result_img_path, stitching_img_path, size_data)
 logger.info(f"左右拼接完成 ---")
 
 # 覆蓋小圖片到拼接圖居中靠下
 synthesis_img_path = os.path.join(root_path, 'synthesis_.png')
 res = image_synthesis(last_img_path, face_img_path, synthesis_img_path, size_data,  # coordinate=(100, 500)  )
 logger.info(f"--- end --- res = {res}")