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)
 	}
 }