MediaCodecPlayer implementation (stage 4 - preroll)

media/base/android/test_data_factory.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 "media/base/android/test_data_factory.h"
 
 #include "base/strings/stringprintf.h"
 #include "media/base/android/demuxer_stream_player_params.h"
 #include "media/base/decoder_buffer.h"
 #include "media/base/test_data_util.h"
@@ skipping 42 matching lines @@
   configs.duration = duration;
 
   return configs;
 }
 
 TestDataFactory::TestDataFactory(const char* file_name_template,
                                  base::TimeDelta duration,
                                  base::TimeDelta frame_period)
     : duration_(duration),
       frame_period_(frame_period),
-      starvation_mode_(false) {
+      starvation_mode_(false),
+      eos_reached_(false) {
   LoadPackets(file_name_template);
 }
 
 TestDataFactory::~TestDataFactory() {}
 
 bool TestDataFactory::CreateChunk(DemuxerData* chunk, base::TimeDelta* delay) {
   DCHECK(chunk);
   DCHECK(delay);
 
+  if (eos_reached_)
+    return false;
+
   *delay = base::TimeDelta();
 
   for (int i = 0; i < 4; ++i) {
     chunk->access_units.push_back(AccessUnit());
     AccessUnit& unit = chunk->access_units.back();
     unit.status = DemuxerStream::kOk;
 
     unit.timestamp = regular_pts_;
     regular_pts_ += frame_period_;
 
     if (unit.timestamp > duration_) {
       if (starvation_mode_)
         return false;
 
       unit.is_end_of_stream = true;
+      eos_reached_ = true;
       break;  // EOS units have no data.
     }
 
     unit.data = packet_[i];
 
     // Allow for modification by subclasses.
     ModifyAccessUnit(i, &unit);
 
     // Maintain last PTS. ModifyAccessUnit() can modify unit's PTS.
     if (last_pts_ < unit.timestamp)
       last_pts_ = unit.timestamp;
   }
 
   return true;
 }
 
 void TestDataFactory::SeekTo(const base::TimeDelta& seek_time) {
   regular_pts_ = seek_time;
   last_pts_ = base::TimeDelta();
+  eos_reached_ = false;
 }
 
 void TestDataFactory::LoadPackets(const char* file_name_template) {
   for (int i = 0; i < 4; ++i) {
     scoped_refptr<DecoderBuffer> buffer =
         ReadTestDataFile(base::StringPrintf(file_name_template, i));
     packet_[i] = std::vector<uint8>(buffer->data(),
                                     buffer->data() + buffer->data_size());
   }
 }
 
 }  // namespace media