Bug 1330918 - Use timestamps for frames in VideoTrackEncoder. r?jesup, r?bechen
This makes VideoTrackEncoder use timestamps when it passes frames to
VP8TrackEncoder. It also rewrites the chunks' durations and bases them on said
timestamps. This should mean that VP8TrackEncoder can continue passing durations
to the encoder as it does today.
MozReview-Commit-ID: GaUsF5PR4ZN
--- a/dom/media/encoder/TrackEncoder.cpp
+++ b/dom/media/encoder/TrackEncoder.cpp
@@ -274,44 +274,83 @@ VideoTrackEncoder::NotifyQueuedTrackChan
nsresult
VideoTrackEncoder::AppendVideoSegment(const VideoSegment& aSegment)
{
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
// Append all video segments from MediaStreamGraph, including null an
// non-null frames.
- VideoSegment::ChunkIterator iter(const_cast<VideoSegment&>(aSegment));
- while (!iter.IsEnded()) {
+ VideoSegment::ConstChunkIterator iter(aSegment);
+ for (; !iter.IsEnded(); iter.Next()) {
VideoChunk chunk = *iter;
- mLastFrameDuration += chunk.GetDuration();
- // Send only the unique video frames for encoding.
- // Or if we got the same video chunks more than 1 seconds,
- // force to send into encoder.
- if ((mLastFrame != chunk.mFrame) ||
- (mLastFrameDuration >= mTrackRate)) {
- RefPtr<layers::Image> image = chunk.mFrame.GetImage();
+
+ if (mLastChunk.mTimeStamp.IsNull()) {
+ if (chunk.IsNull()) {
+ // The start of this track is frameless. We need to track the time
+ // it takes to get the first frame.
+ mLastChunk.mDuration += chunk.mDuration;
+ continue;
+ }
+
+ // This is the first real chunk in the track. Use its timestamp as the
+ // starting point for this track.
+ MOZ_ASSERT(!chunk.mTimeStamp.IsNull());
+ const StreamTime nullDuration = mLastChunk.mDuration;
+ mLastChunk = chunk;
- // Because we may get chunks with a null image (due to input blocking),
- // accumulate duration and give it to the next frame that arrives.
- // Canonically incorrect - the duration should go to the previous frame
- // - but that would require delaying until the next frame arrives.
- // Best would be to do like OMXEncoder and pass an effective timestamp
- // in with each frame.
- if (image) {
- mRawSegment.AppendFrame(image.forget(),
- mLastFrameDuration,
- chunk.mFrame.GetIntrinsicSize(),
- PRINCIPAL_HANDLE_NONE,
- chunk.mFrame.GetForceBlack());
- mLastFrameDuration = 0;
+ // Adapt to the time before the first frame. This extends the first frame
+ // from [start, end] to [0, end], but it'll do for now.
+ mLastChunk.mTimeStamp -=
+ TimeDuration::FromMicroseconds(
+ RateConvertTicksRoundUp(PR_USEC_PER_SEC, mTrackRate, nullDuration));
+ }
+
+ MOZ_ASSERT(!mLastChunk.IsNull());
+ if (mLastChunk.CanCombineWithFollowing(chunk) || chunk.IsNull()) {
+ // This is the same frame as before (or null). We extend the last chunk
+ // with its duration.
+ mLastChunk.mDuration += chunk.mDuration;
+
+ if (mLastChunk.mDuration < mTrackRate) {
+ continue;
+ }
+
+ // If we have gotten dupes for over a second, we force send one
+ // to the encoder to make sure there is some output.
+ chunk.mTimeStamp = mLastChunk.mTimeStamp + TimeDuration::FromSeconds(1);
+
+ if (chunk.IsNull()) {
+ // Ensure that we don't pass null to the encoder by making mLastChunk
+ // null later on.
+ chunk.mFrame = mLastChunk.mFrame;
}
}
- mLastFrame.TakeFrom(&chunk.mFrame);
- iter.Next();
+
+ TimeDuration diff = chunk.mTimeStamp - mLastChunk.mTimeStamp;
+ if (diff <= TimeDuration::FromSeconds(0)) {
+ // The timestamp from mLastChunk is newer than from chunk.
+ // This means the durations reported from MediaStreamGraph for mLastChunk
+ // were larger than the timestamp diff - and durations were used to
+ // trigger the 1-second frame above. This could happen due to drift or
+ // underruns in the graph.
+ chunk.mTimeStamp = mLastChunk.mTimeStamp;
+ } else {
+ RefPtr<layers::Image> lastImage = mLastChunk.mFrame.GetImage();
+ mRawSegment.AppendFrame(lastImage.forget(),
+ RateConvertTicksRoundUp(
+ mTrackRate, PR_USEC_PER_SEC,
+ diff.ToMicroseconds()),
+ mLastChunk.mFrame.GetIntrinsicSize(),
+ PRINCIPAL_HANDLE_NONE,
+ mLastChunk.mFrame.GetForceBlack(),
+ mLastChunk.mTimeStamp);
+ }
+
+ mLastChunk = chunk;
}
if (mRawSegment.GetDuration() > 0) {
mReentrantMonitor.NotifyAll();
}
return NS_OK;
}
--- a/dom/media/encoder/TrackEncoder.h
+++ b/dom/media/encoder/TrackEncoder.h
@@ -250,20 +250,20 @@ class VideoTrackEncoder : public TrackEn
public:
explicit VideoTrackEncoder(TrackRate aTrackRate)
: TrackEncoder()
, mFrameWidth(0)
, mFrameHeight(0)
, mDisplayWidth(0)
, mDisplayHeight(0)
, mTrackRate(aTrackRate)
- , mTotalFrameDuration(0)
- , mLastFrameDuration(0)
, mVideoBitrate(0)
- {}
+ {
+ mLastChunk.mDuration = 0;
+ }
/**
* Notified by the same callback of MediaEncoder when it has received a track
* change from MediaStreamGraph. Called on the MediaStreamGraph thread.
*/
void NotifyQueuedTrackChanges(MediaStreamGraph* aGraph, TrackID aID,
StreamTime aTrackOffset,
uint32_t aTrackEvents,
@@ -334,27 +334,20 @@ protected:
int mDisplayHeight;
/**
* The track rate of source video.
*/
TrackRate mTrackRate;
/**
- * The total duration of frames in encoded video in StreamTime, kept track of
- * in subclasses.
- */
- StreamTime mTotalFrameDuration;
-
- /**
* The last unique frame and duration we've sent to track encoder,
* kept track of in subclasses.
*/
- VideoFrame mLastFrame;
- StreamTime mLastFrameDuration;
+ VideoChunk mLastChunk;
/**
* A segment queue of audio track data, protected by mReentrantMonitor.
*/
VideoSegment mRawSegment;
uint32_t mVideoBitrate;
};