[Ozone-DRI] Listen for swap events

ui/ozone/platform/dri/test/mock_dri_wrapper.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 "ui/ozone/platform/dri/test/mock_dri_wrapper.h"
 
 #include <xf86drm.h>
 #include <xf86drmMode.h>
 
 #include "base/logging.h"
 #include "third_party/skia/include/core/SkCanvas.h"
-#include "ui/ozone/platform/dri/crtc_controller.h"
 #include "ui/ozone/platform/dri/hardware_display_plane_manager_legacy.h"
 
 namespace ui {
 
 namespace {
 
 template<class Object> Object* DrmAllocator() {
   return static_cast<Object*>(drmMalloc(sizeof(Object)));
 }
 
@@ skipping 11 matching lines @@
         planes_.push_back(
             new HardwareDisplayPlane(kPlaneBaseId + i, 1 << crtc_idx));
       }
     }
   }
 };
 
 }  // namespace
 
 MockDriWrapper::MockDriWrapper(int fd)
-    : DriWrapper(""),
+    : DriWrapper("", true),
       get_crtc_call_count_(0),
       set_crtc_call_count_(0),
       restore_crtc_call_count_(0),
       add_framebuffer_call_count_(0),
       remove_framebuffer_call_count_(0),
       page_flip_call_count_(0),
       overlay_flip_call_count_(0),
-      handle_events_count_(0),
       set_crtc_expectation_(true),
       add_framebuffer_expectation_(true),
       page_flip_expectation_(true),
       create_dumb_buffer_expectation_(true),
       current_framebuffer_(0) {
   fd_ = fd;
   plane_manager_.reset(new HardwareDisplayPlaneManagerLegacy());
 }
 
 MockDriWrapper::MockDriWrapper(int fd,
                                std::vector<uint32_t> crtcs,
                                size_t planes_per_crtc)
-    : DriWrapper(""),
+    : DriWrapper("", true),
       get_crtc_call_count_(0),
       set_crtc_call_count_(0),
       restore_crtc_call_count_(0),
       add_framebuffer_call_count_(0),
       remove_framebuffer_call_count_(0),
       page_flip_call_count_(0),
       overlay_flip_call_count_(0),
-      handle_events_count_(0),
       set_crtc_expectation_(true),
       add_framebuffer_expectation_(true),
       page_flip_expectation_(true),
       create_dumb_buffer_expectation_(true),
       current_framebuffer_(0) {
   fd_ = fd;
   plane_manager_.reset(
       new MockHardwareDisplayPlaneManager(this, crtcs, planes_per_crtc));
 }
 
@@ skipping 46 matching lines @@
   remove_framebuffer_call_count_++;
   return true;
 }
 
 ScopedDrmFramebufferPtr MockDriWrapper::GetFramebuffer(uint32_t framebuffer) {
   return ScopedDrmFramebufferPtr();
 }
 
 bool MockDriWrapper::PageFlip(uint32_t crtc_id,
                               uint32_t framebuffer,
-                              void* data) {
+                              const PageFlipCallback& callback) {
   page_flip_call_count_++;
   current_framebuffer_ = framebuffer;
-  controllers_.push(static_cast<ui::CrtcController*>(data));
+  if (page_flip_expectation_)
+    callback.Run(0, 0, 0);
+
   return page_flip_expectation_;
 }
 
 bool MockDriWrapper::PageFlipOverlay(uint32_t crtc_id,
                                      uint32_t framebuffer,
                                      const gfx::Rect& location,
                                      const gfx::Rect& source,
                                      int overlay_plane) {
   overlay_flip_call_count_++;
   return true;
@@ skipping 23 matching lines @@
 bool MockDriWrapper::SetCursor(uint32_t crtc_id,
                                uint32_t handle,
                                const gfx::Size& size) {
   return true;
 }
 
 bool MockDriWrapper::MoveCursor(uint32_t crtc_id, const gfx::Point& point) {
   return true;
 }
 
-void MockDriWrapper::HandleEvent(drmEventContext& event) {
-  CHECK(!controllers_.empty());
-  controllers_.front()->OnPageFlipEvent(0, 0, 0);
-  controllers_.pop();
-  handle_events_count_++;
-}
-
 bool MockDriWrapper::CreateDumbBuffer(const SkImageInfo& info,
                                       uint32_t* handle,
                                       uint32_t* stride,
                                       void** pixels) {
   if (!create_dumb_buffer_expectation_)
     return false;
 
   *handle = 0;
   *stride = info.minRowBytes();
   *pixels = new char[info.getSafeSize(*stride)];
   buffers_.push_back(
       skia::AdoptRef(SkSurface::NewRasterDirect(info, *pixels, *stride)));
   buffers_.back()->getCanvas()->clear(SK_ColorBLACK);
 
   return true;
 }
 
 void MockDriWrapper::DestroyDumbBuffer(const SkImageInfo& info,
                                        uint32_t handle,
                                        uint32_t stride,
                                        void* pixels) {
   delete[] static_cast<char*>(pixels);
 }
 
 }  // namespace ui