2 Copyright 2006 Mark Calderbank
4 This file is part of VOMP.
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.
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.
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
21 #include "demuxerts.h"
23 DemuxerTS::DemuxerTS(int p_vID, int p_aID)
29 void DemuxerTS::flush()
37 int DemuxerTS::scan(UCHAR *buf, int len)
39 // Temporarily, just look for the lowest audio stream and return it
41 UINT LoPattern = 0x100 | PESTYPE_AUD0,
42 HiPattern = 0x100 | PESTYPE_AUDMAX;
47 UINT pattern = *(UINT*)buf;
50 if (pattern < LoPattern || pattern > HiPattern) continue;
53 if (HiPattern == LoPattern) break;
58 void DemuxerTS::setVID(int p_vID)
63 void DemuxerTS::setAID(int p_aID)
68 int DemuxerTS::findPTS(UCHAR* buf, int len, ULLONG* dest)
70 UINT LoPattern = 0x100 | PESTYPE_VID0,
71 HiPattern = 0x100 | PESTYPE_VIDMAX;
75 UINT pattern = *(UINT*)buf;
77 if (pattern < LoPattern || pattern > HiPattern) continue;
79 UINT framelength = ((UINT)buf[3] << 8) | buf[4];
82 if ( buf[1] & 0x80 ) // PTS_DTS_flags indicate that PTS is present
84 *dest = ( (ULLONG)(buf[3] & 0x0E) << 29 ) |
85 ( (ULLONG)(buf[4]) << 22 ) |
86 ( (ULLONG)(buf[5] & 0xFE) << 14 ) |
87 ( (ULLONG)(buf[6]) << 7 ) |
88 ( (ULLONG)(buf[7] & 0xFE) >> 1 );
92 buf += framelength; len -= framelength;
98 int DemuxerTS::put(UCHAR* buf, int len)
100 int ret = 0; // return number of bytes consumed
104 if (len >= TS_SIZE + 1 - partPacket)
105 { // Remainder of partial packet is available, plus one
106 memcpy(store+partPacket, buf, TS_SIZE - partPacket);
107 ret += TS_SIZE - partPacket;
108 buf += TS_SIZE - partPacket;
109 len -= TS_SIZE - partPacket;
110 partPacket = TS_SIZE;
112 { // Packet is properly terminated
113 int rc = processTS(store);
115 partPacket = 0; // Successfully processed
117 return ret; // Try again later.
120 { // Packet not terminated. Find another candidate, and shift store
121 Log::getInstance()->log("TS Demuxer", Log::ERR, "TS Misaligned!");
123 while (search < partPacket && store[search] != TS_SIG)
125 partPacket -= search;
126 if (partPacket) memcpy(store, store+search, partPacket);
130 { // Still don't have complete packet. Consume what we do have.
131 memcpy(store+partPacket, buf, len);
138 // Position ourselves at a candidate TS packet
139 while (*buf != TS_SIG && len)
141 Log::getInstance()->log("TS Demuxer", Log::ERR, "TS Misaligned!");
147 if (len < TS_SIZE + 1)
148 { // Not enough data. Store what we have.
149 memcpy(store, buf, len);
155 if (buf[TS_SIZE] != TS_SIG)
156 { // Not terminated correctly.
157 while (*buf != TS_SIG && len)
164 int rc = processTS(buf);
166 { // Successfully processed
167 buf += TS_SIZE; ret += TS_SIZE; len -= TS_SIZE;
170 { // Processing failed.
178 int DemuxerTS::processTS(UCHAR* buf)
180 int datalen = TS_SIZE - 4;
181 int pid = ( (buf[1] & 0x1F) << 8 ) | buf[2];
182 UCHAR payload = buf[1] & 0x40;
184 if (buf[3] & 0x20) // Adaptation field is present
185 datalen -= (buf[4] + 1);
186 if (datalen < 0) // Error in stream TODO log this
188 if (datalen == 0) // Null packet
190 buf += (TS_SIZE - datalen);
196 if (pid == vID) rc = vPacket.submit();
197 if (pid == aID) rc = aPacket.submit();
198 if (rc == 0) return 0;
203 buf += 6; datalen -= 6;
208 buf += 6; datalen -= 6;
211 // if (datalen >= 4) pesID = buf[3];
214 // case PESTYPE_VID0 ... PESTYPE_VIDMAX:
215 // if (video_current == -1) video_current = pesID;
216 // if (video_current == pesID)
219 // vPacket.init(pesID);
220 // buf += 6; datalen -= 6;
223 // case PESTYPE_AUD0 ... PESTYPE_AUDMAX:
224 // if (audio_current == -1) audio_current = pesID;
225 // if (audio_current == pesID)
228 // aPacket.init(pesID);
229 // buf += 6; datalen -= 6;
235 PESPacket* packet = NULL;
236 if (pid == vID) packet = &vPacket;
237 if (pid == aID) packet = &aPacket;
240 if (packet->write(buf, datalen) == 0)
241 { // Writing to packet failed. It has overflowed.
242 Log::getInstance()->log("TS Demuxer", Log::DEBUG, "Packet overflow, pid %d", pid);
243 if (packet->submit() == 0)
246 packet->write((UCHAR*)"\200\000\000", 3);
247 packet->write(buf, datalen);