summaryrefslogtreecommitdiff
path: root/cleanup/wipesides.go
diff options
context:
space:
mode:
authorNick White <git@njw.name>2019-05-13 17:28:38 +0100
committerNick White <git@njw.name>2019-05-13 17:28:38 +0100
commit6b7cc0cd671db9c8ff89eea1686099c48dba412a (patch)
tree3a184a9c893be8e097e17a1acc1cd9d80e1474be /cleanup/wipesides.go
parentb2f127e33f92d778c9a4700ea5c97ba6fb20fe4a (diff)
Make cleanup a basic library
Diffstat (limited to 'cleanup/wipesides.go')
-rw-r--r--cleanup/wipesides.go122
1 files changed, 122 insertions, 0 deletions
diff --git a/cleanup/wipesides.go b/cleanup/wipesides.go
new file mode 100644
index 0000000..ce3b374
--- /dev/null
+++ b/cleanup/wipesides.go
@@ -0,0 +1,122 @@
+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
+}