mirror of https://bitbucket.org/ausocean/av.git
231 lines
7.6 KiB
Go
231 lines
7.6 KiB
Go
package generator
|
|
|
|
import (
|
|
_"fmt"
|
|
_"os"
|
|
//"bitbucket.org/ausocean/av/mpegts"
|
|
//"bitbucket.org/ausocean/av/pes"
|
|
//"bitbucket.org/ausocean/av/tools"
|
|
//"bitbucket.org/ausocean/av/rtp"
|
|
"../mpegts"
|
|
"../pes"
|
|
"../tools"
|
|
"../rtp"
|
|
)
|
|
|
|
var (
|
|
PatTable = []byte{0, 0, 176, 13, 0, 1, 193, 0, 0, 0, 1, 240, 0, 42, 177, 4, 178, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,}
|
|
|
|
PmtTable = []byte{0, 2, 176, 18, 0, 1, 193, 0, 0, 0xE1, 0x00, 0xF0, 0, 0x1B, 0xE1, 0, 0xF0, 0, 0x15, 0xBD, 0x4D, 0x56, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,}
|
|
)
|
|
|
|
const (
|
|
SdtPid = 17
|
|
PatPid = 0
|
|
PmtPid = 4096
|
|
VideoPid = 256
|
|
)
|
|
|
|
type tsGenerator struct {
|
|
rtpInputChan chan rtp.RtpPacket
|
|
outputChan chan []byte
|
|
nalInputChan chan []byte
|
|
currentTsPacket *mpegts.MpegTsPacket
|
|
payloadByteChan chan byte
|
|
currentCC byte
|
|
currentPtsTime float64
|
|
currentPcrTime float64
|
|
fps uint
|
|
pesPktChan chan []byte
|
|
ccMap map[int]int
|
|
}
|
|
|
|
func (g *tsGenerator)GetInputChan() chan []byte {
|
|
return g.nalInputChan
|
|
}
|
|
|
|
func (g *tsGenerator)GetOutputChan() chan []byte {
|
|
return g.outputChan
|
|
}
|
|
|
|
func NewTsGenerator(fps uint) (g *tsGenerator) {
|
|
g = new(tsGenerator)
|
|
g.outputChan = make(chan []byte, 100)
|
|
g.rtpInputChan = make(chan rtp.RtpPacket, 100)
|
|
g.nalInputChan = make(chan []byte, 10000)
|
|
g.currentCC = 0
|
|
g.fps = fps
|
|
g.currentPcrTime = .0
|
|
g.currentPtsTime = .7
|
|
g.pesPktChan = make(chan []byte, 1000)
|
|
g.payloadByteChan = make(chan byte, 100000)
|
|
g.ccMap = make(map[int]int, 4)
|
|
g.ccMap[SdtPid] = 0
|
|
g.ccMap[PatPid] = 0
|
|
g.ccMap[PmtPid] = 0
|
|
g.ccMap[VideoPid] = 0
|
|
return
|
|
}
|
|
|
|
func (g *tsGenerator) genPts()(pts uint64){
|
|
pts = uint64(g.currentPtsTime * float64(90000))
|
|
g.currentPtsTime += 1.0/float64(g.fps)
|
|
return
|
|
}
|
|
|
|
func (g *tsGenerator) genPcr()(pcr uint64){
|
|
pcr = uint64(g.currentPcrTime * float64(90000))
|
|
g.currentPcrTime += 1.0/float64(g.fps)
|
|
return
|
|
}
|
|
|
|
func (g *tsGenerator) Start(){
|
|
go g.generate()
|
|
}
|
|
|
|
func (g *tsGenerator) generate() {
|
|
var rtpBuffer [](*rtp.RtpPacket)
|
|
for {
|
|
select {
|
|
case rtpPacket := <-g.rtpInputChan:
|
|
rtpBuffer = append(rtpBuffer, &rtpPacket)
|
|
if len(rtpBuffer) > 2 {
|
|
// if there's something weird going on with sequence numbers then
|
|
// insertion sort
|
|
if rtpPacket.SequenceNumber < rtpBuffer[len(rtpBuffer)-2].SequenceNumber {
|
|
for i := 1; i < len(rtpBuffer); i++ {
|
|
for j := i; j > 0 && rtpBuffer[j].SequenceNumber < rtpBuffer[j-1].SequenceNumber; j-- {
|
|
temp := rtpBuffer[j]
|
|
rtpBuffer[j] = rtpBuffer[j-1]
|
|
rtpBuffer[j-1] = temp
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(rtpBuffer) > 200 {
|
|
for tools.GetOctectType(rtpBuffer[0]) != 7 {
|
|
rtpBuffer = rtpBuffer[1:]
|
|
}
|
|
sps := make([]byte, len(rtpBuffer[0].Payload))
|
|
copy(sps[:], rtpBuffer[0].Payload[:])
|
|
rtpBuffer = rtpBuffer[1:]
|
|
pps := make([]byte, len(rtpBuffer[0].Payload))
|
|
copy(pps[:], rtpBuffer[0].Payload[:])
|
|
rtpBuffer = rtpBuffer[1:]
|
|
sei := make([]byte, len(rtpBuffer[0].Payload))
|
|
copy(sei[:], rtpBuffer[0].Payload[:])
|
|
rtpBuffer = rtpBuffer[1:]
|
|
for tools.GetOctectType(rtpBuffer[0]) != 7 {
|
|
switch tools.GetOctectType(rtpBuffer[0]) {
|
|
case 28:
|
|
if tools.GetStartBit(rtpBuffer[0]) == 1 {
|
|
var buffer []byte
|
|
buffer = append(buffer, []byte{0x00, 0x00, 0x01}...)
|
|
buffer = append(buffer, []byte{0x09, 0x10}...)
|
|
buffer = append(buffer, []byte{0x00, 0x00, 0x01}...)
|
|
buffer = append(buffer, sps...)
|
|
buffer = append(buffer, []byte{0x00, 0x00, 0x01}...)
|
|
buffer = append(buffer, pps...)
|
|
buffer = append(buffer, []byte{0x00, 0x00, 0x01}...)
|
|
buffer = append(buffer, sei...)
|
|
buffer = append(buffer, []byte{0x00, 0x00, 0x01}...)
|
|
buffer = append(buffer, rtpBuffer[0].Payload[0]&0xE0|rtpBuffer[0].Payload[1]&0x1F)
|
|
buffer = append(buffer, rtpBuffer[0].Payload[2:]...)
|
|
rtpBuffer = rtpBuffer[1:]
|
|
for {
|
|
buffer = append(buffer, rtpBuffer[0].Payload[2:]...)
|
|
if tools.GetEndBit(rtpBuffer[0]) == 1 {
|
|
rtpBuffer = rtpBuffer[1:]
|
|
g.nalInputChan <- buffer
|
|
break
|
|
}
|
|
rtpBuffer = rtpBuffer[1:]
|
|
}
|
|
}
|
|
case 1:
|
|
var buffer []byte
|
|
buffer = append(buffer, []byte{0x00, 0x00, 0x01}...)
|
|
buffer = append(buffer, []byte{0x09, 0x10}...)
|
|
buffer = append(buffer, []byte{0x00, 0x00, 0x01}...)
|
|
buffer = append(buffer, sps...)
|
|
buffer = append(buffer, []byte{0x00, 0x00, 0x01}...)
|
|
buffer = append(buffer, pps...)
|
|
buffer = append(buffer, []byte{0x00, 0x00, 0x01}...)
|
|
buffer = append(buffer, sei...)
|
|
buffer = append(buffer, []byte{0x00, 0x00, 0x01}...)
|
|
buffer = append(buffer, rtpBuffer[0].Payload[0]&0xE0|rtpBuffer[0].Payload[1]&0x1F)
|
|
buffer = append(buffer, rtpBuffer[0].Payload[2:]...)
|
|
rtpBuffer = rtpBuffer[1:]
|
|
g.nalInputChan <- buffer
|
|
default:
|
|
}
|
|
}
|
|
}
|
|
case nalUnit := <-g.nalInputChan:
|
|
pesPkt := pes.PESPacket{
|
|
StreamID: 0xE0,
|
|
PDI: byte(2),
|
|
PTS: g.genPts(),
|
|
Data: nalUnit,
|
|
HeaderLength: 5,
|
|
}
|
|
g.pesPktChan <- pesPkt.ToByteSlice()
|
|
case pesPkt := <-g.pesPktChan:
|
|
for ii := range pesPkt {
|
|
g.payloadByteChan <- pesPkt[ii]
|
|
}
|
|
pusi := true
|
|
for len(g.payloadByteChan) > 0 {
|
|
pkt := mpegts.MpegTsPacket{
|
|
PUSI: pusi,
|
|
PID: VideoPid,
|
|
RAI: pusi,
|
|
CC: byte(g.getCC(VideoPid)),
|
|
AFC: byte(3),
|
|
PCRF: pusi,
|
|
}
|
|
pkt.FillPayload(g.payloadByteChan)
|
|
|
|
if pusi {
|
|
// Create pat table and send off
|
|
patPkt := mpegts.MpegTsPacket{
|
|
PUSI: pusi,
|
|
PID: PatPid,
|
|
CC: byte(g.getCC(PatPid)),
|
|
AFC: 1,
|
|
Payload: PatTable,
|
|
}
|
|
|
|
patPktAsByteSlice, _ := patPkt.ToByteSlice()
|
|
g.outputChan <- patPktAsByteSlice
|
|
|
|
// Create pmt table and send off
|
|
pmtPkt := mpegts.MpegTsPacket{
|
|
PUSI: pusi,
|
|
PID: PmtPid,
|
|
CC: byte(g.getCC(PmtPid)),
|
|
AFC: 1,
|
|
Payload: PmtTable,
|
|
}
|
|
pmtPktAsByteSlice, _ := pmtPkt.ToByteSlice()
|
|
g.outputChan <- pmtPktAsByteSlice
|
|
|
|
pkt.PCR = g.genPcr()
|
|
pusi = false
|
|
}
|
|
pktAsBytelice, _ := pkt.ToByteSlice()
|
|
g.outputChan<-pktAsBytelice
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (g *tsGenerator) getCC(pid int) int {
|
|
temp := g.ccMap[pid]
|
|
if g.ccMap[pid]++; g.ccMap[pid] > 15 {
|
|
g.ccMap[pid] = 0
|
|
}
|
|
return temp
|
|
}
|