| /* |
| * libjingle |
| * Copyright 2015 Google Inc. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are met: |
| * |
| * 1. Redistributions of source code must retain the above copyright notice, |
| * this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright notice, |
| * this list of conditions and the following disclaimer in the documentation |
| * and/or other materials provided with the distribution. |
| * 3. The name of the author may not be used to endorse or promote products |
| * derived from this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED |
| * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF |
| * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO |
| * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
| * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; |
| * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, |
| * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR |
| * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF |
| * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include <string.h> |
| |
| #include "talk/media/base/videoframe_unittest.h" |
| #include "talk/media/webrtc/webrtcvideoframe.h" |
| #include "talk/media/webrtc/webrtcvideoframefactory.h" |
| |
| class WebRtcVideoFrameFactoryTest |
| : public VideoFrameTest<cricket::WebRtcVideoFrameFactory> { |
| public: |
| WebRtcVideoFrameFactoryTest() {} |
| |
| void InitFrame(webrtc::VideoRotation frame_rotation) { |
| const int frame_width = 1920; |
| const int frame_height = 1080; |
| |
| // Build the CapturedFrame. |
| captured_frame_.fourcc = cricket::FOURCC_I420; |
| captured_frame_.pixel_width = 1; |
| captured_frame_.pixel_height = 1; |
| captured_frame_.time_stamp = 5678; |
| captured_frame_.rotation = frame_rotation; |
| captured_frame_.width = frame_width; |
| captured_frame_.height = frame_height; |
| captured_frame_.data_size = |
| (frame_width * frame_height) + |
| ((frame_width + 1) / 2) * ((frame_height + 1) / 2) * 2; |
| captured_frame_buffer_.reset(new uint8_t[captured_frame_.data_size]); |
| // Initialize memory to satisfy DrMemory tests. |
| memset(captured_frame_buffer_.get(), 0, captured_frame_.data_size); |
| captured_frame_.data = captured_frame_buffer_.get(); |
| } |
| |
| void VerifyFrame(cricket::VideoFrame* dest_frame, |
| webrtc::VideoRotation src_rotation, |
| int src_width, |
| int src_height, |
| bool apply_rotation) { |
| if (!apply_rotation) { |
| EXPECT_EQ(dest_frame->GetRotation(), src_rotation); |
| EXPECT_EQ(dest_frame->GetWidth(), src_width); |
| EXPECT_EQ(dest_frame->GetHeight(), src_height); |
| } else { |
| EXPECT_EQ(dest_frame->GetRotation(), webrtc::kVideoRotation_0); |
| if (src_rotation == webrtc::kVideoRotation_90 || |
| src_rotation == webrtc::kVideoRotation_270) { |
| EXPECT_EQ(dest_frame->GetWidth(), src_height); |
| EXPECT_EQ(dest_frame->GetHeight(), src_width); |
| } else { |
| EXPECT_EQ(dest_frame->GetWidth(), src_width); |
| EXPECT_EQ(dest_frame->GetHeight(), src_height); |
| } |
| } |
| } |
| |
| void TestCreateAliasedFrame(bool apply_rotation) { |
| cricket::VideoFrameFactory& factory = factory_; |
| factory.SetApplyRotation(apply_rotation); |
| InitFrame(webrtc::kVideoRotation_270); |
| const cricket::CapturedFrame& captured_frame = get_captured_frame(); |
| // Create the new frame from the CapturedFrame. |
| rtc::scoped_ptr<cricket::VideoFrame> frame; |
| int new_width = captured_frame.width / 2; |
| int new_height = captured_frame.height / 2; |
| frame.reset(factory.CreateAliasedFrame(&captured_frame, new_width, |
| new_height, new_width, new_height)); |
| VerifyFrame(frame.get(), webrtc::kVideoRotation_270, new_width, new_height, |
| apply_rotation); |
| |
| frame.reset(factory.CreateAliasedFrame( |
| &captured_frame, new_width, new_height, new_width / 2, new_height / 2)); |
| VerifyFrame(frame.get(), webrtc::kVideoRotation_270, new_width / 2, |
| new_height / 2, apply_rotation); |
| |
| // Reset the frame first so it's exclusive hence we could go through the |
| // StretchToFrame code path in CreateAliasedFrame. |
| frame.reset(); |
| frame.reset(factory.CreateAliasedFrame( |
| &captured_frame, new_width, new_height, new_width / 2, new_height / 2)); |
| VerifyFrame(frame.get(), webrtc::kVideoRotation_270, new_width / 2, |
| new_height / 2, apply_rotation); |
| } |
| |
| const cricket::CapturedFrame& get_captured_frame() { return captured_frame_; } |
| |
| private: |
| cricket::CapturedFrame captured_frame_; |
| rtc::scoped_ptr<uint8_t[]> captured_frame_buffer_; |
| cricket::WebRtcVideoFrameFactory factory_; |
| }; |
| |
| TEST_F(WebRtcVideoFrameFactoryTest, NoApplyRotation) { |
| TestCreateAliasedFrame(false); |
| } |
| |
| TEST_F(WebRtcVideoFrameFactoryTest, ApplyRotation) { |
| TestCreateAliasedFrame(true); |
| } |