Add functions for rotate and timestamp

This commit is contained in:
Milan Nikolic
2025-06-13 19:51:23 +02:00
parent 711ad2f102
commit 7ed9c4c442
6 changed files with 89 additions and 79 deletions

View File

@@ -32,7 +32,7 @@ ACaptureSessionOutput *captureSessionOutput;
ACaptureSessionOutputContainer *captureSessionOutputContainer; ACaptureSessionOutputContainer *captureSessionOutputContainer;
void device_on_disconnected(void *context, ACameraDevice *device) { void device_on_disconnected(void *context, ACameraDevice *device) {
LOGI("camera %s is diconnected.\n", ACameraDevice_getId(device)); LOGI("camera %s is disconnected.\n", ACameraDevice_getId(device));
} }
void device_on_error(void *context, ACameraDevice *device, int error) { void device_on_error(void *context, ACameraDevice *device, int error) {
@@ -242,6 +242,7 @@ func New(opts Options) (camera *Camera, err error) {
ret := C.openCamera(C.int(opts.Index), C.int(opts.Width), C.int(opts.Height)) ret := C.openCamera(C.int(opts.Index), C.int(opts.Width), C.int(opts.Height))
if int(ret) != 0 { if int(ret) != 0 {
err = fmt.Errorf("camera: can not open camera %d: error %d", opts.Index, int(ret)) err = fmt.Errorf("camera: can not open camera %d: error %d", opts.Index, int(ret))
return return
} }

View File

@@ -6,14 +6,9 @@ package camera
import ( import (
"fmt" "fmt"
"image" "image"
"image/color"
"image/draw"
"time"
"github.com/anthonynsimon/bild/transform"
"github.com/korandiz/v4l" "github.com/korandiz/v4l"
"github.com/korandiz/v4l/fmt/mjpeg" "github.com/korandiz/v4l/fmt/mjpeg"
"github.com/pbnjay/pixfont"
im "github.com/gen2brain/cam2ip/image" im "github.com/gen2brain/cam2ip/image"
) )
@@ -46,23 +41,27 @@ func New(opts Options) (camera *Camera, err error) {
devices := v4l.FindDevices() devices := v4l.FindDevices()
if len(devices) < opts.Index+1 { if len(devices) < opts.Index+1 {
err = fmt.Errorf("camera: no camera at index %d", opts.Index) err = fmt.Errorf("camera: no camera at index %d", opts.Index)
return return
} }
camera.camera, err = v4l.Open(devices[opts.Index].Path) camera.camera, err = v4l.Open(devices[opts.Index].Path)
if err != nil { if err != nil {
err = fmt.Errorf("camera: %s", err.Error()) err = fmt.Errorf("camera: %w", err)
return return
} }
if camera.camera == nil { if camera.camera == nil {
err = fmt.Errorf("camera: can not open camera %d", opts.Index) err = fmt.Errorf("camera: can not open camera %d", opts.Index)
return return
} }
config, err := camera.camera.GetConfig() config, err := camera.camera.GetConfig()
if err != nil { if err != nil {
err = fmt.Errorf("camera: %s", err.Error()) err = fmt.Errorf("camera: %w", err)
return return
} }
@@ -72,13 +71,15 @@ func New(opts Options) (camera *Camera, err error) {
err = camera.camera.SetConfig(config) err = camera.camera.SetConfig(config)
if err != nil { if err != nil {
err = fmt.Errorf("camera: %s", err.Error()) err = fmt.Errorf("camera: %w", err)
return return
} }
err = camera.camera.TurnOn() err = camera.camera.TurnOn()
if err != nil { if err != nil {
err = fmt.Errorf("camera: %s", err.Error()) err = fmt.Errorf("camera: %w", err)
return return
} }
@@ -87,37 +88,26 @@ func New(opts Options) (camera *Camera, err error) {
// Read reads next frame from camera and returns image. // Read reads next frame from camera and returns image.
func (c *Camera) Read() (img image.Image, err error) { func (c *Camera) Read() (img image.Image, err error) {
buffer, err := c.camera.Capture() buffer, err := c.camera.Capture()
if err != nil { if err != nil {
err = fmt.Errorf("camera: can not grab frame: %s", err.Error()) err = fmt.Errorf("camera: can not grab frame: %w", err)
return return
} }
img, err = im.NewDecoder(buffer).Decode() img, err = im.NewDecoder(buffer).Decode()
if err != nil { if err != nil {
err = fmt.Errorf("camera: %s", err.Error()) err = fmt.Errorf("camera: %w", err)
return return
} }
switch c.opts.Rotate { if c.opts.Rotate != 0 {
case 90: img = im.Rotate(img, c.opts.Rotate)
img = transform.Rotate(img, 90, &transform.RotationOptions{ResizeBounds: true})
case 180:
img = transform.Rotate(img, 180, &transform.RotationOptions{ResizeBounds: true})
case 270:
img = transform.Rotate(img, 270, &transform.RotationOptions{ResizeBounds: true})
} }
if c.opts.Timestamp { if c.opts.Timestamp {
dimg, ok := img.(draw.Image) img, err = im.Timestamp(img, "")
if !ok {
err = fmt.Errorf("camera: %T is not a drawable image type", img)
return
}
pixfont.DrawString(dimg, 10, 10, time.Now().Format("2006-01-02 15:04:05"), color.White)
img = dimg
} }
return return
@@ -126,6 +116,7 @@ func (c *Camera) Read() (img image.Image, err error) {
// GetProperty returns the specified camera property. // GetProperty returns the specified camera property.
func (c *Camera) GetProperty(id int) float64 { func (c *Camera) GetProperty(id int) float64 {
ret, _ := c.camera.GetControl(uint32(id)) ret, _ := c.camera.GetControl(uint32(id))
return float64(ret) return float64(ret)
} }
@@ -138,12 +129,13 @@ func (c *Camera) SetProperty(id int, value float64) {
func (c *Camera) Close() (err error) { func (c *Camera) Close() (err error) {
if c.camera == nil { if c.camera == nil {
err = fmt.Errorf("camera: camera is not opened") err = fmt.Errorf("camera: camera is not opened")
return return
} }
c.camera.TurnOff() c.camera.TurnOff()
c.camera.Close() c.camera.Close()
c.camera = nil c.camera = nil
return return
} }

View File

@@ -1,4 +1,4 @@
//go:build opencv //go:build opencv && !android
// Package camera. // Package camera.
package camera package camera
@@ -6,13 +6,10 @@ package camera
import ( import (
"fmt" "fmt"
"image" "image"
"image/color"
"image/draw"
"time"
"github.com/anthonynsimon/bild/transform"
"github.com/pbnjay/pixfont"
"gocv.io/x/gocv" "gocv.io/x/gocv"
im "github.com/gen2brain/cam2ip/image"
) )
// Property identifiers. // Property identifiers.
@@ -75,7 +72,7 @@ func New(opts Options) (camera *Camera, err error) {
camera.camera, err = gocv.VideoCaptureDevice(opts.Index) camera.camera, err = gocv.VideoCaptureDevice(opts.Index)
if err != nil { if err != nil {
err = fmt.Errorf("camera: can not open camera %d: %s", opts.Index, err.Error()) err = fmt.Errorf("camera: can not open camera %d: %w", opts.Index, err)
} }
camera.SetProperty(PropFrameWidth, opts.Width) camera.SetProperty(PropFrameWidth, opts.Width)
@@ -92,9 +89,9 @@ func (c *Camera) Read() (img image.Image, err error) {
return return
} }
img, e := c.frame.ToImage() img, err = c.frame.ToImage()
if e != nil { if err != nil {
err = fmt.Errorf("camera: %v", e) err = fmt.Errorf("camera: %w", err)
return return
} }
@@ -103,24 +100,12 @@ func (c *Camera) Read() (img image.Image, err error) {
return return
} }
switch c.opts.Rotate { if c.opts.Rotate != 0 {
case 90: img = im.Rotate(img, c.opts.Rotate)
img = transform.Rotate(img, 90, &transform.RotationOptions{ResizeBounds: true})
case 180:
img = transform.Rotate(img, 180, &transform.RotationOptions{ResizeBounds: true})
case 270:
img = transform.Rotate(img, 270, &transform.RotationOptions{ResizeBounds: true})
} }
if c.opts.Timestamp { if c.opts.Timestamp {
dimg, ok := img.(draw.Image) img, err = im.Timestamp(img, "")
if !ok {
err = fmt.Errorf("camera: %T is not a drawable image type", img)
return
}
pixfont.DrawString(dimg, 10, 10, time.Now().Format("2006-01-02 15:04:05"), color.White)
img = dimg
} }
return return
@@ -140,11 +125,19 @@ func (c *Camera) SetProperty(id int, value float64) {
func (c *Camera) Close() (err error) { func (c *Camera) Close() (err error) {
if c.camera == nil { if c.camera == nil {
err = fmt.Errorf("camera: camera is not opened") err = fmt.Errorf("camera: camera is not opened")
return return
} }
c.frame.Close() err = c.frame.Close()
err = c.camera.Close() if err != nil {
c.camera = nil err = fmt.Errorf("camera: %w", err)
return
}
err = c.camera.Close()
c.camera = nil
return return
} }

View File

@@ -7,15 +7,11 @@ import (
"bytes" "bytes"
"fmt" "fmt"
"image" "image"
"image/color"
"image/draw"
"runtime" "runtime"
"syscall" "syscall"
"time"
"unsafe" "unsafe"
"github.com/anthonynsimon/bild/transform" im "github.com/gen2brain/cam2ip/image"
"github.com/pbnjay/pixfont"
) )
func init() { func init() {
@@ -105,7 +101,7 @@ func New(opts Options) (camera *Camera, err error) {
// Read reads next frame from camera and returns image. // Read reads next frame from camera and returns image.
func (c *Camera) Read() (img image.Image, err error) { func (c *Camera) Read() (img image.Image, err error) {
ret := sendMessage(c.camera, wmCapGrabFrameNoStop, 0, 0) ret := sendMessage(c.camera, wmCapGrabFrameNoStop, 0, 0)
if bool(int(ret) == 0) { if int(ret) == 0 {
err = fmt.Errorf("camera: can not grab frame") err = fmt.Errorf("camera: can not grab frame")
return return
} }
@@ -123,7 +119,7 @@ func (c *Camera) Read() (img image.Image, err error) {
for y := height - 1; y >= 0; y-- { for y := height - 1; y >= 0; y-- {
_, err = r.Read(b) _, err = r.Read(b)
if err != nil { if err != nil {
err = fmt.Errorf("camera: can not retrieve frame: %v", err) err = fmt.Errorf("camera: can not retrieve frame: %w", err)
return return
} }
@@ -139,24 +135,12 @@ func (c *Camera) Read() (img image.Image, err error) {
img = c.frame img = c.frame
switch c.opts.Rotate { if c.opts.Rotate != 0 {
case 90: img = im.Rotate(img, c.opts.Rotate)
img = transform.Rotate(img, 90, &transform.RotationOptions{ResizeBounds: true})
case 180:
img = transform.Rotate(img, 180, &transform.RotationOptions{ResizeBounds: true})
case 270:
img = transform.Rotate(img, 270, &transform.RotationOptions{ResizeBounds: true})
} }
if c.opts.Timestamp { if c.opts.Timestamp {
dimg, ok := img.(draw.Image) img, err = im.Timestamp(img, "")
if !ok {
err = fmt.Errorf("camera: %T is not a drawable image type", img)
return
}
pixfont.DrawString(dimg, 10, 10, time.Now().Format("2006-01-02 15:04:05"), color.White)
img = dimg
} }
return return

40
image/image.go Normal file
View File

@@ -0,0 +1,40 @@
package image
import (
"fmt"
"image"
"image/color"
"image/draw"
"time"
"github.com/anthonynsimon/bild/transform"
"github.com/pbnjay/pixfont"
)
func Rotate(img image.Image, angle int) image.Image {
switch angle {
case 90:
img = transform.Rotate(img, 90, &transform.RotationOptions{ResizeBounds: true})
case 180:
img = transform.Rotate(img, 180, &transform.RotationOptions{ResizeBounds: true})
case 270:
img = transform.Rotate(img, 270, &transform.RotationOptions{ResizeBounds: true})
}
return img
}
func Timestamp(img image.Image, format string) (image.Image, error) {
if format == "" {
format = "2006-01-02 15:04:05"
}
dimg, ok := img.(draw.Image)
if !ok {
return img, fmt.Errorf("camera: %T is not a drawable image type", img)
}
pixfont.DrawString(dimg, 10, 10, time.Now().Format(format), color.White)
return dimg, nil
}