add IsCodecSupported

2016-05-26 20:14:21 +08:00 · 2016-05-26 20:14:21 +08:00 · e620324491
commit e620324491
parent d3ba527309
1 changed files with 14 additions and 0 deletions
--- a/muxer.go
+++ b/muxer.go
@ -27,7 +27,21 @@ type Muxer struct {
 	tswPMT       *TSWriter
 }

+func (self *Muxer) IsCodecSupported(codec av.CodecData) bool {
+	switch codec.Type() {
+	case av.H264, av.AAC:
+		return true
+	default:
+		return false
+	}
+}
+
 func (self *Muxer) NewStream(codec av.CodecData) (err error) {
+	if !self.IsCodecSupported(codec) {
+		err = fmt.Errorf("codec type=%x is not supported", codec.Type())
+		return
+	}
+
 	stream := &Stream{
 		muxer: self,
 		CodecData: codec,