rename isom to aacparser
This commit is contained in:
nareix
parent
a5ec0acea0
commit
dfd71d04bf
10
checksum.go
10
checksum.go
@ -1,13 +1,12 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
"io"
|
||||
"fmt"
|
||||
"encoding/hex"
|
||||
"fmt"
|
||||
"io"
|
||||
)
|
||||
|
||||
var ieeeCrc32Tbl = []uint32 {
|
||||
var ieeeCrc32Tbl = []uint32{
|
||||
0x00000000, 0xB71DC104, 0x6E3B8209, 0xD926430D, 0xDC760413, 0x6B6BC517,
|
||||
0xB24D861A, 0x0550471E, 0xB8ED0826, 0x0FF0C922, 0xD6D68A2F, 0x61CB4B2B,
|
||||
0x649B0C35, 0xD386CD31, 0x0AA08E3C, 0xBDBD4F38, 0x70DB114C, 0xC7C6D048,
|
||||
@ -55,7 +54,7 @@ var ieeeCrc32Tbl = []uint32 {
|
||||
|
||||
func updateIeeeCrc32(crc uint32, data []byte) uint32 {
|
||||
for _, b := range data {
|
||||
crc = ieeeCrc32Tbl[b^byte(crc)]^(crc>>8)
|
||||
crc = ieeeCrc32Tbl[b^byte(crc)] ^ (crc >> 8)
|
||||
}
|
||||
return crc
|
||||
}
|
||||
@ -101,4 +100,3 @@ func (self *Crc32Writer) Write(b []byte) (n int, err error) {
|
||||
self.Crc32 = updateIeeeCrc32(self.Crc32, b)
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
@ -1,4 +1,3 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
@ -7,8 +6,7 @@ import (
|
||||
|
||||
func TestChecksum(t *testing.T) {
|
||||
b := []byte("hello world")
|
||||
b = append(b, []byte{0xbb,0x08,0xec,0x87}...)
|
||||
b = append(b, []byte{0xbb, 0x08, 0xec, 0x87}...)
|
||||
crc := updateIeeeCrc32(0xffffffff, b)
|
||||
t.Logf("%x", crc)
|
||||
}
|
||||
|
||||
|
||||
20
demuxer.go
20
demuxer.go
@ -1,11 +1,10 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
"io"
|
||||
"bytes"
|
||||
"fmt"
|
||||
"github.com/nareix/mp4/isom"
|
||||
"github.com/nareix/codec/aacparser"
|
||||
"io"
|
||||
)
|
||||
|
||||
type Demuxer struct {
|
||||
@ -33,7 +32,7 @@ func (self *Demuxer) ReadHeader() (err error) {
|
||||
for {
|
||||
if self.pmt != nil {
|
||||
n := 0
|
||||
for _, track := range(self.Tracks) {
|
||||
for _, track := range self.Tracks {
|
||||
if track.payloadReady {
|
||||
n++
|
||||
}
|
||||
@ -58,7 +57,7 @@ func (self *Demuxer) ReadSample() (track *Track, err error) {
|
||||
}
|
||||
|
||||
for {
|
||||
for _, _track := range(self.Tracks) {
|
||||
for _, _track := range self.Tracks {
|
||||
if _track.payloadReady {
|
||||
track = _track
|
||||
return
|
||||
@ -87,13 +86,13 @@ func (self *Demuxer) readPacket() (err error) {
|
||||
}
|
||||
} else {
|
||||
if self.pmt == nil {
|
||||
for _, entry := range(self.pat.Entries) {
|
||||
for _, entry := range self.pat.Entries {
|
||||
if entry.ProgramMapPID == header.PID {
|
||||
self.pmt = new(PMT)
|
||||
if *self.pmt, err = ReadPMT(bytes.NewReader(payload)); err != nil {
|
||||
return
|
||||
}
|
||||
for _, info := range(self.pmt.ElementaryStreamInfos) {
|
||||
for _, info := range self.pmt.ElementaryStreamInfos {
|
||||
track := &Track{}
|
||||
|
||||
track.demuxer = self
|
||||
@ -113,7 +112,7 @@ func (self *Demuxer) readPacket() (err error) {
|
||||
}
|
||||
} else {
|
||||
|
||||
for _, track := range(self.Tracks) {
|
||||
for _, track := range self.Tracks {
|
||||
if header.PID == track.pid {
|
||||
if err = track.appendPacket(header, payload); err != nil {
|
||||
return
|
||||
@ -126,7 +125,7 @@ func (self *Demuxer) readPacket() (err error) {
|
||||
return
|
||||
}
|
||||
|
||||
func (self *Track) GetMPEG4AudioConfig() isom.MPEG4AudioConfig {
|
||||
func (self *Track) GetMPEG4AudioConfig() aacparser.MPEG4AudioConfig {
|
||||
return self.mpeg4AudioConfig
|
||||
}
|
||||
|
||||
@ -154,7 +153,7 @@ func (self *Track) appendPayload() (err error) {
|
||||
|
||||
if self.Type == AAC {
|
||||
if !self.mpeg4AudioConfig.IsValid() {
|
||||
if self.mpeg4AudioConfig, _, _, _, err = isom.ReadADTSFrame(self.payload); err != nil {
|
||||
if self.mpeg4AudioConfig, _, _, _, err = aacparser.ReadADTSFrame(self.payload); err != nil {
|
||||
return
|
||||
}
|
||||
self.mpeg4AudioConfig = self.mpeg4AudioConfig.Complete()
|
||||
@ -200,4 +199,3 @@ func (self *Track) appendPacket(header TSHeader, payload []byte) (err error) {
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
@ -1,16 +1,15 @@
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"os"
|
||||
"io"
|
||||
ts "../"
|
||||
"fmt"
|
||||
"encoding/hex"
|
||||
"bytes"
|
||||
"encoding/gob"
|
||||
"runtime/pprof"
|
||||
"encoding/hex"
|
||||
"flag"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"runtime/pprof"
|
||||
)
|
||||
|
||||
type GobAllSamples struct {
|
||||
@ -157,7 +156,7 @@ func readSamples(filename string, ch chan Sample) {
|
||||
}
|
||||
}
|
||||
|
||||
for _, entry := range(pat.Entries) {
|
||||
for _, entry := range pat.Entries {
|
||||
if entry.ProgramMapPID == header.PID {
|
||||
//fmt.Println("matchs", entry)
|
||||
if pmt, err = ts.ReadPMT(pr); err != nil {
|
||||
@ -167,7 +166,7 @@ func readSamples(filename string, ch chan Sample) {
|
||||
}
|
||||
}
|
||||
|
||||
for _, info = range(pmt.ElementaryStreamInfos) {
|
||||
for _, info = range pmt.ElementaryStreamInfos {
|
||||
if info.ElementaryPID == header.PID {
|
||||
onStreamPayload()
|
||||
}
|
||||
@ -231,7 +230,7 @@ func testInputGob(pathGob string, pathOut string, testSeg bool, writeM3u8 bool)
|
||||
if sample.Sync {
|
||||
syncCount++
|
||||
if testSeg {
|
||||
if syncCount % 3 == 0 {
|
||||
if syncCount%3 == 0 {
|
||||
filename := fmt.Sprintf("%s.seg%d.ts", pathOut, segCount)
|
||||
|
||||
if debugStream {
|
||||
@ -241,7 +240,7 @@ func testInputGob(pathGob string, pathOut string, testSeg bool, writeM3u8 bool)
|
||||
if m3u8file != nil {
|
||||
info, _ := outfile.Stat()
|
||||
size := info.Size()
|
||||
dur := float64(trackH264.PTS - lastPTS) / float64(allSamples.TimeScale)
|
||||
dur := float64(trackH264.PTS-lastPTS) / float64(allSamples.TimeScale)
|
||||
writeM3U8Item(m3u8file, lastFilename, size, dur)
|
||||
}
|
||||
|
||||
@ -395,4 +394,3 @@ func main() {
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
30
muxer.go
30
muxer.go
@ -1,12 +1,11 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io"
|
||||
"github.com/nareix/mp4/isom"
|
||||
"github.com/nareix/codec/h264parser"
|
||||
"github.com/nareix/codec/aacparser"
|
||||
"io"
|
||||
)
|
||||
|
||||
type Muxer struct {
|
||||
@ -87,7 +86,7 @@ func (self *Muxer) WriteHeader() (err error) {
|
||||
}
|
||||
|
||||
// about to remove
|
||||
for _, track := range(self.Tracks) {
|
||||
for _, track := range self.Tracks {
|
||||
track.spsHasWritten = false
|
||||
}
|
||||
|
||||
@ -106,20 +105,20 @@ func (self *Track) TimeScale() int64 {
|
||||
return self.timeScale
|
||||
}
|
||||
|
||||
func (self *Track) SetMPEG4AudioConfig(config isom.MPEG4AudioConfig) {
|
||||
func (self *Track) SetMPEG4AudioConfig(config aacparser.MPEG4AudioConfig) {
|
||||
self.mpeg4AudioConfig = config
|
||||
}
|
||||
|
||||
func (self *Track) tsToPesTs(ts int64) uint64 {
|
||||
return uint64(ts)*PTS_HZ/uint64(self.timeScale)+PTS_HZ
|
||||
return uint64(ts)*PTS_HZ/uint64(self.timeScale) + PTS_HZ
|
||||
}
|
||||
|
||||
func (self *Track) tsToPCR(ts int64) uint64 {
|
||||
return uint64(ts)*PCR_HZ/uint64(self.timeScale)+PCR_HZ
|
||||
return uint64(ts)*PCR_HZ/uint64(self.timeScale) + PCR_HZ
|
||||
}
|
||||
|
||||
func (self *Track) tsToTime(ts int64) float64 {
|
||||
return float64(ts)/float64(self.timeScale)
|
||||
return float64(ts) / float64(self.timeScale)
|
||||
}
|
||||
|
||||
func (self *Track) WriteSample(pts int64, dts int64, isKeyFrame bool, data []byte) (err error) {
|
||||
@ -129,8 +128,8 @@ func (self *Track) WriteSample(pts int64, dts int64, isKeyFrame bool, data []byt
|
||||
|
||||
if self.Type == AAC {
|
||||
|
||||
if !isom.IsADTSFrame(data) {
|
||||
data = append(isom.MakeADTSHeader(self.mpeg4AudioConfig, 1024, len(data)), data...)
|
||||
if !aacparser.IsADTSFrame(data) {
|
||||
data = append(aacparser.MakeADTSHeader(self.mpeg4AudioConfig, 1024, len(data)), data...)
|
||||
}
|
||||
if false {
|
||||
fmt.Printf("WriteSample=%x\n", data[:5])
|
||||
@ -182,17 +181,17 @@ func (self *Track) WriteSample(pts int64, dts int64, isKeyFrame bool, data []byt
|
||||
/* about to remove */
|
||||
|
||||
func (self *Track) setPCR() {
|
||||
self.tsw.PCR = uint64(self.PTS)*PCR_HZ/uint64(self.timeScale)
|
||||
self.tsw.PCR = uint64(self.PTS) * PCR_HZ / uint64(self.timeScale)
|
||||
}
|
||||
|
||||
func (self *Track) getPesHeader(dataLength int) (data []byte){
|
||||
func (self *Track) getPesHeader(dataLength int) (data []byte) {
|
||||
if self.PTS == 0 {
|
||||
self.PTS = self.timeScale
|
||||
}
|
||||
buf := &bytes.Buffer{}
|
||||
pes := PESHeader{
|
||||
StreamId: self.streamId,
|
||||
PTS: uint64(self.PTS)*PTS_HZ/uint64(self.timeScale),
|
||||
PTS: uint64(self.PTS) * PTS_HZ / uint64(self.timeScale),
|
||||
}
|
||||
WritePESHeader(buf, pes, dataLength)
|
||||
return buf.Bytes()
|
||||
@ -216,9 +215,9 @@ func (self *Track) WriteH264NALU(sync bool, duration int, nalu []byte) (err erro
|
||||
for i, nalu := range nalus {
|
||||
var startCode []byte
|
||||
if i == 0 {
|
||||
startCode = []byte{0,0,0,1,0x9,0xf0,0,0,0,1} // AUD
|
||||
startCode = []byte{0, 0, 0, 1, 0x9, 0xf0, 0, 0, 0, 1} // AUD
|
||||
} else {
|
||||
startCode = []byte{0,0,1}
|
||||
startCode = []byte{0, 0, 1}
|
||||
}
|
||||
data.Append(startCode)
|
||||
data.Append(nalu)
|
||||
@ -252,4 +251,3 @@ func (self *Track) WriteADTSAACFrame(duration int, frame []byte) (err error) {
|
||||
self.incPTS(duration)
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
63
reader.go
63
reader.go
@ -1,4 +1,3 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
@ -32,7 +31,7 @@ func ReadUInt64(r io.Reader, n int) (res uint64, err error) {
|
||||
if res32, err = ReadUInt(r, n-4); err != nil {
|
||||
return
|
||||
}
|
||||
res |= uint64(res32)<<32
|
||||
res |= uint64(res32) << 32
|
||||
n = 4
|
||||
}
|
||||
if res32, err = ReadUInt(r, n); err != nil {
|
||||
@ -59,7 +58,7 @@ func ReadTSHeader(r io.Reader) (self TSHeader, err error) {
|
||||
|
||||
if DebugReader {
|
||||
fmt.Printf("ts: flags %s\n", FieldsDumper{
|
||||
Fields: []struct{
|
||||
Fields: []struct {
|
||||
Length int
|
||||
Desc string
|
||||
}{
|
||||
@ -78,12 +77,12 @@ func ReadTSHeader(r io.Reader) (self TSHeader, err error) {
|
||||
})
|
||||
}
|
||||
|
||||
if flags & 0x400000 != 0 {
|
||||
if flags&0x400000 != 0 {
|
||||
// When set to '1' it indicates that this TS packet contains the first PES packet.
|
||||
self.PayloadUnitStart = true
|
||||
}
|
||||
|
||||
if (flags & 0xff000000) >> 24 != 0x47 {
|
||||
if (flags&0xff000000)>>24 != 0x47 {
|
||||
err = fmt.Errorf("invalid sync")
|
||||
return
|
||||
}
|
||||
@ -91,7 +90,7 @@ func ReadTSHeader(r io.Reader) (self TSHeader, err error) {
|
||||
self.PID = (flags & 0x1fff00) >> 8
|
||||
self.ContinuityCounter = flags & 0xf
|
||||
|
||||
if flags & 0x20 != 0 {
|
||||
if flags&0x20 != 0 {
|
||||
var flags, length uint
|
||||
if length, err = ReadUInt(r, 1); err != nil {
|
||||
return
|
||||
@ -104,7 +103,7 @@ func ReadTSHeader(r io.Reader) (self TSHeader, err error) {
|
||||
|
||||
if DebugReader {
|
||||
fmt.Printf("ts: ext_flags %s\n", FieldsDumper{
|
||||
Fields: []struct{
|
||||
Fields: []struct {
|
||||
Length int
|
||||
Desc string
|
||||
}{
|
||||
@ -123,12 +122,12 @@ func ReadTSHeader(r io.Reader) (self TSHeader, err error) {
|
||||
}
|
||||
|
||||
// random_access_indicator
|
||||
if flags & 0x40 != 0 {
|
||||
if flags&0x40 != 0 {
|
||||
self.RandomAccessIndicator = true
|
||||
}
|
||||
|
||||
// PCR
|
||||
if flags & 0x10 != 0 {
|
||||
if flags&0x10 != 0 {
|
||||
var v uint64
|
||||
if v, err = ReadUInt64(lr, 6); err != nil {
|
||||
return
|
||||
@ -141,7 +140,7 @@ func ReadTSHeader(r io.Reader) (self TSHeader, err error) {
|
||||
}
|
||||
|
||||
// OPCR
|
||||
if flags & 0x08 != 0 {
|
||||
if flags&0x08 != 0 {
|
||||
var v uint64
|
||||
if v, err = ReadUInt64(lr, 6); err != nil {
|
||||
return
|
||||
@ -150,14 +149,14 @@ func ReadTSHeader(r io.Reader) (self TSHeader, err error) {
|
||||
}
|
||||
|
||||
// Splice countdown
|
||||
if flags & 0x04 != 0 {
|
||||
if flags&0x04 != 0 {
|
||||
if _, err = ReadUInt(lr, 1); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// Transport private data
|
||||
if flags & 0x02 != 0 {
|
||||
if flags&0x02 != 0 {
|
||||
var length uint
|
||||
if length, err = ReadUInt(lr, 1); err != nil {
|
||||
return
|
||||
@ -212,7 +211,6 @@ func ReadPSI(r io.Reader) (self PSI, lr *io.LimitedReader, cr *Crc32Reader, err
|
||||
fmt.Printf("psi: pointer=%d\n", pointer)
|
||||
}
|
||||
|
||||
|
||||
if pointer != 0 {
|
||||
if err = ReadDummy(r, int(pointer)); err != nil {
|
||||
return
|
||||
@ -253,7 +251,7 @@ func ReadPSI(r io.Reader) (self PSI, lr *io.LimitedReader, cr *Crc32Reader, err
|
||||
|
||||
if DebugReader {
|
||||
fmt.Printf("psi: %s\n", FieldsDumper{
|
||||
Fields: []struct{
|
||||
Fields: []struct {
|
||||
Length int
|
||||
Desc string
|
||||
}{
|
||||
@ -328,7 +326,7 @@ func ReadPMT(r io.Reader) (self PMT, err error) {
|
||||
|
||||
if DebugReader {
|
||||
fmt.Printf("pmt: %s\n", FieldsDumper{
|
||||
Fields: []struct{
|
||||
Fields: []struct {
|
||||
Length int
|
||||
Desc string
|
||||
}{
|
||||
@ -374,7 +372,7 @@ func ReadPMT(r io.Reader) (self PMT, err error) {
|
||||
|
||||
if DebugReader {
|
||||
fmt.Printf("pmt: info1 %s\n", FieldsDumper{
|
||||
Fields: []struct{
|
||||
Fields: []struct {
|
||||
Length int
|
||||
Desc string
|
||||
}{
|
||||
@ -395,7 +393,7 @@ func ReadPMT(r io.Reader) (self PMT, err error) {
|
||||
|
||||
if DebugReader {
|
||||
fmt.Printf("pmt: info2 %s\n", FieldsDumper{
|
||||
Fields: []struct{
|
||||
Fields: []struct {
|
||||
Length int
|
||||
Desc string
|
||||
}{
|
||||
@ -500,7 +498,7 @@ func ReadPESHeader(r io.Reader) (res *PESHeader, err error) {
|
||||
}
|
||||
|
||||
if length == 0 {
|
||||
length = 1<<31
|
||||
length = 1 << 31
|
||||
}
|
||||
lrAll := &io.LimitedReader{R: r, N: int64(length)}
|
||||
lr := lrAll
|
||||
@ -517,7 +515,7 @@ func ReadPESHeader(r io.Reader) (res *PESHeader, err error) {
|
||||
|
||||
if DebugReader {
|
||||
fmt.Printf("pes: %s\n", FieldsDumper{
|
||||
Fields: []struct{
|
||||
Fields: []struct {
|
||||
Length int
|
||||
Desc string
|
||||
}{
|
||||
@ -545,7 +543,7 @@ func ReadPESHeader(r io.Reader) (res *PESHeader, err error) {
|
||||
|
||||
if DebugReader {
|
||||
fmt.Printf("pes: %s\n", FieldsDumper{
|
||||
Fields: []struct{
|
||||
Fields: []struct {
|
||||
Length int
|
||||
Desc string
|
||||
}{
|
||||
@ -568,7 +566,7 @@ func ReadPESHeader(r io.Reader) (res *PESHeader, err error) {
|
||||
}
|
||||
lr = &io.LimitedReader{R: lr, N: int64(length)}
|
||||
|
||||
if flags & 0x80 != 0 {
|
||||
if flags&0x80 != 0 {
|
||||
var v uint64
|
||||
if v, err = ReadUInt64(lr, 5); err != nil {
|
||||
return
|
||||
@ -580,7 +578,7 @@ func ReadPESHeader(r io.Reader) (res *PESHeader, err error) {
|
||||
}
|
||||
}
|
||||
|
||||
if flags & 0x40 != 0 && flags & 0x80 != 0 {
|
||||
if flags&0x40 != 0 && flags&0x80 != 0 {
|
||||
var v uint64
|
||||
if v, err = ReadUInt64(lr, 5); err != nil {
|
||||
return
|
||||
@ -592,35 +590,35 @@ func ReadPESHeader(r io.Reader) (res *PESHeader, err error) {
|
||||
}
|
||||
|
||||
// ESCR flag
|
||||
if flags & 0x20 != 0 {
|
||||
if flags&0x20 != 0 {
|
||||
if _, err = ReadUInt64(lr, 6); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// ES rate flag
|
||||
if flags & 0x10 != 0 {
|
||||
if flags&0x10 != 0 {
|
||||
if _, err = ReadUInt64(lr, 3); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// additional copy info flag
|
||||
if flags & 0x04 != 0 {
|
||||
if flags&0x04 != 0 {
|
||||
if _, err = ReadUInt(lr, 1); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// PES CRC flag
|
||||
if flags & 0x02 != 0 {
|
||||
if flags&0x02 != 0 {
|
||||
if _, err = ReadUInt(lr, 2); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// PES extension flag
|
||||
if flags & 0x01 != 0 {
|
||||
if flags&0x01 != 0 {
|
||||
var flags uint
|
||||
|
||||
// PES private data flag(1)
|
||||
@ -634,7 +632,7 @@ func ReadPESHeader(r io.Reader) (res *PESHeader, err error) {
|
||||
}
|
||||
|
||||
// PES private data flag(1)
|
||||
if flags & 0x80 != 0 {
|
||||
if flags&0x80 != 0 {
|
||||
// if set to 1 16 bytes of user defined data is appended to the header data field
|
||||
if err = ReadDummy(lr, 16); err != nil {
|
||||
return
|
||||
@ -642,7 +640,7 @@ func ReadPESHeader(r io.Reader) (res *PESHeader, err error) {
|
||||
}
|
||||
|
||||
// pack header field flag(1)
|
||||
if flags & 0x40 != 0 {
|
||||
if flags&0x40 != 0 {
|
||||
// if set to 1 the 8-bit pack field length value is appended to the header data field
|
||||
if err = ReadDummy(lr, 1); err != nil {
|
||||
return
|
||||
@ -650,21 +648,21 @@ func ReadPESHeader(r io.Reader) (res *PESHeader, err error) {
|
||||
}
|
||||
|
||||
// program packet sequence counter flag(1)
|
||||
if flags & 0x20 != 0 {
|
||||
if flags&0x20 != 0 {
|
||||
if err = ReadDummy(lr, 2); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// P-STD buffer flag(1)
|
||||
if flags & 0x10 != 0 {
|
||||
if flags&0x10 != 0 {
|
||||
if err = ReadDummy(lr, 2); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// PES extension flag 2(1)
|
||||
if flags & 0x01 != 0 {
|
||||
if flags&0x01 != 0 {
|
||||
if err = ReadDummy(lr, 2); err != nil {
|
||||
return
|
||||
}
|
||||
@ -684,4 +682,3 @@ func ReadPESHeader(r io.Reader) (res *PESHeader, err error) {
|
||||
res = self
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
6
track.go
6
track.go
@ -1,9 +1,8 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"github.com/nareix/mp4/isom"
|
||||
"github.com/nareix/codec/aacparser"
|
||||
)
|
||||
|
||||
type Track struct {
|
||||
@ -16,7 +15,7 @@ type Track struct {
|
||||
PTS int64
|
||||
timeScale int64
|
||||
|
||||
mpeg4AudioConfig isom.MPEG4AudioConfig
|
||||
mpeg4AudioConfig aacparser.MPEG4AudioConfig
|
||||
buf bytes.Buffer
|
||||
payload []byte
|
||||
peshdr *PESHeader
|
||||
@ -36,4 +35,3 @@ const (
|
||||
H264 = 1
|
||||
AAC = 2
|
||||
)
|
||||
|
||||
|
||||
18
ts.go
18
ts.go
@ -1,4 +1,3 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
@ -70,7 +69,7 @@ type PESHeader struct {
|
||||
|
||||
func PESUIntToTs(v uint64) (ts uint64) {
|
||||
// 0010 PTS 32..30 1 PTS 29..15 1 PTS 14..00 1
|
||||
return (((v>>33)&0x7)<<30) | (((v>>17)&0x7fff)<<15) | ((v>>1)&0x7fff)
|
||||
return (((v >> 33) & 0x7) << 30) | (((v >> 17) & 0x7fff) << 15) | ((v >> 1) & 0x7fff)
|
||||
}
|
||||
|
||||
func PESTsToUInt(ts uint64) (v uint64) {
|
||||
@ -85,15 +84,15 @@ const (
|
||||
|
||||
func UIntToPCR(v uint64) uint64 {
|
||||
// base(33)+resverd(6)+ext(9)
|
||||
base := v>>15
|
||||
ext := v&0x1ff
|
||||
return base*300+ext
|
||||
base := v >> 15
|
||||
ext := v & 0x1ff
|
||||
return base*300 + ext
|
||||
}
|
||||
|
||||
func PCRToUInt(pcr uint64) uint64 {
|
||||
base := pcr/300
|
||||
ext := pcr%300
|
||||
return base<<15|0x3f<<9|ext
|
||||
base := pcr / 300
|
||||
ext := pcr % 300
|
||||
return base<<15 | 0x3f<<9 | ext
|
||||
}
|
||||
|
||||
type FieldsDumper struct {
|
||||
@ -109,11 +108,10 @@ func (self FieldsDumper) String() (res string) {
|
||||
pos := uint(self.Length)
|
||||
for _, field := range self.Fields {
|
||||
pos -= uint(field.Length)
|
||||
val := (self.Val>>pos)&(1<<uint(field.Length)-1)
|
||||
val := (self.Val >> pos) & (1<<uint(field.Length) - 1)
|
||||
if val != 0 {
|
||||
res += fmt.Sprintf("%s=%x ", field.Desc, val)
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
@ -1,4 +1,3 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
@ -8,4 +7,3 @@ import (
|
||||
func TestPESTsConv(t *testing.T) {
|
||||
t.Logf("%x", PESTsToUInt(0x123))
|
||||
}
|
||||
|
||||
|
||||
10
vecio.go
10
vecio.go
@ -1,13 +1,12 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
"io"
|
||||
"os"
|
||||
"net"
|
||||
"fmt"
|
||||
"unsafe"
|
||||
"io"
|
||||
"net"
|
||||
"os"
|
||||
"syscall"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
type iovec struct {
|
||||
@ -134,4 +133,3 @@ func newVecWriter(w io.Writer) (vecw *vecWriter) {
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
58
writer.go
58
writer.go
@ -1,17 +1,16 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io"
|
||||
"bytes"
|
||||
)
|
||||
|
||||
var DebugWriter = false
|
||||
|
||||
func WriteUInt64(w io.Writer, val uint64, n int) (err error) {
|
||||
var b [8]byte
|
||||
for i := n-1; i >= 0; i-- {
|
||||
for i := n - 1; i >= 0; i-- {
|
||||
b[i] = byte(val)
|
||||
val >>= 8
|
||||
}
|
||||
@ -51,13 +50,13 @@ func WriteTSHeader(w io.Writer, self TSHeader, dataLength int) (written int, err
|
||||
// Payload flag(1) 0x10
|
||||
// Continuity counter(4)
|
||||
|
||||
flags = 0x47<<24
|
||||
flags = 0x47 << 24
|
||||
flags |= 0x10
|
||||
if self.PayloadUnitStart {
|
||||
flags |= 0x400000
|
||||
}
|
||||
flags |= (self.PID&0x1fff)<<8
|
||||
flags |= self.ContinuityCounter&0xf
|
||||
flags |= (self.PID & 0x1fff) << 8
|
||||
flags |= self.ContinuityCounter & 0xf
|
||||
|
||||
if DebugWriter {
|
||||
fmt.Printf("tsw: pid=%x\n", self.PID)
|
||||
@ -94,7 +93,7 @@ func WriteTSHeader(w io.Writer, self TSHeader, dataLength int) (written int, err
|
||||
}
|
||||
written += 4
|
||||
|
||||
if flags & EXT != 0 {
|
||||
if flags&EXT != 0 {
|
||||
var length uint
|
||||
|
||||
// Discontinuity indicator 1 0x80
|
||||
@ -104,16 +103,16 @@ func WriteTSHeader(w io.Writer, self TSHeader, dataLength int) (written int, err
|
||||
// OPCR flag 1 0x08
|
||||
|
||||
length = 1 // extFlags
|
||||
if extFlags & PCR != 0 {
|
||||
if extFlags&PCR != 0 {
|
||||
length += 6
|
||||
}
|
||||
if extFlags & OPCR != 0 {
|
||||
if extFlags&OPCR != 0 {
|
||||
length += 6
|
||||
}
|
||||
|
||||
paddingLength := 0
|
||||
// need padding
|
||||
if int(length) + 5 + dataLength < 188 {
|
||||
if int(length)+5+dataLength < 188 {
|
||||
paddingLength = 188 - dataLength - 5 - int(length)
|
||||
length = 188 - uint(dataLength) - 5
|
||||
}
|
||||
@ -129,13 +128,13 @@ func WriteTSHeader(w io.Writer, self TSHeader, dataLength int) (written int, err
|
||||
return
|
||||
}
|
||||
|
||||
if extFlags & PCR != 0 {
|
||||
if extFlags&PCR != 0 {
|
||||
if err = WriteUInt64(w, PCRToUInt(self.PCR), 6); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
if extFlags & OPCR != 0 {
|
||||
if extFlags&OPCR != 0 {
|
||||
if err = WriteUInt64(w, PCRToUInt(self.OPCR), 6); err != nil {
|
||||
return
|
||||
}
|
||||
@ -147,7 +146,7 @@ func WriteTSHeader(w io.Writer, self TSHeader, dataLength int) (written int, err
|
||||
}
|
||||
}
|
||||
|
||||
written += int(length)+1
|
||||
written += int(length) + 1
|
||||
}
|
||||
|
||||
return
|
||||
@ -214,7 +213,7 @@ func (self *TSWriter) WriteIovecTo(w io.Writer, data *iovec) (err error) {
|
||||
}
|
||||
payloadLength := 188 - headerLength
|
||||
if self.DisableHeaderPadding && data.Len < payloadLength {
|
||||
data.Append(makeRepeatValBytes(0xff, payloadLength - data.Len))
|
||||
data.Append(makeRepeatValBytes(0xff, payloadLength-data.Len))
|
||||
}
|
||||
|
||||
if DebugWriter {
|
||||
@ -267,8 +266,8 @@ func WritePSI(w io.Writer, self PSI, data []byte) (err error) {
|
||||
|
||||
// section_syntax_indicator(1)=1,private_bit(1)=0,reserved(2)=3,unused(2)=0,section_length(10)
|
||||
var flags, length uint
|
||||
length = 2+3+4+uint(len(data))
|
||||
flags = 0xa<<12|length
|
||||
length = 2 + 3 + 4 + uint(len(data))
|
||||
flags = 0xa<<12 | length
|
||||
if err = WriteUInt(cw, flags, 2); err != nil {
|
||||
return
|
||||
}
|
||||
@ -283,7 +282,7 @@ func WritePSI(w io.Writer, self PSI, data []byte) (err error) {
|
||||
}
|
||||
|
||||
// resverd(2)=3,version(5)=0,Current_next_indicator(1)=1
|
||||
flags = 0x3<<6|1
|
||||
flags = 0x3<<6 | 1
|
||||
if err = WriteUInt(cw, flags, 1); err != nil {
|
||||
return
|
||||
}
|
||||
@ -312,7 +311,7 @@ func WritePSI(w io.Writer, self PSI, data []byte) (err error) {
|
||||
}
|
||||
|
||||
func bswap32(v uint) uint {
|
||||
return (v>>24)|((v>>16)&0xff)<<8|((v>>8)&0xff)<<16|(v&0xff)<<24
|
||||
return (v >> 24) | ((v>>16)&0xff)<<8 | ((v>>8)&0xff)<<16 | (v&0xff)<<24
|
||||
}
|
||||
|
||||
func WritePESHeader(w io.Writer, self PESHeader, dataLength int) (err error) {
|
||||
@ -340,8 +339,8 @@ func WritePESHeader(w io.Writer, self PESHeader, dataLength int) (err error) {
|
||||
return
|
||||
}
|
||||
|
||||
const PTS = 1<<7
|
||||
const DTS = 1<<6
|
||||
const PTS = 1 << 7
|
||||
const DTS = 1 << 6
|
||||
|
||||
if self.PTS != 0 {
|
||||
pts_dts_flags |= PTS
|
||||
@ -350,10 +349,10 @@ func WritePESHeader(w io.Writer, self PESHeader, dataLength int) (err error) {
|
||||
}
|
||||
}
|
||||
|
||||
if pts_dts_flags & PTS != 0 {
|
||||
if pts_dts_flags&PTS != 0 {
|
||||
header_length += 5
|
||||
}
|
||||
if pts_dts_flags & DTS != 0 {
|
||||
if pts_dts_flags&DTS != 0 {
|
||||
header_length += 5
|
||||
}
|
||||
|
||||
@ -385,8 +384,8 @@ func WritePESHeader(w io.Writer, self PESHeader, dataLength int) (err error) {
|
||||
|
||||
// pts(40)?
|
||||
// dts(40)?
|
||||
if pts_dts_flags & PTS != 0 {
|
||||
if pts_dts_flags & DTS != 0 {
|
||||
if pts_dts_flags&PTS != 0 {
|
||||
if pts_dts_flags&DTS != 0 {
|
||||
if err = WriteUInt64(w, PESTsToUInt(self.PTS)|3<<36, 5); err != nil {
|
||||
return
|
||||
}
|
||||
@ -529,7 +528,7 @@ func WritePMT(w io.Writer, self PMT) (err error) {
|
||||
return
|
||||
}
|
||||
|
||||
psi := PSI {
|
||||
psi := PSI{
|
||||
TableId: 2,
|
||||
TableIdExtension: 1,
|
||||
}
|
||||
@ -668,7 +667,7 @@ func (self *SimpleH264Writer) WriteNALU(sync bool, duration int, nalu []byte) (e
|
||||
|
||||
pes := PESHeader{
|
||||
StreamId: StreamIdH264,
|
||||
PTS: uint64(self.PTS)*PTS_HZ/uint64(self.TimeScale),
|
||||
PTS: uint64(self.PTS) * PTS_HZ / uint64(self.TimeScale),
|
||||
}
|
||||
if err = WritePESHeader(self.pesBuf, pes, 0); err != nil {
|
||||
return
|
||||
@ -679,16 +678,16 @@ func (self *SimpleH264Writer) WriteNALU(sync bool, duration int, nalu []byte) (e
|
||||
for i, nalu := range nalus {
|
||||
var startCode []byte
|
||||
if i == 0 {
|
||||
startCode = []byte{0,0,0,1,0x9,0xf0,0,0,0,1} // AUD
|
||||
startCode = []byte{0, 0, 0, 1, 0x9, 0xf0, 0, 0, 0, 1} // AUD
|
||||
} else {
|
||||
startCode = []byte{0,0,1}
|
||||
startCode = []byte{0, 0, 1}
|
||||
}
|
||||
data.Append(startCode)
|
||||
data.Append(nalu)
|
||||
}
|
||||
|
||||
self.tswH264.RandomAccessIndicator = sync
|
||||
self.tswH264.PCR = uint64(self.PCR)*PCR_HZ/uint64(self.TimeScale)
|
||||
self.tswH264.PCR = uint64(self.PCR) * PCR_HZ / uint64(self.TimeScale)
|
||||
|
||||
self.tswH264.W = self.W
|
||||
if err = self.tswH264.WriteIovec(data); err != nil {
|
||||
@ -701,4 +700,3 @@ func (self *SimpleH264Writer) WriteNALU(sync bool, duration int, nalu []byte) (e
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
@ -1,10 +1,9 @@
|
||||
|
||||
package ts
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"encoding/hex"
|
||||
"bytes"
|
||||
"encoding/hex"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestWriteTSHeader(t *testing.T) {
|
||||
@ -13,7 +12,6 @@ func TestWriteTSHeader(t *testing.T) {
|
||||
W: bw,
|
||||
PCR: 0x12345678,
|
||||
}
|
||||
w.Write([]byte{'h','e','l','o'}[:], false)
|
||||
w.Write([]byte{'h', 'e', 'l', 'o'}[:], false)
|
||||
t.Logf("\n%s", hex.Dump(bw.Bytes()))
|
||||
}
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user