astra/Vidio
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

SetFrameBuffer returns error

Browse source
This commit is contained in:
Alex Eidt 2022-08-02 19:54:28 -07:00
parent 45952ddbd0
commit 9e805e183b
4 changed files with 41 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
README.md
View file

@ -32,7 +32,7 @@ FPS() float64
Codec() string Codec() string
AudioCodec() string AudioCodec() string
FrameBuffer() []byte FrameBuffer() []byte
SetFrameBuffer(buffer []byte) SetFrameBuffer(buffer []byte) error
Read() bool Read() bool
Close() Close()
@ -54,7 +54,7 @@ Depth() int
FPS() float64 FPS() float64
Codec() string Codec() string
FrameBuffer() []byte FrameBuffer() []byte
SetFrameBuffer(buffer []byte) SetFrameBuffer(buffer []byte) error
Read() bool Read() bool
Close() Close()

13
camera.go
View file

@ -51,10 +51,14 @@ func (camera *Camera) FrameBuffer() []byte {
return camera.framebuffer return camera.framebuffer
} }
// Sets the framebuffer to the given byte array. Note that "buffer" must be large enough func (camera *Camera) SetFrameBuffer(buffer []byte) error {
// to store one frame of video data which is width*height*3. size := camera.width * camera.height * camera.depth
func (camera *Camera) SetFrameBuffer(buffer []byte) { if len(buffer) < size {
return fmt.Errorf("buffer size %d is smaller than frame size %d", len(buffer), size)
}
camera.framebuffer = buffer camera.framebuffer = buffer
return nil
} }
// Returns the webcam device name. // Returns the webcam device name.
@ -200,7 +204,10 @@ func initCamera(camera *Camera) error {
return err return err
} }
if camera.framebuffer == nil {
camera.framebuffer = make([]byte, camera.width*camera.height*camera.depth) camera.framebuffer = make([]byte, camera.width*camera.height*camera.depth)
}
return nil return nil
} }

13
video.go
View file

@ -72,10 +72,14 @@ func (video *Video) FrameBuffer() []byte {
return video.framebuffer return video.framebuffer
} }
// Sets the framebuffer to the given byte array. Note that "buffer" must be large enough func (video *Video) SetFrameBuffer(buffer []byte) error {
// to store one frame of video data which is width*height*3. size := video.width * video.height * video.depth
func (video *Video) SetFrameBuffer(buffer []byte) { if len(buffer) < size {
return fmt.Errorf("buffer size %d is smaller than frame size %d", len(buffer), size)
}
video.framebuffer = buffer video.framebuffer = buffer
return nil
} }
// Creates a new Video struct. // Creates a new Video struct.
@ -136,7 +140,10 @@ func initVideo(video *Video) error {
return err return err
} }
if video.framebuffer == nil {
video.framebuffer = make([]byte, video.width*video.height*video.depth) video.framebuffer = make([]byte, video.width*video.height*video.depth)
}
return nil return nil
} }

17
vidio_test.go
View file

@ -12,6 +12,23 @@ func assertEquals(actual, expected interface{}) {
} }
} }
func TestSetBuffer(t *testing.T) {
video, err := NewVideo("test/koala.mp4")
if err != nil {
panic(err)
}
defer video.Close()
size := video.width*video.height*video.depth + 101
video.SetFrameBuffer(make([]uint8, size))
video.Read()
assertEquals(len(video.framebuffer), size)
fmt.Println("Set Buffer Test Passed")
}
func TestVideoMetaData(t *testing.T) { func TestVideoMetaData(t *testing.T) {
video, err := NewVideo("test/koala.mp4") video, err := NewVideo("test/koala.mp4")
if err != nil { if err != nil {