mean wpm added

arman-aminian · Feb 6, 2021 · f75a8a6 · f75a8a6
1 parent a727189
commit f75a8a6
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/parser.go b/parser.go
@@ -13,6 +13,9 @@ func Parse(path string) error {
 	}
 	max := parsers.CalculateMaxWpm(srt, 0, len(srt.Lines))
 	fmt.Println(max)
+	mean := parsers.CalculateMeanWpm(srt, 0, len(srt.Lines))
+	fmt.Println(mean)
+
 	//fmt.Println(srt.Lines[2].WPM)
 	//fmt.Println(srt.Lines[2].Text)
 	return nil

diff --git a/parsers/parser.go b/parsers/parser.go
@@ -15,3 +15,12 @@ func CalculateMaxWpm(p Parser, from, to int) int {
 	}
 	return maxWpm
 }
+
+func CalculateMeanWpm(p Parser, from, to int) int {
+	lines := p.GetLines()[from:to]
+	sum := 0
+	for _, line := range lines {
+		sum += line.WPM
+	}
+	return sum / len(lines)
+}
diff --git a/parsers/srt.go b/parsers/srt.go
@@ -60,7 +60,11 @@ func (s *Srt) Parse(path string) error {
 			}
 			wn := WordsCount(l.Text)
 			t := l.End.Sub(l.Start).Minutes()
-			l.WPM = int(math.Round(float64(wn) / t))
+			if t == 0 {
+				l.WPM = 0
+			} else {
+				l.WPM = int(math.Round(float64(wn) / t))
+			}
 
 			lines = append(lines, l)
 		}