| // 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 LIB_ESCHER_VK_NAIVE_GPU_ALLOCATOR_H_ |
| #define LIB_ESCHER_VK_NAIVE_GPU_ALLOCATOR_H_ |
| |
| #include <vulkan/vulkan.hpp> |
| |
| #include "lib/escher/impl/gpu_mem_slab.h" |
| #include "lib/escher/vk/gpu_allocator.h" |
| #include "lib/escher/vk/vulkan_context.h" |
| |
| namespace escher { |
| |
| // NaiveGpuAllocator uses a separate GpuMemSlab for each GpuMem that it |
| // allocates. This ignores Vulkan best practices, and is a placeholder until a |
| // more sophisticated allocator is written. |
| class NaiveGpuAllocator : public GpuAllocator { |
| public: |
| NaiveGpuAllocator(const VulkanContext& context); |
| ~NaiveGpuAllocator(); |
| |
| // |GpuAllocator| |
| GpuMemPtr AllocateMemory(vk::MemoryRequirements reqs, |
| vk::MemoryPropertyFlags flags) override; |
| |
| // |GpuAllocator| |
| BufferPtr AllocateBuffer(ResourceManager* manager, vk::DeviceSize size, |
| vk::BufferUsageFlags usage_flags, |
| vk::MemoryPropertyFlags memory_property_flags, |
| GpuMemPtr* out_ptr) override; |
| |
| // |GpuAllocator| |
| ImagePtr AllocateImage(ResourceManager* manager, |
| const escher::ImageInfo& info, |
| GpuMemPtr* out_ptr) override; |
| |
| // |GpuAllocator| |
| uint32_t GetTotalBytesAllocated() const override; |
| |
| private: |
| // Callbacks to allow a GpuMemSlab to notify its GpuAllocator of changes. |
| friend class impl::GpuMemSlab; |
| void OnSlabCreated(vk::DeviceSize slab_size); |
| void OnSlabDestroyed(vk::DeviceSize slab_size); |
| |
| vk::PhysicalDevice physical_device_; |
| vk::Device device_; |
| vk::DeviceSize total_slab_bytes_ = 0; |
| size_t slab_count_ = 0; |
| }; |
| |
| } // namespace escher |
| |
| #endif // LIB_ESCHER_VK_NAIVE_GPU_ALLOCATOR_H_ |