|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
import cppy.cp_util as util
|
|
|
|
|
from collections import Counter
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class WordFrequencyStateMachine:
|
|
|
|
|
def __init__(self, file_path):
|
|
|
|
|
self.file_path = file_path
|
|
|
|
|
self.content = None
|
|
|
|
|
self.words = None
|
|
|
|
|
self.word_freq = None
|
|
|
|
|
self.state = 'IDLE'
|
|
|
|
|
|
|
|
|
|
def transition_to_read_file(self):
|
|
|
|
|
try:
|
|
|
|
|
with open(self.file_path, 'r', encoding='utf-8') as file:
|
|
|
|
|
self.content = file.read()
|
|
|
|
|
self.state = 'WORDS_SPLIT'
|
|
|
|
|
except FileNotFoundError:
|
|
|
|
|
print(f"文件 {self.file_path} 未找到。")
|
|
|
|
|
except Exception as e:
|
|
|
|
|
print(f"读取文件时发生错误: {e}")
|
|
|
|
|
|
|
|
|
|
def transition_to_split_words(self):
|
|
|
|
|
if self.content is not None:
|
|
|
|
|
self.words = util.extract_str_words(self.content)
|
|
|
|
|
self.state = 'CALCULATE_FREQ'
|
|
|
|
|
else:
|
|
|
|
|
print("文件内容为空，无法分割单词。")
|
|
|
|
|
|
|
|
|
|
def transition_to_calculate_freq(self):
|
|
|
|
|
if self.words is not None:
|
|
|
|
|
self.word_freq = Counter(self.words)
|
|
|
|
|
self.state = 'DONE'
|
|
|
|
|
else:
|
|
|
|
|
print("单词列表为空，无法计算词频。")
|
|
|
|
|
|
|
|
|
|
def run(self):
|
|
|
|
|
while self.state != 'DONE':
|
|
|
|
|
if self.state == 'IDLE':
|
|
|
|
|
self.transition_to_read_file()
|
|
|
|
|
elif self.state == 'WORDS_SPLIT':
|
|
|
|
|
self.transition_to_split_words()
|
|
|
|
|
elif self.state == 'CALCULATE_FREQ':
|
|
|
|
|
self.transition_to_calculate_freq()
|
|
|
|
|
else:
|
|
|
|
|
print(f"未知状态: {self.state}")
|
|
|
|
|
break
|
|
|
|
|
|
|
|
|
|
return self.word_freq
|
|
|
|
|
|
|
|
|
|
# 使用状态机计算词频
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
state_machine = WordFrequencyStateMachine(util.testfilepath)
|
|
|
|
|
word_frequencies = state_machine.run()
|
|
|
|
|
|
|
|
|
|
# 打印结果
|
|
|
|
|
util.print_word_freqs(word_frequencies.most_common(10))
|
