diff options
| author | Nick White <git@njw.name> | 2019-01-25 17:26:43 +0000 |
|---|---|---|
| committer | Nick White <git@njw.name> | 2019-01-25 17:26:43 +0000 |
| commit | 40c1e4956c0f4cd2c5f139aba7cb7bc04c57fc0f (patch) | |
| tree | bf660d77242213d2a910d1b491672187d07c1d7f | |
| parent | 97be6f42d53322118e45073f67cbdcb35441422a (diff) | |
Add html output including all images, by writing them to an html directory
| -rw-r--r-- | avg-lines/html.go | 61 | ||||
| -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) } } |