demuxer.h

   1 /*
   2     Copyright 2005 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
  30 #ifndef DEMUXER_H
  31 #define DEMUXER_H
  32
  33 #include <stdio.h>
  34 #include <memory.h>
  35 #include "stream.h"
  36 #include "log.h"
  37 #include "defines.h"
  38 #include "callback.h"
  39 #include "draintarget.h"
  40
  41 #include "audio.h"
  42 #include "video.h"
  43
  44
  45 class Demuxer
  46 {
  47   public:
  48     Demuxer();
  49     ~Demuxer();
  50     static Demuxer* getInstance();
  51     int init(Callback* callback);
  52     void reset();
  53     void flush();
  54     void flushAudio();
  55     void seek();
  56     void setVideoStream(int id);
  57     void setAudioStream(int id);
  58 #ifndef NEW_DEMUXER
  59     int writeAudio(int fd);
  60     int writeVideo(int fd);
  61 #else
  62   int writeAudio(DrainTarget* dt);
  63     int writeVideo(DrainTarget* dt);
  64 #endif
  65
  66     int scan(UCHAR* buf, int len);
  67     int findVideoPTS(UCHAR* buf, int len, ULLONG* dest);
  68     int put(UCHAR* buf, int len);
  69
  70     int getHorizontalSize() { return horizontal_size; }
  71     int getVerticalSize() { return vertical_size; }
  72     int getAspectRatio() { return aspect_ratio; }
  73     int getFrameRate() { return frame_rate; }
  74     int getBitRate() { return bit_rate; }
  75     ULLONG getVideoPTS() { return video_pts; }
  76
  77     enum AspectRatio
  78     {
  79       ASPECT_4_3  = 2,
  80       ASPECT_16_9 = 3
  81     };
  82
  83   private:
  84     static Demuxer* instance;
  85     Stream videostream;
  86     Stream audiostream;
  87     int shutdown();
  88     int initted;
  89     int seeking;
  90     UCHAR* inbuf;
  91
  92     void setAspectRatio(enum AspectRatio);
  93     Callback* callback;
  94
  95     int video_current, audio_current;
  96     int state_frametype, state_framepos;
  97     int state_stream_fill, state_vid_parsed;
  98     int frame_length;
  99     int parse_find_frame(int len);
 100     int parse_video_frame(int len, int* full);
 101     int parse_audio_frame(int len, int* full);
 102     int parse_private1_frame(int len, int* full);
 103     void parse_video_details(UCHAR* buf, int len);
 104
 105     UCHAR* local_frame;
 106     static const int demuxMemoryV = 2097152;
 107     static const int demuxMemoryA = 524288;
 108     static const int FrameRates[9];
 109
 110     enum FRAMETYPE
 111     {
 112       FRAMETYPE_PRIVATE_1 = 0xbd,
 113
 114       FRAMETYPE_AUD0 = 0xc0,
 115       FRAMETYPE_AUD1,
 116       FRAMETYPE_AUD2,
 117       FRAMETYPE_AUD3,
 118       FRAMETYPE_AUD4,
 119       FRAMETYPE_AUD5,
 120       FRAMETYPE_AUD6,
 121       FRAMETYPE_AUD7,
 122       FRAMETYPE_AUD8,
 123       FRAMETYPE_AUD9,
 124       FRAMETYPE_AUD10,
 125       FRAMETYPE_AUD11,
 126       FRAMETYPE_AUD12,
 127       FRAMETYPE_AUD13,
 128       FRAMETYPE_AUD14,
 129       FRAMETYPE_AUD15,
 130       FRAMETYPE_AUD16,
 131       FRAMETYPE_AUD17,
 132       FRAMETYPE_AUD18,
 133       FRAMETYPE_AUD19,
 134       FRAMETYPE_AUD20,
 135       FRAMETYPE_AUD21,
 136       FRAMETYPE_AUD22,
 137       FRAMETYPE_AUD23,
 138       FRAMETYPE_AUD24,
 139       FRAMETYPE_AUD25,
 140       FRAMETYPE_AUD26,
 141       FRAMETYPE_AUD27,
 142       FRAMETYPE_AUD28,
 143       FRAMETYPE_AUD29,
 144       FRAMETYPE_AUD30,
 145       FRAMETYPE_AUD31,
 146       FRAMETYPE_AUDMAX = FRAMETYPE_AUD31,
 147
 148       FRAMETYPE_VID0 = 0xe0,
 149       FRAMETYPE_VID1,
 150       FRAMETYPE_VID2,
 151       FRAMETYPE_VID3,
 152       FRAMETYPE_VID4,
 153       FRAMETYPE_VID5,
 154       FRAMETYPE_VID6,
 155       FRAMETYPE_VID7,
 156       FRAMETYPE_VID8,
 157       FRAMETYPE_VID9,
 158       FRAMETYPE_VID10,
 159       FRAMETYPE_VID11,
 160       FRAMETYPE_VID12,
 161       FRAMETYPE_VID13,
 162       FRAMETYPE_VID14,
 163       FRAMETYPE_VID15,
 164       FRAMETYPE_VIDMAX = FRAMETYPE_VID15
 165     };
 166
 167     int horizontal_size;
 168     int vertical_size;
 169     enum AspectRatio aspect_ratio;
 170     int arcnt;
 171     int frame_rate;
 172     int bit_rate;
 173     ULLONG video_pts;
 174 };
 175
 176 #endif