mirror of https://bitbucket.org/ausocean/av.git
113 lines
2.2 KiB
Go
113 lines
2.2 KiB
Go
/*
|
|
NAME
|
|
lex.go
|
|
|
|
DESCRIPTION
|
|
lex.go provides a lexer to extract separate JPEG images from a JPEG stream.
|
|
This could either be a series of descrete JPEG images, or an MJPEG stream.
|
|
|
|
AUTHOR
|
|
Dan Kortschak <dan@ausocean.org>
|
|
Saxon Nelson-Milton <saxon@ausocean.org>
|
|
|
|
LICENSE
|
|
lex.go is Copyright (C) 2017 the Australian Ocean Lab (AusOcean)
|
|
|
|
It is free software: you can redistribute it and/or modify them
|
|
under the terms of the GNU General Public License as published by the
|
|
Free Software Foundation, either version 3 of the License, or (at your
|
|
option) any later version.
|
|
|
|
It is distributed in the hope that it will be useful, but WITHOUT
|
|
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with revid in gpl.txt. If not, see http://www.gnu.org/licenses.
|
|
*/
|
|
|
|
package mjpeg
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"time"
|
|
|
|
"bitbucket.org/ausocean/utils/logger"
|
|
)
|
|
|
|
var Log logger.LoggerIF
|
|
|
|
var noDelay = make(chan time.Time)
|
|
|
|
func init() {
|
|
close(noDelay)
|
|
}
|
|
|
|
// Lex parses JPEG frames read from src into separate writes to dst with
|
|
// successive writes being performed not earlier than the specified delay.
|
|
func Lex(dst io.Writer, src io.Reader, delay time.Duration) error {
|
|
var tick <-chan time.Time
|
|
if delay == 0 {
|
|
tick = noDelay
|
|
} else {
|
|
ticker := time.NewTicker(delay)
|
|
defer ticker.Stop()
|
|
tick = ticker.C
|
|
}
|
|
|
|
r := bufio.NewReader(src)
|
|
for {
|
|
buf := make([]byte, 2, 4<<10)
|
|
n, err := r.Read(buf)
|
|
if n < 2 {
|
|
return io.ErrUnexpectedEOF
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !bytes.Equal(buf, []byte{0xff, 0xd8}) {
|
|
return fmt.Errorf("parser: not JPEG frame start: %#v", buf)
|
|
}
|
|
|
|
nImg := 1
|
|
|
|
var last byte
|
|
for {
|
|
b, err := r.ReadByte()
|
|
if err != nil {
|
|
if err == io.EOF {
|
|
return io.ErrUnexpectedEOF
|
|
}
|
|
return err
|
|
}
|
|
|
|
buf = append(buf, b)
|
|
|
|
if last == 0xff && b == 0xd8 {
|
|
nImg++
|
|
}
|
|
|
|
if last == 0xff && b == 0xd9 {
|
|
nImg--
|
|
}
|
|
|
|
if nImg == 0 {
|
|
<-tick
|
|
Log.Debug("writing buf","len(buf)",len(buf))
|
|
_, err = dst.Write(buf)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
break
|
|
}
|
|
|
|
last = b
|
|
}
|
|
}
|
|
}
|