summaryrefslogtreecommitdiff
path: root/avg-lines
diff options
context:
space:
mode:
Diffstat (limited to 'avg-lines')
-rw-r--r--avg-lines/html.go61
-rw-r--r--avg-lines/main.go (renamed from avg-lines/avg-lines.go)20
2 files changed, 65 insertions, 16 deletions
diff --git a/avg-lines/html.go b/avg-lines/html.go
new file mode 100644
index 0000000..f299830
--- /dev/null
+++ b/avg-lines/html.go
@@ -0,0 +1,61 @@
+package main
+
+import (
+ "fmt"
+ "os"
+ "path/filepath"
+
+ "rescribe.xyz/go.git/lib/line"
+)
+
+func copylineimg(fn string, l line.Detail) error {
+ f, err := os.Create(fn)
+ if err != nil {
+ return err
+ }
+ defer f.Close()
+
+ return l.Img.CopyLineTo(f)
+}
+
+func htmlout(dir string, lines line.Details) error {
+ err := os.MkdirAll(dir, 0700)
+ if err != nil {
+ return err
+ }
+
+ fn := filepath.Join(dir, "index.html")
+ f, err := os.Create(fn)
+ if err != nil {
+ return err
+ }
+ defer f.Close()
+
+ _, err = fmt.Fprintf(f, "<!DOCTYPE html><html><head><meta charset='UTF-8'><title></title>" +
+ "<style>td {border: 1px solid #444}</style></head><body>\n<table>\n")
+ if err != nil {
+ return err
+ }
+ for _, l := range lines {
+ fn = filepath.Base(l.OcrName) + "_" + l.Name + ".png"
+ err = copylineimg(filepath.Join(dir, fn), l)
+ if err != nil {
+ return err
+ }
+ _, err = fmt.Fprintf(f, "<tr>\n" +
+ "<td><h1>%.4f%%</h1></td>\n" +
+ "<td>%s %s</td>\n" +
+ "<td><img src='%s' width='100%%' /><br />%s</td>\n" +
+ "</tr>\n",
+ l.Avgconf, l.OcrName, l.Name, fn, l.Text)
+ if err != nil {
+ return err
+ }
+ }
+ _, err = fmt.Fprintf(f, "</table>\n</body></html>\n")
+ if err != nil {
+ return err
+ }
+
+ return nil
+}
diff --git a/avg-lines/avg-lines.go b/avg-lines/main.go
index 160dd7c..a953598 100644
--- a/avg-lines/avg-lines.go
+++ b/avg-lines/main.go
@@ -15,7 +15,7 @@ import (
func main() {
flag.Usage = func() {
- fmt.Fprintf(os.Stderr, "Usage: avg-lines [-html] [-nosort] [prob1] [hocr1] [prob2] [...]\n")
+ fmt.Fprintf(os.Stderr, "Usage: avg-lines [-html dir] [-nosort] [prob1] [hocr1] [prob2] [...]\n")
fmt.Fprintf(os.Stderr, "Prints a report of the average confidence for each line, sorted\n")
fmt.Fprintf(os.Stderr, "from worst to best.\n")
fmt.Fprintf(os.Stderr, "Both .hocr and .prob files can be processed.\n")
@@ -24,7 +24,7 @@ func main() {
fmt.Fprintf(os.Stderr, "option.\n\n")
flag.PrintDefaults()
}
- var usehtml = flag.Bool("html", false, "Output html page")
+ var html = flag.String("html", "", "Output in html format to the specified directory")
var nosort = flag.Bool("nosort", false, "Don't sort lines by confidence")
flag.Parse()
if flag.NArg() < 1 {
@@ -59,23 +59,11 @@ func main() {
sort.Sort(lines)
}
- if *usehtml == false {
+ if *html == "" {
for _, l := range lines {
fmt.Printf("%s %s: %.2f%%\n", l.OcrName, l.Name, l.Avgconf)
}
} else {
- fmt.Printf("<!DOCTYPE html><html><head><meta charset='UTF-8'><title></title><style>td {border: 1px solid #444}</style></head><body>\n")
- fmt.Printf("<table>\n")
- for _, l := range lines {
- fmt.Printf("<tr>\n")
- fmt.Printf("<td><h1>%.4f%%</h1></td>\n", l.Avgconf)
- fmt.Printf("<td>%s %s</td>\n", l.OcrName, l.Name)
- // TODO: think about this, what do we want to do here? if showing imgs is important,
- // will need to copy them somewhere, so works with hocr too
- //fmt.Printf("<td><img src='%s' /><br />%s</td>\n", l.Filebase + ".bin.png", l.Fulltext)
- fmt.Printf("</tr>\n")
- }
- fmt.Printf("</table>\n")
- fmt.Printf("</body></html>\n")
+ htmlout(*html, lines)
}
}