summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--integralimg_test.go47
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
+}