cc/test/test_shared_bitmap_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 "cc/test/test_shared_bitmap_manager.h"
   6
   7 #include "base/memory/shared_memory.h"
   8
   9 namespace cc {
  10
  11 namespace {
  12 class OwnedSharedBitmap : public SharedBitmap {
  13  public:
  14   OwnedSharedBitmap(scoped_ptr<base::SharedMemory> shared_memory,
  15                     const SharedBitmapId& id)
  16       : SharedBitmap(static_cast<uint8*>(shared_memory->memory()), id),
  17         shared_memory_(shared_memory.Pass()) {}
  18
  19   ~OwnedSharedBitmap() override {}
  20
  21   base::SharedMemory* memory() override { return shared_memory_.get(); }
  22
  23  private:
  24   scoped_ptr<base::SharedMemory> shared_memory_;
  25 };
  26
  27 class UnownedSharedBitmap : public SharedBitmap {
  28  public:
  29   UnownedSharedBitmap(base::SharedMemory* shared_memory,
  30                       const SharedBitmapId& id)
  31       : SharedBitmap(static_cast<uint8*>(shared_memory->memory()), id),
  32         shared_memory_(shared_memory) {}
  33
  34   ~UnownedSharedBitmap() override {}
  35
  36   base::SharedMemory* memory() override { return shared_memory_; }
  37
  38  private:
  39   base::SharedMemory* shared_memory_;
  40 };
  41 }  // namespace
  42
  43 TestSharedBitmapManager::TestSharedBitmapManager() {}
  44
  45 TestSharedBitmapManager::~TestSharedBitmapManager() {}
  46
  47 scoped_ptr<SharedBitmap> TestSharedBitmapManager::AllocateSharedBitmap(
  48     const gfx::Size& size) {
  49   base::AutoLock lock(lock_);
  50   scoped_ptr<base::SharedMemory> memory(new base::SharedMemory);
  51   memory->CreateAndMapAnonymous(size.GetArea() * 4);
  52   SharedBitmapId id = SharedBitmap::GenerateId();
  53   bitmap_map_[id] = memory.get();
  54   return make_scoped_ptr(new OwnedSharedBitmap(memory.Pass(), id));
  55 }
  56
  57 scoped_ptr<SharedBitmap> TestSharedBitmapManager::GetSharedBitmapFromId(
  58     const gfx::Size&,
  59     const SharedBitmapId& id) {
  60   base::AutoLock lock(lock_);
  61   if (bitmap_map_.find(id) == bitmap_map_.end())
  62     return nullptr;
  63   return make_scoped_ptr(new UnownedSharedBitmap(bitmap_map_[id], id));
  64 }
  65
  66 }  // namespace cc