2 Copyright 2005-2008 Mark Calderbank
3 Copyright 2007 Marten Richter (AC3 support)
5 This file is part of VOMP.
7 VOMP is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or
10 (at your option) any later version.
12 VOMP is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with VOMP; if not, write to the Free Software Foundation, Inc.,
19 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
24 Thanks go to Jon Gettler of the MVPMC project and Stephen Rice for
25 the demuxer in mvpmc. It was used in the creation of this demuxer;
26 however, no code was copied verbatim.
44 PESPacket(const PESPacket& packet);
45 PESPacket& operator=(const PESPacket& packet);
47 void init(UCHAR type, UCHAR sub=0);
49 int write(const UCHAR* buf, int len);
51 UCHAR operator[] (UINT index) const;
52 // return data[index] if in bounds, else 0
53 // so no proper error condition but never mind for now
54 const UCHAR* getData() const { return data; }
55 UINT getLength() const { return length; }
56 UINT getSize() const { return size; }
57 UCHAR getPacketType() const { return packetType; }
58 void setSubstream(UCHAR s) { substream = s; }
59 UCHAR getSubstream() const { return substream; }
60 ULLONG getPTS() const;
61 bool hasPTS() const { return (getPTS() != PTS_INVALID); }
63 UINT findPictureHeader(bool h264) const;
64 UINT findSeqHeader(bool h264) const;
65 UINT countPictureHeaders(bool h264) const;
66 static const ULLONG PTS_INVALID = (1LL << 33);
68 void copyFrom(const PESPacket& packet);
74 UINT mutable seq_header; // 0 = no, 1 = unknown, else = header offset
82 static Demuxer* getInstance();
83 int init(Callback* tcallback, DrainTarget* audio, DrainTarget* video,
84 DrainTarget* teletext,
85 ULONG demuxMemoryV, ULONG demuxMemoryA, ULONG demuxMemoryT, double fps=25.,
86 DVBSubtitles* tsubtitles=NULL);
91 void setVideoStream(int id);
92 //TODO HANS next virtual necessary?
93 //virtual void setAudioStream(int id);
94 void setAudioStream(int id);
95 void setTeletextStream(int id);
96 void setDVBSubtitleStream(int id);
101 virtual int scan(UCHAR* buf, int len) = 0;
102 virtual int findPTS(UCHAR* buf, int len, ULLONG* dest) = 0;
103 virtual int put(UCHAR* buf, int len) = 0;
104 virtual void setFrameNum(ULONG frame) {}
105 virtual void setPacketNum(ULONG packet) {}
106 virtual ULONG getFrameNumFromPTS(ULLONG pts) {return 0;}
107 virtual ULONG getPacketNum() {return 0;}
109 bool* getmpAudioChannels(); //Maybe virtual ?
110 bool* getac3AudioChannels(); //Maybe virtual ?
111 bool* getSubtitleChannels();
112 int getselAudioChannel();
113 int getselSubtitleChannel();
114 bool ish264() {return h264;}
115 void seth264(bool newh264){h264=newh264;}
117 int getHorizontalSize() { return horizontal_size; }
118 int getVerticalSize() { return vertical_size; }
119 int getAspectRatio() { return aspect_ratio; }
120 int getFrameRate() { return frame_rate; }
121 int getBitRate() { return bit_rate; }
122 ULLONG getVideoPTS() { return video_pts; }
123 ULLONG getAudioPTS() { return audio_pts; }
131 // Remove all data from a buffer apart from video PES packets.
132 // Returns the length of the reduced data.
133 // removed *static function*, due to DemuxerTS
134 virtual UINT stripAudio(UCHAR* buf, UINT len);
136 // Scan a buffer to see if video packets are present.
137 // Returns true if video exists; false if not.
138 // removed *static function* for h264 detection
139 static bool scanForVideo(UCHAR* buf, UINT len, bool &ish264);
143 PESTYPE_PRIVATE_1 = 0xBD,
146 PESTYPE_AUD1, PESTYPE_AUD2, PESTYPE_AUD3, PESTYPE_AUD4,
147 PESTYPE_AUD5, PESTYPE_AUD6, PESTYPE_AUD7, PESTYPE_AUD8,
148 PESTYPE_AUD9, PESTYPE_AUD10, PESTYPE_AUD11, PESTYPE_AUD12,
149 PESTYPE_AUD13, PESTYPE_AUD14, PESTYPE_AUD15, PESTYPE_AUD16,
150 PESTYPE_AUD17, PESTYPE_AUD18, PESTYPE_AUD19, PESTYPE_AUD20,
151 PESTYPE_AUD21, PESTYPE_AUD22, PESTYPE_AUD23, PESTYPE_AUD24,
152 PESTYPE_AUD25, PESTYPE_AUD26, PESTYPE_AUD27, PESTYPE_AUD28,
153 PESTYPE_AUD29, PESTYPE_AUD30, PESTYPE_AUD31,
154 PESTYPE_AUDMAX = PESTYPE_AUD31,
157 PESTYPE_VID1, PESTYPE_VID2, PESTYPE_VID3, PESTYPE_VID4,
158 PESTYPE_VID5, PESTYPE_VID6, PESTYPE_VID7, PESTYPE_VID8,
159 PESTYPE_VID9, PESTYPE_VID10, PESTYPE_VID11, PESTYPE_VID12,
160 PESTYPE_VID13, PESTYPE_VID14, PESTYPE_VID15,
161 PESTYPE_VIDMAX = PESTYPE_VID15
163 enum PESTYPE_SUBSTREAM
165 PESTYPE_SUBSTREAM_TELETEXT0 = 0x10,
166 PESTYPE_SUBSTREAM_TELETEXT1,PESTYPE_SUBSTREAM_TELETEXT2, PESTYPE_SUBSTREAM_TELETEXT3,
167 PESTYPE_SUBSTREAM_TELETEXT4,PESTYPE_SUBSTREAM_TELETEXT5,PESTYPE_SUBSTREAM_TELETEXT6,
168 PESTYPE_SUBSTREAM_TELETEXT7, PESTYPE_SUBSTREAM_TELETEXT8, PESTYPE_SUBSTREAM_TELETEXT9,
169 PESTYPE_SUBSTREAM_TELETEXT10,PESTYPE_SUBSTREAM_TELETEXT11,PESTYPE_SUBSTREAM_TELETEXT12,
170 PESTYPE_SUBSTREAM_TELETEXT13,PESTYPE_SUBSTREAM_TELETEXT14,PESTYPE_SUBSTREAM_TELETEXT15,
171 PESTYPE_SUBSTREAM_TELETEXTMAX=PESTYPE_SUBSTREAM_TELETEXT15,
172 PESTYPE_SUBSTREAM_DVBSUBTITLE0=0x20,
173 PESTYPE_SUBSTREAM_DVBSUBTITLE1,PESTYPE_SUBSTREAM_DVBSUBTITLE2,PESTYPE_SUBSTREAM_DVBSUBTITLE3,
174 PESTYPE_SUBSTREAM_DVBSUBTITLE4,PESTYPE_SUBSTREAM_DVBSUBTITLE5,PESTYPE_SUBSTREAM_DVBSUBTITLE6,
175 PESTYPE_SUBSTREAM_DVBSUBTITLE7,
176 PESTYPE_SUBSTREAM_DVBSUBTITLEMAX=PESTYPE_SUBSTREAM_DVBSUBTITLE7,
177 PESTYPE_SUBSTREAM_AC30 = 0x80,
178 PESTYPE_SUBSTREAM_AC31,PESTYPE_SUBSTREAM_AC32, PESTYPE_SUBSTREAM_AC33,
179 PESTYPE_SUBSTREAM_AC34,PESTYPE_SUBSTREAM_AC35,PESTYPE_SUBSTREAM_AC36,
180 PESTYPE_SUBSTREAM_AC37,
181 PESTYPE_SUBSTREAM_AC3MAX = PESTYPE_SUBSTREAM_AC37
185 // Operations on PES packets
186 bool submitPacket(PESPacket&);
187 void parsePacketDetails(PESPacket&);
189 // General demuxer objects and status indicators
190 static Demuxer* instance;
193 DVBSubtitles* subtitles;
194 Stream teletextstream;
200 int video_current, audio_current, teletext_current, subtitle_current;
202 // Video stream information
203 void setAspectRatio(enum AspectRatio);
209 enum AspectRatio aspect_ratio;
214 ULLONG video_pts_seek;
217 bool isteletextdecoded;
220 unsigned int packetnum;
223 static const int FrameRates[9];
228 bool avail_mpaudchan[PESTYPE_AUDMAX-PESTYPE_AUD0+1];
229 bool avail_ac3audchan[PESTYPE_SUBSTREAM_AC3MAX-PESTYPE_SUBSTREAM_AC30+1];
230 bool avail_dvbsubtitlechan[PESTYPE_SUBSTREAM_DVBSUBTITLEMAX-PESTYPE_SUBSTREAM_DVBSUBTITLE0+1];