remux/ts2es.c

   1 #include "ts2es.h"
   2
   3 // from VDR's remux.c
   4 #define MAXNONUSEFULDATA (10*1024*1024)
   5
   6 class cTS2ES: public ipack {
   7         friend void PutES(uint8_t *Buffer, int Size, void *Data);
   8
   9 private:
  10         uint8_t *m_ResultBuffer;
  11         int *m_ResultCount;
  12
  13 public:
  14         cTS2ES(uint8_t *ResultBuffer, int *ResultCount);
  15         ~cTS2ES();
  16
  17         void PutTSPacket(const uint8_t *Buffer);
  18 };
  19
  20 void PutES(uint8_t *Buffer, int Size, void *Data) {
  21         cTS2ES *This = (cTS2ES*)Data;
  22         uint8_t payl = Buffer[8] + 9 + This->start - 1;
  23         int count = Size - payl;
  24
  25         if (*This->m_ResultCount + count > RESULTBUFFERSIZE) {
  26                 esyslog("ERROR: result buffer overflow (%d + %d > %d)",
  27                                 *This->m_ResultCount, count, RESULTBUFFERSIZE);
  28                 count = RESULTBUFFERSIZE - *This->m_ResultCount;
  29         }
  30         memcpy(This->m_ResultBuffer + *This->m_ResultCount, Buffer + payl, count);
  31         *This->m_ResultCount += count;
  32         This->start = 1;
  33 }
  34
  35 cTS2ES::cTS2ES(uint8_t *ResultBuffer, int *ResultCount) {
  36         m_ResultBuffer = ResultBuffer;
  37         m_ResultCount = ResultCount;
  38
  39         init_ipack(this, IPACKS, PutES, 0);
  40         data = (void*)this;
  41 }
  42
  43 cTS2ES::~cTS2ES() {
  44 }
  45
  46 void cTS2ES::PutTSPacket(const uint8_t *Buffer) {
  47   if (!Buffer)
  48      return;
  49
  50   if (Buffer[1] & 0x80) { // ts error
  51                 // TODO
  52         }
  53
  54   if (Buffer[1] & 0x40) { // payload start
  55                 if (plength == MMAX_PLENGTH - 6) {
  56             plength = found - 6;
  57       found = 0;
  58       send_ipack(this);
  59       reset_ipack(this);
  60     }
  61   }
  62
  63         uint8_t off = 0;
  64
  65   if (Buffer[3] & 0x20) {  // adaptation field?
  66                 off = Buffer[4] + 1;
  67     if (off + 4 > TS_SIZE - 1)
  68       return;
  69   }
  70
  71   instant_repack((uint8_t*)(Buffer + 4 + off), TS_SIZE - 4 - off, this);
  72 }
  73
  74 cTS2ESRemux::cTS2ESRemux(int Pid):
  75                 cTSRemux(false) {
  76         m_Pid = Pid;
  77   m_Remux = new cTS2ES(m_ResultBuffer, &m_ResultCount);
  78 }
  79
  80 cTS2ESRemux::~cTS2ESRemux() {
  81         delete m_Remux;
  82 }
  83
  84 void cTS2ESRemux::PutTSPacket(int Pid, const uint8_t *Data) {
  85         if (Pid == m_Pid) m_Remux->PutTSPacket(Data);
  86 }
  87