WebmMuxer-MediaRecorderHandler: thread hopping and data ownership

content/renderer/media/media_recorder_handler.cc

Index: content/renderer/media/media_recorder_handler.cc
diff --git a/content/renderer/media/media_recorder_handler.cc b/content/renderer/media/media_recorder_handler.cc
index 5934ef9fd11ee121a05e63fa12dcb506736b2ce8..8b559a63721585aa74b8f5eaa9511aafab25a63c 100644
--- a/content/renderer/media/media_recorder_handler.cc
+++ b/content/renderer/media/media_recorder_handler.cc
@@ -7,8 +7,10 @@
 #include "base/bind.h"
 #include "base/location.h"
 #include "base/logging.h"
+#include "base/strings/stringprintf.h"

[miu, 2015/09/22 04:11:08]

Is this needed?  Doesn't seem to be used.

[mcasas, 2015/09/22 16:58:40]

Done.
(Argh! I've tried hard to remove unused includes but
missed this one).

 #include "content/renderer/media/video_track_recorder.h"
 #include "media/base/bind_to_current_loop.h"
+#include "media/base/video_frame.h"
 #include "media/capture/webm_muxer.h"
 #include "third_party/WebKit/public/platform/WebMediaRecorderHandlerClient.h"
 #include "third_party/WebKit/public/platform/WebString.h"
@@ -60,10 +62,10 @@ bool MediaRecorderHandler::start(int timeslice) {
   DCHECK(main_render_thread_checker_.CalledOnValidThread());
   DCHECK(!recording_);
   DCHECK(!media_stream_.isNull());
+  DCHECK(!webm_muxer_);
 
-  webm_muxer_.reset(new media::WebmMuxer(media::BindToCurrentLoop(base::Bind(
-      &MediaRecorderHandler::WriteData, weak_factory_.GetWeakPtr()))));
-  DCHECK(webm_muxer_);
+  webm_muxer_.reset(new media::WebmMuxer(base::Bind(
+      &MediaRecorderHandler::WriteData, weak_factory_.GetWeakPtr())));
 
   blink::WebVector<blink::WebMediaStreamTrack> video_tracks;
   media_stream_.videoTracks(video_tracks);
@@ -84,8 +86,8 @@ bool MediaRecorderHandler::start(int timeslice) {
     return false;
 
   const VideoTrackRecorder::OnEncodedVideoCB on_encoded_video_cb =
-      base::Bind(&media::WebmMuxer::OnEncodedVideo,
-                 base::Unretained(webm_muxer_.get()));
+      media::BindToCurrentLoop(base::Bind(&MediaRecorderHandler::OnEncodedVideo,
+                                          weak_factory_.GetWeakPtr()));
 
   video_recorders_.push_back(new VideoTrackRecorder(video_track,
                                                     on_encoded_video_cb));
@@ -117,9 +119,22 @@ void MediaRecorderHandler::resume() {
   NOTIMPLEMENTED();
 }
 
-void MediaRecorderHandler::WriteData(const base::StringPiece& data) {
+void MediaRecorderHandler::OnEncodedVideo(
+    const scoped_refptr<media::VideoFrame>& video_frame,
+    scoped_ptr<std::string> encoded_data,
+    base::TimeTicks timestamp,
+    bool is_key_frame) {
   DCHECK(main_render_thread_checker_.CalledOnValidThread());
-  client_->writeData(data.data(), data.size(), false  /* lastInSlice */);
+  if (webm_muxer_) {
+    webm_muxer_->OnEncodedVideo(video_frame, encoded_data.Pass(), timestamp,
+                                is_key_frame);
+  }
+}
+
+
+void MediaRecorderHandler::WriteData(scoped_ptr<std::string> data) {
+  DCHECK(main_render_thread_checker_.CalledOnValidThread());
+  client_->writeData(data->data(), data->length(), false  /* lastInSlice */);
 }
 
 void MediaRecorderHandler::OnVideoFrameForTesting(