From 323969e58df323ddce392be9abee5fd4019b79fc Mon Sep 17 00:00:00 2001 From: Klaus Schmidinger Date: Fri, 27 Mar 2009 13:38:59 +0100 Subject: Adapted cFrameDetector::Analyze() to HD NTSC broadcasts that split frames over several payload units --- remux.c | 129 +++++++++++++++++++++++++++++++++++++++++++--------------------- 1 file changed, 88 insertions(+), 41 deletions(-) (limited to 'remux.c') diff --git a/remux.c b/remux.c index a22c024b..2a1f7a94 100644 --- a/remux.c +++ b/remux.c @@ -4,7 +4,7 @@ * See the main source file 'vdr.c' for copyright information and * how to reach the author. * - * $Id: remux.c 2.14 2009/03/08 12:12:17 kls Exp $ + * $Id: remux.c 2.15 2009/03/27 13:32:11 kls Exp $ */ #include "remux.h" @@ -676,15 +676,25 @@ cFrameDetector::cFrameDetector(int Pid, int Type) { pid = Pid; type = Type; + synced = false; newFrame = independentFrame = false; - lastPts = 0; + numPtsValues = 0; + numIFrames = 0; isVideo = type == 0x02 || type == 0x1B; // MPEG 2 or MPEG 4 frameDuration = 0; - framesPerPayloadUnit = 0; + framesInPayloadUnit = framesPerPayloadUnit = 0; + payloadUnitOfFrame = 0; scanning = false; scanner = 0; } +static int CmpUint32(const void *p1, const void *p2) +{ + if (*(uint32_t *)p1 < *(uint32_t *)p2) return -1; + if (*(uint32_t *)p1 > *(uint32_t *)p2) return 1; + return 0; +} + int cFrameDetector::Analyze(const uchar *Data, int Length) { newFrame = independentFrame = false; @@ -692,31 +702,47 @@ int cFrameDetector::Analyze(const uchar *Data, int Length) if (TsHasPayload(Data) && !TsIsScrambled(Data) && TsPid(Data) == pid) { if (TsPayloadStart(Data)) { if (!frameDuration) { - const uchar *Pes = Data + TsPayloadOffset(Data); - if (PesHasPts(Pes)) { - int64_t Pts = PesGetPts(Pes); - if (Pts < lastPts) { // avoid wrapping - lastPts = 0; - framesPerPayloadUnit = 0; + // frame duration unknown, so collect a sequenece of PTS values: + if (numPtsValues < MaxPtsValues && numIFrames < 2) { // collect a sequence containing at least two I-frames + const uchar *Pes = Data + TsPayloadOffset(Data); + if (PesHasPts(Pes)) { + ptsValues[numPtsValues] = PesGetPts(Pes); + // check for rollover: + if (numPtsValues && ptsValues[numPtsValues - 1] > 0xF0000000 && ptsValues[numPtsValues] < 0x10000000) { + dbgframes("#"); + numPtsValues = 0; + numIFrames = 0; + } + else + numPtsValues++; } - if ((!lastPts || !framesPerPayloadUnit) && Pts != lastPts) - lastPts = Pts; - else { - int64_t Delta = Pts - lastPts; - if (isVideo) { - if (Delta % 3600 == 0) - frameDuration = 3600; // PAL, 25 fps - else if (Delta % 3003 == 0) - frameDuration = 3003; // NTSC, 29.97 fps - else { - frameDuration = 3600; // unknown, assuming 25 fps - dsyslog("unknown frame duration, assuming 25 fps (PTS: %lld - %lld = %lld FPPU = %d)\n", Pts, lastPts, Delta, framesPerPayloadUnit); - } + } + else { + // find the smallest PTS delta: + qsort(ptsValues, numPtsValues, sizeof(uint32_t), CmpUint32); + numPtsValues--; + for (int i = 0; i < numPtsValues; i++) + ptsValues[i] = ptsValues[i + 1] - ptsValues[i]; + qsort(ptsValues, numPtsValues, sizeof(uint32_t), CmpUint32); + uint32_t Delta = ptsValues[0]; + // determine frame info: + if (isVideo) { + if (Delta % 3600 == 0) + frameDuration = 3600; // PAL, 25 fps + else if (Delta % 3003 == 0) + frameDuration = 3003; // NTSC, 29.97 fps + else if (Delta == 1501) { + frameDuration = 3003; // NTSC, 29.97 fps + framesPerPayloadUnit = -2; + } + else { + frameDuration = 3600; // unknown, assuming 25 fps + dsyslog("unknown frame duration (%d), assuming 25 fps", Delta); } - else // audio - frameDuration = Delta; // PTS of audio frames is always increasing - dbgframes("PTS: %lld - %lld = %lld -> FD = %d FPS = %5.2f FPPU = %d\n", Pts, lastPts, Delta, frameDuration, 90000.0 / frameDuration, framesPerPayloadUnit); } + else // audio + frameDuration = Delta; // PTS of audio frames is always increasing + dbgframes("\nframe duration = %d FPS = %5.2f FPPU = %d\n", frameDuration, 90000.0 / frameDuration, framesPerPayloadUnit); } } scanner = 0; @@ -724,24 +750,31 @@ int cFrameDetector::Analyze(const uchar *Data, int Length) } if (scanning) { int PayloadOffset = TsPayloadOffset(Data); - if (TsPayloadStart(Data)) + if (TsPayloadStart(Data)) { PayloadOffset += PesPayloadOffset(Data + PayloadOffset); + if (!framesPerPayloadUnit) + framesPerPayloadUnit = framesInPayloadUnit; + if (DebugFrames && !synced) + dbgframes("/"); + } for (int i = PayloadOffset; i < TS_SIZE; i++) { scanner <<= 8; scanner |= Data[i]; switch (type) { case 0x02: // MPEG 2 video if (scanner == 0x00000100) { // Picture Start Code - if (frameDuration) { - newFrame = true; - independentFrame = ((Data[i + 2] >> 3) & 0x07) == 1; // I-Frame - if (framesPerPayloadUnit == 1) { + newFrame = true; + independentFrame = ((Data[i + 2] >> 3) & 0x07) == 1; // I-Frame + if (synced) { + if (framesPerPayloadUnit <= 1) { scanning = false; return TS_SIZE; } } else { - framesPerPayloadUnit++; + framesInPayloadUnit++; + if (independentFrame) + numIFrames++; dbgframes("%d ", (Data[i + 2] >> 3) & 0x07); } scanner = 0; @@ -749,16 +782,25 @@ int cFrameDetector::Analyze(const uchar *Data, int Length) break; case 0x1B: // MPEG 4 video if (scanner == 0x00000109) { // Access Unit Delimiter - if (frameDuration) { - newFrame = true; - independentFrame = Data[i + 1] == 0x10; - if (framesPerPayloadUnit == 1) { + newFrame = true; + independentFrame = Data[i + 1] == 0x10; + if (synced) { + if (framesPerPayloadUnit < 0) { + payloadUnitOfFrame = (payloadUnitOfFrame + 1) % -framesPerPayloadUnit; + if (payloadUnitOfFrame != 0 && independentFrame) + payloadUnitOfFrame = 0; + if (payloadUnitOfFrame) + newFrame = false; + } + if (framesPerPayloadUnit <= 1) { scanning = false; return TS_SIZE; } } else { - framesPerPayloadUnit++; + framesInPayloadUnit++; + if (independentFrame) + numIFrames++; dbgframes("%02X ", Data[i + 1]); } scanner = 0; @@ -766,18 +808,23 @@ int cFrameDetector::Analyze(const uchar *Data, int Length) break; case 0x04: // MPEG audio case 0x06: // AC3 audio - if (frameDuration) { - newFrame = true; - independentFrame = true; + newFrame = true; + independentFrame = true; + if (synced) scanning = false; + else { + framesInPayloadUnit = 1; + numIFrames++; } - else - framesPerPayloadUnit = 1; break; default: esyslog("ERROR: unknown stream type %d (PID %d) in frame detector", type, pid); pid = 0; // let's just ignore any further data } } + if (!synced && frameDuration && independentFrame) { + synced = true; + dbgframes("*"); + } } } return TS_SIZE; -- cgit v1.2.3