Implement all resolution change policies for desktop and tab capture.

content/browser/media/capture/web_contents_video_capture_device_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/browser/media/capture/web_contents_video_capture_device.h"
 
 #include "base/bind_helpers.h"
 #include "base/debug/debugger.h"
 #include "base/run_loop.h"
 #include "base/test/test_timeouts.h"
@@ skipping 144 matching lines @@
 
 // A stub implementation which returns solid-color bitmaps in calls to
 // CopyFromCompositingSurfaceToVideoFrame(), and which allows the video-frame
 // readback path to be switched on and off. The behavior is controlled by a
 // CaptureTestSourceController.
 class CaptureTestView : public TestRenderWidgetHostView {
  public:
   explicit CaptureTestView(RenderWidgetHostImpl* rwh,
                            CaptureTestSourceController* controller)
       : TestRenderWidgetHostView(rwh),
-        controller_(controller) {}
+        rwh_(rwh),
+        controller_(controller),
+        fake_bounds_(100, 100, 100 + kTestWidth, 100 + kTestHeight) {}
 
   ~CaptureTestView() override {}
 
   // TestRenderWidgetHostView overrides.
   gfx::Rect GetViewBounds() const override {
-    return gfx::Rect(100, 100, 100 + kTestWidth, 100 + kTestHeight);
+    return fake_bounds_;
+  }
+
+  void SetSize(const gfx::Size& size) override {
+    fake_bounds_.set_size(size);
+    rwh_->WasResized();
+  }
+
+  void SetBounds(const gfx::Rect& rect) override {
+    fake_bounds_ = rect;
+    rwh_->WasResized();
   }
 
   bool CanCopyToVideoFrame() const override {
     return controller_->CanCopyToVideoFrame();
   }
 
   void CopyFromCompositingSurfaceToVideoFrame(
       const gfx::Rect& src_subrect,
       const scoped_refptr<media::VideoFrame>& target,
       const base::Callback<void(bool)>& callback) override {
@@ skipping 23 matching lines @@
           target.get(), SkColorGetR(c), SkColorGetG(c), SkColorGetB(c));
       BrowserThread::PostTask(BrowserThread::UI,
                               FROM_HERE,
                               base::Bind(callback, present_time, true));
       controller_->SignalCopy();
     }
   }
 
  private:
   scoped_ptr<RenderWidgetHostViewFrameSubscriber> subscriber_;
+  // TODO(miu): Investigate why TestRenderWidgetHostView::GetRenderWidgetHost()
+  // always returns null, and perhaps fix it in order to remove this |rwh_|
+  // member.
+  RenderWidgetHostImpl* const rwh_;
   CaptureTestSourceController* const controller_;
+  gfx::Rect fake_bounds_;
 
   DISALLOW_IMPLICIT_CONSTRUCTORS(CaptureTestView);
 };
 
 #if defined(COMPILER_MSVC)
 // MSVC warns on diamond inheritance. See comment for same warning on
 // RenderViewHostImpl.
 #pragma warning(push)
 #pragma warning(disable: 4250)
 #endif
@@ skipping 78 matching lines @@
   DISALLOW_IMPLICIT_CONSTRUCTORS(CaptureTestRenderViewHostFactory);
 };
 
 // A stub consumer of captured video frames, which checks the output of
 // WebContentsVideoCaptureDevice.
 class StubClient : public media::VideoCaptureDevice::Client {
  public:
   StubClient(const base::Callback<void(SkColor)>& color_callback,
              const base::Closure& error_callback)
       : color_callback_(color_callback),
-        error_callback_(error_callback) {
+        error_callback_(error_callback),
+        expected_frame_size_(kTestWidth, kTestHeight),
+        resolution_change_policy_(media::RESOLUTION_POLICY_FIXED_RESOLUTION) {
     buffer_pool_ = new VideoCaptureBufferPool(2);
   }
   ~StubClient() override {}
 
   MOCK_METHOD5(OnIncomingCapturedData,
                void(const uint8* data,
                     int length,
                     const media::VideoCaptureFormat& frame_format,
                     int rotation,
                     const base::TimeTicks& timestamp));
   MOCK_METHOD9(OnIncomingCapturedYuvData,
                void (const uint8* y_data,
                      const uint8* u_data,
                      const uint8* v_data,
                      size_t y_stride,
                      size_t u_stride,
                      size_t v_stride,
                      const media::VideoCaptureFormat& frame_format,
                      int clockwise_rotation,
                      const base::TimeTicks& timestamp));
 
   MOCK_METHOD0(DoOnIncomingCapturedBuffer, void(void));
 
+  void SetExpectedFrameSizeProperties(const gfx::Size& max_size,
+                                      media::ResolutionChangePolicy policy) {
+    expected_frame_size_ = max_size;
+    resolution_change_policy_ = policy;
+  }
+
   scoped_ptr<media::VideoCaptureDevice::Client::Buffer> ReserveOutputBuffer(
       media::VideoPixelFormat format,
       const gfx::Size& dimensions) override {
     CHECK_EQ(format, media::PIXEL_FORMAT_I420);
     int buffer_id_to_drop = VideoCaptureBufferPool::kInvalidId;  // Ignored.
     int buffer_id = buffer_pool_->ReserveForProducer(format, dimensions,
                                                      &buffer_id_to_drop);
     if (buffer_id == VideoCaptureBufferPool::kInvalidId)
       return NULL;
 
     return scoped_ptr<media::VideoCaptureDevice::Client::Buffer>(
         new AutoReleaseBuffer(
             buffer_pool_, buffer_pool_->GetBufferHandle(buffer_id), buffer_id));
   }
   // Trampoline method to workaround GMOCK problems with scoped_ptr<>.
   void OnIncomingCapturedBuffer(scoped_ptr<Buffer> buffer,
                                 const media::VideoCaptureFormat& frame_format,
                                 const base::TimeTicks& timestamp) override {
     DoOnIncomingCapturedBuffer();
   }
 
   void OnIncomingCapturedVideoFrame(
       scoped_ptr<Buffer> buffer,
       const scoped_refptr<media::VideoFrame>& frame,
       const base::TimeTicks& timestamp) override {
-    EXPECT_EQ(gfx::Size(kTestWidth, kTestHeight), frame->visible_rect().size());
+    switch (resolution_change_policy_) {
+      case media::RESOLUTION_POLICY_FIXED_RESOLUTION:
+        EXPECT_EQ(expected_frame_size_, frame->visible_rect().size());
+        break;
+      case media::RESOLUTION_POLICY_FIXED_ASPECT_RATIO:
+        EXPECT_FALSE(frame->visible_rect().IsEmpty());
+        EXPECT_GE(expected_frame_size_.width(), frame->visible_rect().width());
+        EXPECT_GE(expected_frame_size_.height(),
+                  frame->visible_rect().height());
+        EXPECT_NEAR(static_cast<double>(expected_frame_size_.width()) /
+                        expected_frame_size_.height(),
+                    static_cast<double>(frame->visible_rect().width()) /
+                        frame->visible_rect().height(),
+                    0.01);
+        break;
+      case media::RESOLUTION_POLICY_ANY_WITHIN_LIMIT:
+        EXPECT_FALSE(frame->visible_rect().IsEmpty());
+        EXPECT_GE(expected_frame_size_.width(), frame->visible_rect().width());
+        EXPECT_GE(expected_frame_size_.height(),
+                  frame->visible_rect().height());
+        break;
+      default:
+        FAIL();
+        break;
+    }
     EXPECT_EQ(media::VideoFrame::I420, frame->format());
     double frame_rate = 0;
     EXPECT_TRUE(
         frame->metadata()->GetDouble(media::VideoFrameMetadata::FRAME_RATE,
                                      &frame_rate));
     EXPECT_EQ(kTestFramesPerSecond, frame_rate);
     uint8 yuv[3];
     for (int plane = 0; plane < 3; ++plane)
       yuv[plane] = frame->visible_data(plane)[0];
     // TODO(nick): We just look at the first pixel presently, because if
@@ skipping 29 matching lines @@
 
     const int id_;
     const scoped_refptr<VideoCaptureBufferPool> pool_;
     const scoped_ptr<VideoCaptureBufferPool::BufferHandle> buffer_handle_;
   };
 
   scoped_refptr<VideoCaptureBufferPool> buffer_pool_;
   base::Callback<void(SkColor)> color_callback_;
   base::Closure error_callback_;
 
+  gfx::Size expected_frame_size_;
+  media::ResolutionChangePolicy resolution_change_policy_;
+
   DISALLOW_COPY_AND_ASSIGN(StubClient);
 };
 
 class StubClientObserver {
  public:
   StubClientObserver()
       : error_encountered_(false),
         wait_color_yuv_(0xcafe1950) {
     client_.reset(new StubClient(
         base::Bind(&StubClientObserver::OnColor, base::Unretained(this)),
@@ skipping 192 matching lines @@
       test_view->SimulateUpdate();
     } else {
       // Simulate a non-accelerated paint.
       NotificationService::current()->Notify(
           NOTIFICATION_RENDER_WIDGET_HOST_DID_UPDATE_BACKING_STORE,
           Source<RenderWidgetHost>(web_contents_->GetRenderViewHost()),
           NotificationService::NoDetails());
     }
   }
 
+  void SimulateSourceSizeChange(const gfx::Size& size) {
+    DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+    CaptureTestView* test_view = static_cast<CaptureTestView*>(
+        web_contents_->GetRenderViewHost()->GetView());
+    test_view->SetSize(size);
+  }
+
   void DestroyVideoCaptureDevice() { device_.reset(); }
 
   StubClientObserver* client_observer() {
     return &client_observer_;
   }
 
  private:
   FakeScreen fake_screen_;
 
   StubClientObserver client_observer_;
@@ skipping 237 matching lines @@
   // Now push some good frames through; they should be processed normally.
   source()->SetCopyResultSize(kTestWidth, kTestHeight);
   source()->SetSolidColor(SK_ColorGREEN);
   ASSERT_NO_FATAL_FAILURE(client_observer()->WaitForNextColor(SK_ColorGREEN));
   source()->SetSolidColor(SK_ColorRED);
   ASSERT_NO_FATAL_FAILURE(client_observer()->WaitForNextColor(SK_ColorRED));
 
   device()->StopAndDeAllocate();
 }
 
+// Tests that, when configured with the FIXED_ASPECT_RATIO resolution change
+// policy, the source size changes result in video frames of possibly varying
+// resolutions, but all with the same aspect ratio.
+TEST_F(WebContentsVideoCaptureDeviceTest, VariableResolution_FixedAspectRatio) {
+  media::VideoCaptureParams capture_params;
+  capture_params.requested_format.frame_size.SetSize(kTestWidth, kTestHeight);
+  capture_params.requested_format.frame_rate = kTestFramesPerSecond;
+  capture_params.requested_format.pixel_format = media::PIXEL_FORMAT_I420;
+  capture_params.resolution_change_policy =
+      media::RESOLUTION_POLICY_FIXED_ASPECT_RATIO;
+
+  scoped_ptr<media::VideoCaptureDevice::Client> client =
+      client_observer()->PassClient();
+  StubClient* const stub_client = static_cast<StubClient*>(client.get());
+  stub_client->SetExpectedFrameSizeProperties(
+      gfx::Size(kTestWidth, kTestHeight),
+      media::RESOLUTION_POLICY_FIXED_ASPECT_RATIO);
+  device()->AllocateAndStart(capture_params, client.Pass());
+
+  for (int i = 0; i < 6; i++) {
+    const char* name = NULL;
+    switch (i % 3) {
+      case 0:
+        source()->SetCanCopyToVideoFrame(true);
+        source()->SetUseFrameSubscriber(false);
+        name = "VideoFrame";
+        break;
+      case 1:
+        source()->SetCanCopyToVideoFrame(false);
+        source()->SetUseFrameSubscriber(true);
+        name = "Subscriber";
+        break;
+      case 2:
+        source()->SetCanCopyToVideoFrame(false);
+        source()->SetUseFrameSubscriber(false);
+        name = "SkBitmap";
+        break;
+      default:
+        FAIL();
+    }
+
+    SCOPED_TRACE(base::StringPrintf("Using %s path, iteration #%d", name, i));
+
+    // Source size equals maximum size.  StubClient will expect frames to be
+    // kTestWidth by kTestHeight.
+    SimulateSourceSizeChange(gfx::Size(kTestWidth, kTestHeight));
+    source()->SetSolidColor(SK_ColorRED);
+    SimulateDrawEvent();
+    ASSERT_NO_FATAL_FAILURE(client_observer()->WaitForNextColor(SK_ColorRED));
+
+    // Source size is half in both dimensions.  StubClient will confirm the
+    // aspect ratio of the resulting frames is correct.
+    SimulateSourceSizeChange(gfx::Size(kTestWidth / 2, kTestHeight / 2));
+    source()->SetSolidColor(SK_ColorGREEN);
+    SimulateDrawEvent();
+    ASSERT_NO_FATAL_FAILURE(client_observer()->WaitForNextColor(SK_ColorGREEN));
+
+    // Source size changes aspect ratio.  StubClient will confirm the aspect
+    // ratio of the resulting frames is correct.
+    SimulateSourceSizeChange(gfx::Size(kTestWidth / 2, kTestHeight));
+    source()->SetSolidColor(SK_ColorBLUE);
+    SimulateDrawEvent();
+    ASSERT_NO_FATAL_FAILURE(client_observer()->WaitForNextColor(SK_ColorBLUE));
+
+    // Source size changes aspect ratio again.  StubClient will confirm the
+    // aspect ratio of the resulting frames is correct.
+    SimulateSourceSizeChange(gfx::Size(kTestWidth, kTestHeight / 2));
+    source()->SetSolidColor(SK_ColorBLACK);
+    SimulateDrawEvent();
+    ASSERT_NO_FATAL_FAILURE(client_observer()->WaitForNextColor(SK_ColorBLACK));
+  }
+
+  device()->StopAndDeAllocate();
+}
+
+// Tests that, when configured with the ANY_WITHIN_LIMIT resolution change
+// policy, the source size changes result in video frames of possibly varying
+// resolutions.
+TEST_F(WebContentsVideoCaptureDeviceTest, VariableResolution_AnyWithinLimits) {
+  media::VideoCaptureParams capture_params;
+  capture_params.requested_format.frame_size.SetSize(kTestWidth, kTestHeight);
+  capture_params.requested_format.frame_rate = kTestFramesPerSecond;
+  capture_params.requested_format.pixel_format = media::PIXEL_FORMAT_I420;
+  capture_params.resolution_change_policy =
+      media::RESOLUTION_POLICY_ANY_WITHIN_LIMIT;
+
+  scoped_ptr<media::VideoCaptureDevice::Client> client =
+      client_observer()->PassClient();
+  StubClient* const stub_client = static_cast<StubClient*>(client.get());
+  stub_client->SetExpectedFrameSizeProperties(
+      gfx::Size(kTestWidth, kTestHeight),
+      media::RESOLUTION_POLICY_ANY_WITHIN_LIMIT);
+  device()->AllocateAndStart(capture_params, client.Pass());
+
+  for (int i = 0; i < 6; i++) {
+    const char* name = NULL;
+    switch (i % 3) {
+      case 0:
+        source()->SetCanCopyToVideoFrame(true);
+        source()->SetUseFrameSubscriber(false);
+        name = "VideoFrame";
+        break;
+      case 1:
+        source()->SetCanCopyToVideoFrame(false);
+        source()->SetUseFrameSubscriber(true);
+        name = "Subscriber";
+        break;
+      case 2:
+        source()->SetCanCopyToVideoFrame(false);
+        source()->SetUseFrameSubscriber(false);
+        name = "SkBitmap";
+        break;
+      default:
+        FAIL();
+    }
+
+    SCOPED_TRACE(base::StringPrintf("Using %s path, iteration #%d", name, i));
+
+    // Source size equals maximum size.  StubClient will expect frames to be
+    // kTestWidth by kTestHeight.
+    SimulateSourceSizeChange(gfx::Size(kTestWidth, kTestHeight));
+    source()->SetSolidColor(SK_ColorRED);
+    SimulateDrawEvent();
+    ASSERT_NO_FATAL_FAILURE(client_observer()->WaitForNextColor(SK_ColorRED));
+
+    // Source size is half in both dimensions.  StubClient will confirm the
+    // resulting frame is also half in both dimensions.
+    SimulateSourceSizeChange(gfx::Size(kTestWidth / 2, kTestHeight / 2));
+    source()->SetSolidColor(SK_ColorGREEN);
+    SimulateDrawEvent();
+    ASSERT_NO_FATAL_FAILURE(client_observer()->WaitForNextColor(SK_ColorGREEN));
+
+    // Source size changes to something arbitrary.  StubClient will confirm the
+    // resulting frame has the same dimensions.
+    SimulateSourceSizeChange(gfx::Size(kTestWidth / 2 + 42,
+                                       kTestHeight * 2 - 10));
+    source()->SetSolidColor(SK_ColorBLUE);
+    SimulateDrawEvent();
+    ASSERT_NO_FATAL_FAILURE(client_observer()->WaitForNextColor(SK_ColorBLUE));
+
+    // Source size changes to something arbitrary that exceeds the maximum frame
+    // size.  StubClient will confirm the resulting frame has dimensions within
+    // the maximum limit.
+    SimulateSourceSizeChange(gfx::Size(kTestWidth * 2 + 99, kTestHeight / 2));
+    source()->SetSolidColor(SK_ColorBLACK);
+    SimulateDrawEvent();
+    ASSERT_NO_FATAL_FAILURE(client_observer()->WaitForNextColor(SK_ColorBLACK));
+  }
+
+  device()->StopAndDeAllocate();
+}
+
 }  // namespace
 }  // namespace content