diff --git a/talk/app/webrtc/objc/RTCI420Frame.mm b/talk/app/webrtc/objc/RTCI420Frame.mm index 1646510b37..6c6c564582 100644 --- a/talk/app/webrtc/objc/RTCI420Frame.mm +++ b/talk/app/webrtc/objc/RTCI420Frame.mm @@ -29,7 +29,7 @@ #include -#include "webrtc/media/engine/webrtcvideoframe.h" +#include "webrtc/media/base/videoframe.h" @implementation RTCI420Frame { std::unique_ptr _videoFrame; @@ -98,10 +98,7 @@ if (self = [super init]) { // Keep a shallow copy of the video frame. The underlying frame buffer is // not copied. - _videoFrame.reset(new cricket::WebRtcVideoFrame( - videoFrame->video_frame_buffer(), - videoFrame->rotation(), - videoFrame->timestamp_us())); + _videoFrame.reset(videoFrame->Copy()); } return self; } diff --git a/webrtc/api/android/jni/peerconnection_jni.cc b/webrtc/api/android/jni/peerconnection_jni.cc index 5095b152ba..9efe037c9f 100644 --- a/webrtc/api/android/jni/peerconnection_jni.cc +++ b/webrtc/api/android/jni/peerconnection_jni.cc @@ -770,8 +770,7 @@ class JavaVideoRendererWrapper // ownership of the frame, and the frame should be released with // VideoRenderer.releaseNativeFrame(). static jlong javaShallowCopy(const cricket::VideoFrame* frame) { - return jlongFromPointer(new cricket::WebRtcVideoFrame( - frame->video_frame_buffer(), frame->rotation(), frame->timestamp_us())); + return jlongFromPointer(frame->Copy()); } // Return a VideoRenderer.I420Frame referring to the data in |frame|. diff --git a/webrtc/media/base/videoframe.h b/webrtc/media/base/videoframe.h index 3dcbb4efa5..a1393499f9 100644 --- a/webrtc/media/base/videoframe.h +++ b/webrtc/media/base/videoframe.h @@ -58,6 +58,16 @@ class VideoFrame { // Indicates the rotation angle in degrees. virtual webrtc::VideoRotation rotation() const = 0; + // Make a shallow copy of the frame. The frame buffer itself is not copied. + // Both the current and new VideoFrame will share a single reference-counted + // frame buffer. + + // TODO(nisse): Deprecated, to be deleted in the cricket::VideoFrame and + // webrtc::VideoFrame merge. To make a copy, use the cricket::WebRtcVideoFrame + // constructor passing video_frame_buffer(), rotation() and timestamp_us() as + // arguments. + virtual VideoFrame *Copy() const = 0; + // Return a copy of frame which has its pending rotation applied. The // ownership of the returned frame is held by this frame. diff --git a/webrtc/media/base/videoframe_unittest.h b/webrtc/media/base/videoframe_unittest.h index 13fb790c1a..f19d0325d6 100644 --- a/webrtc/media/base/videoframe_unittest.h +++ b/webrtc/media/base/videoframe_unittest.h @@ -1808,6 +1808,31 @@ class VideoFrameTest : public testing::Test { EXPECT_TRUE(IsEqual(frame1, frame2, 1)); } + /////////////////// + // General tests // + /////////////////// + + void Copy() { + std::unique_ptr source(new T); + std::unique_ptr target; + ASSERT_TRUE(LoadFrameNoRepeat(source.get())); + target.reset(source->Copy()); + EXPECT_TRUE(IsEqual(*source, *target, 0)); + source.reset(); + ASSERT_TRUE(target->video_frame_buffer() != NULL); + EXPECT_TRUE(target->video_frame_buffer()->DataY() != NULL); + } + + void CopyIsRef() { + std::unique_ptr source(new T); + std::unique_ptr target; + ASSERT_TRUE(LoadFrameNoRepeat(source.get())); + target.reset(source->Copy()); + EXPECT_TRUE(IsEqual(*source, *target, 0)); + const T* const_source = source.get(); + EXPECT_EQ(const_source->video_frame_buffer(), target->video_frame_buffer()); + } + int repeat_; }; diff --git a/webrtc/media/engine/webrtcvideoframe.cc b/webrtc/media/engine/webrtcvideoframe.cc index 3102b05f77..71219d13f9 100644 --- a/webrtc/media/engine/webrtcvideoframe.cc +++ b/webrtc/media/engine/webrtcvideoframe.cc @@ -103,6 +103,11 @@ webrtc::VideoRotation WebRtcVideoFrame::rotation() const { return rotation_; } +VideoFrame* WebRtcVideoFrame::Copy() const { + return new WebRtcVideoFrame(video_frame_buffer_, rotation_, timestamp_us_, + transport_frame_id_); +} + size_t WebRtcVideoFrame::ConvertToRgbBuffer(uint32_t to_fourcc, uint8_t* buffer, size_t size, diff --git a/webrtc/media/engine/webrtcvideoframe.h b/webrtc/media/engine/webrtcvideoframe.h index cfbef519aa..bdf5a02a62 100644 --- a/webrtc/media/engine/webrtcvideoframe.h +++ b/webrtc/media/engine/webrtcvideoframe.h @@ -93,6 +93,8 @@ class WebRtcVideoFrame : public VideoFrame { webrtc::VideoRotation rotation() const override; + VideoFrame* Copy() const override; + size_t ConvertToRgbBuffer(uint32_t to_fourcc, uint8_t* buffer, size_t size, diff --git a/webrtc/media/engine/webrtcvideoframe_unittest.cc b/webrtc/media/engine/webrtcvideoframe_unittest.cc index 9d6bd5f7b5..fb7be22c53 100644 --- a/webrtc/media/engine/webrtcvideoframe_unittest.cc +++ b/webrtc/media/engine/webrtcvideoframe_unittest.cc @@ -223,6 +223,9 @@ TEST_WEBRTCVIDEOFRAME(ConvertFromUYVYBufferInverted) // TEST_WEBRTCVIDEOFRAME(ConvertToI422Buffer) // TEST_WEBRTCVIDEOFRAME(ConstructARGBBlackWhitePixel) +TEST_WEBRTCVIDEOFRAME(Copy) +TEST_WEBRTCVIDEOFRAME(CopyIsRef) + // These functions test implementation-specific details. // Tests the Init function with different cropped size. TEST_F(WebRtcVideoFrameTest, InitEvenSize) { @@ -267,6 +270,24 @@ TEST_F(WebRtcVideoFrameTest, TextureInitialValues) { EXPECT_EQ(40, frame.timestamp_us()); } +TEST_F(WebRtcVideoFrameTest, CopyTextureFrame) { + webrtc::test::FakeNativeHandle* dummy_handle = + new webrtc::test::FakeNativeHandle(); + webrtc::NativeHandleBuffer* buffer = + new rtc::RefCountedObject( + dummy_handle, 640, 480); + // Timestamp is converted from ns to us, so last three digits are lost. + WebRtcVideoFrame frame1(buffer, 20000, webrtc::kVideoRotation_0); + VideoFrame* frame2 = frame1.Copy(); + EXPECT_EQ(frame1.video_frame_buffer()->native_handle(), + frame2->video_frame_buffer()->native_handle()); + EXPECT_EQ(frame1.width(), frame2->width()); + EXPECT_EQ(frame1.height(), frame2->height()); + EXPECT_EQ(frame1.GetTimeStamp(), frame2->GetTimeStamp()); + EXPECT_EQ(frame1.timestamp_us(), frame2->timestamp_us()); + delete frame2; +} + TEST_F(WebRtcVideoFrameTest, ApplyRotationToFrame) { WebRtcVideoFrame applied0; EXPECT_TRUE(IsNull(applied0));