#!/usr/bin/env python
# encoding: utf-8

import jieba
import numpy as np
import PIL.Image as Image
from wordcloud import WordCloud

class wordCloud:
    '''
     This is a main Class, the file contains all documents.
     One document contains paragraphs that have several sentences
     It loads the original file and converts the original file to new content
     Then the new content will be saved by this class
    '''
    def __init__(self):
        self.bg_img = 'assets/picture.jpeg'
        self.word_path = 'assets/word.txt'

    def hello(self):
        '''
        This is a welcome speech
        :return: self
        '''
        print('*' * 50)
        print(' ' * 20 + '詞云制作')
        print(' ' * 5 + 'Author: autofelix  Date: 2022-01-17 13:14')
        print('*' * 50)
        return self

    def run(self):
        '''
        The program entry
        '''
        with open(self.word_path, 'r') as f:
            word = f.read()

        cut_word = ' '.join(jieba.cut(word))
        color_mask = np.array(Image.open(self.bg_img))

        word_cloud = WordCloud(
            # 設(shè)置字體，不指定就會(huì)出現(xiàn)亂碼
            font_path='/System/Library/Fonts/PingFang.ttc',
            # 設(shè)置背景色
            background_color='white',
            # 詞云形狀
            mask=color_mask,
            # 允許最大詞匯
            max_words=120,
            # 最大號(hào)字體
            max_font_size=2000
        ).generate(cut_word)

        word_cloud.to_file('word_cloud.jpg')
        im = word_cloud.to_image()
        im.show()


if __name__ == '__main__':
    wordCloud().hello().run()