'''
Python制作透明背景的電子印章
'''
# 導入包
from PIL import Image,ImageFont,ImageDraw, ImageFilter
from math import pi, cos, sin, tan
from random import randint
 
# 定義方法
def is_Chinese(ch):
    '''判斷字符是否為中文'''
    if '\u4e00' <= ch <= '\u9fff':
        return True
    return False
 
 
def pentagram(x, y, R, yDegree=0):
    '''
    計算五角星各個頂點
    int R:五角星的長軸
    int x, y:五角星的中心點
    int yDegree:長軸與y軸的夾角
    '''
    rad = pi / 180   # 每度的弧度值
    r = R * sin(18 * rad) / cos(36 * rad)   # 五角星短軸的長度
 
    # 求取外圈點坐標
    RVertex = [(x - (R * cos((90 + k * 72 + yDegree) * rad)), y - (R * sin((90 + k * 72 + yDegree) * rad))) for k in range(5)]
    # 求取內圈點坐標
    rVertex = [(x - (r * cos((90 + 36 + k * 72 + yDegree) * rad)), y - (r * sin((90 + 36 + k * 72 + yDegree) * rad))) for k in range(5)]
 
    # 頂點左邊交叉合并
    vertex = [x for y in zip(RVertex, rVertex) for x in y]
    return vertex
 
 
def circle(x, y, r):
    '''計算圓的上下左右切點'''
    return (x - r, y - r, x + r, y + r)
 
 
# 定義類
class Stamp:
    def __init__(self,  edge = 5,               # 圖片邊緣空白的距離
                        H = 160,                # 圓心到中層文字下邊緣的距離
                        R = 250,                # 圓半徑
                        border = 13,            # 字到圓圈內側的距離
                        r = 90,                 # 五星外接圓半徑
                        fill = (255, 0, 0, 120),# 印章顏色， 默認純紅色， 透明度0-255，建議90-180
 
                        words_up = "", # 上部文字
                        angle_up = 270,         # 上部文字弧形角度
                        font_size_up = 80,      # 上部文字大小
                        font_xratio_up = 0.66,  # 上部文字橫向變形比例
                        stroke_width_up = 2,    # 上部文字粗細，一般取值0,1,2,3
 
                        words_mid="",   # 中部文字
                        angle_mid = 72,         # 中部文字弧形角度
                        font_size_mid = 60,     # 中部文字大小
                        font_xratio_mid = 0.7,  # 中部文字橫向變形比例
                        stroke_width_mid=1,     # 中部文字粗細，一般取值0,1,2
 
                        words_down="",# 下部文字
                        angle_down = 60,        # 下部文字弧形角度
                        font_size_down = 20,    # 下部文字大小
                        font_xratio_down = 1,   # 下部文字橫向變形比例
                        stroke_width_down=1,    # 下部文字粗細，一般取值0,1,2
 
                        img_wl_path = "", # 紋理圖路徑
                        save_path = "" # 保存圖片路徑
                 ):
 
        # 打開紋理圖像，隨機截取旋轉
        self.img_wl = Image.open(img_wl_path)
 
        # 圖像初始設置為None
        self.img = None
        self.save_path = save_path
 
        self.fill = fill                       # 印章顏色
        self.edge = edge                       # 圖片邊緣空白的距離
        self.H = H                             # 圓心到中層文字下邊緣的距離
        self.R = R                             # 圓半徑
        self.r = r                             # 五星外接圓半徑
        self.border = border                   # 字到圓圈內側的距離
 
        self.words_up = words_up               # 上部文字
        self.angle_up = angle_up               # 上部文字弧形角度
        self.font_size_up = font_size_up       # 上部文字大小
        self.font_xratio_up = font_xratio_up   # 上部文字橫向變形比例
        self.stroke_width_up = stroke_width_up     # 上部文字粗細，一般取值0,1,2,3
 
        self.words_mid = words_mid             # 中部文字
        self.angle_mid = angle_mid             # 中部文字弧形角度
        self.font_size_mid = font_size_mid     # 中部文字大小
        self.font_xratio_mid = font_xratio_mid # 中部文字橫向變形比例
        self.stroke_width_mid = stroke_width_mid   # 中部文字粗細，一般取值0,1,2,3
 
        self.words_down = words_down           # 下部文字
        self.angle_down = angle_down           # 下部文字弧形角度
        self.font_size_down = font_size_down   # 下部文字大小
        self.font_xratio_down = font_xratio_down  # 下部文字橫向變形比例
        self.stroke_width_down = stroke_width_down   # 中部文字粗細，一般取值0,1,2,3
 
 
    def draw_rotated_text(self, image, angle, xy, r, word, fill, font_size, font_xratio, stroke_width, font_flip = False, *args, **kwargs):
        """
        image:底層圖片
        angle：旋轉角度
        xy：旋轉中心
        r:旋轉半徑
        text：繪制的文字
        fill：文字顏色
        font_size：字體大小
        font_xratio：x方向縮放比例（印章字體寬度較標準宋體偏窄）
        stroke_width： 文字筆畫粗細
        font_flip:文字是否垂直翻轉（印章下部文字與上部是相反的）
        """
 
        # 加載字體文件-直接使用windows自帶字體，中文用simsun， 英文用arial
        if is_Chinese(word):
            font = ImageFont.truetype("C:/Windows/Fonts/simsun.ttc", font_size, encoding="utf-8")
        else:
            font = ImageFont.truetype("C:/Windows/Fonts/arial.ttf", font_size, encoding="utf-8")
 
        # 獲取底層圖片的size
        width, height = image.size
        max_dim = max(width, height)
 
        # 創(chuàng)建透明背景的文字層，大小4倍的底層圖片
        mask_size = (max_dim * 2, max_dim * 2)
        # 印章通常使用較窄的字體，這里將繪制文字的圖層x方向壓縮到font_xratio的比例
        mask_resize = (int(max_dim * 2 * font_xratio), max_dim * 2)
        mask = Image.new('L', mask_size, 0)
 
        # 在上面文字層的中心處寫字，字的左上角與中心對其
        draw = ImageDraw.Draw(mask)
 
        # 獲取當前設置字體的寬高
        bd = draw.textbbox((max_dim, max_dim), word, font=font, align="center", *args, **kwargs)
        font_width = bd[2] - bd[0]
        font_hight = bd[3] - bd[1]
 
        # 文字在圓圈上下的方向，需要通過文字所在文字圖層的位置修正，保證所看到的文字不會上下顛倒
        if font_flip:
            word_pos = (int(max_dim-font_width/2), max_dim+r-font_hight)
        else:
            word_pos = (int(max_dim-font_width/2), max_dim-r)
 
        # 寫字， 以xy為中心，r為半徑，文字上邊中點為圓周點，繪制文字
        draw.text(word_pos, word, 255, font=font, align="center",  stroke_width = stroke_width, *args, **kwargs)
 
        # 調整角度,對于Π*n/2的角度，直接rotate即可，對于非Π*n/2的角度，需要先放大圖片以減少旋轉帶來的鋸齒
        if angle % 90 == 0:
            rotated_mask = mask.resize(mask_resize).rotate(angle)
        else:
            bigger_mask = mask.resize((int(max_dim*8*font_xratio), max_dim*8),
                                      resample=Image.BICUBIC)
            rotated_mask = bigger_mask.rotate(angle).resize(mask_resize, resample=Image.LANCZOS)
 
        # 切割文字的圖片
        mask_xy = (max_dim*font_xratio - xy[0], max_dim - xy[1])
        b_box = mask_xy + (mask_xy[0] + width, mask_xy[1] + height)
        mask = rotated_mask.crop(b_box)
 
        # 粘貼到目標圖片上
        color_image = Image.new('RGBA', image.size, fill)
        image.paste(color_image, mask)
 
 
    def draw_stamp(self):
        #創(chuàng)建一張底圖,用來繪制文字
        img = Image.new("RGBA",(2*(self.R+self.edge),2*(self.R+self.edge)),(255,255,255,0))
        draw = ImageDraw.Draw(img)
 
 
        #繪制圓弧， R為外邊緣，width往圓心算
        draw.arc(circle(self.R+self.edge, self.R+self.edge, self.R), start=0, end=360, fill=self.fill, width = self.border)
 
 
        #繪制多邊形
        draw.polygon(pentagram(self.R+self.edge,self.R+self.edge,self.r), fill=self.fill, outline=self.fill)
 
 
        #繪制上圈文字
        angle_word = self.angle_up / len(self.words_up)
        angle_word_curr = ((len(self.words_up)-1) / 2) * angle_word
 
        for word in self.words_up:
            self.draw_rotated_text(img, angle_word_curr, (self.R+self.edge,self.R+self.edge), self.R-self.border*2, word, self.fill, self.font_size_up, self.font_xratio_up, self.stroke_width_up)
            angle_word_curr = angle_word_curr - angle_word
 
 
        #繪制中層文字
        angle_word = self.angle_mid / len(self.words_mid)
        angle_word_curr = -((len(self.words_mid)-1) / 2) * angle_word
 
        for word in self.words_mid:
            self.draw_rotated_text(img, 0, (self.R+self.edge+ self.H*tan(angle_word_curr*pi/180),self.R+self.edge), self.H, word, self.fill, self.font_size_mid, self.font_xratio_mid, self.stroke_width_mid, font_flip = True)
            angle_word_curr = angle_word_curr + angle_word
 
 
        #繪制下圈文字
        angle_word = self.angle_down / len(self.words_down)
        angle_word_curr = -((len(self.words_down)-1) / 2) * angle_word
 
        for word in self.words_down:
            self.draw_rotated_text(img, angle_word_curr, (self.R+self.edge,self.R+self.edge), self.R-self.border*2, word, self.fill, self.font_size_down, self.font_xratio_down, self.stroke_width_down, font_flip = True)
            angle_word_curr = angle_word_curr + angle_word
 
        # 隨機圈一部分紋理圖
        pos_random = (randint(0,200), randint(0,100))
        box = (pos_random[0], pos_random[1], pos_random[0]+300, pos_random[1]+300)
        img_wl_random = self.img_wl.crop(box).rotate(randint(0,360))
        # 重新設置im2的大小，并進行一次高斯模糊
        img_wl_random = img_wl_random.resize(img.size).convert('L').filter(ImageFilter.GaussianBlur(1))
        # 將紋理圖的灰度映射到原圖的透明度，由于紋理圖片自帶灰度，映射后會有透明效果，所以fill的透明度不能太低
        L, H = img.size
        for h in range(H):
            for l in range(L):
                dot = (l, h)
                img.putpixel(dot, img.getpixel(dot)[:3]+(int(img_wl_random.getpixel(dot)/255*img.getpixel(dot)[3]),))
        # 進行一次高斯模糊，提高真實度
        self.img = img.filter(ImageFilter.GaussianBlur(0.6))
 
    def show_stamp(self):
        if self.img:
            self.img.show()
 
    def save_stamp(self):
        if self.img:
            self.img.save(self.save_path)
 
 
if __name__ == '__main__':
 
    # 上面的字
    words_up = "小劉帶你學py技術社區(qū)" 
    # 中間的字
    words_mid="社區(qū)專用章"
    # 下面的字
    words_down="2022092415"
    # 背景圖片路徑
    img_wl_path = "D:\Back1.jpg"
    # 印章保存路徑
    save_path = "D:\A\Project_1\stamp.png"
 
    # 執(zhí)行方法
    stamp = Stamp(words_up=words_up,words_mid=words_mid,words_down=words_down,img_wl_path=img_wl_path,save_path=save_path)
    stamp.draw_stamp()
    stamp.show_stamp()
    stamp.save_stamp()