Chrome-side implementation of media source timestamp offset

media/filters/chunk_demuxer.cc

Index: media/filters/chunk_demuxer.cc
diff --git a/media/filters/chunk_demuxer.cc b/media/filters/chunk_demuxer.cc
index e28a2d313062f98d03e01817df9ba54d39e2db9c..ea8f96cab08a8b89a569323f4ddbf115370651e2 100644
--- a/media/filters/chunk_demuxer.cc
+++ b/media/filters/chunk_demuxer.cc
@@ -111,7 +111,7 @@ static bool IsSupported(const std::string& type,
   *has_audio = false;
   *has_video = false;
 
-  // Search for the SupportedTypeInfo for |type|
+  // Search for the SupportedTypeInfo for |type|.
   for (size_t i = 0; i < arraysize(kSupportedTypeInfo); ++i) {
     const SupportedTypeInfo& type_info = kSupportedTypeInfo[i];
     if (type == type_info.type) {
@@ -661,9 +661,13 @@ ChunkDemuxer::Status ChunkDemuxer::AddId(const std::string& id,
       audio_cb,
       video_cb,
       base::Bind(&ChunkDemuxer::OnNeedKey, base::Unretained(this)),
-      base::Bind(&ChunkDemuxer::OnNewMediaSegment, base::Unretained(this), id));
+      base::Bind(&ChunkDemuxer::OnNewMediaSegment, base::Unretained(this), id),
+      base::Bind(&ChunkDemuxer::OnEndOfMediaSegment,
+                 base::Unretained(this), id));
 
   stream_parser_map_[id] = stream_parser.release();
+  SourceInfo info = { base::TimeDelta(), true };
+  source_info_map_[id] = info;
 
   return kOk;
 }
@@ -674,6 +678,7 @@ void ChunkDemuxer::RemoveId(const std::string& id) {
 
   delete stream_parser_map_[id];
   stream_parser_map_.erase(id);
+  source_info_map_.erase(id);
 
   if (source_id_audio_ == id && audio_)
     audio_->Shutdown();
@@ -809,6 +814,20 @@ void ChunkDemuxer::Abort(const std::string& id) {
   stream_parser_map_[id]->Flush();
 }
 
+bool ChunkDemuxer::SetTimestampOffset(const std::string& id, float offset) {
+  DVLOG(1) << "TimestampOffset(" << id << ", " << offset << ")";

[acolwell GONE FROM CHROMIUM, 2012/07/25 18:22:43]

nit: Add Set

[vrk (LEFT CHROMIUM), 2012/07/25 22:56:50]

Done.

+  DCHECK_GT(stream_parser_map_.count(id), 0u);
+
+  TimeDelta time_offset = TimeDelta::FromMicroseconds(
+      offset * base::Time::kMicrosecondsPerSecond);
+
+  if (!source_info_map_[id].can_update_offset)

[acolwell GONE FROM CHROMIUM, 2012/07/25 18:22:43]

nit: move this above time_offset since we don't need to compute this if we can't
update the offset.

[vrk (LEFT CHROMIUM), 2012/07/25 22:56:50]

Done.

+    return false;
+
+  source_info_map_[id].timestamp_offset = time_offset;
+  return true;
+}
+
 bool ChunkDemuxer::EndOfStream(PipelineStatus status) {
   DVLOG(1) << "EndOfStream(" << status << ")";
   base::AutoLock auto_lock(lock_);
@@ -1013,6 +1032,10 @@ bool ChunkDemuxer::OnAudioBuffers(const StreamParser::BufferQueue& buffers) {
   if (!audio_)
     return false;
 
+  CHECK_GT(source_info_map_.count(source_id_audio_), 0u);
+  AdjustBufferTimestamps(
+      buffers, source_info_map_[source_id_audio_].timestamp_offset);
+
   return audio_->Append(buffers);
 }
 
@@ -1023,6 +1046,10 @@ bool ChunkDemuxer::OnVideoBuffers(const StreamParser::BufferQueue& buffers) {
   if (!video_)
     return false;
 
+  CHECK_GT(source_info_map_.count(source_id_video_), 0u);
+  AdjustBufferTimestamps(
+      buffers, source_info_map_[source_id_video_].timestamp_offset);
+
   return video_->Append(buffers);
 }
 
@@ -1033,11 +1060,16 @@ bool ChunkDemuxer::OnNeedKey(scoped_array<uint8> init_data,
 }
 
 void ChunkDemuxer::OnNewMediaSegment(const std::string& source_id,
-                                     TimeDelta start_timestamp) {
+                                     TimeDelta timestamp) {
   DVLOG(2) << "OnNewMediaSegment(" << source_id << ", "
-           << start_timestamp.InSecondsF() << ")";
+           << timestamp.InSecondsF() << ")";
   lock_.AssertAcquired();
 
+  CHECK_GT(source_info_map_.count(source_id), 0u);

[acolwell GONE FROM CHROMIUM, 2012/07/25 18:22:43]

nit: Create a bool IsValidId(source_id) helper so we don't have this count()
construct everywhere?

[vrk (LEFT CHROMIUM), 2012/07/25 22:56:50]

Done.

+  source_info_map_[source_id].can_update_offset = false;
+  base::TimeDelta start_timestamp =
+      timestamp + source_info_map_[source_id].timestamp_offset;
+
   if (start_time_ == kNoTimestamp()) {
     DCHECK(state_ == INITIALIZING || state_ == WAITING_FOR_START_TIME);
     // Use the first reported media segment start time as the |start_time_|
@@ -1067,4 +1099,24 @@ void ChunkDemuxer::OnNewMediaSegment(const std::string& source_id,
   base::ResetAndReturn(&init_cb_).Run(PIPELINE_OK);
 }
 
+void ChunkDemuxer::OnEndOfMediaSegment(const std::string& source_id) {
+  DVLOG(2) << "OnEndOfMediaSegment(" << source_id << ")";
+  CHECK_GT(source_info_map_.count(source_id), 0u);
+  source_info_map_[source_id].can_update_offset = true;
+}
+
+void ChunkDemuxer::AdjustBufferTimestamps(
+    const StreamParser::BufferQueue& buffers,
+    base::TimeDelta timestamp_offset) {
+  if (timestamp_offset == base::TimeDelta())
+    return;
+
+  for (StreamParser::BufferQueue::const_iterator itr = buffers.begin();
+       itr != buffers.end(); ++itr) {
+    (*itr)->SetDecodeTimestamp(
+        (*itr)->GetDecodeTimestamp() + timestamp_offset);
+    (*itr)->SetTimestamp((*itr)->GetTimestamp() + timestamp_offset);
+  }
+}
+
 }  // namespace media