ui/compositor/test/test_compositor_host_mac.mm

   1 // Copyright (c) 2012 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/compositor/test/test_compositor_host.h"
   6
   7 #import <AppKit/NSApplication.h>
   8 #import <AppKit/NSOpenGL.h>
   9 #import <AppKit/NSView.h>
  10 #import <AppKit/NSWindow.h>
  11 #import <Foundation/NSAutoreleasePool.h>
  12
  13 #include "base/compiler_specific.h"
  14 #include "base/mac/scoped_nsobject.h"
  15 #include "base/memory/scoped_ptr.h"
  16 #include "base/thread_task_runner_handle.h"
  17 #include "ui/compositor/compositor.h"
  18 #include "ui/gfx/geometry/rect.h"
  19
  20 // AcceleratedTestView provides an NSView class that delegates drawing to a
  21 // ui::Compositor delegate, setting up the NSOpenGLContext as required.
  22 @interface AcceleratedTestView : NSView {
  23   ui::Compositor* compositor_;
  24 }
  25 // Designated initializer.
  26 -(id)init;
  27 -(void)setCompositor:(ui::Compositor*)compositor;
  28 @end
  29
  30 @implementation AcceleratedTestView
  31 -(id)init {
  32   // The frame will be resized when reparented into the window's view hierarchy.
  33   self = [super initWithFrame:NSZeroRect];
  34   return self;
  35 }
  36
  37 -(void)setCompositor:(ui::Compositor*)compositor {
  38   compositor_ = compositor;
  39 }
  40
  41 - (void)drawRect:(NSRect)rect {
  42   DCHECK(compositor_) << "Drawing with no compositor set.";
  43   compositor_->Draw();
  44 }
  45 @end
  46
  47 namespace ui {
  48
  49 // Tests that use Objective-C memory semantics need to have a top-level
  50 // NSAutoreleasePool set up and initialized prior to execution and drained upon
  51 // exit.  The tests will leak otherwise.
  52 class FoundationHost {
  53  protected:
  54   FoundationHost() {
  55     pool_ = [[NSAutoreleasePool alloc] init];
  56   }
  57   virtual ~FoundationHost() {
  58     [pool_ drain];
  59   }
  60
  61  private:
  62   NSAutoreleasePool* pool_;
  63   DISALLOW_COPY_AND_ASSIGN(FoundationHost);
  64 };
  65
  66 // Tests that use the AppKit framework need to have the NSApplication
  67 // initialized prior to doing anything with display objects such as windows,
  68 // views, or controls.
  69 class AppKitHost : public FoundationHost {
  70  protected:
  71   AppKitHost() {
  72     [NSApplication sharedApplication];
  73   }
  74   ~AppKitHost() override {}
  75  private:
  76   DISALLOW_COPY_AND_ASSIGN(AppKitHost);
  77 };
  78
  79 // TestCompositorHostMac provides a window surface and a coordinated compositor
  80 // for use in the compositor unit tests.
  81 class TestCompositorHostMac : public TestCompositorHost,
  82                               public AppKitHost {
  83  public:
  84   TestCompositorHostMac(const gfx::Rect& bounds,
  85                         ui::ContextFactory* context_factory);
  86   ~TestCompositorHostMac() override;
  87
  88  private:
  89   // TestCompositorHost:
  90   void Show() override;
  91   ui::Compositor* GetCompositor() override;
  92
  93   gfx::Rect bounds_;
  94
  95   ui::ContextFactory* context_factory_;
  96
  97   scoped_ptr<ui::Compositor> compositor_;
  98
  99   // Owned.  Released when window is closed.
 100   NSWindow* window_;
 101
 102   DISALLOW_COPY_AND_ASSIGN(TestCompositorHostMac);
 103 };
 104
 105 TestCompositorHostMac::TestCompositorHostMac(
 106     const gfx::Rect& bounds,
 107     ui::ContextFactory* context_factory)
 108     : bounds_(bounds), context_factory_(context_factory), window_(nil) {
 109 }
 110
 111 TestCompositorHostMac::~TestCompositorHostMac() {
 112   // Release reference to |compositor_|.  Important because the |compositor_|
 113   // holds |this| as its delegate, so that reference must be removed here.
 114   [[window_ contentView] setCompositor:NULL];
 115   [window_ setContentView:nil];
 116
 117   [window_ orderOut:nil];
 118   [window_ close];
 119 }
 120
 121 void TestCompositorHostMac::Show() {
 122   DCHECK(!window_);
 123   window_ = [[NSWindow alloc]
 124                 initWithContentRect:NSMakeRect(bounds_.x(),
 125                                                bounds_.y(),
 126                                                bounds_.width(),
 127                                                bounds_.height())
 128                           styleMask:NSBorderlessWindowMask
 129                             backing:NSBackingStoreBuffered
 130                               defer:NO];
 131   base::scoped_nsobject<AcceleratedTestView> view(
 132       [[AcceleratedTestView alloc] init]);
 133   compositor_.reset(new ui::Compositor(view,
 134                                        context_factory_,
 135                                        base::ThreadTaskRunnerHandle::Get()));
 136   compositor_->SetScaleAndSize(1.0f, bounds_.size());
 137   [view setCompositor:compositor_.get()];
 138   [window_ setContentView:view];
 139   [window_ orderFront:nil];
 140 }
 141
 142 ui::Compositor* TestCompositorHostMac::GetCompositor() {
 143   return compositor_.get();
 144 }
 145
 146 // static
 147 TestCompositorHost* TestCompositorHost::Create(
 148     const gfx::Rect& bounds,
 149     ui::ContextFactory* context_factory) {
 150   return new TestCompositorHostMac(bounds, context_factory);
 151 }
 152
 153 }  // namespace ui