demuxerts.h

   1 /*
   2     Copyright 2006-2007 Mark Calderbank
   3
   4     This file is part of VOMP.
   5
   6     VOMP is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     VOMP is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with VOMP; if not, write to the Free Software
  18     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19 */
  20
  21 #ifndef DEMUXERTS_H
  22 #define DEMUXERTS_H
  23
  24 #include "demuxer.h"
  25 #include "defines.h"
  26
  27 #define TS_SIZE 188
  28 #define TS_SIG  0x47
  29
  30 class DemuxerTS : public Demuxer
  31 {
  32   public:
  33     DemuxerTS(int p_vID = 0, int p_aID = 0, int p_subID = 0);
  34     void flush();
  35     int scan(UCHAR* buf, int len);
  36     int findPTS(UCHAR* buf, int len, ULLONG* dest);
  37     void setVID(int p_vID);
  38     void setAID(int p_aID, int type);
  39     void setSubID(int p_subID);
  40     int  getVID() { return vID; }
  41     int  getAID() { return aID; }
  42     int  getSubID() { return subID; }
  43     int put(UCHAR* buf, int len);
  44
  45   private:
  46     int processTS(UCHAR* buf);
  47
  48     UCHAR store[TS_SIZE]; // Storage for partial packets
  49     int partPacket;    // Length of partial packet stored from previous put()
  50     bool parsed;       // Whether PES packet to be submitted has been parsed yet
  51     PESPacket vPacket; // Video PES packet under construction
  52     PESPacket aPacket; // Audio PES packet under construction
  53     PESPacket subPacket; // Subtitles PES packet under construction
  54     int vID, aID, subID; // TS IDs for video/audio/subtitles
  55     int atype;
  56     bool vActive, aActive; // Whether video/audio is actively being captured
  57     bool subActive;        // Same for subtitles
  58     UINT subLength;        // Expected length of subtitle packet
  59 };
  60
  61 #endif