Move encoder

This commit is contained in:
Milan Nikolic
2018-01-27 01:47:04 +01:00
parent f65a7cf1aa
commit 1c9dfcb84c
7 changed files with 13 additions and 15 deletions

View File

@@ -38,7 +38,7 @@ func main() {
} }
} }
srv.Camera, err = camera.NewCamera(srv.Index) srv.Camera, err = camera.New(srv.Index)
if err != nil { if err != nil {
fmt.Fprintf(os.Stderr, "%s\n", err.Error()) fmt.Fprintf(os.Stderr, "%s\n", err.Error())
os.Exit(1) os.Exit(1)

View File

@@ -57,8 +57,8 @@ type Camera struct {
camera *opencv.Capture camera *opencv.Capture
} }
// NewCamera returns new Camera for given camera index. // New returns new Camera for given camera index.
func NewCamera(index int) (camera *Camera, err error) { func New(index int) (camera *Camera, err error) {
camera = &Camera{} camera = &Camera{}
camera.Index = index camera.Index = index

View File

@@ -11,7 +11,7 @@ import (
) )
func TestCamera(t *testing.T) { func TestCamera(t *testing.T) {
camera, err := NewCamera(1) camera, err := New(1)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }

View File

@@ -1,4 +1,4 @@
package camera package encoder
import ( import (
"image" "image"
@@ -8,8 +8,8 @@ import (
jpeg "github.com/antonini/golibjpegturbo" jpeg "github.com/antonini/golibjpegturbo"
) )
// NewEncoder returns a new Encoder. // New returns a new Encoder.
func NewEncoder(w io.Writer) *Encoder { func New(w io.Writer) *Encoder {
return &Encoder{w} return &Encoder{w}
} }

View File

@@ -5,6 +5,7 @@ import (
"net/http" "net/http"
"github.com/gen2brain/cam2ip/camera" "github.com/gen2brain/cam2ip/camera"
"github.com/gen2brain/cam2ip/encoder"
) )
// JPEG handler. // JPEG handler.
@@ -34,9 +35,7 @@ func (j *JPEG) ServeHTTP(w http.ResponseWriter, r *http.Request) {
return return
} }
enc := camera.NewEncoder(w) err = encoder.New(w).Encode(img)
err = enc.Encode(img)
if err != nil { if err != nil {
log.Printf("jpeg: encode: %v", err) log.Printf("jpeg: encode: %v", err)
return return

View File

@@ -10,6 +10,7 @@ import (
"time" "time"
"github.com/gen2brain/cam2ip/camera" "github.com/gen2brain/cam2ip/camera"
"github.com/gen2brain/cam2ip/encoder"
) )
// MJPEG handler. // MJPEG handler.
@@ -61,9 +62,7 @@ loop:
continue continue
} }
enc := camera.NewEncoder(partWriter) err = encoder.New(partWriter).Encode(img)
err = enc.Encode(img)
if err != nil { if err != nil {
log.Printf("mjpeg: encode: %v", err) log.Printf("mjpeg: encode: %v", err)
continue continue

View File

@@ -9,6 +9,7 @@ import (
"golang.org/x/net/websocket" "golang.org/x/net/websocket"
"github.com/gen2brain/cam2ip/camera" "github.com/gen2brain/cam2ip/camera"
"github.com/gen2brain/cam2ip/encoder"
) )
// Socket handler. // Socket handler.
@@ -33,9 +34,8 @@ func (s *Socket) write(ws *websocket.Conn) {
} }
w := new(bytes.Buffer) w := new(bytes.Buffer)
enc := camera.NewEncoder(w)
err = enc.Encode(img) err = encoder.New(w).Encode(img)
if err != nil { if err != nil {
log.Printf("socket: encode: %v", err) log.Printf("socket: encode: %v", err)
continue continue