EmojiText/src.py at master · WiiliamC/EmojiText · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
from PIL import Image, ImageDraw, ImageFont
import matplotlib.pyplot as plt
import numpy as np


class TextProcessor:

    def __init__(self, back="😘", fore="💘", col_num=12, debug=False):
        self.background = back  # 背景表情
        self.foreground = fore  # 前景表情
        self.col_num = col_num  # 微信-11，qq-12
        self.debug = debug  # 是否为调试模式
        self._del_set = {' ', '\n'}  # 不进行处理的字符集合

    def _draw_word(self, word):
        """将字符绘制为图像"""
        # 根据中英文确定参数
        if word.encode("UTF-8").isalnum():
            pos = (1, 0)  # 文字位置
            font_size = self.col_num  # 字体大小
            font = ImageFont.truetype("tahoma", size=font_size)  # 文字设置
        else:
            pos = (0, 0)  # 文字位置
            font_size = self.col_num  # 字体大小
            font = ImageFont.truetype("simhei", font_size, encoding='utf-8')  # 文字设置
        temp_size = (self.col_num, round(self.col_num * 1.5))  # 临时图像大小
        image_size = (self.col_num, self.col_num)  # 图像大小
        # 灰度图
        image = Image.new('1', temp_size, 'white')
        draw = ImageDraw.Draw(image)
        draw.text(pos, word, font=font)
        if self.debug:
            plt.imshow(image)  # 使用matplotlib显示
            plt.show()
            print(np.array(image, dtype=bool))  # 转数组
        # 居中平移。不同的字符在同一位置绘制的位置不同，因此需要将他们平移到上下居中、左右居中的位置上。
        image = self._move(image, image_size)
        return image

    @staticmethod
    def _move(image, size):
        """
        居中平移。不同的字符在同一位置绘制的位置不同，因此需要将他们平移到上下居中、左右居中的位置上。
        输入PIL的Image类，返回bool型ndarray类。
        """
        image = np.array(image, dtype=bool)
        bottom, top, left, right = -1, -1, -1, -1
        # 行开始
        for i in range(image.shape[0]):
            if np.all(image[i, :]) and not np.all(image[i + 1, :]):
                top = i
                break
            elif not image[i, :].all():
                top = i
                break
        # 行结束
        for i in range(image.shape[0] - 1, 0, -1):
            if image[i, :].all() and not image[i - 1, :].all():
                bottom = i
                break
            elif not image[i, :].all():
                bottom = i
                break
        # 列开始
        for i in range(image.shape[1]):
            if image[:, i].all() and not image[:, i + 1].all():
                left = i
                break
            elif not image[:, i].all():
                left = i
                break
        # 列结束
        for i in range(image.shape[1] - 1, 0, -1):
            if np.all(image[:, i]) and not np.all(image[:, i - 1]):
                right = i
                break
            elif not image[:, i].all():
                right = i
                break
        # 左右上下居中
        new_image = np.ones(size, dtype=bool)
        h, w = bottom - top - 1, right - left - 1
        new_top, new_left = round((size[0] - h) / 2), round((size[1] - w) / 2)
        new_image[new_top:new_top + h, new_left:new_left + w] = image[top + 1:bottom, left + 1:right]
        return new_image

    def process(self, text):
        """将文本处理为emoji字符串"""
        emojis = ""
        # 遍历文字
        for word in text:
            if word in self._del_set:
                continue
            # 画图
            image = self._draw_word(word)
            for r in range(image.shape[0]):
                for c in range(image.shape[1]):
                    if image[r][c]:
                        emojis += self.background
                    else:
                        emojis += self.foreground
                emojis += '\n'
            emojis += '\n\n'
        return emojis


if __name__ == '__main__':
    tp = TextProcessor(debug=True)
    print(tp.process("！"))