src/ui/ozone/platform/dri/dri_surface_unittest.cc

   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 "base/message_loop/message_loop.h"
   6 #include "testing/gtest/include/gtest/gtest.h"
   7 #include "third_party/skia/include/core/SkCanvas.h"
   8 #include "third_party/skia/include/core/SkColor.h"
   9 #include "third_party/skia/include/core/SkDevice.h"
  10 #include "ui/ozone/platform/dri/crtc_state.h"
  11 #include "ui/ozone/platform/dri/dri_buffer.h"
  12 #include "ui/ozone/platform/dri/dri_surface.h"
  13 #include "ui/ozone/platform/dri/hardware_display_controller.h"
  14 #include "ui/ozone/platform/dri/test/mock_dri_wrapper.h"
  15
  16 namespace {
  17
  18 // Create a basic mode for a 6x4 screen.
  19 const drmModeModeInfo kDefaultMode =
  20     {0, 6, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, {'\0'}};
  21
  22 const uint32_t kDefaultCrtc = 1;
  23 const uint32_t kDefaultConnector = 2;
  24
  25 }  // namespace
  26
  27 class DriSurfaceTest : public testing::Test {
  28  public:
  29   DriSurfaceTest() {}
  30
  31   virtual void SetUp() OVERRIDE;
  32   virtual void TearDown() OVERRIDE;
  33
  34  protected:
  35   scoped_ptr<base::MessageLoop> message_loop_;
  36   scoped_ptr<ui::MockDriWrapper> drm_;
  37   scoped_ptr<ui::HardwareDisplayController> controller_;
  38   scoped_ptr<ui::DriSurface> surface_;
  39
  40  private:
  41   DISALLOW_COPY_AND_ASSIGN(DriSurfaceTest);
  42 };
  43
  44 void DriSurfaceTest::SetUp() {
  45   message_loop_.reset(new base::MessageLoopForUI);
  46   drm_.reset(new ui::MockDriWrapper(3));
  47
  48   controller_.reset(new ui::HardwareDisplayController(
  49       drm_.get(),
  50       scoped_ptr<ui::CrtcState>(
  51           new ui::CrtcState(drm_.get(), kDefaultCrtc, kDefaultConnector))));
  52   scoped_refptr<ui::DriBuffer> buffer(new ui::DriBuffer(drm_.get()));
  53   SkImageInfo info = SkImageInfo::MakeN32Premul(kDefaultMode.hdisplay,
  54                                                 kDefaultMode.vdisplay);
  55   EXPECT_TRUE(buffer->Initialize(info));
  56   EXPECT_TRUE(controller_->Modeset(ui::OverlayPlane(buffer), kDefaultMode));
  57
  58   surface_.reset(new ui::DriSurface(drm_.get(), controller_->AsWeakPtr()));
  59   surface_->ResizeCanvas(gfx::Size(kDefaultMode.hdisplay,
  60                                    kDefaultMode.vdisplay));
  61 }
  62
  63 void DriSurfaceTest::TearDown() {
  64   surface_.reset();
  65   controller_.reset();
  66   drm_.reset();
  67   message_loop_.reset();
  68 }
  69
  70 TEST_F(DriSurfaceTest, CheckFBIDOnSwap) {
  71   surface_->PresentCanvas(gfx::Rect());
  72   // Framebuffer ID 1 is allocated in SetUp for the buffer used to modeset.
  73   EXPECT_EQ(3u, drm_->current_framebuffer());
  74   surface_->PresentCanvas(gfx::Rect());
  75   EXPECT_EQ(2u, drm_->current_framebuffer());
  76 }
  77
  78 TEST_F(DriSurfaceTest, CheckSurfaceContents) {
  79   SkPaint paint;
  80   paint.setColor(SK_ColorWHITE);
  81   SkRect rect = SkRect::MakeWH(kDefaultMode.hdisplay / 2,
  82                                kDefaultMode.vdisplay / 2);
  83   surface_->GetCanvas()->drawRect(rect, paint);
  84   surface_->PresentCanvas(
  85       gfx::Rect(0, 0, kDefaultMode.hdisplay / 2, kDefaultMode.vdisplay / 2));
  86
  87   SkBitmap image;
  88   // Buffer 0 is the buffer used in SetUp for modesetting and buffer 1 is the
  89   // frontbuffer.
  90   // Buffer 2 is the backbuffer we just painted in, so we want to make sure its
  91   // contents are correct.
  92   image.setInfo(drm_->buffers()[2]->getCanvas()->imageInfo());
  93   EXPECT_TRUE(drm_->buffers()[2]->getCanvas()->readPixels(&image, 0, 0));
  94
  95   EXPECT_EQ(kDefaultMode.hdisplay, image.width());
  96   EXPECT_EQ(kDefaultMode.vdisplay, image.height());
  97
  98   // Make sure the updates are correctly propagated to the native surface.
  99   for (int i = 0; i < image.height(); ++i) {
 100     for (int j = 0; j < image.width(); ++j) {
 101       if (j < kDefaultMode.hdisplay / 2 && i < kDefaultMode.vdisplay / 2)
 102         EXPECT_EQ(SK_ColorWHITE, image.getColor(j, i));
 103       else
 104         EXPECT_EQ(SK_ColorBLACK, image.getColor(j, i));
 105     }
 106   }
 107 }