diff --git a/flv.go b/flv.go index 3637306..05c24fd 100644 --- a/flv.go +++ b/flv.go @@ -4,6 +4,7 @@ import ( "time" "fmt" "github.com/nareix/av" + "github.com/nareix/av/avutil" "github.com/nareix/codec/h264parser" "github.com/nareix/codec/aacparser" "github.com/nareix/pio" @@ -109,6 +110,10 @@ func (self *Muxer) WritePacket(pkt av.Packet) (err error) { return } +func (self *Muxer) WriteTrailer() (err error) { + return +} + type flvStream struct { av.CodecData lastts int32 @@ -269,3 +274,16 @@ func (self *Demuxer) ReadPacket() (pkt av.Packet, err error) { return } +func Handler(h *avutil.RegisterHandler) { + h.Probe = func(b []byte) bool { + return b[0] == 'F' && b[1] == 'L' && b[2] == 'V' + } + h.Ext = ".flv" + h.ReaderDemuxer = func(r io.Reader) av.Demuxer { + return NewDemuxer(r) + } + h.WriterMuxer = func(w io.Writer) av.Muxer { + return NewMuxer(w) + } +} + diff --git a/flvio/flvio.go b/flvio/flvio.go index 60a06e8..5152942 100644 --- a/flvio/flvio.go +++ b/flvio/flvio.go @@ -456,3 +456,4 @@ func WriteFileHeader(w *pio.Writer, flags uint8) (err error) { return } +