diff options
Diffstat (limited to 'cleanup')
-rw-r--r-- | cleanup/testdata/pg1.png | bin | 30803 -> 0 bytes | |||
-rw-r--r-- | cleanup/testdata/pg1_integralwipesides_t0.02_w5.png | bin | 33595 -> 0 bytes | |||
-rw-r--r-- | cleanup/testdata/pg1_integralwipesides_t0.05_w25.png | bin | 33432 -> 0 bytes | |||
-rw-r--r-- | cleanup/testdata/pg1_integralwipesides_t0.05_w5.png | bin | 14546 -> 0 bytes | |||
-rw-r--r-- | cleanup/wipesides.go | 122 | ||||
-rw-r--r-- | cleanup/wipesides_test.go | 105 |
6 files changed, 0 insertions, 227 deletions
diff --git a/cleanup/testdata/pg1.png b/cleanup/testdata/pg1.png Binary files differdeleted file mode 100644 index c7c4249..0000000 --- a/cleanup/testdata/pg1.png +++ /dev/null diff --git a/cleanup/testdata/pg1_integralwipesides_t0.02_w5.png b/cleanup/testdata/pg1_integralwipesides_t0.02_w5.png Binary files differdeleted file mode 100644 index 6b4ccb2..0000000 --- a/cleanup/testdata/pg1_integralwipesides_t0.02_w5.png +++ /dev/null diff --git a/cleanup/testdata/pg1_integralwipesides_t0.05_w25.png b/cleanup/testdata/pg1_integralwipesides_t0.05_w25.png Binary files differdeleted file mode 100644 index 39dc88d..0000000 --- a/cleanup/testdata/pg1_integralwipesides_t0.05_w25.png +++ /dev/null diff --git a/cleanup/testdata/pg1_integralwipesides_t0.05_w5.png b/cleanup/testdata/pg1_integralwipesides_t0.05_w5.png Binary files differdeleted file mode 100644 index 50df855..0000000 --- a/cleanup/testdata/pg1_integralwipesides_t0.05_w5.png +++ /dev/null diff --git a/cleanup/wipesides.go b/cleanup/wipesides.go deleted file mode 100644 index ce3b374..0000000 --- a/cleanup/wipesides.go +++ /dev/null @@ -1,122 +0,0 @@ -package cleanup - -// TODO: add minimum size variable (default ~30%?) -// TODO: have the integral image specific stuff done by interface functions - -import ( - "image" - "image/color" -) - -type windowslice struct { - topleft uint64 - topright uint64 - bottomleft uint64 - bottomright uint64 -} - -func getwindowslice(i [][]uint64, x int, size int) windowslice { - maxy := len(i) - 1 - maxx := x + size - if maxx > len(i[0])-1 { - maxx = len(i[0]) - 1 - } - - return windowslice{i[0][x], i[0][maxx], i[maxy][x], i[maxy][maxx]} -} - -// checkwindow checks the window from x to see whether more than -// thresh proportion of the pixels are white, if so it returns true. -func checkwindow(integral [][]uint64, x int, size int, thresh float64) bool { - height := len(integral) - window := getwindowslice(integral, x, size) - // divide by 255 as each on pixel has the value of 255 - sum := (window.bottomright + window.topleft - window.topright - window.bottomleft) / 255 - area := size * height - proportion := float64(area)/float64(sum) - 1 - return proportion <= thresh -} - -// returns the proportion of the given window that is black pixels -func proportion(integral [][]uint64, x int, size int) float64 { - height := len(integral) - window := getwindowslice(integral, x, size) - // divide by 255 as each on pixel has the value of 255 - sum := (window.bottomright + window.topleft - window.topright - window.bottomleft) / 255 - area := size * height - return float64(area)/float64(sum) - 1 -} - -// findbestedge goes through every vertical line from x to x+w to -// find the one with the lowest proportion of black pixels. -func findbestedge(integral [][]uint64, x int, w int) int { - var bestx int - var best float64 - - if w == 1 { - return x - } - - right := x + w - for ; x < right; x++ { - prop := proportion(integral, x, 1) - if prop > best { - best = prop - bestx = x - } - } - - return bestx -} - -// Findedges finds the edges of the main content, by moving a window of wsize -// from the middle of the image to the left and right, stopping when it reaches -// a point at which there is a lower proportion of black pixels than thresh. -func Findedges(integral [][]uint64, wsize int, thresh float64) (int, int) { - maxx := len(integral[0]) - 1 - var lowedge, highedge int = 0, maxx - - for x := maxx / 2; x < maxx-wsize; x++ { - if checkwindow(integral, x, wsize, thresh) { - highedge = findbestedge(integral, x, wsize) - break - } - } - - for x := maxx / 2; x > 0; x-- { - if checkwindow(integral, x, wsize, thresh) { - lowedge = findbestedge(integral, x, wsize) - break - } - } - - return lowedge, highedge -} - -// Wipesides fills the sections of image not within the boundaries -// of lowedge and highedge with white -func Wipesides(img *image.Gray, lowedge int, highedge int) *image.Gray { - b := img.Bounds() - new := image.NewGray(b) - - // set left edge white - for x := b.Min.X; x < lowedge; x++ { - for y := b.Min.Y; y < b.Max.Y; y++ { - new.SetGray(x, y, color.Gray{255}) - } - } - // copy middle - for x := lowedge; x < highedge; x++ { - for y := b.Min.Y; y < b.Max.Y; y++ { - new.SetGray(x, y, img.GrayAt(x, y)) - } - } - // set right edge white - for x := highedge; x < b.Max.X; x++ { - for y := b.Min.Y; y < b.Max.Y; y++ { - new.SetGray(x, y, color.Gray{255}) - } - } - - return new -} diff --git a/cleanup/wipesides_test.go b/cleanup/wipesides_test.go deleted file mode 100644 index aa3e590..0000000 --- a/cleanup/wipesides_test.go +++ /dev/null @@ -1,105 +0,0 @@ -package cleanup - -// TODO: add different pages as test cases -// TODO: test non integral img version - -import ( - "flag" - "fmt" - "image" - "image/draw" - "image/png" - "os" - "testing" - - "rescribe.xyz/go.git/binarize" -) - -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 - orig string - golden string - 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}, - } - - for _, c := range cases { - t.Run(fmt.Sprintf("%s_%0.2f_%d", c.name, c.thresh, 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) - } - integral := binarize.Integralimg(orig) - lowedge, highedge := Findedges(integral, c.wsize, c.thresh) - actual = Wipesides(orig, lowedge, highedge) - 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("Processed %s differs to %s\n", c.orig, c.golden) - } - }) - } -} |