diff options
| -rw-r--r-- | integralimg_test.go | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/integralimg_test.go b/integralimg_test.go index 627249f..bd8a983 100644 --- a/integralimg_test.go +++ b/integralimg_test.go @@ -52,6 +52,32 @@ func TestSqFromPNG(t *testing.T) { } } +func TestSum(t *testing.T) { + f, err := os.Open("testdata/in.png") + if err != nil { + t.Fatalf("Could not open file %s: %v\n", "testdata/in.png", err) + } + defer f.Close() + img, _, err := image.Decode(f) + if err != nil { + t.Fatalf("Could not decode image: %v\n", err) + } + b := img.Bounds() + + imgplus := newGray16Plus(b) + integral := NewImage(b) + + draw.Draw(imgplus, b, img, b.Min, draw.Src) + draw.Draw(integral, b, img, b.Min, draw.Src) + + sumimg := imgplus.sum(b) + sumint := integral.Sum(b) + + if sumimg != sumint { + t.Errorf("Sum of integral image differs to regular image: regular: %d, integral: %d\n", sumimg, sumint) + } +} + func imgsequal(img1, img2 image.Image) bool { b := img1.Bounds() if !b.Eq(img2.Bounds()) { @@ -77,3 +103,24 @@ func imgsequal(img1, img2 image.Image) bool { } return true } + +type grayPlus struct { + image.Gray16 +} + +func newGray16Plus(r image.Rectangle) *grayPlus { + var g grayPlus + g.Gray16 = *image.NewGray16(r) + return &g +} + +func (i grayPlus) sum(r image.Rectangle) uint64 { + var sum uint64 + for y := r.Min.Y; y < r.Max.Y; y++ { + for x := r.Min.X; x < r.Max.X; x++ { + c := i.Gray16At(x, y).Y + sum += uint64(c) + } + } + return sum +} |