// Copyright 2016 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SRC_UI_LIB_ESCHER_VK_GPU_MEM_H_
#define SRC_UI_LIB_ESCHER_VK_GPU_MEM_H_

#include <vulkan/vulkan.hpp>

#include "src/lib/fxl/macros.h"
#include "src/lib/fxl/memory/ref_counted.h"

namespace escher {

class GpuMem;
typedef fxl::RefPtr<GpuMem> GpuMemPtr;

// Ref-counted wrapper around a vk::DeviceMemory.  Supports sub-allocation.
class GpuMem : public fxl::RefCountedThreadSafe<GpuMem> {
 public:
  // Create a GpuMem that takes ownership of |mem|, which will be destroyed when
  // the GpuMem dies. Guaranteed to return a non-null result.
  static GpuMemPtr AdoptVkMemory(vk::Device device, vk::DeviceMemory mem, vk::DeviceSize size,
                                 bool needs_mapped_ptr);

  // Sub-allocate a GpuMem that represents a sub-range of the memory in this
  // GpuMem.  Since these sub-allocations reference the parent GpuMem, the
  // parent will not be destroyed while outstanding sub-allocations exist.
  // Returns nullptr if the requested offset/size do not fit within the current
  // GpuMem.
  // Note: no bookkeeping ensures that sub-allocations do not overlap!
  GpuMemPtr Suballocate(vk::DeviceSize size, vk::DeviceSize offset);

  vk::DeviceMemory base() const { return base_; }
  vk::DeviceSize size() const { return size_; }
  vk::DeviceSize offset() const { return offset_; }
  uint8_t* mapped_ptr() const { return mapped_ptr_; }

 protected:
  // |offset| + |size| must be <= the size of |base|. This class does not
  // takes ownership of |base| by default.
  GpuMem(vk::DeviceMemory base, vk::DeviceSize size, vk::DeviceSize offset, uint8_t* mapped_ptr);

  FRIEND_REF_COUNTED_THREAD_SAFE(GpuMem);
  virtual ~GpuMem();

 private:
  vk::DeviceMemory base_;
  vk::DeviceSize size_;
  vk::DeviceSize offset_;
  uint8_t* mapped_ptr_;

  FXL_DISALLOW_COPY_AND_ASSIGN(GpuMem);
};

}  // namespace escher

#endif  // SRC_UI_LIB_ESCHER_VK_GPU_MEM_H_