Initial commit

camera/camera.go

@@ -0,0 +1,104 @@
+// Package camera.
+package camera
+
+import (
+	"fmt"
+	"image"
+
+	"github.com/lazywei/go-opencv/opencv"
+)
+
+// Property identifiers.
+const (
+	PropPosMsec = iota
+	PropPosFrames
+	PropPosAviRatio
+	PropFrameWidth
+	PropFrameHeight
+	PropFps
+	PropFourcc
+	PropFrameCount
+	PropFormat
+	PropMode
+	PropBrightness
+	PropContrast
+	PropSaturation
+	PropHue
+	PropGain
+	PropExposure
+	PropConvertRgb
+	PropWhiteBalanceU
+	PropRectification
+	PropMonocrome
+	PropSharpness
+	PropAutoExposure
+	PropGamma
+	PropTemperature
+	PropTrigger
+	PropTriggerDelay
+	PropWhiteBalanceV
+	PropZoom
+	PropFocus
+	PropGuid
+	PropIsoSpeed
+	PropMaxDc1394
+	PropBacklight
+	PropPan
+	PropTilt
+	PropRoll
+	PropIris
+	PropSettings
+	PropBuffersize
+)
+
+// Camera represents camera.
+type Camera struct {
+	Index  int
+	camera *opencv.Capture
+}
+
+// NewCamera returns new Camera for given camera index.
+func NewCamera(index int) (camera *Camera, err error) {
+	camera = &Camera{}
+	camera.Index = index
+
+	camera.camera = opencv.NewCameraCapture(index)
+	if camera.camera == nil {
+		err = fmt.Errorf("camera: can not open camera %d", index)
+	}
+
+	return
+}
+
+// Read reads next frame from camera and returns image.
+func (c *Camera) Read() (img image.Image, err error) {
+	if c.camera.GrabFrame() {
+		frame := c.camera.RetrieveFrame(1)
+		img = frame.ToImage()
+	} else {
+		err = fmt.Errorf("camera: can not grab frame")
+	}
+
+	return
+}
+
+// GetProperty returns the specified camera property.
+func (c *Camera) GetProperty(id int) float64 {
+	return c.camera.GetProperty(id)
+}
+
+// SetProperty sets a camera property.
+func (c *Camera) SetProperty(id int, value float64) int {
+	return c.camera.SetProperty(id, value)
+}
+
+// Close closes camera.
+func (c *Camera) Close() (err error) {
+	if c.camera == nil {
+		err = fmt.Errorf("camera: camera is not opened")
+	}
+
+	c.camera.Release()
+	c.camera = nil
+	return
+}

camera/camera_test.go

@@ -0,0 +1,74 @@
+package camera
+
+import (
+	"fmt"
+	"image/jpeg"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+)
+
+func TestCamera(t *testing.T) {
+	camera, err := NewCamera(1)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	defer camera.Close()
+
+	tmpdir, err := ioutil.TempDir(os.TempDir(), "cam2ip")
+	if err != nil {
+		t.Error(err)
+	}
+
+	defer os.RemoveAll(tmpdir)
+
+	var width, height float64 = 640, 480
+	camera.SetProperty(PropFrameWidth, width)
+	camera.SetProperty(PropFrameHeight, height)
+
+	if camera.GetProperty(PropFrameWidth) != width {
+		t.Error("FrameWidth not correct")
+	}
+
+	if camera.GetProperty(PropFrameHeight) != height {
+		t.Error("FrameHeight not correct")
+	}
+
+	var i int
+	var n int = 10
+
+	timeout := time.After(time.Duration(n) * time.Second)
+
+	for {
+		select {
+		case <-timeout:
+			//fmt.Printf("Fps: %d\n", i/n)
+			return
+		default:
+			i += 1
+
+			img, err := camera.Read()
+			if err != nil {
+				t.Error(err)
+			}
+
+			file, err := os.Create(filepath.Join(tmpdir, fmt.Sprintf("%03d.jpg", i)))
+			if err != nil {
+				t.Error(err)
+			}
+
+			err = jpeg.Encode(file, img, &jpeg.Options{Quality: 75})
+			if err != nil {
+				t.Error(err)
+			}
+
+			err = file.Close()
+			if err != nil {
+				t.Error(err)
+			}
+		}
+	}
+}

camera/encode.go

@@ -0,0 +1,29 @@
+package camera
+
+import (
+	"image"
+	//"image/jpeg"
+	"io"
+
+	jpeg "github.com/kjk/golibjpegturbo"
+)
+
+// NewEncoder returns a new Encoder.
+func NewEncoder(w io.Writer) *Encoder {
+	return &Encoder{w}
+}
+
+// Encoder struct.
+type Encoder struct {
+	w io.Writer
+}
+
+// Encode encodes image to JPEG.
+func (e Encoder) Encode(img image.Image) error {
+	err := jpeg.Encode(e.w, img, &jpeg.Options{Quality: 75})
+	if err != nil {
+		return err
+	}
+
+	return nil
+}