Index: content/renderer/media/rtc_video_encoder.cc |
diff --git a/content/renderer/media/rtc_video_encoder.cc b/content/renderer/media/rtc_video_encoder.cc |
index cdadae55cd560cc65a60864302399a9afb21df07..c2a3e3efb6e773cdf16835ae36b84ad6affce798 100644 |
--- a/content/renderer/media/rtc_video_encoder.cc |
+++ b/content/renderer/media/rtc_video_encoder.cc |
@@ -13,6 +13,7 @@ |
#include "base/single_thread_task_runner.h" |
#include "base/synchronization/waitable_event.h" |
#include "base/thread_task_runner_handle.h" |
+#include "media/base/bind_to_current_loop.h" |
#include "media/base/bitstream_buffer.h" |
#include "media/base/video_frame.h" |
#include "media/base/video_util.h" |
@@ -491,48 +492,51 @@ void RTCVideoEncoder::Impl::EncodeOneFrame() { |
} |
const int index = input_buffers_free_.back(); |
- base::SharedMemory* input_buffer = input_buffers_[index]; |
- scoped_refptr<media::VideoFrame> frame = |
- media::VideoFrame::WrapExternalSharedMemory( |
- media::VideoFrame::I420, |
- input_frame_coded_size_, |
- gfx::Rect(input_visible_size_), |
- input_visible_size_, |
- reinterpret_cast<uint8*>(input_buffer->memory()), |
- input_buffer->mapped_size(), |
- input_buffer->handle(), |
- 0, |
- base::TimeDelta()); |
- frame->AddDestructionObserver( |
- base::Bind(&RTCVideoEncoder::Impl::EncodeFrameFinished, this, index)); |
- if (!frame.get()) { |
- DLOG(ERROR) << "Impl::EncodeOneFrame(): failed to create frame"; |
- NOTIFY_ERROR(media::VideoEncodeAccelerator::kPlatformFailureError); |
- return; |
- } |
- |
- // Do a strided copy of the input frame to match the input requirements for |
- // the encoder. |
- // TODO(sheu): support zero-copy from WebRTC. http://crbug.com/269312 |
- if (libyuv::I420Copy(next_frame->buffer(webrtc::kYPlane), |
- next_frame->stride(webrtc::kYPlane), |
- next_frame->buffer(webrtc::kUPlane), |
- next_frame->stride(webrtc::kUPlane), |
- next_frame->buffer(webrtc::kVPlane), |
- next_frame->stride(webrtc::kVPlane), |
- frame->data(media::VideoFrame::kYPlane), |
- frame->stride(media::VideoFrame::kYPlane), |
- frame->data(media::VideoFrame::kUPlane), |
- frame->stride(media::VideoFrame::kUPlane), |
- frame->data(media::VideoFrame::kVPlane), |
- frame->stride(media::VideoFrame::kVPlane), |
- next_frame->width(), |
- next_frame->height())) { |
- DLOG(ERROR) << "Failed to copy buffer"; |
- NOTIFY_ERROR(media::VideoEncodeAccelerator::kPlatformFailureError); |
- return; |
+ scoped_refptr<media::VideoFrame> frame; |
+ if (next_frame->native_handle()) { |
+ frame = static_cast<media::VideoFrame*>(next_frame->native_handle()); |
+ } else { |
+ base::SharedMemory* input_buffer = input_buffers_[index]; |
+ frame = media::VideoFrame::WrapExternalSharedMemory( |
+ media::VideoFrame::I420, |
+ input_frame_coded_size_, |
+ gfx::Rect(input_visible_size_), |
+ input_visible_size_, |
+ reinterpret_cast<uint8*>(input_buffer->memory()), |
+ input_buffer->mapped_size(), |
+ input_buffer->handle(), |
+ 0, |
+ base::TimeDelta()); |
+ frame->AddDestructionObserver( |
+ base::Bind(&RTCVideoEncoder::Impl::EncodeFrameFinished, this, index)); |
+ if (!frame.get()) { |
+ DLOG(ERROR) << "Impl::EncodeOneFrame(): failed to create frame"; |
+ NOTIFY_ERROR(media::VideoEncodeAccelerator::kPlatformFailureError); |
+ return; |
+ } |
+ // Do a strided copy of the input frame to match the input requirements for |
+ // the encoder. |
+ // TODO(sheu): support zero-copy from WebRTC. http://crbug.com/269312 |
+ if (libyuv::I420Copy(next_frame->buffer(webrtc::kYPlane), |
+ next_frame->stride(webrtc::kYPlane), |
+ next_frame->buffer(webrtc::kUPlane), |
+ next_frame->stride(webrtc::kUPlane), |
+ next_frame->buffer(webrtc::kVPlane), |
+ next_frame->stride(webrtc::kVPlane), |
+ frame->data(media::VideoFrame::kYPlane), |
+ frame->stride(media::VideoFrame::kYPlane), |
+ frame->data(media::VideoFrame::kUPlane), |
+ frame->stride(media::VideoFrame::kUPlane), |
+ frame->data(media::VideoFrame::kVPlane), |
+ frame->stride(media::VideoFrame::kVPlane), |
+ next_frame->width(), next_frame->height())) { |
+ DLOG(ERROR) << "Failed to copy buffer"; |
+ NOTIFY_ERROR(media::VideoEncodeAccelerator::kPlatformFailureError); |
+ return; |
+ } |
} |
- |
+ frame->AddDestructionObserver(media::BindToCurrentLoop( |
+ base::Bind(&RTCVideoEncoder::Impl::EncodeFrameFinished, this, index))); |
video_encoder_->Encode(frame, next_frame_keyframe); |
input_buffers_free_.pop_back(); |
SignalAsyncWaiter(WEBRTC_VIDEO_CODEC_OK); |