add demuxer.go
This commit is contained in:
nareix
parent
b69852c281
commit
e060a3e79d
212
demuxer.go
Normal file
212
demuxer.go
Normal file
@ -0,0 +1,212 @@
|
|||||||
|
|
||||||
|
package mp4
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/nareix/mp4/atom"
|
||||||
|
_ "os"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
_ "log"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Demuxer struct {
|
||||||
|
R io.ReadSeeker
|
||||||
|
Tracks []*Track
|
||||||
|
TrackH264 *Track
|
||||||
|
TrackAAC *Track
|
||||||
|
MovieAtom *atom.Movie
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Demuxer) ReadHeader() (err error) {
|
||||||
|
var N int64
|
||||||
|
var moov *atom.Movie
|
||||||
|
|
||||||
|
if N, err = self.R.Seek(0, 2); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if _, err = self.R.Seek(0, 0); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
lr := &io.LimitedReader{R: self.R, N: N}
|
||||||
|
for lr.N > 0 {
|
||||||
|
var ar *io.LimitedReader
|
||||||
|
|
||||||
|
var cc4 string
|
||||||
|
if ar, cc4, err = atom.ReadAtomHeader(lr, ""); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if cc4 == "moov" {
|
||||||
|
if moov, err = atom.ReadMovie(ar); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err = atom.ReadDummy(lr, int(ar.N)); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if moov == nil {
|
||||||
|
err = fmt.Errorf("'moov' atom not found")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
self.MovieAtom = moov
|
||||||
|
|
||||||
|
self.Tracks = []*Track{}
|
||||||
|
for _, atrack := range(moov.Tracks) {
|
||||||
|
track := &Track{
|
||||||
|
TrackAtom: atrack,
|
||||||
|
r: self.R,
|
||||||
|
}
|
||||||
|
if atrack.Media != nil && atrack.Media.Info != nil && atrack.Media.Info.Sample != nil {
|
||||||
|
track.sample = atrack.Media.Info.Sample
|
||||||
|
} else {
|
||||||
|
err = fmt.Errorf("sample table not found")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if record := atom.GetAVCDecoderConfRecordByTrack(atrack); record != nil {
|
||||||
|
track.Type = H264
|
||||||
|
self.TrackH264 = track
|
||||||
|
if len(record.PPS) > 0 {
|
||||||
|
track.PPS = record.PPS[0]
|
||||||
|
}
|
||||||
|
if len(record.SPS) > 0 {
|
||||||
|
track.SPS = record.SPS[0]
|
||||||
|
}
|
||||||
|
self.Tracks = append(self.Tracks, track)
|
||||||
|
} else if mp4a := atom.GetMp4aDescByTrack(atrack); mp4a != nil {
|
||||||
|
self.TrackAAC = track
|
||||||
|
track.Type = AAC
|
||||||
|
self.Tracks = append(self.Tracks, track)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Track) setSampleIndex(index int) (err error) {
|
||||||
|
found := false
|
||||||
|
start := 0
|
||||||
|
self.chunkGroupIndex = 0
|
||||||
|
|
||||||
|
for self.chunkIndex = range(self.sample.ChunkOffset.Entries) {
|
||||||
|
n := self.sample.SampleToChunk.Entries[self.chunkGroupIndex].SamplesPerChunk
|
||||||
|
if index >= start && index < start+n {
|
||||||
|
found = true
|
||||||
|
self.sampleIndexInChunk = index-start
|
||||||
|
break
|
||||||
|
}
|
||||||
|
start += n
|
||||||
|
if self.chunkGroupIndex+1 < len(self.sample.SampleToChunk.Entries) &&
|
||||||
|
self.chunkIndex+1 == self.sample.SampleToChunk.Entries[self.chunkGroupIndex+1].FirstChunk {
|
||||||
|
self.chunkGroupIndex++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !found {
|
||||||
|
err = io.EOF
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
self.sampleIndex = index
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Track) incSampleIndex() {
|
||||||
|
self.sampleIndexInChunk++
|
||||||
|
if self.sampleIndexInChunk == self.sample.SampleToChunk.Entries[self.chunkGroupIndex].SamplesPerChunk {
|
||||||
|
self.chunkIndex++
|
||||||
|
self.sampleIndexInChunk = 0
|
||||||
|
}
|
||||||
|
if self.chunkGroupIndex+1 < len(self.sample.SampleToChunk.Entries) &&
|
||||||
|
self.chunkIndex+1 == self.sample.SampleToChunk.Entries[self.chunkGroupIndex+1].FirstChunk {
|
||||||
|
self.chunkGroupIndex++
|
||||||
|
}
|
||||||
|
self.sampleIndex++
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Track) SampleCount() int {
|
||||||
|
if self.sample.SampleSize.SampleSize == 0 {
|
||||||
|
chunkGroupIndex := 0
|
||||||
|
count := 0
|
||||||
|
for chunkIndex := range(self.sample.ChunkOffset.Entries) {
|
||||||
|
n := self.sample.SampleToChunk.Entries[chunkGroupIndex].SamplesPerChunk
|
||||||
|
count += n
|
||||||
|
if chunkGroupIndex+1 < len(self.sample.SampleToChunk.Entries) &&
|
||||||
|
chunkIndex+1 == self.sample.SampleToChunk.Entries[chunkGroupIndex+1].FirstChunk {
|
||||||
|
chunkGroupIndex++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return count
|
||||||
|
} else {
|
||||||
|
return len(self.sample.SampleSize.Entries)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Track) ReadSample() (pts int64, dts int64, isKeyFrame bool, data []byte, err error) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Track) ReadSampleAtIndex(index int) (pts int64, dts int64, isKeyFrame bool, data []byte, err error) {
|
||||||
|
if self.sampleIndex+1 == index {
|
||||||
|
self.incSampleIndex()
|
||||||
|
} else if self.sampleIndex != index {
|
||||||
|
if err = self.setSampleIndex(index); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if self.chunkIndex > len(self.sample.ChunkOffset.Entries) {
|
||||||
|
err = io.EOF
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if self.chunkGroupIndex >= len(self.sample.SampleToChunk.Entries) {
|
||||||
|
err = io.EOF
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
chunkOffset := self.sample.ChunkOffset.Entries[self.chunkIndex]
|
||||||
|
sampleOffset := 0
|
||||||
|
sampleSize := 0
|
||||||
|
|
||||||
|
if self.sample.SampleSize.SampleSize != 0 {
|
||||||
|
sampleOffset = chunkOffset + self.sampleIndexInChunk*self.sample.SampleSize.SampleSize
|
||||||
|
} else {
|
||||||
|
sampleOffset = chunkOffset
|
||||||
|
for i := self.sampleIndex-self.sampleIndexInChunk; i < self.sampleIndex; i++ {
|
||||||
|
sampleOffset += self.sample.SampleSize.Entries[i]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err = self.r.Seek(int64(sampleOffset), 0); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
data = make([]byte, sampleSize)
|
||||||
|
if _, err = self.r.Read(data); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Track) Duration() float32 {
|
||||||
|
total := int64(0)
|
||||||
|
for _, entry := range(self.sample.TimeToSample.Entries) {
|
||||||
|
total += int64(entry.Duration*entry.Count)
|
||||||
|
}
|
||||||
|
return float32(total)/float32(self.TrackAtom.Media.Header.TimeScale)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Track) TimeToSampleIndex(second float32) int {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Track) TimeStampToTime(ts int64) float32 {
|
||||||
|
return 0.0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self *Track) WriteSample(pts int64, dts int64, data []byte) (err error) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
26
example/example.go
Normal file
26
example/example.go
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/nareix/mp4"
|
||||||
|
"os"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
func DemuxExample() {
|
||||||
|
file, _ := os.Open("test.mp4")
|
||||||
|
|
||||||
|
demuxer := &mp4.Demuxer{
|
||||||
|
R: file,
|
||||||
|
}
|
||||||
|
demuxer.ReadHeader()
|
||||||
|
|
||||||
|
fmt.Println("Duration: ", demuxer.TrackH264.Duration())
|
||||||
|
count := demuxer.TrackH264.SampleCount()
|
||||||
|
fmt.Println("SampleCount: ", count)
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
DemuxExample()
|
||||||
|
}
|
||||||
|
|
||||||
@ -1,30 +0,0 @@
|
|||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
mp4 "./.."
|
|
||||||
"flag"
|
|
||||||
"log"
|
|
||||||
)
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
testconv := flag.Bool("testconv", false, "")
|
|
||||||
probe := flag.Bool("probe", false, "")
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
if *testconv {
|
|
||||||
if err := mp4.TestConvert(flag.Arg(0)); err != nil {
|
|
||||||
log.Println(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if *probe {
|
|
||||||
if err := mp4.ProbeFile(flag.Arg(0)); err != nil {
|
|
||||||
log.Println(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
200
test.go
200
test.go
@ -1,200 +0,0 @@
|
|||||||
|
|
||||||
package mp4
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/nareix/mp4/atom"
|
|
||||||
"os"
|
|
||||||
"io"
|
|
||||||
"fmt"
|
|
||||||
"log"
|
|
||||||
"encoding/hex"
|
|
||||||
)
|
|
||||||
|
|
||||||
/*
|
|
||||||
func TestRewrite(filename string) (err error) {
|
|
||||||
var infile *os.File
|
|
||||||
if infile, err = os.Open(filename); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
var outfile *os.File
|
|
||||||
if outfile, err = os.Open(filename+".out.mp4"); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
|
|
||||||
func getAVCDecoderConfRecordByTrack(track *atom.Track) (record *atom.AVCDecoderConfRecord) {
|
|
||||||
if media := track.Media; media != nil {
|
|
||||||
if info := media.Info; info != nil {
|
|
||||||
if sample := info.Sample; sample != nil {
|
|
||||||
if desc := sample.SampleDesc; desc != nil {
|
|
||||||
if avc1 := desc.Avc1Desc; avc1 != nil {
|
|
||||||
if conf := avc1.Conf; conf != nil {
|
|
||||||
return &conf.Record
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func ProbeFile(filename string) (err error) {
|
|
||||||
var osfile *os.File
|
|
||||||
if osfile, err = os.Open(filename); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
var finfo os.FileInfo
|
|
||||||
if finfo, err = osfile.Stat(); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
dumper := &atom.Dumper{}
|
|
||||||
var moov *atom.Movie
|
|
||||||
|
|
||||||
lr := &io.LimitedReader{R: osfile, N: finfo.Size()}
|
|
||||||
for lr.N > 0 {
|
|
||||||
var ar *io.LimitedReader
|
|
||||||
|
|
||||||
var cc4 string
|
|
||||||
if ar, cc4, err = atom.ReadAtomHeader(lr, ""); err != nil {
|
|
||||||
log.Println("read atom failed")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if cc4 == "moov" {
|
|
||||||
if moov, err = atom.ReadMovie(ar); err != nil {
|
|
||||||
log.Println("read '%s' atom failed", cc4)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if false {
|
|
||||||
atom.WalkMovie(dumper, moov)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
fmt.Println("atom:", cc4)
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err = atom.ReadDummy(lr, int(ar.N)); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if moov == nil {
|
|
||||||
err = fmt.Errorf("'moov' atom not found")
|
|
||||||
log.Println("'moov' atom not found")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(moov.Tracks) > 0 {
|
|
||||||
track := moov.Tracks[0]
|
|
||||||
record := getAVCDecoderConfRecordByTrack(track)
|
|
||||||
|
|
||||||
if record != nil && len(record.SPS) > 0 {
|
|
||||||
sps := record.SPS[0]
|
|
||||||
if len(sps) > 1 {
|
|
||||||
sps = sps[1:]
|
|
||||||
log.Println(hex.Dump(sps))
|
|
||||||
var info *atom.H264SPSInfo
|
|
||||||
if info, err = atom.ParseH264SPS(sps); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
log.Println(info)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestConvert(filename string) (err error) {
|
|
||||||
var osfile *os.File
|
|
||||||
if osfile, err = os.Open(filename); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
var finfo os.FileInfo
|
|
||||||
if finfo, err = osfile.Stat(); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
log.Println("filesize", finfo.Size())
|
|
||||||
|
|
||||||
lr := &io.LimitedReader{R: osfile, N: finfo.Size()}
|
|
||||||
|
|
||||||
var outfile *os.File
|
|
||||||
if outfile, err = os.Create(filename+".out.mp4"); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
for lr.N > 0 {
|
|
||||||
var ar *io.LimitedReader
|
|
||||||
|
|
||||||
var cc4 string
|
|
||||||
if ar, cc4, err = atom.ReadAtomHeader(lr, ""); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if cc4 == "moov" {
|
|
||||||
|
|
||||||
curPos, _ := outfile.Seek(0, 1)
|
|
||||||
origSize := ar.N+8
|
|
||||||
var moov *atom.Movie
|
|
||||||
if moov, err = atom.ReadMovie(ar); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
//log.Println(moov.Tracks[0].Media.Info.Data.Refer)
|
|
||||||
|
|
||||||
if err = atom.WriteMovie(outfile, moov); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
curPosAfterRead, _ := outfile.Seek(0, 1)
|
|
||||||
bytesWritten := curPosAfterRead - curPos
|
|
||||||
|
|
||||||
log.Println("regen moov", "tracks nr", len(moov.Tracks),
|
|
||||||
"origSize", origSize, "bytesWritten", bytesWritten,
|
|
||||||
)
|
|
||||||
|
|
||||||
padSize := origSize - bytesWritten - 8
|
|
||||||
aw, _ := atom.WriteAtomHeader(outfile, "free")
|
|
||||||
atom.WriteDummy(outfile, int(padSize))
|
|
||||||
aw.Close()
|
|
||||||
|
|
||||||
} else {
|
|
||||||
|
|
||||||
outcc4 := cc4
|
|
||||||
if outcc4 != "mdat" {
|
|
||||||
log.Println("omit", cc4)
|
|
||||||
outcc4 = "free"
|
|
||||||
} else {
|
|
||||||
log.Println("copy", cc4)
|
|
||||||
}
|
|
||||||
|
|
||||||
var aw *atom.Writer
|
|
||||||
if aw, err = atom.WriteAtomHeader(outfile, outcc4); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err = io.CopyN(aw, ar, ar.N); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if err = aw.Close(); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//log.Println("atom", cc4, "left", lr.N)
|
|
||||||
//atom.ReadDummy(ar, int(ar.N))
|
|
||||||
}
|
|
||||||
|
|
||||||
if err = outfile.Close(); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
Loading…
x
Reference in New Issue
Block a user