| // Copyright 2018 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_VMA_GPU_ALLOCATOR_H_ |
| #define SRC_UI_LIB_ESCHER_VK_VMA_GPU_ALLOCATOR_H_ |
| |
| #include <optional> |
| |
| #include "src/ui/lib/escher/third_party/VulkanMemoryAllocator/vk_mem_alloc.h" |
| #include "src/ui/lib/escher/vk/gpu_allocator.h" |
| #include "src/ui/lib/escher/vk/vulkan_context.h" |
| |
| #include <vulkan/vulkan.hpp> |
| |
| namespace escher { |
| |
| // This class implements the GpuAllocator interface on top of a vk_mem_alloc |
| // instance. |
| class VmaGpuAllocator : public GpuAllocator { |
| public: |
| VmaGpuAllocator(const VulkanContext& context); |
| ~VmaGpuAllocator(); |
| |
| // |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| |
| size_t GetTotalBytesAllocated() const override; |
| |
| // |GpuAllocator| |
| size_t GetUnusedBytesAllocated() const override; |
| |
| private: |
| virtual bool CreateImage(const VkImageCreateInfo& image_create_info, |
| const VmaAllocationCreateInfo& allocation_create_info, VkImage* image, |
| VmaAllocation* vma_allocation, VmaAllocationInfo* vma_allocation_info); |
| |
| // Some platforms / ICDs may have extra requirements on memory types for |
| // specific image type / formats. |
| // Returns a correct memory type mask bits for VmaAllocationCreateInfo. |
| // Returns 0 if there is no restrictions on memory type. |
| uint32_t GetMemoryTypeBitsMask(const escher::ImageInfo& info); |
| #ifdef VK_USE_PLATFORM_MACOS_MVK |
| // Currently there are memory type requirements on MoltenVK devices, and we |
| // only store the value on these devices to avoid duplicated calculation of |
| // memory type bits. |
| std::optional<uint32_t> memory_type_bits_mask_; |
| #endif |
| |
| vk::PhysicalDevice physical_device_; |
| VmaAllocator allocator_; |
| }; |
| |
| } // namespace escher |
| |
| #endif // SRC_UI_LIB_ESCHER_VK_VMA_GPU_ALLOCATOR_H_ |