1 #include <lib/dvb/pvrparse.h>
2 #include <lib/base/eerror.h>
6 #error no byte order defined!
9 int eMPEGStreamInformation::save(const char *filename)
11 FILE *f = fopen(filename, "wb");
15 for (std::map<off_t, pts_t>::const_iterator i(m_access_points.begin()); i != m_access_points.end(); ++i)
17 unsigned long long d[2];
18 #if BYTE_ORDER == BIG_ENDIAN
22 d[0] = bswap_64(i->first);
23 d[1] = bswap_64(i->second);
25 fwrite(d, sizeof(d), 1, f);
32 int eMPEGStreamInformation::load(const char *filename)
34 FILE *f = fopen(filename, "rb");
37 m_access_points.clear();
38 m_pts_to_offset.clear();
41 unsigned long long d[2];
42 if (fread(d, sizeof(d), 1, f) < 1)
45 #if BYTE_ORDER == LITTLE_ENDIAN
46 d[0] = bswap_64(d[0]);
47 d[1] = bswap_64(d[1]);
49 m_access_points[d[0]] = d[1];
50 m_pts_to_offset.insert(std::pair<pts_t,off_t>(d[1], d[0]));
53 fixupDiscontinuties();
57 bool eMPEGStreamInformation::empty()
59 return m_access_points.empty();
62 void eMPEGStreamInformation::fixupDiscontinuties()
64 m_timestamp_deltas.clear();
65 if (!m_access_points.size())
68 // eDebug("Fixing discontinuities ...");
70 /* if we have no delta at the beginning, extrapolate it */
71 if ((m_access_points.find(0) == m_access_points.end()) && (m_access_points.size() > 1))
73 std::map<off_t,pts_t>::const_iterator second = m_access_points.begin();
74 std::map<off_t,pts_t>::const_iterator first = second++;
75 if (first->first < second->first) /* i.e., not equal or broken */
77 off_t diff = second->first - first->first;
78 pts_t tdiff = second->second - first->second;
79 tdiff *= first->first;
81 m_timestamp_deltas[0] = first->second - tdiff;
82 // eDebug("first delta is %08llx", first->second - tdiff);
86 if (m_timestamp_deltas.empty())
87 m_timestamp_deltas[m_access_points.begin()->first] = m_access_points.begin()->second;
89 pts_t currentDelta = m_timestamp_deltas.begin()->second, lastpts_t = 0;
90 for (std::map<off_t,pts_t>::const_iterator i(m_access_points.begin()); i != m_access_points.end(); ++i)
92 pts_t current = i->second - currentDelta;
93 pts_t diff = current - lastpts_t;
95 if (llabs(diff) > (90000*5)) // 5sec diff
97 // eDebug("%llx < %llx, have discont. new timestamp is %llx (diff is %llx)!", current, lastpts_t, i->second, diff);
98 currentDelta = i->second - lastpts_t; /* FIXME: should be the extrapolated new timestamp, based on the current rate */
99 // eDebug("current delta now %llx, making current to %llx", currentDelta, i->second - currentDelta);
100 m_timestamp_deltas[i->first] = currentDelta;
102 lastpts_t = i->second - currentDelta;
106 // eDebug("ok, found %d disconts.", m_timestamp_deltas.size());
109 for (off_t x=0x25807E34ULL; x < 0x25B3CF70; x+= 100000)
113 int r = getPTS(o, p);
114 eDebug("%08llx -> %08llx | %08llx, %d, %08llx %08llx", x, getDelta(x), getInterpolated(x), r, o, p);
119 pts_t eMPEGStreamInformation::getDelta(off_t offset)
121 if (!m_timestamp_deltas.size())
123 std::map<off_t,pts_t>::iterator i = m_timestamp_deltas.upper_bound(offset);
125 /* i can be the first when you query for something before the first PTS */
126 if (i != m_timestamp_deltas.begin())
132 int eMPEGStreamInformation::fixupPTS(const off_t &offset, pts_t &ts)
134 if (!m_timestamp_deltas.size())
137 std::multimap<pts_t, off_t>::const_iterator
138 l = m_pts_to_offset.upper_bound(ts - 60 * 90000),
139 u = m_pts_to_offset.upper_bound(ts + 60 * 90000),
140 nearest = m_pts_to_offset.end();
144 if ((nearest == m_pts_to_offset.end()) || (llabs(l->first - ts) < llabs(nearest->first - ts)))
148 if (nearest == m_pts_to_offset.end())
151 ts -= getDelta(nearest->second);
156 int eMPEGStreamInformation::getPTS(off_t &offset, pts_t &pts)
158 std::map<off_t,pts_t>::iterator before = m_access_points.lower_bound(offset);
160 /* usually, we prefer the AP before the given offset. however if there is none, we take any. */
161 if (before != m_access_points.begin())
164 if (before == m_access_points.end())
170 offset = before->first;
171 pts = before->second - getDelta(offset);
176 pts_t eMPEGStreamInformation::getInterpolated(off_t offset)
178 /* get the PTS values before and after the offset. */
179 std::map<off_t,pts_t>::iterator before, after;
180 after = m_access_points.upper_bound(offset);
183 if (before != m_access_points.begin())
185 else /* we query before the first known timestamp ... FIXME */
189 if (before == m_access_points.end())
192 /* if after == end, then we need to extrapolate ... FIXME */
193 if ((before->first == offset) || (after == m_access_points.end()))
194 return before->second - getDelta(offset);
196 pts_t before_ts = before->second - getDelta(before->first);
197 pts_t after_ts = after->second - getDelta(after->first);
199 // eDebug("%08llx .. ? .. %08llx", before_ts, after_ts);
200 // eDebug("%08llx .. %08llx .. %08llx", before->first, offset, after->first);
202 pts_t diff = after_ts - before_ts;
203 off_t diff_off = after->first - before->first;
205 diff = (offset - before->first) * diff / diff_off;
206 // eDebug("%08llx .. %08llx .. %08llx", before_ts, before_ts + diff, after_ts);
207 return before_ts + diff;
210 off_t eMPEGStreamInformation::getAccessPoint(pts_t ts, int marg)
212 /* FIXME: more efficient implementation */
216 for (std::map<off_t, pts_t>::const_iterator i(m_access_points.begin()); i != m_access_points.end(); ++i)
218 pts_t delta = getDelta(i->first);
219 pts_t c = i->second - delta;
222 return (last + i->first)/376*188;
224 return (last + last2)/376*188;
233 return (last + last2)/376*188;
238 int eMPEGStreamInformation::getNextAccessPoint(pts_t &ts, const pts_t &start, int direction)
240 off_t offset = getAccessPoint(start);
242 std::map<off_t, pts_t>::const_iterator i = m_access_points.find(offset);
243 if (i == m_access_points.end())
245 eDebug("getNextAccessPoint: initial AP not found");
248 c1 = i->second - getDelta(i->first);
253 if (i == m_access_points.end())
256 c2 = i->second - getDelta(i->first);
257 if (c1 == c2) { // Discontinuity
259 c2 = i->second - getDelta(i->first);
266 if (i == m_access_points.begin())
272 c2 = i->second - getDelta(i->first);
273 if (c1 == c2) { // Discontinuity
275 c2 = i->second - getDelta(i->first);
281 ts = i->second - getDelta(i->first);
282 eDebug("fine, at %llx - %llx = %llx", ts, i->second, getDelta(i->first));
283 eDebug("fine, at %lld - %lld = %lld", ts, i->second, getDelta(i->first));
287 eMPEGStreamParserTS::eMPEGStreamParserTS(eMPEGStreamInformation &streaminfo): m_streaminfo(streaminfo), m_pktptr(0), m_pid(-1), m_need_next_packet(0), m_skip(0), m_last_pts_valid(0)
291 int eMPEGStreamParserTS::processPacket(const unsigned char *pkt, off_t offset)
293 if (!wantPacket(pkt))
294 eWarning("something's wrong.");
296 const unsigned char *end = pkt + 188;
298 if (!(pkt[3] & 0x10))
300 eWarning("[TSPARSE] PUSI set but no payload.");
304 if (pkt[3] & 0x20) // adaption field present?
305 pkt += pkt[4] + 4 + 1; /* skip adaption field and header */
307 pkt += 4; /* skip header */
311 eWarning("[TSPARSE] dropping huge adaption field");
315 // ok, we now have the start of the payload, aligned with the PES packet start.
316 if (pkt[0] || pkt[1] || (pkt[2] != 1))
318 eWarning("broken startcode");
326 if (pkt[7] & 0x80) // PTS present?
328 pts = ((unsigned long long)(pkt[ 9]&0xE)) << 29;
329 pts |= ((unsigned long long)(pkt[10]&0xFF)) << 22;
330 pts |= ((unsigned long long)(pkt[11]&0xFE)) << 14;
331 pts |= ((unsigned long long)(pkt[12]&0xFF)) << 7;
332 pts |= ((unsigned long long)(pkt[13]&0xFE)) >> 1;
336 m_last_pts_valid = 1;
339 int sec = pts / 90000;
340 int frm = pts % 90000;
348 eDebug("pts: %016llx %d:%02d:%02d:%02d:%05d", pts, d, hr, min, sec, frm);
352 /* advance to payload */
355 /* sometimes, there are zeros before the startcode. */
356 while (pkt < (end-4))
357 if (pkt[0] || pkt[1] || pkt[2])
362 /* if startcode found */
363 // eDebug("%02x %02x %02x %02x", pkt[0], pkt[1], pkt[2], pkt[3]);
364 if (!(pkt[0] || pkt[1] || (pkt[2] != 1)))
366 if (pkt[3] == 0xb3) /* sequence header */
370 m_streaminfo.m_access_points[offset] = pts;
371 // eDebug("Sequence header at %llx, pts %llx", offset, pts);
373 /*eDebug("Sequence header but no valid PTS value.")*/;
376 if (pkt[3] == 0x09 && /* MPEG4 AVC unit access delimiter */
377 (pkt[4] >> 5) == 0) /* and I-frame */
381 m_streaminfo.m_access_points[offset] = pts;
382 // eDebug("MPEG4 AVC UAD at %llx, pts %llx", offset, pts);
384 /*eDebug("MPEG4 AVC UAD but no valid PTS value.")*/;
390 inline int eMPEGStreamParserTS::wantPacket(const unsigned char *hdr) const
394 eDebug("missing sync!");
397 int ppid = ((hdr[1]&0x1F) << 8) | hdr[2];
402 if (m_need_next_packet) /* next packet (on this pid) was required? */
405 if (hdr[1] & 0x40) /* pusi set: yes. */
411 void eMPEGStreamParserTS::parseData(off_t offset, const void *data, unsigned int len)
413 const unsigned char *packet = (const unsigned char*)data;
414 const unsigned char *packet_start = packet;
416 /* sorry for the redundant code here, but there are too many special cases... */
419 /* emergency resync. usually, this should not happen, because the data should
422 to make this code work for non-strictly-sync-aligned data, (for example, bad
423 files) we fix a possible resync here by skipping data until the next 0x47.
425 if this is a false 0x47, the packet will be dropped by wantPacket, and the
426 next time, sync will be re-established. */
428 while (!m_pktptr && len)
430 if (packet[0] == 0x47)
438 eDebug("SYNC LOST: skipped %d bytes.", skipped);
445 /* skip last packet */
448 unsigned int skiplen = -m_pktptr;
455 } else if (m_pktptr < 4) /* header not complete, thus we don't know if we want this packet */
457 unsigned int storelen = 4 - m_pktptr;
460 memcpy(m_pkt + m_pktptr, packet, storelen);
462 m_pktptr += storelen;
467 if (!wantPacket(m_pkt))
476 /* otherwise we complete up to the full packet */
477 unsigned int storelen = 188 - m_pktptr;
480 memcpy(m_pkt + m_pktptr, packet, storelen);
481 m_pktptr += storelen;
487 m_need_next_packet = processPacket(m_pkt, offset + (packet - packet_start));
490 } else if (len >= 4) /* if we have a full header... */
492 if (wantPacket(packet)) /* decide wheter we need it ... */
494 if (len >= 188) /* packet complete? */
496 m_need_next_packet = processPacket(packet, offset + (packet - packet_start)); /* process it now. */
499 memcpy(m_pkt, packet, len); /* otherwise queue it up */
508 else if (!m_pktptr) /* we dont want this packet, otherwise m_pktptr = sk (=len) > 4 */
513 } else /* if we don't have a complete header */
515 memcpy(m_pkt, packet, len); /* complete header next time */
523 void eMPEGStreamParserTS::setPid(int _pid)
529 int eMPEGStreamParserTS::getLastPTS(pts_t &last_pts)
531 if (!m_last_pts_valid)
536 last_pts = m_last_pts;