extend interface functions [v2]

Author: LdDl

v2/blob/blob.go

@@ -8,8 +8,11 @@ import (
 
 type Blobie interface {
 	GetCenter() image.Point
+	GetCurrentRect() image.Rectange
 	GetPredictedNextPosition() image.Point
 	GetDiagonal() float64
+	GetClassID() int
+	GetClassName() string
 	Exists() bool
 	NoMatchTimes() int
 	IncrementNoMatchTimes()
@@ -18,4 +21,6 @@ type Blobie interface {
 	SetID(id uuid.UUID)
 	PredictNextPosition(n int)
 	Update(newb Blobie) error
+	SetDraw(drawOptions *DrawOptions)
+	DrawTrack(mat *gocv.Mat, optionalText string)
 }

v2/blob/kalman_blob.go

@@ -146,6 +146,10 @@ func (sb *KalmanBlobie) GetCenter() image.Point {
 	return sb.Center
 }
 
+func (sb *KalmanBlobie) GetCurrentRect() image.Rectange {
+	return sb.CurrentRect
+}
+
 func (sb *KalmanBlobie) GetDiagonal() float64 {
 	return sb.Diagonal
 }
@@ -177,3 +181,38 @@ func (sb *KalmanBlobie) IncrementNoMatchTimes() {
 func (sb *KalmanBlobie) SetExists(isExists bool) {
 	sb.isExists = isExists
 }
+
+// GetClassID Returns class identifier [KalmanBlobie]
+func (b *KalmanBlobie) GetClassID() int {
+	return b.classID
+}
+
+// GetClassName Returns class name [KalmanBlobie]
+func (b *KalmanBlobie) GetClassName() string {
+	return b.className
+}
+
+// SetDraw Sets options for drawing [KalmanBlobie]
+func (b *KalmanBlobie) SetDraw(drawOptions *DrawOptions) {
+	b.drawingOptions = drawOptions
+}
+
+// DrawTrack Draws blob's track [KalmanBlobie]
+func (b *KalmanBlobie) DrawTrack(mat *gocv.Mat, optionalText string) {
+	if b.drawingOptions == nil {
+		b.drawingOptions = NewDrawOptionsDefault()
+	}
+	gocv.Rectangle(mat, b.CurrentRect, b.drawingOptions.BBoxColor.Color, b.drawingOptions.BBoxColor.Thickness)
+	if b.isStillBeingTracked {
+		for i := range b.Track {
+			gocv.Circle(mat, b.Track[i], b.drawingOptions.CentroidColor.Radius, b.drawingOptions.CentroidColor.Color, b.drawingOptions.CentroidColor.Thickness)
+		}
+		if optionalText != "" {
+			pt := image.Pt(b.CurrentRect.Min.X, b.CurrentRect.Min.Y)
+			textSize := gocv.GetTextSize(optionalText, b.drawingOptions.TextColor.Font, b.drawingOptions.TextColor.Scale, b.drawingOptions.TextColor.Thickness)
+			textRect := image.Rectangle{Min: image.Point{X: pt.X, Y: pt.Y - textSize.Y}, Max: image.Point{X: pt.X + textSize.X, Y: pt.Y}}
+			gocv.Rectangle(mat, textRect, b.drawingOptions.BBoxColor.Color, b.drawingOptions.BBoxColor.Thickness)
+			gocv.PutText(mat, optionalText, pt, b.drawingOptions.TextColor.Font, b.drawingOptions.TextColor.Scale, b.drawingOptions.TextColor.Color, b.drawingOptions.TextColor.Thickness)
+		}
+	}
+}

v2/blob/simple_blob.go

@@ -38,6 +38,10 @@ func (sb *SimpleBlobie) GetCenter() image.Point {
 	return sb.Center
 }
 
+func (sb *SimpleBlobie) GetCurrentRect() image.Rectange {
+	return sb.CurrentRect
+}
+
 func (sb *SimpleBlobie) GetDiagonal() float64 {
 	return sb.Diagonal
 }
@@ -138,17 +142,17 @@ func (b *SimpleBlobie) SetClassName(className string) {
 	b.className = className
 }
 
-// GetClassID - Return class identifier
+// GetClassID Returns class identifier [SimpleBlobie]
 func (b *SimpleBlobie) GetClassID() int {
 	return b.classID
 }
 
-// GetClassName - Return class name
+// GetClassName Returns class name [SimpleBlobie]
 func (b *SimpleBlobie) GetClassName() string {
 	return b.className
 }
 
-// SetDraw - Set options for drawing
+// SetDraw Sets options for drawing [SimpleBlobie]
 func (b *SimpleBlobie) SetDraw(drawOptions *DrawOptions) {
 	b.drawingOptions = drawOptions
 }
@@ -209,7 +213,7 @@ func (b *SimpleBlobie) PredictNextPosition(n int) {
 	(*b).PredictedNextPosition.Y = (*b).Track[len((*b).Track)-1].Y + deltaY
 }
 
-// DrawTrack - Draw blob's track
+// DrawTrack Draws blob's track [SimpleBlobie]
 func (b *SimpleBlobie) DrawTrack(mat *gocv.Mat, optionalText string) {
 	if b.drawingOptions == nil {
 		b.drawingOptions = NewDrawOptionsDefault()