import concurrent.futures  
from collections import Counter  
import cppy.cp_util as util


class WordFrequencyAgent:  
    def __init__(self, words):          
        self.words = words  
  
    def compute_word_frequency(self):          
        self.word_freq = Counter(self.words)  
  
    def get_word_frequency(self):  
        return self.word_freq

    
# 将文本分割成多个部分，并为每个部分创建一个Agent
def create_agents(words, num_agents = 4 ):  
    text_chunks = [ words[i::num_agents] for i in range(num_agents) ]  
    agents = [ WordFrequencyAgent(chunk) for chunk in text_chunks ]
    return agents    


def compute_all_word_frequencies(agents):  
    with concurrent.futures.ThreadPoolExecutor() as executor:  
        # 使用线程池来并行计算词频  
        future_to_agent = {executor.submit(agent.compute_word_frequency): agent for agent in agents}  
        for future in concurrent.futures.as_completed(future_to_agent):  
            agent = future_to_agent[future]  
            data = future.result()   # 词频被保存在agent中


# 所有Agent计算完成后，合并它们的词频结果
def merge_word_frequencies(agents):  
    merged_freq = Counter()  
    for agent in agents:  
        merged_freq.update(agent.get_word_frequency())  
    return merged_freq  


if __name__ == '__main__':       
    words = util.extract_file_words(util.testfilepath) # 从文本抽词
    agents = create_agents(words)  # 创建代理
    compute_all_word_frequencies(agents)  # 计算
    merged_word_freq = merge_word_frequencies(agents)   # 合并结果    
    util.print_word_freqs(merged_word_freq.most_common(10))  # 排序输出