Add support for OpenCV 3

video/video.go

@@ -1,3 +1,5 @@
+// +build !cv3
+
 // Package video.
 package video
 

video/video_cv3.go

@@ -0,0 +1,60 @@
+// +build cv3
+
+// Package video.
+package video
+
+import (
+	"fmt"
+	"image"
+
+	"gocv.io/x/gocv"
+)
+
+// Video represents video.
+type Video struct {
+	video *gocv.VideoCapture
+}
+
+// New returns new Video for given path.
+func New(filename string) (video *Video, err error) {
+	video = &Video{}
+
+	video.video, err = gocv.VideoCaptureFile(filename)
+	if err != nil {
+		err = fmt.Errorf("video: can not open video %s: %s", filename, err.Error())
+	}
+
+	return
+}
+
+// Read reads next frame from video and returns image.
+func (v *Video) Read() (img image.Image, err error) {
+	mat := gocv.NewMat()
+	defer mat.Close()
+
+	ok := v.video.Read(mat)
+	if !ok {
+		err = fmt.Errorf("video: can not grab frame")
+		return
+	}
+
+	img, e := mat.ToImage()
+	if e != nil {
+		err = fmt.Errorf("video: %v", e)
+		return
+	}
+
+	return
+}
+
+// Close closes video.
+func (v *Video) Close() (err error) {
+	if v.video == nil {
+		err = fmt.Errorf("video: video is not opened")
+		return
+	}
+
+	err = v.video.Close()
+	v.video = nil
+	return
+}