#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import turtle
 
##全局變量##
#詞頻排列顯示個(gè)數(shù)
count = 10
#單詞頻率數(shù)組-作為y軸數(shù)據(jù)
data = []
#單詞數(shù)組-作為x軸數(shù)據(jù)
words = []
#y軸顯示放大倍數(shù)-可以根據(jù)詞頻數(shù)量進(jìn)行調(diào)節(jié)
yScale = 6
#x軸顯示放大倍數(shù)-可以根據(jù)count數(shù)量進(jìn)行調(diào)節(jié)
xScale = 30
 
################# Turtle Start #################### 
#從點(diǎn)(x1,y1)到(x2,y2)繪制線段
def drawLine(t, x1, y1, x2, y2):
  t.penup()
  t.goto (x1, y1)
  t.pendown()
  t.goto (x2, y2)
 
# 在坐標(biāo)(x,y)處寫文字
def drawText(t, x, y, text):
  t.penup()
  t.goto (x, y)
  t.pendown()
  t.write(text)
 
def drawGraph(t):
  #繪制x/y軸線
  drawLine (t, 0, 0, 360, 0)
  drawLine (t, 0, 300, 0, 0)
 
  #x軸: 坐標(biāo)及描述
  for x in range(count):
    x=x+1 #向右移一位,為了不畫在原點(diǎn)上
    drawText(t, x*xScale-4, -20, (words[x-1]))
    drawText(t, x*xScale-4, data[x-1]*yScale+10, data[x-1])
  drawBar(t)
 
#繪制一個(gè)柱體
def drawRectangle(t, x, y):
  x = x*xScale
  y = y*yScale#放大倍數(shù)顯示
  drawLine(t, x-5, 0, x-5, y)
  drawLine(t, x-5, y, x+5, y)
  drawLine(t, x+5, y, x+5, 0)
  drawLine(t, x+5, 0, x-5, 0)
   
#繪制多個(gè)柱體
def drawBar(t):
  for i in range(count):
    drawRectangle(t, i+1, data[i])  
################# Turtle End ####################
 
     
#對(duì)文本的每一行計(jì)算詞頻的函數(shù)
def processLine(line, wordCounts):
  #用空格替換標(biāo)點(diǎn)符號(hào)
  line = replacePunctuations(line)
  #從每一行獲取每個(gè)詞
  words = line.split() 
  for word in words:
    if word in wordCounts:
      wordCounts[word] += 1
    else:
      wordCounts[word] = 1
 
#空格替換標(biāo)點(diǎn)的函數(shù)
def replacePunctuations(line):
  for ch in line:
    if ch in "~@#$%^&*()_-+=<>?/,.:;{}[]|\'""":
      line = line.replace(ch, " ")
  return line
 
def main():
  #用戶輸入一個(gè)文件名
  filename = input("enter a filename:").strip()
  infile = open(filename, "r")
   
  #建立用于計(jì)算詞頻的空字典
  wordCounts = {}
  for line in infile:
    processLine(line.lower(), wordCounts)
     
  #從字典中獲取數(shù)據(jù)對(duì)
  pairs = list(wordCounts.items())
 
  #列表中的數(shù)據(jù)對(duì)交換位置,數(shù)據(jù)對(duì)排序
  items = [[x,y]for (y,x)in pairs] 
  items.sort() 
 
  #輸出count個(gè)數(shù)詞頻結(jié)果
  for i in range(len(items)-1, len(items)-count-1, -1):
    print(items[i][1]+"\t"+str(items[i][0]))
    data.append(items[i][0])
    words.append(items[i][1])
     
  infile.close()
   
  #根據(jù)詞頻結(jié)果繪制柱狀圖
  turtle.title('詞頻結(jié)果柱狀圖')
  turtle.setup(900, 750, 0, 0)
  t = turtle.Turtle()
  t.hideturtle()
  t.width(3)
  drawGraph(t)
  turtle.done()
     
#調(diào)用main()函數(shù)
if __name__ == '__main__':
  main()