Method Feed

codec/mpegts/mpegts.go:522–578

(ts io.Reader)

	return
}
func (s *MpegTsStream) Feed(ts io.Reader) (err error) {
	var reader bytes.Reader
	var lr io.LimitedReader
	lr.R = &reader
	var tsHeader MpegTsHeader
	tsData := make([]byte, TS_PACKET_SIZE)
	for {
		_, err = io.ReadFull(ts, tsData)
		if err == io.EOF {
			// 文件结尾 把最后面的数据发出去
			for _, pesPkt := range s.PESBuffer {
				if pesPkt != nil {
					s.PESChan <- pesPkt
				}
			}
			return nil
		} else if err != nil {
			return
		}
		reader.Reset(tsData)
		lr.N = TS_PACKET_SIZE
		if tsHeader, err = ReadTsHeader(&lr); err != nil {
			return
		}
		if tsHeader.Pid == PID_PAT {
			if s.PAT, err = ReadPAT(&lr); err != nil {
				return
			}
			continue
		}
		if len(s.PMT.Stream) == 0 {
			for _, v := range s.PAT.Program {
				if v.ProgramMapPID == tsHeader.Pid {
					if s.PMT, err = ReadPMT(&lr); err != nil {
						return
					}
					for _, v := range s.PMT.Stream {
						s.PESBuffer[v.ElementaryPID] = nil
					}
				}
				continue
			}
		} else if pesPkt, ok := s.PESBuffer[tsHeader.Pid]; ok {
			if tsHeader.PayloadUnitStartIndicator == 1 {
				if pesPkt != nil {
					s.PESChan <- pesPkt
				}
				pesPkt = &MpegTsPESPacket{}
				s.PESBuffer[tsHeader.Pid] = pesPkt
				if pesPkt.Header, err = ReadPESHeader(&lr); err != nil {
					return
				}
			}
			io.Copy(&pesPkt.Payload, &lr)
		}
	}
}