diff options
author | Nick White <git@njw.name> | 2019-05-13 19:23:03 +0100 |
---|---|---|
committer | Nick White <git@njw.name> | 2019-05-13 19:23:03 +0100 |
commit | 02a6c66eb77a5b455bcf2d0547d2383074eb7e41 (patch) | |
tree | 00f2bde1dfa7a1b23e55478309f26ba1b54fdab9 /binarize/sauvola_test.go | |
parent | d94fb9f7e74aaad335a587030ed2b4ce44c24cbf (diff) |
Reorganise image manipulation to separate integral image parts
Also unify everything else under preproc/
Note that the UsefulImg interface should be used by the main
functions, to simplify things, but this hasn't been done yet.
Diffstat (limited to 'binarize/sauvola_test.go')
-rw-r--r-- | binarize/sauvola_test.go | 108 |
1 files changed, 0 insertions, 108 deletions
diff --git a/binarize/sauvola_test.go b/binarize/sauvola_test.go deleted file mode 100644 index 5faeb61..0000000 --- a/binarize/sauvola_test.go +++ /dev/null @@ -1,108 +0,0 @@ -package binarize - -import ( - "flag" - "fmt" - "image" - "image/draw" - "image/png" - "os" - "testing" -) - -var update = flag.Bool("update", false, "update golden files") - -func decode(s string) (*image.Gray, error) { - f, err := os.Open(s) - defer f.Close() - if err != nil { - return nil, err - } - img, err := png.Decode(f) - if err != nil { - return nil, err - } - b := img.Bounds() - gray := image.NewGray(image.Rect(0, 0, b.Dx(), b.Dy())) - draw.Draw(gray, b, img, b.Min, draw.Src) - return gray, nil -} - -func imgsequal(img1 *image.Gray, img2 *image.Gray) bool { - b := img1.Bounds() - if ! b.Eq(img2.Bounds()) { - return false - } - for y := b.Min.Y; y < b.Max.Y; y++ { - for x := b.Min.X; x < b.Max.X; x++ { - r0, g0, b0, a0 := img1.At(x, y).RGBA() - r1, g1, b1, a1 := img2.At(x, y).RGBA() - if r0 != r1 { - return false - } - if g0 != g1 { - return false - } - if b0 != b1 { - return false - } - if a0 != a1 { - return false - } - } - } - return true -} - -func TestBinarization(t *testing.T) { - cases := []struct { - name string - orig string - golden string - ksize float64 - wsize int - }{ - {"integralsauvola", "testdata/pg1.png", "testdata/pg1_integralsauvola_k0.5_w41.png", 0.5, 41}, - {"integralsauvola", "testdata/pg1.png", "testdata/pg1_integralsauvola_k0.5_w19.png", 0.5, 19}, - {"integralsauvola", "testdata/pg1.png", "testdata/pg1_integralsauvola_k0.3_w19.png", 0.3, 19}, - {"sauvola", "testdata/pg1.png", "testdata/pg1_sauvola_k0.5_w41.png", 0.5, 41}, - {"sauvola", "testdata/pg1.png", "testdata/pg1_sauvola_k0.5_w19.png", 0.5, 19}, - {"sauvola", "testdata/pg1.png", "testdata/pg1_sauvola_k0.3_w19.png", 0.3, 19}, - } - - for _, c := range cases { - t.Run(fmt.Sprintf("%s_%0.1f_%d", c.name, c.ksize, c.wsize), func(t *testing.T) { - var actual *image.Gray - orig, err := decode(c.orig) - if err != nil { - t.Fatalf("Could not open file %s: %v\n", c.orig, err) - } - switch c.name { - case "integralsauvola": - actual = IntegralSauvola(orig, c.ksize, c.wsize) - case "sauvola": - actual = Sauvola(orig, c.ksize, c.wsize) - default: - t.Fatalf("No method %s\n", c.name) - } - if *update { - f, err := os.Create(c.golden) - defer f.Close() - if err != nil { - t.Fatalf("Could not open file %s to update: %v\n", c.golden, err) - } - err = png.Encode(f, actual) - if err != nil { - t.Fatalf("Could not encode update of %s: %v\n", c.golden, err) - } - } - golden, err := decode(c.golden) - if err != nil { - t.Fatalf("Could not open file %s: %v\n", c.golden, err) - } - if ! imgsequal(golden, actual) { - t.Errorf("Binarized %s differs to %s\n", c.orig, c.golden) - } - }) - } -} |