From fb77852acbdbbcedcdb9771770cb6771da002851 Mon Sep 17 00:00:00 2001 From: Nick White Date: Wed, 23 Jan 2019 21:54:09 +0000 Subject: Update line-conf-buckets to mostly use package functions too. Working now, but needs more consolidation to be worth it. --- parse/prob/prob.go | 72 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 72 insertions(+) create mode 100644 parse/prob/prob.go (limited to 'parse/prob') diff --git a/parse/prob/prob.go b/parse/prob/prob.go new file mode 100644 index 0000000..5a84567 --- /dev/null +++ b/parse/prob/prob.go @@ -0,0 +1,72 @@ +package prob + +import ( + "bufio" + "io/ioutil" + "path/filepath" + "strconv" + "strings" + + "git.rescribe.xyz/testingtools/parse" +) + +// TODO: probably switch to just relying on io.Reader +func getLineAvg(r *bufio.Reader) (float64, error) { + var err error + + totalconf := float64(0) + num := 0 + + err = nil + for err == nil { + var line string + line, err = r.ReadString('\n') + fields := strings.Fields(line) + + if len(fields) == 2 { + conf, converr := strconv.ParseFloat(fields[1], 64) + if converr != nil { + continue + } + totalconf += conf + num += 1 + } + } + if num <= 0 { + return 0, nil + } + avg := totalconf / float64(num) + return avg, nil +} + +// TODO: probably switch to just relying on io.Reader +// Note this only processes one line at a time +func GetLineDetails(name string, r *bufio.Reader) (parse.LineDetails, error) { + var line parse.LineDetail + lines := make(parse.LineDetails, 0) + + avg, err := getLineAvg(r) + if err != nil { + return lines, err + } + + filebase := strings.Replace(name, ".prob", "", 1) + + txt, err := ioutil.ReadFile(filebase + ".txt") + if err != nil { + return lines, err + } + + line.Name = name + line.Avgconf = avg + line.Text = string(txt) + line.OcrName = filepath.Dir(filebase) + + var imgfn parse.ImgPath + imgfn.Path = filebase + ".bin.png" + line.Img = imgfn + + lines = append(lines, line) + + return lines, nil +} -- cgit v1.2.1-24-ge1ad