test hls ll

format/nvr/muxer.go

@@ -0,0 +1,59 @@
+package nvr
+
+import (
+	"log"
+	"os"
+	"time"
+
+	"github.com/deepch/vdk/av"
+)
+
+type Muxer struct {
+	name      string
+	patch     string
+	started   bool
+	file      *os.File
+	codec     []av.CodecData
+	buffer    []*av.Packet
+	bufferDur time.Duration
+	seqDur    time.Duration
+}
+
+//NewMuxer func
+func NewMuxer(codec []av.CodecData, name, patch string, seqDur time.Duration) *Muxer {
+	return &Muxer{
+		codec:  codec,
+		name:   name,
+		patch:  patch,
+		seqDur: seqDur,
+	}
+}
+
+//WritePacket func
+func (obj *Muxer) CodecUpdate(val []av.CodecData) {
+	obj.codec = val
+}
+
+//WritePacket func
+func (obj *Muxer) WritePacket(pkt *av.Packet) (err error) {
+	if !obj.started && pkt.IsKeyFrame {
+		obj.started = true
+	}
+	if obj.started {
+		if pkt.IsKeyFrame && obj.bufferDur >= obj.seqDur {
+			log.Println("write to drive", len(obj.buffer), obj.bufferDur)
+			obj.buffer = nil
+			obj.bufferDur = 0
+		}
+		obj.buffer = append(obj.buffer, pkt)
+		if pkt.Idx == 0 {
+			obj.bufferDur += pkt.Duration
+		}
+	}
+	return nil
+}
+
+//Close func
+func (obj *Muxer) Close() {
+	return
+}