Home | History | Annotate | Download | only in webrtc 
      1 // Copyright 2014 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include <algorithm>
      6 
      7 #include "content/renderer/media/webrtc/webrtc_video_capturer_adapter.h"
      8 #include "media/base/video_frame.h"
      9 #include "testing/gtest/include/gtest/gtest.h"
     10 
     11 namespace content {
     12 
     13 class WebRtcVideoCapturerAdapterTest
     14     : public sigslot::has_slots<>,
     15       public ::testing::Test {
     16  public:
     17   WebRtcVideoCapturerAdapterTest()
     18       : adapter_(false),
     19         output_frame_width_(0),
     20         output_frame_height_(0) {
     21     adapter_.SignalFrameCaptured.connect(
     22         this, &WebRtcVideoCapturerAdapterTest::OnFrameCaptured);
     23   }
     24   virtual ~WebRtcVideoCapturerAdapterTest() {}
     25 
     26   void TestSourceCropFrame(int capture_width,
     27                            int capture_height,
     28                            int cropped_width,
     29                            int cropped_height,
     30                            int natural_width,
     31                            int natural_height) {
     32     const int horiz_crop = ((capture_width - cropped_width) / 2);
     33     const int vert_crop = ((capture_height - cropped_height) / 2);
     34 
     35     gfx::Size coded_size(capture_width, capture_height);
     36     gfx::Size natural_size(natural_width, natural_height);
     37     gfx::Rect view_rect(horiz_crop, vert_crop, cropped_width, cropped_height);
     38     scoped_refptr<media::VideoFrame> frame =
     39         media::VideoFrame::CreateFrame(media::VideoFrame::I420,
     40                                        coded_size, view_rect, natural_size,
     41                                        base::TimeDelta());
     42     adapter_.OnFrameCaptured(frame);
     43     EXPECT_EQ(natural_width, output_frame_width_);
     44     EXPECT_EQ(natural_height, output_frame_height_);
     45   }
     46  protected:
     47   void OnFrameCaptured(cricket::VideoCapturer* capturer,
     48                        const cricket::CapturedFrame* frame) {
     49     output_frame_width_ = frame->width;
     50     output_frame_height_ = frame->height;
     51   }
     52 
     53  private:
     54   WebRtcVideoCapturerAdapter adapter_;
     55   int output_frame_width_;
     56   int output_frame_height_;
     57 };
     58 
     59 TEST_F(WebRtcVideoCapturerAdapterTest, CropFrameTo640360) {
     60   TestSourceCropFrame(640, 480, 640, 360, 640, 360);
     61 }
     62 
     63 TEST_F(WebRtcVideoCapturerAdapterTest, CropFrameTo320320) {
     64   TestSourceCropFrame(640, 480, 480, 480, 320, 320);
     65 }
     66 
     67 TEST_F(WebRtcVideoCapturerAdapterTest, Scale720To640360) {
     68   TestSourceCropFrame(1280, 720, 1280, 720, 640, 360);
     69 }
     70 
     71 }  // namespace content
     72