demuxer.h

   1 /*
   2     Copyright 2005-2006 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19 */
  20
  21 /*
  22
  23 Thanks goes to Jon Gettler of the MVPMC project and Stephen Rice for
  24 the demuxer in mvpmc. It was used in the creation of this Demuxer,
  25 however, no code was copied verbatim.
  26
  27 */
  28
  29 #ifndef DEMUXER_H
  30 #define DEMUXER_H
  31
  32 #include <stdio.h>
  33 #include <memory.h>
  34 #include "stream.h"
  35 #include "log.h"
  36 #include "defines.h"
  37 #include "callback.h"
  38 #include "draintarget.h"
  39
  40 class Demuxer
  41 {
  42 protected:
  43   class PESPacket
  44   {
  45     public:
  46       PESPacket();
  47       void init(UCHAR type);
  48       int  write(UCHAR* buf, int len);
  49       int  submit();
  50     protected:
  51       UCHAR data[0x10000];
  52       UINT length, size;
  53       bool closed;
  54       UCHAR packetType;
  55       ULLONG pts;
  56       bool seq_header;
  57       UINT submitted;
  58       virtual void parseDetails();
  59       UINT findPictureHeader();
  60
  61       static const ULLONG PTS_INVALID;
  62   };
  63   friend class PESPacket;
  64
  65   public:
  66     Demuxer();
  67     virtual ~Demuxer();
  68     static Demuxer* getInstance();
  69     int init(Callback* callback, DrainTarget* audio, DrainTarget* video);
  70     virtual void reset();
  71     virtual void flush();
  72     void flushAudio();
  73     void seek();
  74     void setVideoStream(int id);
  75     void setAudioStream(int id);
  76     bool writeAudio();
  77     bool writeVideo();
  78
  79     virtual int scan(UCHAR* buf, int len) = 0;
  80     virtual int findVideoPTS(UCHAR* buf, int len, ULLONG* dest) = 0;
  81     virtual int put(UCHAR* buf, int len) = 0;
  82     virtual void setFrameNum(ULONG frame) {}
  83     virtual ULONG getFrameNumFromPTS(ULLONG pts) {return 0;}
  84
  85     int getHorizontalSize() { return horizontal_size; }
  86     int getVerticalSize() { return vertical_size; }
  87     int getAspectRatio() { return aspect_ratio; }
  88     int getFrameRate() { return frame_rate; }
  89     int getBitRate() { return bit_rate; }
  90     ULLONG getVideoPTS() { return video_pts; }
  91
  92     enum AspectRatio
  93     {
  94       ASPECT_4_3  = 2,
  95       ASPECT_16_9 = 3
  96     };
  97
  98   protected:
  99     // General demuxer objects and status indicators
 100     static Demuxer* instance;
 101     Stream videostream;
 102     Stream audiostream;
 103     int shutdown();
 104     bool initted;
 105     bool vid_seeking;
 106     bool aud_seeking;
 107     int video_current, audio_current;
 108
 109     // Video stream information
 110     void setAspectRatio(enum AspectRatio);
 111     Callback* callback;
 112
 113     int horizontal_size;
 114     int vertical_size;
 115     enum AspectRatio aspect_ratio;
 116     int arcnt;
 117     int frame_rate;
 118     int bit_rate;
 119     ULLONG video_pts;
 120     ULLONG video_pts_seek;
 121     ULLONG audio_pts;
 122
 123     // Constants
 124     static const int demuxMemoryV = 2097152;
 125     static const int demuxMemoryA = 524288;
 126     static const int FrameRates[9];
 127
 128     enum PESTYPE
 129     {
 130       PESTYPE_PRIVATE_1 = 0xBD,
 131
 132       PESTYPE_AUD0 = 0xC0,
 133       PESTYPE_AUD1,  PESTYPE_AUD2,  PESTYPE_AUD3,  PESTYPE_AUD4,
 134       PESTYPE_AUD5,  PESTYPE_AUD6,  PESTYPE_AUD7,  PESTYPE_AUD8,
 135       PESTYPE_AUD9,  PESTYPE_AUD10, PESTYPE_AUD11, PESTYPE_AUD12,
 136       PESTYPE_AUD13, PESTYPE_AUD14, PESTYPE_AUD15, PESTYPE_AUD16,
 137       PESTYPE_AUD17, PESTYPE_AUD18, PESTYPE_AUD19, PESTYPE_AUD20,
 138       PESTYPE_AUD21, PESTYPE_AUD22, PESTYPE_AUD23, PESTYPE_AUD24,
 139       PESTYPE_AUD25, PESTYPE_AUD26, PESTYPE_AUD27, PESTYPE_AUD28,
 140       PESTYPE_AUD29, PESTYPE_AUD30, PESTYPE_AUD31,
 141       PESTYPE_AUDMAX = PESTYPE_AUD31,
 142
 143       PESTYPE_VID0 = 0xE0,
 144       PESTYPE_VID1,  PESTYPE_VID2,  PESTYPE_VID3,  PESTYPE_VID4,
 145       PESTYPE_VID5,  PESTYPE_VID6,  PESTYPE_VID7,  PESTYPE_VID8,
 146       PESTYPE_VID9,  PESTYPE_VID10, PESTYPE_VID11, PESTYPE_VID12,
 147       PESTYPE_VID13, PESTYPE_VID14, PESTYPE_VID15,
 148       PESTYPE_VIDMAX = PESTYPE_VID15
 149     };
 150
 151 };
 152
 153 #endif