go-opencv/samples/player.go
2013-10-08 10:13:28 +08:00

77 lines
1.6 KiB
Go

// Copyright 2011 <chaishushan@gmail.com>. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package main
import (
"fmt"
"os"
"code.google.com/p/go-opencv/opencv"
)
func main() {
filename := "../data/???.avi"
if len(os.Args) == 2 {
filename = os.Args[1]
} else {
fmt.Printf("Usage: go run player.go videoname\n")
os.Exit(0)
}
cap := opencv.NewFileCapture(filename)
if cap == nil {
panic("can not open video")
}
defer cap.Release()
win := opencv.NewWindow("GoOpenCV: VideoPlayer")
defer win.Destroy()
fps := int(cap.GetProperty(opencv.CV_CAP_PROP_FPS))
frames := int(cap.GetProperty(opencv.CV_CAP_PROP_FRAME_COUNT))
stop := false
win.SetMouseCallback(func(event, x, y, flags int) {
if flags&opencv.CV_EVENT_LBUTTONDOWN != 0 {
stop = !stop
if stop {
fmt.Printf("status: stop")
} else {
fmt.Printf("status: palying")
}
}
})
win.CreateTrackbar("Seek", 1, frames, func(pos int) {
cur_pos := int(cap.GetProperty(opencv.CV_CAP_PROP_POS_FRAMES))
if pos != cur_pos {
cap.SetProperty(opencv.CV_CAP_PROP_POS_FRAMES, float64(pos))
fmt.Printf("Seek to %d(%d)\n", pos, frames)
}
})
for {
if !stop {
img := cap.QueryFrame()
if img == nil { break }
frame_pos := int(cap.GetProperty(opencv.CV_CAP_PROP_POS_FRAMES))
if frame_pos >= frames { break }
win.SetTrackbarPos("Seek", frame_pos)
win.ShowImage(img)
key := opencv.WaitKey(1000/fps)
if key == 27 { os.Exit(0) }
} else {
key := opencv.WaitKey(20)
if key == 27 { os.Exit(0) }
}
}
opencv.WaitKey(0)
}