ui/ozone/platform/test/test_window_manager.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 "ui/ozone/platform/test/test_window_manager.h"
   6
   7 #include "base/bind.h"
   8 #include "base/files/file_util.h"
   9 #include "base/location.h"
  10 #include "base/stl_util.h"
  11 #include "base/threading/worker_pool.h"
  12 #include "third_party/skia/include/core/SkCanvas.h"
  13 #include "third_party/skia/include/core/SkSurface.h"
  14 #include "ui/gfx/codec/png_codec.h"
  15 #include "ui/gfx/skia_util.h"
  16 #include "ui/gfx/vsync_provider.h"
  17 #include "ui/ozone/public/surface_ozone_canvas.h"
  18
  19 namespace ui {
  20
  21 namespace {
  22
  23 void WriteDataToFile(const base::FilePath& location, const SkBitmap& bitmap) {
  24   DCHECK(!location.empty());
  25   std::vector<unsigned char> png_data;
  26   gfx::PNGCodec::FastEncodeBGRASkBitmap(bitmap, true, &png_data);
  27   base::WriteFile(location,
  28                   reinterpret_cast<const char*>(vector_as_array(&png_data)),
  29                   png_data.size());
  30 }
  31
  32 class FileSurface : public SurfaceOzoneCanvas {
  33  public:
  34   FileSurface(const base::FilePath& location) : location_(location) {}
  35   ~FileSurface() override {}
  36
  37   // SurfaceOzoneCanvas overrides:
  38   void ResizeCanvas(const gfx::Size& viewport_size) override {
  39     surface_ = skia::AdoptRef(SkSurface::NewRaster(SkImageInfo::MakeN32Premul(
  40         viewport_size.width(), viewport_size.height())));
  41   }
  42   skia::RefPtr<SkSurface> GetSurface() override { return surface_; }
  43   void PresentCanvas(const gfx::Rect& damage) override {
  44     if (location_.empty())
  45       return;
  46     SkBitmap bitmap;
  47     bitmap.setInfo(surface_->getCanvas()->imageInfo());
  48
  49     // TODO(dnicoara) Use SkImage instead to potentially avoid a copy.
  50     // See crbug.com/361605 for details.
  51     if (surface_->getCanvas()->readPixels(&bitmap, 0, 0)) {
  52       base::WorkerPool::PostTask(
  53           FROM_HERE, base::Bind(&WriteDataToFile, location_, bitmap), true);
  54     }
  55   }
  56   scoped_ptr<gfx::VSyncProvider> CreateVSyncProvider() override {
  57     return nullptr;
  58   }
  59
  60  private:
  61   base::FilePath location_;
  62   skia::RefPtr<SkSurface> surface_;
  63 };
  64
  65 }  // namespace
  66
  67 TestWindowManager::TestWindowManager(const base::FilePath& dump_location)
  68     : location_(dump_location) {
  69 }
  70
  71 TestWindowManager::~TestWindowManager() {
  72   DCHECK(thread_checker_.CalledOnValidThread());
  73 }
  74
  75 void TestWindowManager::Initialize() {
  76   if (location_.empty())
  77     return;
  78   if (!DirectoryExists(location_) && !base::CreateDirectory(location_) &&
  79       location_ != base::FilePath("/dev/null"))
  80     PLOG(FATAL) << "unable to create output directory";
  81   if (!base::PathIsWritable(location_))
  82     PLOG(FATAL) << "unable to write to output location";
  83 }
  84
  85 int32_t TestWindowManager::AddWindow(TestWindow* window) {
  86   return windows_.Add(window);
  87 }
  88
  89 void TestWindowManager::RemoveWindow(int32_t window_id, TestWindow* window) {
  90   DCHECK_EQ(window, windows_.Lookup(window_id));
  91   windows_.Remove(window_id);
  92 }
  93
  94 base::FilePath TestWindowManager::base_path() const {
  95   return location_;
  96 }
  97
  98 scoped_ptr<SurfaceOzoneCanvas> TestWindowManager::CreateCanvasForWidget(
  99     gfx::AcceleratedWidget widget) {
 100   DCHECK(thread_checker_.CalledOnValidThread());
 101   TestWindow* window = windows_.Lookup(widget);
 102   DCHECK(window);
 103   return make_scoped_ptr<SurfaceOzoneCanvas>(new FileSurface(window->path()));
 104 }
 105
 106 bool TestWindowManager::LoadEGLGLES2Bindings(
 107     AddGLLibraryCallback add_gl_library,
 108     SetGLGetProcAddressProcCallback set_gl_get_proc_address) {
 109   DCHECK(thread_checker_.CalledOnValidThread());
 110   return false;
 111 }
 112
 113 }  // namespace ui