112 lines
2.0 KiB
Go
112 lines
2.0 KiB
Go
|
|
package atom
|
|
|
|
import (
|
|
"io"
|
|
"bytes"
|
|
"log"
|
|
"encoding/hex"
|
|
)
|
|
|
|
type VideoSampleDesc struct {
|
|
VideoSampleDescHeader
|
|
AVCDecoderConf []byte
|
|
}
|
|
|
|
func ReadVideoSampleDesc(r *io.LimitedReader) (res *VideoSampleDesc, err error) {
|
|
self := &VideoSampleDesc{}
|
|
|
|
if self.VideoSampleDescHeader, err = ReadVideoSampleDescHeader(r); err != nil {
|
|
return
|
|
}
|
|
|
|
for r.N > 0 {
|
|
var cc4 string
|
|
var ar *io.LimitedReader
|
|
if ar, cc4, err = ReadAtomHeader(r, ""); err != nil {
|
|
return
|
|
}
|
|
|
|
if false {
|
|
log.Println("VideoSampleDesc:", cc4, ar.N)
|
|
//log.Println("VideoSampleDesc:", "avcC", len(self.AVCDecoderConf))
|
|
}
|
|
|
|
switch cc4 {
|
|
case "avcC": {
|
|
if self.AVCDecoderConf, err = ReadBytes(ar, int(ar.N)); err != nil {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
if _, err = ReadDummy(ar, int(ar.N)); err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
res = self
|
|
return
|
|
}
|
|
|
|
type SampleDescEntry struct {
|
|
Format string
|
|
DataRefIndex int
|
|
Data []byte
|
|
|
|
Video *VideoSampleDesc
|
|
}
|
|
|
|
func ReadSampleDescEntry(r *io.LimitedReader) (res *SampleDescEntry, err error) {
|
|
self := &SampleDescEntry{}
|
|
if r, self.Format, err = ReadAtomHeader(r, ""); err != nil {
|
|
return
|
|
}
|
|
if _, err = ReadDummy(r, 6); err != nil {
|
|
return
|
|
}
|
|
if self.DataRefIndex, err = ReadInt(r, 2); err != nil {
|
|
return
|
|
}
|
|
|
|
if self.Data, err = ReadBytes(r, int(r.N)); err != nil {
|
|
return
|
|
}
|
|
|
|
if self.Format == "avc1" {
|
|
br := bytes.NewReader(self.Data)
|
|
var err error
|
|
self.Video, err = ReadVideoSampleDesc(&io.LimitedReader{R: br, N: int64(len(self.Data))})
|
|
if false {
|
|
log.Println("ReadSampleDescEntry:", hex.Dump(self.Data))
|
|
log.Println("ReadSampleDescEntry:", err)
|
|
}
|
|
} else if self.Format == "mp4a" {
|
|
}
|
|
|
|
res = self
|
|
return
|
|
}
|
|
|
|
func WriteSampleDescEntry(w io.WriteSeeker, self *SampleDescEntry) (err error) {
|
|
var aw *Writer
|
|
if aw, err = WriteAtomHeader(w, self.Format); err != nil {
|
|
return
|
|
}
|
|
w = aw
|
|
if err = WriteDummy(w, 6); err != nil {
|
|
return
|
|
}
|
|
if err = WriteInt(w, self.DataRefIndex, 2); err != nil {
|
|
return
|
|
}
|
|
if err = WriteBytes(w, self.Data); err != nil {
|
|
return
|
|
}
|
|
if err = aw.Close(); err != nil {
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|