summaryrefslogtreecommitdiff
path: root/pkg/prob/prob.go
diff options
context:
space:
mode:
Diffstat (limited to 'pkg/prob/prob.go')
-rw-r--r--pkg/prob/prob.go69
1 files changed, 69 insertions, 0 deletions
diff --git a/pkg/prob/prob.go b/pkg/prob/prob.go
new file mode 100644
index 0000000..8bdb3d5
--- /dev/null
+++ b/pkg/prob/prob.go
@@ -0,0 +1,69 @@
+package prob
+
+import (
+ "io/ioutil"
+ "path/filepath"
+ "strconv"
+ "strings"
+
+ "rescribe.xyz/utils/pkg/line"
+)
+
+func getLineAvg(f string) (float64, error) {
+ totalconf := float64(0)
+ num := 0
+
+ prob, err := ioutil.ReadFile(f)
+ if err != nil {
+ return 0, err
+ }
+
+ for _, l := range strings.Split(string(prob), "\n") {
+ fields := strings.Fields(l)
+
+ if len(fields) == 2 {
+ conf, err := strconv.ParseFloat(fields[1], 64)
+ if err != nil {
+ continue
+ }
+ totalconf += conf
+ num += 1
+ }
+ }
+ if num <= 0 {
+ return 0, nil
+ }
+ avg := totalconf / float64(num)
+ return avg, nil
+}
+
+// Note this only processes one line at a time
+func GetLineDetails(probfn string) (line.Details, error) {
+ var l line.Detail
+ lines := make(line.Details, 0)
+
+ avg, err := getLineAvg(probfn)
+ if err != nil {
+ return lines, err
+ }
+
+ filebase := strings.Replace(probfn, ".prob", "", 1)
+
+ txt, err := ioutil.ReadFile(filebase + ".txt")
+ if err != nil {
+ return lines, err
+ }
+
+ l.Name = filepath.Base(filebase)
+ l.Avgconf = avg
+ l.Text = string(txt)
+ l.OcrName = filepath.Base(filepath.Dir(filebase))
+
+ var imgfn line.ImgPath
+ imgfn.Path = filebase + ".bin.png"
+ l.Img = imgfn
+
+ lines = append(lines, l)
+
+ return lines, nil
+}