Method WriteRTPFrame

track/h264.go:113–197

(rtpItem *LIRTP)

}

func (vt *H264) WriteRTPFrame(rtpItem *LIRTP) {
	defer func() {
		err := recover()
		if err != nil {
			vt.Error("WriteRTPFrame panic", zap.Any("err", err))
			vt.Publisher.Stop(zap.Any("err", err))
		}
	}()
	if vt.lastSeq != vt.lastSeq2+1 && vt.lastSeq2 != 0 {
		vt.lostFlag = true
		vt.Warn("lost rtp packet", zap.Uint16("lastSeq", vt.lastSeq), zap.Uint16("lastSeq2", vt.lastSeq2))
	}
	frame := &rtpItem.Value
	pts := frame.Timestamp
	rv := vt.Value
	// 有些流的 rtp 包中没有设置 marker 导致无法判断是否是最后一个包，此时通过时间戳变化判断，先 flush 之前的帧
	if rv.PTS != time.Duration(pts) {
		if rv.AUList.ByteLength > 0 {
			if !vt.dcChanged && rv.IFrame {
				vt.insertDCRtp()
			}
			vt.Flush()
			rv = vt.Value
		}
		vt.generateTimestamp(pts)
	}
	rv.RTP.Push(rtpItem)
	if naluType := frame.H264Type(); naluType < 24 {
		vt.WriteSliceBytes(frame.Payload)
	} else {
		offset := naluType.Offset()
		switch naluType {
		case codec.NALU_STAPA, codec.NALU_STAPB:
			if len(frame.Payload) <= offset {
				vt.Error("invalid nalu size", zap.Int("naluType", int(naluType)))
				return
			}
			for buffer := util.Buffer(frame.Payload[offset:]); buffer.CanRead(); {
				nextSize := int(buffer.ReadUint16())
				if buffer.Len() >= nextSize {
					vt.WriteSliceBytes(buffer.ReadN(nextSize))
				} else {
					vt.Error("invalid nalu size", zap.Int("naluType", int(naluType)))
					return
				}
			}
		case codec.NALU_FUA, codec.NALU_FUB:
			b1 := frame.Payload[1]
			if util.Bit1(b1, 0) {
				naluType = naluType.Parse(b1)
				firstByte := naluType.Or(frame.Payload[0] & 0x60)
				switch naluType {
				case codec.NALU_SPS, codec.NALU_PPS:
					vt.buf.WriteByte(firstByte)
				default:
					vt.WriteSliceByte(firstByte)
				}
			}