little tests for oblique and horizontal only line

Author: LdDl

v2/blob/line_cross.go

@@ -241,6 +241,7 @@ func (b *KalmanBlobie) IsCrossedTheObliqueLine(leftX, leftY, rightX, rightY int,
 		// First segment is: P1 = (b.Track[prevFrame].X, b.Track[prevFrame].Y), Q1 = (b.Track[currFrame].X, b.Track[currFrame].Y)
 		// Second segment is: P2 = (leftX, leftY), Q2 = (rightX, rightY)
 		if isIntersects(b.Track[prevFrame].X, b.Track[prevFrame].Y, b.Track[currFrame].X, b.Track[currFrame].Y, leftX, leftY, rightX, rightY) {
+			
 			if direction {
 				if b.Track[currFrame].Y > b.Track[prevFrame].Y { // TO us
 					b.crossedLine = true

v2/blob/line_cross_test.go

@@ -21,19 +21,61 @@ func TestHorizontalLineTest(t *testing.T) {
 
 	for _, b := range allblobies.Objects {
 		if b.IsCrossedTheLine(horizontalLine[0][1], horizontalLine[0][0], horizontalLine[1][0], direction) {
-			t.Logf("Correct when direction is TO US")
+			t.Logf("[HORIZONTAL] Correct when direction is TO US")
 		} else {
-			t.Logf("Incorrect when direction is TO US")
+			t.Error("[HORIZONTAL] Incorrect when direction is TO US")
 		}
 
 		if b.IsCrossedTheLine(horizontalLine[0][1], horizontalLine[0][0], horizontalLine[1][0], !direction) {
-			t.Logf("Incorrect when direction is FROM US")
+			t.Error("[HORIZONTAL] Incorrect when direction is FROM US")
 		} else {
-			t.Logf("Correct when direction is FROM US")
+			t.Logf("[HORIZONTAL] Correct when direction is FROM US")
 		}
 	}
 }
 
 func TestObliqueLineTest(t *testing.T) {
+	shoudlCrossObliqueLine := [][]int{
+		[]int{4, 35},
+		[]int{71, 31},
+	}
+	shoudlNOTCrossObliqueLine := [][]int{
+		[]int{4, 35},
+		[]int{71, 45},
+	}
+
+	direction := true // true - TO us
+	allblobies := NewBlobiesDefaults()
+
+	simpleB_time0 := NewSimpleBlobie(image.Rect(26, 8, 44, 18), nil)
+	simpleB_time1 := NewSimpleBlobie(image.Rect(26, 20, 44, 30), nil)
+	simpleB_time2 := NewSimpleBlobie(image.Rect(26, 32, 44, 42), nil)
+
+	allblobies.MatchToExisting([]Blobie{simpleB_time0, simpleB_time1, simpleB_time2})
 
+	for _, b := range allblobies.Objects {
+		if b.IsCrossedTheObliqueLine(shoudlCrossObliqueLine[0][0], shoudlCrossObliqueLine[0][1], shoudlCrossObliqueLine[1][0], shoudlCrossObliqueLine[1][1], direction) {
+			t.Logf("[OBLIQUE] Correct when direction is TO US")
+		} else {
+			t.Error("[OBLIQUE ERR] Incorrect when direction is TO US")
+		}
+
+		if b.IsCrossedTheObliqueLine(shoudlCrossObliqueLine[0][0], shoudlCrossObliqueLine[0][1], shoudlCrossObliqueLine[1][0], shoudlCrossObliqueLine[1][1], !direction) {
+			t.Error("[OBLIQUE ERR] Incorrect when direction is FROM US")
+		} else {
+			t.Logf("[OBLIQUE] Correct when direction is FROM US")
+		}
+
+		if b.IsCrossedTheObliqueLine(shoudlNOTCrossObliqueLine[0][0], shoudlNOTCrossObliqueLine[0][1], shoudlNOTCrossObliqueLine[1][0], shoudlNOTCrossObliqueLine[1][1], direction) {
+			t.Error("[OBLIQUE NOT CROSS ERR] Incorrect when direction is TO US")
+		} else {
+			t.Logf("[OBLIQUE NOT CROSS] Correct when direction is TO US") 
+		}
+
+		if b.IsCrossedTheObliqueLine(shoudlNOTCrossObliqueLine[0][0], shoudlNOTCrossObliqueLine[0][1], shoudlNOTCrossObliqueLine[1][0], shoudlNOTCrossObliqueLine[1][1], !direction) {
+			t.Error("[OBLIQUE NOT CROSS ERR] Incorrect when direction is FROM US")
+		} else {
+			t.Logf("[OBLIQUE NOT CROSS] Correct when direction is FROM US") 
+		}
+	}
 }

v2/go.mod

@@ -4,9 +4,9 @@ go 1.16
 
 require (
 	github.com/LdDl/kalman-filter v0.1.1
-	github.com/pkg/errors v0.9.1 // indirect
+	github.com/pkg/errors v0.9.1
 	github.com/satori/go.uuid v1.2.0
 	gocv.io/x/gocv v0.23.0
-	gonum.org/v1/gonum v0.9.1 // indirect
+	gonum.org/v1/gonum v0.9.1
 	gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect
 )