ui/ozone/public/ozone_gpu_test_helper.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/public/ozone_gpu_test_helper.h"
   6
   7 #include "base/thread_task_runner_handle.h"
   8 #include "ipc/ipc_listener.h"
   9 #include "ipc/ipc_message.h"
  10 #include "ipc/ipc_sender.h"
  11 #include "ipc/message_filter.h"
  12 #include "ui/ozone/public/gpu_platform_support.h"
  13 #include "ui/ozone/public/gpu_platform_support_host.h"
  14 #include "ui/ozone/public/ozone_platform.h"
  15
  16 namespace ui {
  17
  18 namespace {
  19
  20 const int kGpuProcessHostId = 1;
  21
  22 }  // namespace
  23
  24 static void DispatchToGpuPlatformSupportHostTask(IPC::Message* msg) {
  25   ui::OzonePlatform::GetInstance()
  26       ->GetGpuPlatformSupportHost()
  27       ->OnMessageReceived(*msg);
  28   delete msg;
  29 }
  30
  31 class FakeGpuProcess : public IPC::Sender {
  32  public:
  33   FakeGpuProcess(
  34       const scoped_refptr<base::SingleThreadTaskRunner>& ui_task_runner)
  35       : ui_task_runner_(ui_task_runner) {}
  36   ~FakeGpuProcess() override {}
  37
  38   void Init() {
  39     ui::OzonePlatform::GetInstance()
  40         ->GetGpuPlatformSupport()
  41         ->OnChannelEstablished(this);
  42   }
  43
  44   void InitOnIO() {
  45     IPC::MessageFilter* filter = ui::OzonePlatform::GetInstance()
  46                                      ->GetGpuPlatformSupport()
  47                                      ->GetMessageFilter();
  48
  49     if (filter)
  50       filter->OnFilterAdded(this);
  51   }
  52
  53   bool Send(IPC::Message* msg) override {
  54     ui_task_runner_->PostTask(
  55         FROM_HERE, base::Bind(&DispatchToGpuPlatformSupportHostTask, msg));
  56     return true;
  57   }
  58
  59  private:
  60   scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner_;
  61 };
  62
  63 static void DispatchToGpuPlatformSupportTask(IPC::Message* msg) {
  64   ui::OzonePlatform::GetInstance()->GetGpuPlatformSupport()->OnMessageReceived(
  65       *msg);
  66   delete msg;
  67 }
  68
  69 class FakeGpuProcessHost {
  70  public:
  71   FakeGpuProcessHost(
  72       const scoped_refptr<base::SingleThreadTaskRunner>& gpu_task_runner)
  73       : gpu_task_runner_(gpu_task_runner) {}
  74   ~FakeGpuProcessHost() {}
  75
  76   void Init() {
  77     base::Callback<void(IPC::Message*)> sender =
  78         base::Bind(&DispatchToGpuPlatformSupportTask);
  79
  80     ui::OzonePlatform::GetInstance()
  81         ->GetGpuPlatformSupportHost()
  82         ->OnChannelEstablished(kGpuProcessHostId, gpu_task_runner_, sender);
  83   }
  84
  85  private:
  86   scoped_refptr<base::SingleThreadTaskRunner> gpu_task_runner_;
  87 };
  88
  89 OzoneGpuTestHelper::OzoneGpuTestHelper() {
  90 }
  91
  92 OzoneGpuTestHelper::~OzoneGpuTestHelper() {
  93 }
  94
  95 bool OzoneGpuTestHelper::Initialize(
  96     const scoped_refptr<base::SingleThreadTaskRunner>& ui_task_runner,
  97     const scoped_refptr<base::SingleThreadTaskRunner>& gpu_task_runner) {
  98   io_helper_thread_.reset(new base::Thread("IOHelperThread"));
  99   if (!io_helper_thread_->StartWithOptions(
 100           base::Thread::Options(base::MessageLoop::TYPE_IO, 0)))
 101     return false;
 102
 103   fake_gpu_process_.reset(new FakeGpuProcess(ui_task_runner));
 104   io_helper_thread_->task_runner()->PostTask(
 105       FROM_HERE, base::Bind(&FakeGpuProcess::InitOnIO,
 106                             base::Unretained(fake_gpu_process_.get())));
 107   fake_gpu_process_->Init();
 108
 109   fake_gpu_process_host_.reset(new FakeGpuProcessHost(gpu_task_runner));
 110   fake_gpu_process_host_->Init();
 111
 112   return true;
 113 }
 114
 115 }  // namespace ui