vdr-plugin-streamdev/remux/ts2ps.c
2004-12-30 22:43:55 +00:00

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);
}