zishuo/webrtc
1
0
Fork 0
mirror of https://github.com/pion/webrtc.git synced 2025-10-05 23:26:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add support for RTP header extensions, and IVF videos now fully work!

Browse Source
This commit is contained in:
Sean DuBois
2018-06-11 01:42:41 -07:00
parent 882b87b346
commit d1608eda58
3 changed files with 47 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
examples/save-to-disk/ivf-writer.go
View File

@@ -9,8 +9,6 @@ import (
"github.com/pions/webrtc/rtp"
)
var le = binary.LittleEndian
type IVFWriter struct {
fd *os.File
time time.Time
@@ -49,16 +47,16 @@ func NewIVFWriter(fileName string) (*IVFWriter, error) {
}
header := make([]byte, 32)
copy(header[0:], []byte("DKIF")) // DKIF
le.PutUint16(header[4:], 0) // Version
le.PutUint16(header[6:], 32) // Header Size
copy(header[8:], []byte("VP80")) // FOURCC
le.PutUint16(header[12:], 640) // Version
le.PutUint16(header[14:], 480) // Header Size
le.PutUint32(header[16:], 30) // Framerate numerator
le.PutUint32(header[20:], 1) // Framerate Denominator
le.PutUint32(header[24:], 900) // Frame count
le.PutUint32(header[28:], 0) // Unused
copy(header[0:], []byte("DKIF")) // DKIF
binary.LittleEndian.PutUint16(header[4:], 0) // Version
binary.LittleEndian.PutUint16(header[6:], 32) // Header Size
copy(header[8:], []byte("VP80")) // FOURCC
binary.LittleEndian.PutUint16(header[12:], 640) // Version
binary.LittleEndian.PutUint16(header[14:], 480) // Header Size
binary.LittleEndian.PutUint32(header[16:], 30) // Framerate numerator
binary.LittleEndian.PutUint32(header[20:], 1) // Framerate Denominator
binary.LittleEndian.PutUint32(header[24:], 900) // Frame count
binary.LittleEndian.PutUint32(header[28:], 0) // Unused
panicWrite(f, header)
@@ -103,7 +101,6 @@ func (i *IVFWriter) DecodeVP8RTPPacket(packet *rtp.Packet) (*VP8RTPPacket, error
payloadIndex++
}
fmt.Print(vp8Packet)
vp8Packet.Payload = p[payloadIndex:]
return vp8Packet, nil
@@ -123,9 +120,8 @@ func (i *IVFWriter) AddPacket(packet *rtp.Packet) {
}
frameHeader := make([]byte, 12)
le.PutUint32(frameHeader[0:], uint32(len(i.currentFrame))) // Frame length
le.PutUint64(frameHeader[4:], i.count) // PTS
binary.LittleEndian.PutUint32(frameHeader[0:], uint32(len(i.currentFrame))) // Frame length
binary.LittleEndian.PutUint64(frameHeader[4:], i.count) // PTS
i.count += 1