[Ozone-DRM] Modeset when re-mapping windows to controllers

ui/ozone/platform/drm/gpu/drm_surface_unittest.cc

 // Copyright 2014 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 "base/message_loop/message_loop.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/skia/include/core/SkCanvas.h"
 #include "third_party/skia/include/core/SkColor.h"
 #include "third_party/skia/include/core/SkDevice.h"
 #include "ui/ozone/platform/drm/gpu/crtc_controller.h"
@@ skipping 11 matching lines @@
 // Create a basic mode for a 6x4 screen.
 const drmModeModeInfo kDefaultMode =
     {0, 6, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, {'\0'}};
 
 const gfx::AcceleratedWidget kDefaultWidgetHandle = 1;
 const uint32_t kDefaultCrtc = 1;
 const uint32_t kDefaultConnector = 2;
 const size_t kPlanesPerCrtc = 1;
 const uint32_t kDefaultCursorSize = 64;
 
+std::vector<skia::RefPtr<SkSurface>> GetFramebuffers(ui::MockDrmDevice* drm) {
+  std::vector<skia::RefPtr<SkSurface>> framebuffers;
+  for (const auto& buffer : drm->buffers()) {
+    // Skip destroyed buffers and cursor buffers.
+    if (!buffer || (buffer->width() == kDefaultCursorSize &&
+                    buffer->height() == kDefaultCursorSize))
+      continue;
+
+    framebuffers.push_back(buffer);
+  }
+
+  return framebuffers;
+}
+
 }  // namespace
 
 class DrmSurfaceTest : public testing::Test {
  public:
   DrmSurfaceTest() {}
 
   void SetUp() override;
   void TearDown() override;
 
  protected:
@@ skipping 42 matching lines @@
   scoped_ptr<ui::DrmWindow> window =
       screen_manager_->RemoveWindow(kDefaultWidgetHandle);
   window->Shutdown();
   drm_ = nullptr;
   message_loop_.reset();
 }
 
 TEST_F(DrmSurfaceTest, CheckFBIDOnSwap) {
   surface_->PresentCanvas(gfx::Rect());
   drm_->RunCallbacks();
-  // Framebuffer ID 1 is allocated in SetUp for the buffer used to modeset.
-  EXPECT_EQ(2u, drm_->current_framebuffer());
+
+  // Framebuffer ID 1 is allocated in SetUp for the buffer used to modeset and
+  // framebuffer ID 2 is used when the window to display mapping is done.
+  EXPECT_EQ(3u, drm_->current_framebuffer());
   surface_->PresentCanvas(gfx::Rect());
   drm_->RunCallbacks();
-  EXPECT_EQ(3u, drm_->current_framebuffer());
+  EXPECT_EQ(4u, drm_->current_framebuffer());
 }
 
 TEST_F(DrmSurfaceTest, CheckSurfaceContents) {
   SkPaint paint;
   paint.setColor(SK_ColorWHITE);
   SkRect rect =
       SkRect::MakeWH(kDefaultMode.hdisplay / 2, kDefaultMode.vdisplay / 2);
   surface_->GetSurface()->getCanvas()->drawRect(rect, paint);
   surface_->PresentCanvas(
       gfx::Rect(0, 0, kDefaultMode.hdisplay / 2, kDefaultMode.vdisplay / 2));
   drm_->RunCallbacks();
 
   SkBitmap image;
-  std::vector<skia::RefPtr<SkSurface>> framebuffers;
-  for (const auto& buffer : drm_->buffers()) {
-    // Skip cursor buffers.
-    if (buffer->width() == kDefaultCursorSize &&
-        buffer->height() == kDefaultCursorSize)
-      continue;
-
-    framebuffers.push_back(buffer);
-  }
+  std::vector<skia::RefPtr<SkSurface>> framebuffers =
+      GetFramebuffers(drm_.get());
 
   // Buffer 0 is the modesetting buffer, buffer 2 is the frontbuffer and buffer
   // 1 is the backbuffer.
   EXPECT_EQ(3u, framebuffers.size());
 
   image.setInfo(framebuffers[1]->getCanvas()->imageInfo());
   EXPECT_TRUE(framebuffers[1]->getCanvas()->readPixels(&image, 0, 0));
 
   EXPECT_EQ(kDefaultMode.hdisplay, image.width());
   EXPECT_EQ(kDefaultMode.vdisplay, image.height());
@@ skipping 23 matching lines @@
 
   paint.setColor(SK_ColorRED);
   rect.SetRect(0, kDefaultMode.vdisplay / 2, kDefaultMode.hdisplay / 2,
                kDefaultMode.vdisplay / 2);
   surface_->GetSurface()->getCanvas()->drawRect(RectToSkRect(rect), paint);
   surface_->PresentCanvas(rect);
 
   drm_->RunCallbacks();
 
   SkBitmap image;
-  std::vector<skia::RefPtr<SkSurface>> framebuffers;
-  for (const auto& buffer : drm_->buffers()) {
-    // Skip cursor buffers.
-    if (buffer->width() == kDefaultCursorSize &&
-        buffer->height() == kDefaultCursorSize)
-      continue;
-
-    framebuffers.push_back(buffer);
-  }
+  std::vector<skia::RefPtr<SkSurface>> framebuffers =
+      GetFramebuffers(drm_.get());
 
   // Buffer 0 is the modesetting buffer, buffer 2 is the backbuffer and buffer
   // 1 is the frontbuffer.
   EXPECT_EQ(3u, framebuffers.size());
 
   image.setInfo(framebuffers[2]->getCanvas()->imageInfo());
   EXPECT_TRUE(framebuffers[2]->getCanvas()->readPixels(&image, 0, 0));
 
   EXPECT_EQ(kDefaultMode.hdisplay, image.width());
   EXPECT_EQ(kDefaultMode.vdisplay, image.height());
 
   // Make sure the updates are correctly propagated to the native surface.
   for (int i = 0; i < image.height(); ++i) {
     for (int j = 0; j < image.width(); ++j) {
       if (j < kDefaultMode.hdisplay / 2 && i < kDefaultMode.vdisplay / 2)
         EXPECT_EQ(SK_ColorWHITE, image.getColor(j, i));
       else if (j < kDefaultMode.hdisplay / 2)
         EXPECT_EQ(SK_ColorRED, image.getColor(j, i));
       else
         EXPECT_EQ(SK_ColorBLACK, image.getColor(j, i));
     }
   }
 }