Allow setting memory limits on media::DemuxerStream

media/blink/webmediaplayer_impl.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 "media/blink/webmediaplayer_impl.h"
 
 #include <algorithm>
 #include <cmath>
 #include <limits>
 
@@ skipping 577 matching lines @@
   return stats.audio_bytes_decoded;
 }
 
 unsigned WebMediaPlayerImpl::videoDecodedByteCount() const {
   DCHECK(main_task_runner_->BelongsToCurrentThread());
 
   PipelineStatistics stats = pipeline_.GetStatistics();
   return stats.video_bytes_decoded;
 }
 
+size_t WebMediaPlayerImpl::getAudioBufferSize() const {
+  if (demuxer_ && demuxer_->GetStream(DemuxerStream::AUDIO)) {
+    return demuxer_->GetStream(DemuxerStream::AUDIO)->GetMemoryLimit();
+  }
+  return 0;
+}
+
+void WebMediaPlayerImpl::setAudioBufferSize(size_t size) {
+  if (demuxer_ && demuxer_->GetStream(DemuxerStream::AUDIO)) {
+    demuxer_->GetStream(DemuxerStream::AUDIO)->SetMemoryLimit(size);
+  }
+}
+
+size_t WebMediaPlayerImpl::getVideoBufferSize() const {
+  if (demuxer_ && demuxer_->GetStream(DemuxerStream::VIDEO)) {
+    return demuxer_->GetStream(DemuxerStream::VIDEO)->GetMemoryLimit();
+  }
+  return 0;
+}
+
+void WebMediaPlayerImpl::setVideoBufferSize(size_t size) {
+  if (demuxer_ && demuxer_->GetStream(DemuxerStream::VIDEO)) {
+    demuxer_->GetStream(DemuxerStream::VIDEO)->SetMemoryLimit(size);
+  }
+}
+
 bool WebMediaPlayerImpl::copyVideoTextureToPlatformTexture(
     blink::WebGraphicsContext3D* web_graphics_context,
     unsigned int texture,
     unsigned int level,
     unsigned int internal_format,
     unsigned int type,
     bool premultiply_alpha,
     bool flip_y) {
   return copyVideoTextureToPlatformTexture(web_graphics_context, texture,
                                            internal_format, type,
@@ skipping 425 matching lines @@
 
   // pause() may be called after playback has ended and the HTMLMediaElement
   // requires that currentTime() == duration() after ending.  We want to ensure
   // |paused_time_| matches currentTime() in this case or a future seek() may
   // incorrectly discard what it thinks is a seek to the existing time.
   paused_time_ =
       ended_ ? pipeline_.GetMediaDuration() : pipeline_.GetMediaTime();
 }
 
 }  // namespace media