121 lines
2.7 KiB
Go
121 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/dariubs/percent"
|
|
"gonum.org/v1/plot"
|
|
"gonum.org/v1/plot/plotter"
|
|
"gonum.org/v1/plot/plotutil"
|
|
"gonum.org/v1/plot/vg"
|
|
"io/ioutil"
|
|
"log"
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
type letter struct {
|
|
upperCase string
|
|
lowerCase string
|
|
count int
|
|
frequency float64
|
|
}
|
|
|
|
type letterList []letter
|
|
|
|
func (l letterList) Len() int {
|
|
return len(l)
|
|
}
|
|
|
|
func (l letterList) Less(i, j int) bool {
|
|
return l[i].frequency > l[j].frequency
|
|
}
|
|
|
|
func (l letterList) Swap(i, j int) {
|
|
l[i], l[j] = l[j], l[i]
|
|
}
|
|
|
|
const LettersInTheAlphabet = 26
|
|
|
|
var alphabet = [LettersInTheAlphabet]rune{
|
|
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M',
|
|
'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
|
|
}
|
|
|
|
var letters = make([]letter, LettersInTheAlphabet)
|
|
|
|
func initLetterStruct() {
|
|
for index := range alphabet {
|
|
letters[index].upperCase = string(alphabet[index])
|
|
letters[index].lowerCase = string(alphabet[index] + 32)
|
|
}
|
|
}
|
|
|
|
func readFile(relativePath string) string {
|
|
content, err := ioutil.ReadFile(relativePath)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
return string(content)
|
|
}
|
|
|
|
func countLetters(inputText string) int {
|
|
totalCount := 0
|
|
for index := range alphabet {
|
|
letters[index].count +=
|
|
strings.Count(inputText, letters[index].lowerCase) +
|
|
strings.Count(inputText, letters[index].upperCase)
|
|
totalCount += letters[index].count
|
|
}
|
|
return totalCount
|
|
}
|
|
|
|
func calculateFrequency(totalCount int) {
|
|
for index := range alphabet {
|
|
letters[index].frequency = percent.PercentOf(letters[index].count, totalCount)
|
|
}
|
|
}
|
|
|
|
func printResult() {
|
|
for index := range alphabet {
|
|
l := letters[index]
|
|
fmt.Printf(
|
|
"The letter %s (%s) occurs %d times in the text and the frequencyArray in percent is %0.2f\n",
|
|
l.upperCase, l.lowerCase, l.count, l.frequency,
|
|
)
|
|
}
|
|
}
|
|
|
|
func plotResult() {
|
|
var count plotter.Values
|
|
for index := range alphabet {
|
|
count = append(count, float64(letters[index].count))
|
|
}
|
|
p := plot.New()
|
|
p.Title.Text = "Letter count"
|
|
p.X.Label.Text = "letter"
|
|
p.Y.Label.Text = "count"
|
|
countBars, err := plotter.NewBarChart(count, vg.Points(LettersInTheAlphabet/2))
|
|
countBars.Color = plotutil.Color(2)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
p.Add(countBars)
|
|
var xAxisNaming []string
|
|
for index := range letters {
|
|
xAxisNaming = append(xAxisNaming, letters[index].upperCase)
|
|
}
|
|
p.NominalX(xAxisNaming...)
|
|
if err := p.Save(25*vg.Centimeter, 15*vg.Centimeter, "barChart.png"); err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
initLetterStruct()
|
|
//totalCount := countLetters(readFile("plaintext.txt"))
|
|
totalCount := countLetters(readFile("../part01-encrypt/output.txt"))
|
|
calculateFrequency(totalCount)
|
|
plotResult()
|
|
sort.Sort(letterList(letters))
|
|
printResult()
|
|
}
|