2019-12-25 22:12:17 +01:00
|
|
|
// Copyright 2019 yuzu Emulator Project
|
|
|
|
// Licensed under GPLv2 or any later version
|
|
|
|
// Refer to the license.txt file included.
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <climits>
|
|
|
|
#include <vector>
|
|
|
|
|
|
|
|
#include "common/common_types.h"
|
|
|
|
|
2021-01-03 22:17:57 +01:00
|
|
|
#include "video_core/vulkan_common/vulkan_memory_allocator.h"
|
2020-12-25 01:30:11 +01:00
|
|
|
#include "video_core/vulkan_common/vulkan_wrapper.h"
|
2019-12-25 22:12:17 +01:00
|
|
|
|
|
|
|
namespace Vulkan {
|
|
|
|
|
2020-12-26 05:10:53 +01:00
|
|
|
class Device;
|
2019-12-25 22:12:17 +01:00
|
|
|
class VKScheduler;
|
|
|
|
|
2020-12-31 02:58:05 +01:00
|
|
|
struct StagingBufferRef {
|
|
|
|
VkBuffer buffer;
|
2021-01-16 20:20:18 +01:00
|
|
|
VkDeviceSize offset;
|
2020-12-31 02:58:05 +01:00
|
|
|
std::span<u8> mapped_span;
|
2019-12-25 22:12:17 +01:00
|
|
|
};
|
|
|
|
|
2020-12-31 02:58:05 +01:00
|
|
|
class StagingBufferPool {
|
2019-12-25 22:12:17 +01:00
|
|
|
public:
|
2021-01-16 20:20:18 +01:00
|
|
|
static constexpr size_t NUM_SYNCS = 16;
|
|
|
|
|
2021-01-03 22:11:01 +01:00
|
|
|
explicit StagingBufferPool(const Device& device, MemoryAllocator& memory_allocator,
|
2020-12-31 02:58:05 +01:00
|
|
|
VKScheduler& scheduler);
|
|
|
|
~StagingBufferPool();
|
2019-12-25 22:12:17 +01:00
|
|
|
|
2021-01-03 22:38:15 +01:00
|
|
|
StagingBufferRef Request(size_t size, MemoryUsage usage);
|
2019-12-25 22:12:17 +01:00
|
|
|
|
|
|
|
void TickFrame();
|
|
|
|
|
|
|
|
private:
|
2021-01-16 20:20:18 +01:00
|
|
|
struct StreamBufferCommit {
|
|
|
|
size_t upper_bound;
|
|
|
|
u64 tick;
|
|
|
|
};
|
|
|
|
|
2020-12-31 02:58:05 +01:00
|
|
|
struct StagingBuffer {
|
|
|
|
vk::Buffer buffer;
|
|
|
|
MemoryCommit commit;
|
|
|
|
std::span<u8> mapped_span;
|
2020-09-10 08:43:30 +02:00
|
|
|
u64 tick = 0;
|
2020-12-31 02:58:05 +01:00
|
|
|
|
|
|
|
StagingBufferRef Ref() const noexcept {
|
|
|
|
return {
|
|
|
|
.buffer = *buffer,
|
2021-01-16 20:20:18 +01:00
|
|
|
.offset = 0,
|
2020-12-31 02:58:05 +01:00
|
|
|
.mapped_span = mapped_span,
|
|
|
|
};
|
|
|
|
}
|
2019-12-25 22:12:17 +01:00
|
|
|
};
|
|
|
|
|
2020-12-31 02:58:05 +01:00
|
|
|
struct StagingBuffers {
|
2019-12-25 22:12:17 +01:00
|
|
|
std::vector<StagingBuffer> entries;
|
2020-12-31 02:58:05 +01:00
|
|
|
size_t delete_index = 0;
|
|
|
|
size_t iterate_index = 0;
|
2019-12-25 22:12:17 +01:00
|
|
|
};
|
|
|
|
|
2020-12-31 02:58:05 +01:00
|
|
|
static constexpr size_t NUM_LEVELS = sizeof(size_t) * CHAR_BIT;
|
|
|
|
using StagingBuffersCache = std::array<StagingBuffers, NUM_LEVELS>;
|
2019-12-25 22:12:17 +01:00
|
|
|
|
2021-01-16 20:20:18 +01:00
|
|
|
StagingBufferRef GetStreamBuffer(size_t size);
|
|
|
|
|
2021-01-23 21:59:32 +01:00
|
|
|
bool AreRegionsActive(size_t region_begin, size_t region_end) const;
|
|
|
|
|
2021-01-21 05:07:11 +01:00
|
|
|
StagingBufferRef GetStagingBuffer(size_t size, MemoryUsage usage);
|
|
|
|
|
2021-01-03 22:38:15 +01:00
|
|
|
std::optional<StagingBufferRef> TryGetReservedBuffer(size_t size, MemoryUsage usage);
|
2019-12-25 22:12:17 +01:00
|
|
|
|
2021-01-03 22:38:15 +01:00
|
|
|
StagingBufferRef CreateStagingBuffer(size_t size, MemoryUsage usage);
|
2019-12-25 22:12:17 +01:00
|
|
|
|
2021-01-03 22:38:15 +01:00
|
|
|
StagingBuffersCache& GetCache(MemoryUsage usage);
|
2019-12-25 22:12:17 +01:00
|
|
|
|
2021-01-03 22:38:15 +01:00
|
|
|
void ReleaseCache(MemoryUsage usage);
|
2019-12-25 22:12:17 +01:00
|
|
|
|
2020-12-31 02:58:05 +01:00
|
|
|
void ReleaseLevel(StagingBuffersCache& cache, size_t log2);
|
2019-12-25 22:12:17 +01:00
|
|
|
|
2020-12-26 05:10:53 +01:00
|
|
|
const Device& device;
|
2021-01-03 22:11:01 +01:00
|
|
|
MemoryAllocator& memory_allocator;
|
2019-12-25 22:12:17 +01:00
|
|
|
VKScheduler& scheduler;
|
|
|
|
|
2021-01-16 20:20:18 +01:00
|
|
|
vk::Buffer stream_buffer;
|
|
|
|
vk::DeviceMemory stream_memory;
|
|
|
|
u8* stream_pointer = nullptr;
|
|
|
|
|
|
|
|
size_t iterator = 0;
|
|
|
|
size_t used_iterator = 0;
|
|
|
|
size_t free_iterator = 0;
|
|
|
|
std::array<u64, NUM_SYNCS> sync_ticks{};
|
|
|
|
|
2021-01-03 22:38:15 +01:00
|
|
|
StagingBuffersCache device_local_cache;
|
|
|
|
StagingBuffersCache upload_cache;
|
|
|
|
StagingBuffersCache download_cache;
|
2019-12-25 22:12:17 +01:00
|
|
|
|
2020-12-31 02:58:05 +01:00
|
|
|
size_t current_delete_level = 0;
|
|
|
|
u64 buffer_index = 0;
|
2019-12-25 22:12:17 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace Vulkan
|