Allow setting memory limits on media::DemuxerStream

content/renderer/media/android/media_source_delegate.cc

 // Copyright 2013 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 "content/renderer/media/android/media_source_delegate.h"
 
 #include <limits>
 #include <string>
 #include <vector>
 
@@ skipping 186 matching lines @@
 }
 
 size_t MediaSourceDelegate::AudioDecodedByteCount() const {
   return statistics_.audio_bytes_decoded;
 }
 
 size_t MediaSourceDelegate::VideoDecodedByteCount() const {
   return statistics_.video_bytes_decoded;
 }
 
+size_t MediaSourceDelegate::getAudioBufferSize() const {
+  media::DemuxerStream* stream = nullptr;
+  if (chunk_demuxer_)
+    stream = chunk_demuxer_->GetStream(DemuxerStream::AUDIO);
+  if (stream)
+    return stream->GetMemoryLimit();
+  return 0;
+}
+
+void MediaSourceDelegate::setAudioBufferSize(size_t size) {
+  media::DemuxerStream* stream = nullptr;
+  if (chunk_demuxer_)
+    stream = chunk_demuxer_->GetStream(DemuxerStream::AUDIO);
+  if (stream)
+    return stream->SetMemoryLimit(size);
+}
+
+size_t MediaSourceDelegate::getVideoBufferSize() const {
+  media::DemuxerStream* stream = nullptr;
+  if (chunk_demuxer_)
+    stream = chunk_demuxer_->GetStream(DemuxerStream::VIDEO);
+  if (stream)
+    return stream->GetMemoryLimit();
+  return 0;
+}
+
+void MediaSourceDelegate::setVideoBufferSize(size_t size) {
+  media::DemuxerStream* stream = nullptr;
+  if (chunk_demuxer_)
+    stream = chunk_demuxer_->GetStream(DemuxerStream::VIDEO);
+  if (stream)
+    return stream->SetMemoryLimit(size);
+}
+
 void MediaSourceDelegate::CancelPendingSeek(const base::TimeDelta& seek_time) {
   DCHECK(main_task_runner_->BelongsToCurrentThread());
   DVLOG(1) << __FUNCTION__ << "(" << seek_time.InSecondsF() << ") : "
            << demuxer_client_id_;
 
   if (!chunk_demuxer_)
     return;
 
   {
     // Remember to trivially finish any newly arriving browser seek requests
@@ skipping 537 matching lines @@
     configs->video_size = config.natural_size();
     configs->is_video_encrypted = config.is_encrypted();
     configs->video_extra_data = std::vector<uint8>(
         config.extra_data(), config.extra_data() + config.extra_data_size());
     return true;
   }
   return false;
 }
 
 }  // namespace content