# -*- coding: utf-8 -*-

import cv2

from PIL import Image, ImageOps, ImageFilter


# 轉(zhuǎn)換成漫畫風格
def toCarttonStyle(picturePath):
 # 設置輸入輸出路徑和文件名稱
 imgInput_FileName = picturePath
 imgOutput_FileName = picturePath.split(".")[0] + '_cartoon.' + picturePath.split(".")[1]

 # 屬性設置
 num_down = 2 # 縮減像素采樣的數(shù)目
 num_bilateral = 7 # 定義雙邊濾波的數(shù)目

 # 讀取圖片
 img_rgb = cv2.imread(imgInput_FileName)

 # 用高斯金字塔降低取樣
 img_color = img_rgb
 for _ in range(num_down):
 img_color = cv2.pyrDown(img_color)

 # 重復使用小的雙邊濾波代替一個大的濾波
 for _ in range(num_bilateral):
 img_color = cv2.bilateralFilter(img_color, d=9, sigmaColor=9, sigmaSpace=7)

 # 升采樣圖片到原始大小
 for _ in range(num_down):
 img_color = cv2.pyrUp(img_color)

 # 轉(zhuǎn)換為灰度并且使其產(chǎn)生中等的模糊
 img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_RGB2GRAY)
 img_blur = cv2.medianBlur(img_gray, 7)

 # 檢測到邊緣并且增強其效果
 img_edge = cv2.adaptiveThreshold(img_blur, 255,
     cv2.ADAPTIVE_THRESH_MEAN_C,
     cv2.THRESH_BINARY,
     blockSize=9,
     C=2)
 
 # 算法處理后，照片的尺寸可能會不統(tǒng)一
 # 把照片的尺寸統(tǒng)一化
 height=img_rgb.shape[0]
 width = img_rgb.shape[1]
 img_color=cv2.resize(img_color,(width,height))
 
 # 轉(zhuǎn)換回彩色圖像
 img_edge = cv2.cvtColor(img_edge, cv2.COLOR_GRAY2RGB)
 img_cartoon = cv2.bitwise_and(img_color, img_edge)

 # 保存轉(zhuǎn)換后的圖片
 cv2.imwrite(imgOutput_FileName, img_cartoon)
 print('文件轉(zhuǎn)換成漫畫成功，保存在' + imgOutput_FileName)


# 透明度轉(zhuǎn)換 素描轉(zhuǎn)換的一部分
def dodge(a, b, alpha):
 # alpha為圖片透明度
 return min(int(a * 255 / (256 - b * alpha)), 255)


# 圖片轉(zhuǎn)換為素描
def toSketchStyle(picturePath, blur=25, alpha=1.0):
 # 設置輸入輸出路徑和文件名稱
 imgInput_FileName = picturePath
 imgOutput_FileName = picturePath.split(".")[0] + '_Sketch.' + picturePath.split(".")[1]

 # 轉(zhuǎn)化成ima對象
 img = Image.open(picturePath)
 # 將文件轉(zhuǎn)成灰色
 img1 = img.convert('L')

 img2 = img1.copy()

 img2 = ImageOps.invert(img2)

 # 模糊度
 for i in range(blur):
 img2 = img2.filter(ImageFilter.BLUR)
 width, height = img1.size
 for x in range(width):
 for y in range(height):
  a = img1.getpixel((x, y))
  b = img2.getpixel((x, y))
  img1.putpixel((x, y), dodge(a, b, alpha))

 # 保存轉(zhuǎn)換后文件
 img1.save(imgOutput_FileName)
 print('文件轉(zhuǎn)換成漫畫成功，保存在' + imgOutput_FileName)


if __name__ == '__main__':
 imgInput_FileName = input('輸入文件路徑：')
 while True:
 print('1、漫畫風格')
 print('2、素描風格')
 userChoose = input('請選擇風格（輸入序號即可）：')
 if userChoose.__eq__('1'):
  toCarttonStyle(imgInput_FileName)
  break
 elif userChoose.__eq__('2'):
  toSketchStyle(imgInput_FileName)
  break
 else:
  print('違法輸入（請輸入序號）')
  break