From 02a6c66eb77a5b455bcf2d0547d2383074eb7e41 Mon Sep 17 00:00:00 2001 From: Nick White Date: Mon, 13 May 2019 19:23:03 +0100 Subject: 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. --- preproc/wipesides_test.go | 52 +++-------------------------------------------- 1 file changed, 3 insertions(+), 49 deletions(-) (limited to 'preproc/wipesides_test.go') diff --git a/preproc/wipesides_test.go b/preproc/wipesides_test.go index b0ada4e..f66f39b 100644 --- a/preproc/wipesides_test.go +++ b/preproc/wipesides_test.go @@ -4,59 +4,13 @@ package preproc // TODO: test non integral img version 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 TestWipeSides(t *testing.T) { cases := []struct { name string @@ -65,9 +19,9 @@ func TestWipeSides(t *testing.T) { thresh float64 wsize int }{ - {"integralwipesides", "testdata/pg1.png", "testdata/pg1_integralwipesides_t0.02_w5.png", 0.02, 5}, - {"integralwipesides", "testdata/pg1.png", "testdata/pg1_integralwipesides_t0.05_w5.png", 0.05, 5}, - {"integralwipesides", "testdata/pg1.png", "testdata/pg1_integralwipesides_t0.05_w25.png", 0.05, 25}, + {"integralwipesides", "testdata/pg2.png", "testdata/pg2_integralwipesides_t0.02_w5.png", 0.02, 5}, + {"integralwipesides", "testdata/pg2.png", "testdata/pg2_integralwipesides_t0.05_w5.png", 0.05, 5}, + {"integralwipesides", "testdata/pg2.png", "testdata/pg2_integralwipesides_t0.05_w25.png", 0.05, 25}, } for _, c := range cases { -- cgit v1.2.1-24-ge1ad