remux/ts2ps.c

   1 #include "ts2ps.h"
   2
   3 class cTS2PS {
   4         friend void PutPES(uint8_t *Buffer, int Size, void *Data);
   5
   6 private:
   7         ipack m_Ipack;
   8         uint8_t *m_ResultBuffer;
   9         int *m_ResultCount;
  10
  11 public:
  12         cTS2PS(uint8_t *ResultBuffer, int *ResultCount, uint8_t AudioCid = 0x00,
  13                         bool PS = false);
  14         ~cTS2PS();
  15
  16         void PutTSPacket(const uint8_t *Buffer);
  17 };
  18
  19 void PutPES(uint8_t *Buffer, int Size, void *Data) {
  20         cTS2PS *This = (cTS2PS*)Data;
  21         if (Size <= 0 || !Buffer || !This ) {
  22                 esyslog("ERROR: negative size or null pointers");
  23                 return;
  24         }
  25         if (*This->m_ResultCount + Size > RESULTBUFFERSIZE) {
  26                 esyslog("ERROR: result buffer overflow (%d + %d > %d)",
  27                                 *This->m_ResultCount, Size, RESULTBUFFERSIZE);
  28                 Size = RESULTBUFFERSIZE - *This->m_ResultCount;
  29         }
  30         memcpy(This->m_ResultBuffer + *This->m_ResultCount, Buffer, Size);
  31         *This->m_ResultCount += Size;
  32 }
  33
  34 cTS2PS::cTS2PS(uint8_t *ResultBuffer, int *ResultCount, uint8_t AudioCid,
  35                 bool PS) {
  36         m_ResultBuffer = ResultBuffer;
  37         m_ResultCount = ResultCount;
  38
  39         init_ipack(&m_Ipack, IPACKS, PutPES, PS);
  40         m_Ipack.cid = AudioCid;
  41         m_Ipack.data = (void*)this;
  42 }
  43
  44 cTS2PS::~cTS2PS() {
  45 }
  46
  47 void cTS2PS::PutTSPacket(const uint8_t *Buffer) {
  48   if (!Buffer)
  49      return;
  50
  51   if (Buffer[1] & 0x80) { // ts error
  52                 // TODO
  53         }
  54
  55   if (Buffer[1] & 0x40) { // payload start
  56                 if (m_Ipack.plength == MMAX_PLENGTH - 6 && m_Ipack.found > 6) {
  57             m_Ipack.plength = m_Ipack.found - 6;
  58       m_Ipack.found = 0;
  59       send_ipack(&m_Ipack);
  60       reset_ipack(&m_Ipack);
  61     }
  62   }
  63
  64         uint8_t off = 0;
  65
  66   if (Buffer[3] & 0x20) {  // adaptation field?
  67                 off = Buffer[4] + 1;
  68     if (off + 4 > TS_SIZE - 1)
  69       return;
  70   }
  71
  72   instant_repack((uint8_t*)(Buffer + 4 + off), TS_SIZE - 4 - off, &m_Ipack);
  73 }
  74
  75 cTS2PSRemux::cTS2PSRemux(int VPid, int APid1, int APid2, int DPid1,
  76                 int DPid2, bool PS) {
  77         m_VPid  = VPid;
  78         m_APid1 = APid1;
  79         m_APid2 = APid2;
  80         m_DPid1 = DPid1;
  81         m_DPid2 = DPid2;
  82   m_VRemux  =         new cTS2PS(m_ResultBuffer, &m_ResultCount, 0x00, PS);
  83   m_ARemux1 =         new cTS2PS(m_ResultBuffer, &m_ResultCount, 0xC0, PS);
  84   m_ARemux2 = APid2 ? new cTS2PS(m_ResultBuffer, &m_ResultCount, 0xC1, PS)
  85                           : NULL;
  86   m_DRemux1 = DPid1 ? new cTS2PS(m_ResultBuffer, &m_ResultCount, 0x00, PS)
  87                           : NULL;
  88   //XXX don't yet know how to tell apart primary and secondary DD data...
  89   m_DRemux2 = /*XXX m_DPid2 ? new cTS2PS(m_ResultBuffer, &m_ResultCount,
  90                         0x00, PS) : XXX*/ NULL;
  91 }
  92
  93 cTS2PSRemux::~cTS2PSRemux() {
  94         if (m_DRemux2) delete m_DRemux2;
  95         if (m_DRemux1) delete m_DRemux1;
  96         if (m_ARemux2) delete m_ARemux2;
  97         delete m_ARemux1;
  98         delete m_VRemux;
  99 }
 100
 101 void cTS2PSRemux::PutTSPacket(int Pid, const uint8_t *Data) {
 102         if      (Pid == m_VPid)               m_VRemux->PutTSPacket(Data);
 103         else if (Pid == m_APid1)              m_ARemux1->PutTSPacket(Data);
 104         else if (Pid == m_APid2 && m_ARemux2) m_ARemux2->PutTSPacket(Data);
 105         else if (Pid == m_DPid1 && m_DRemux1) m_DRemux1->PutTSPacket(Data);
 106         else if (Pid == m_DPid2 && m_DRemux2) m_DRemux2->PutTSPacket(Data);
 107 }
 108