MediaCodecPlayer implementation (stage 4 - preroll)

media/base/android/media_codec_decoder_unittest.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/thread_task_runner_handle.h"
 #include "base/timer/timer.h"
 #include "media/base/android/media_codec_audio_decoder.h"
 #include "media/base/android/media_codec_bridge.h"
@@ skipping 131 matching lines @@
   void CreateAudioDecoder();
   void CreateVideoDecoder();
   void SetVideoSurface();
   void SetStopRequestAtTime(const base::TimeDelta& time) {
     stop_request_time_ = time;
   }
 
   // Decoder callbacks.
   void OnDataRequested();
   void OnStarvation() { is_starved_ = true; }
+  void OnPrerollDone() { decoder_->ResumeAfterPreroll(); }
   void OnStopDone() { is_stopped_ = true; }
   void OnError() {}
   void OnUpdateCurrentTime(base::TimeDelta now_playing,
                            base::TimeDelta last_buffered) {
     // Add the |last_buffered| value for PTS. For video it is the same as
     // |now_playing| and is equal to PTS, for audio |last_buffered| should
     // exceed PTS.
     pts_stat_.AddValue(last_buffered);
 
     if (stop_request_time_ != kNoTimestamp() &&
@@ skipping 62 matching lines @@
 
   DCHECK(!timer.IsRunning());
   return false;
 }
 
 void MediaCodecDecoderTest::CreateAudioDecoder() {
   decoder_ = scoped_ptr<MediaCodecDecoder>(new MediaCodecAudioDecoder(
       task_runner_, base::Bind(&MediaCodecDecoderTest::OnDataRequested,
                                base::Unretained(this)),
       base::Bind(&MediaCodecDecoderTest::OnStarvation, base::Unretained(this)),
+      base::Bind(&MediaCodecDecoderTest::OnPrerollDone, base::Unretained(this)),
       base::Bind(&MediaCodecDecoderTest::OnStopDone, base::Unretained(this)),
       base::Bind(&MediaCodecDecoderTest::OnError, base::Unretained(this)),
       base::Bind(&MediaCodecDecoderTest::OnUpdateCurrentTime,
                  base::Unretained(this))));
 
   data_available_cb_ = base::Bind(&MediaCodecDecoder::OnDemuxerDataAvailable,
                                   base::Unretained(decoder_.get()));
 }
 
 void MediaCodecDecoderTest::CreateVideoDecoder() {
   decoder_ = scoped_ptr<MediaCodecDecoder>(new MediaCodecVideoDecoder(
       task_runner_, base::Bind(&MediaCodecDecoderTest::OnDataRequested,
                                base::Unretained(this)),
       base::Bind(&MediaCodecDecoderTest::OnStarvation, base::Unretained(this)),
+      base::Bind(&MediaCodecDecoderTest::OnPrerollDone, base::Unretained(this)),
       base::Bind(&MediaCodecDecoderTest::OnStopDone, base::Unretained(this)),
       base::Bind(&MediaCodecDecoderTest::OnError, base::Unretained(this)),
       base::Bind(&MediaCodecDecoderTest::OnUpdateCurrentTime,
                  base::Unretained(this)),
       base::Bind(&MediaCodecDecoderTest::OnVideoSizeChanged,
                  base::Unretained(this)),
       base::Bind(&MediaCodecDecoderTest::OnVideoCodecCreated,
                  base::Unretained(this))));
 
   data_available_cb_ = base::Bind(&MediaCodecDecoder::OnDemuxerDataAvailable,
@@ skipping 225 matching lines @@
   EXPECT_TRUE(decoder_->Start(base::TimeDelta::FromMilliseconds(0)));
 
   EXPECT_TRUE(WaitForCondition(
       base::Bind(&MediaCodecDecoderTest::is_stopped, base::Unretained(this)),
       timeout));
 
   EXPECT_TRUE(decoder_->IsStopped());
   EXPECT_TRUE(decoder_->IsCompleted());
 
   // Last buffered timestamp should be no less than PTS.
-  EXPECT_EQ(22, pts_stat_.num_values());
+  // The number of hits in pts_stat_ depends on the preroll implementation.
+  // We might not report the time for the first buffer after preroll that
+  // is written to the audio track. pts_stat_.num_values() is either 21 or 22.
+  EXPECT_LE(21, pts_stat_.num_values());
   EXPECT_LE(data_factory_->last_pts(), pts_stat_.max());
 }
 
 TEST_F(MediaCodecDecoderTest, VideoPlayTillCompletion) {
   SKIP_TEST_IF_MEDIA_CODEC_BRIDGE_IS_NOT_AVAILABLE();
 
   CreateVideoDecoder();
 
   base::TimeDelta duration = base::TimeDelta::FromMilliseconds(500);
   // The first output frame might come out with significant delay. Apparently
@@ skipping 137 matching lines @@
   decoder_->RequestToStop();
 
   EXPECT_TRUE(WaitForCondition(
       base::Bind(&MediaCodecDecoderTest::is_stopped, base::Unretained(this))));
 
   EXPECT_TRUE(decoder_->IsStopped());
   EXPECT_FALSE(decoder_->IsCompleted());
 }
 
 }  // namespace media