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
11namespace content {
12
13class 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
59TEST_F(WebRtcVideoCapturerAdapterTest, CropFrameTo640360) {
60  TestSourceCropFrame(640, 480, 640, 360, 640, 360);
61}
62
63TEST_F(WebRtcVideoCapturerAdapterTest, CropFrameTo320320) {
64  TestSourceCropFrame(640, 480, 480, 480, 320, 320);
65}
66
67TEST_F(WebRtcVideoCapturerAdapterTest, Scale720To640360) {
68  TestSourceCropFrame(1280, 720, 1280, 720, 640, 360);
69}
70
71}  // namespace content
72