105 lines
2.8 KiB
C
105 lines
2.8 KiB
C
#include "remux/ts2ps.h"
|
|
|
|
class cTS2PS {
|
|
friend void PutPES(uint8_t *Buffer, int Size, void *Data);
|
|
|
|
private:
|
|
ipack m_Ipack;
|
|
uint8_t *m_ResultBuffer;
|
|
int *m_ResultCount;
|
|
|
|
public:
|
|
cTS2PS(uint8_t *ResultBuffer, int *ResultCount, uint8_t AudioCid = 0x00,
|
|
bool PS = false);
|
|
~cTS2PS();
|
|
|
|
void PutTSPacket(const uint8_t *Buffer);
|
|
};
|
|
|
|
void PutPES(uint8_t *Buffer, int Size, void *Data) {
|
|
cTS2PS *This = (cTS2PS*)Data;
|
|
if (*This->m_ResultCount + Size > RESULTBUFFERSIZE) {
|
|
esyslog("ERROR: result buffer overflow (%d + %d > %d)",
|
|
*This->m_ResultCount, Size, RESULTBUFFERSIZE);
|
|
Size = RESULTBUFFERSIZE - *This->m_ResultCount;
|
|
}
|
|
memcpy(This->m_ResultBuffer + *This->m_ResultCount, Buffer, Size);
|
|
*This->m_ResultCount += Size;
|
|
}
|
|
|
|
cTS2PS::cTS2PS(uint8_t *ResultBuffer, int *ResultCount, uint8_t AudioCid,
|
|
bool PS) {
|
|
m_ResultBuffer = ResultBuffer;
|
|
m_ResultCount = ResultCount;
|
|
|
|
init_ipack(&m_Ipack, IPACKS, PutPES, PS);
|
|
m_Ipack.cid = AudioCid;
|
|
m_Ipack.data = (void*)this;
|
|
}
|
|
|
|
cTS2PS::~cTS2PS() {
|
|
}
|
|
|
|
void cTS2PS::PutTSPacket(const uint8_t *Buffer) {
|
|
if (!Buffer)
|
|
return;
|
|
|
|
if (Buffer[1] & 0x80) { // ts error
|
|
// TODO
|
|
}
|
|
|
|
if (Buffer[1] & 0x40) { // payload start
|
|
if (m_Ipack.plength == MMAX_PLENGTH - 6 && m_Ipack.found > 6) {
|
|
m_Ipack.plength = m_Ipack.found - 6;
|
|
m_Ipack.found = 0;
|
|
send_ipack(&m_Ipack);
|
|
reset_ipack(&m_Ipack);
|
|
}
|
|
}
|
|
|
|
uint8_t off = 0;
|
|
|
|
if (Buffer[3] & 0x20) { // adaptation field?
|
|
off = Buffer[4] + 1;
|
|
if (off + 4 > TS_SIZE - 1)
|
|
return;
|
|
}
|
|
|
|
instant_repack((uint8_t*)(Buffer + 4 + off), TS_SIZE - 4 - off, &m_Ipack);
|
|
}
|
|
|
|
cTS2PSRemux::cTS2PSRemux(int VPid, int APid1, int APid2, int DPid1,
|
|
int DPid2, bool PS) {
|
|
m_VPid = VPid;
|
|
m_APid1 = APid1;
|
|
m_APid2 = APid2;
|
|
m_DPid1 = DPid1;
|
|
m_DPid2 = DPid2;
|
|
m_VRemux = new cTS2PS(m_ResultBuffer, &m_ResultCount, 0x00, PS);
|
|
m_ARemux1 = new cTS2PS(m_ResultBuffer, &m_ResultCount, 0xC0, PS);
|
|
m_ARemux2 = APid2 ? new cTS2PS(m_ResultBuffer, &m_ResultCount, 0xC1, PS)
|
|
: NULL;
|
|
m_DRemux1 = DPid1 ? new cTS2PS(m_ResultBuffer, &m_ResultCount, 0x00, PS)
|
|
: NULL;
|
|
//XXX don't yet know how to tell apart primary and secondary DD data...
|
|
m_DRemux2 = /*XXX m_DPid2 ? new cTS2PS(m_ResultBuffer, &m_ResultCount,
|
|
0x00, PS) : XXX*/ NULL;
|
|
}
|
|
|
|
cTS2PSRemux::~cTS2PSRemux() {
|
|
if (m_DRemux2) delete m_DRemux2;
|
|
if (m_DRemux1) delete m_DRemux1;
|
|
if (m_ARemux2) delete m_ARemux2;
|
|
delete m_ARemux1;
|
|
delete m_VRemux;
|
|
}
|
|
|
|
void cTS2PSRemux::PutTSPacket(int Pid, const uint8_t *Data) {
|
|
if (Pid == m_VPid) m_VRemux->PutTSPacket(Data);
|
|
else if (Pid == m_APid1) m_ARemux1->PutTSPacket(Data);
|
|
else if (Pid == m_APid2 && m_ARemux2) m_ARemux2->PutTSPacket(Data);
|
|
else if (Pid == m_DPid1 && m_DRemux1) m_DRemux1->PutTSPacket(Data);
|
|
else if (Pid == m_DPid2 && m_DRemux2) m_DRemux2->PutTSPacket(Data);
|
|
}
|
|
|