cc/test/test_shared_bitmap_manager.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 "cc/test/test_shared_bitmap_manager.h"

#include <stdint.h>

#include "base/memory/ptr_util.h"
#include "base/memory/shared_memory.h"

namespace cc {

namespace {

static uint32_t g_next_sequence_number = 1;

class OwnedSharedBitmap : public viz::SharedBitmap {
 public:
  OwnedSharedBitmap(std::unique_ptr<base::SharedMemory> shared_memory,
                    const viz::SharedBitmapId& id)
      : viz::SharedBitmap(static_cast<uint8_t*>(shared_memory->memory()),
                          id,
                          g_next_sequence_number++),
        shared_memory_(std::move(shared_memory)) {}

  ~OwnedSharedBitmap() override = default;

  // viz::SharedBitmap:
  base::SharedMemoryHandle GetSharedMemoryHandle() const override {
    return shared_memory_->handle();
  }

 private:
  std::unique_ptr<base::SharedMemory> shared_memory_;
};

class UnownedSharedBitmap : public viz::SharedBitmap {
 public:
  UnownedSharedBitmap(uint8_t* pixels, const viz::SharedBitmapId& id)
      : viz::SharedBitmap(pixels, id, g_next_sequence_number++) {}

  // viz::SharedBitmap:
  base::SharedMemoryHandle GetSharedMemoryHandle() const override {
    return base::SharedMemoryHandle();
  }
};

}  // namespace

TestSharedBitmapManager::TestSharedBitmapManager() = default;

TestSharedBitmapManager::~TestSharedBitmapManager() = default;

std::unique_ptr<viz::SharedBitmap>
TestSharedBitmapManager::AllocateSharedBitmap(const gfx::Size& size) {
  base::AutoLock lock(lock_);
  std::unique_ptr<base::SharedMemory> memory(new base::SharedMemory);
  memory->CreateAndMapAnonymous(size.GetArea() * 4);
  viz::SharedBitmapId id = viz::SharedBitmap::GenerateId();
  bitmap_map_[id] = memory.get();
  return std::make_unique<OwnedSharedBitmap>(std::move(memory), id);
}

std::unique_ptr<viz::SharedBitmap>
TestSharedBitmapManager::GetSharedBitmapFromId(const gfx::Size&,
                                               const viz::SharedBitmapId& id) {
  base::AutoLock lock(lock_);
  if (bitmap_map_.find(id) == bitmap_map_.end())
    return nullptr;
  uint8_t* pixels = static_cast<uint8_t*>(bitmap_map_[id]->memory());
  return std::make_unique<UnownedSharedBitmap>(pixels, id);
}

}  // namespace cc