-
-
Notifications
You must be signed in to change notification settings - Fork 1.9k
/
VKFramebuffer.cpp
94 lines (77 loc) · 2.46 KB
/
VKFramebuffer.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include "stdafx.h"
#include "VKFramebuffer.h"
#include "vkutils/image.h"
#include "vkutils/image_helpers.h"
#include <unordered_map>
namespace vk
{
std::unordered_map<u64, std::vector<std::unique_ptr<vk::framebuffer_holder>>> g_framebuffers_cache;
vk::framebuffer_holder* get_framebuffer(VkDevice dev, u16 width, u16 height, VkRenderPass renderpass, const std::vector<vk::image*>& image_list)
{
u64 key = u64(width) | (u64(height) << 16);
auto &queue = g_framebuffers_cache[key];
for (auto &fbo : queue)
{
if (fbo->matches(image_list, width, height))
{
return fbo.get();
}
}
std::vector<std::unique_ptr<vk::image_view>> image_views;
image_views.reserve(image_list.size());
for (auto &e : image_list)
{
const VkImageSubresourceRange subres = { e->aspect(), 0, 1, 0, 1 };
image_views.push_back(std::make_unique<vk::image_view>(dev, e, VK_IMAGE_VIEW_TYPE_2D, vk::default_component_map, subres));
}
auto value = std::make_unique<vk::framebuffer_holder>(dev, renderpass, width, height, std::move(image_views));
auto ret = value.get();
queue.push_back(std::move(value));
return ret;
}
vk::framebuffer_holder* get_framebuffer(VkDevice dev, u16 width, u16 height, VkRenderPass renderpass, VkFormat format, VkImage attachment)
{
u64 key = u64(width) | (u64(height) << 16);
auto &queue = g_framebuffers_cache[key];
for (const auto &e : queue)
{
if (e->attachments[0]->info.image == attachment)
{
return e.get();
}
}
VkImageSubresourceRange range = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 1, 0, 1 };
std::vector<std::unique_ptr<vk::image_view>> views;
views.push_back(std::make_unique<vk::image_view>(dev, attachment, VK_IMAGE_VIEW_TYPE_2D, format, vk::default_component_map, range));
auto value = std::make_unique<vk::framebuffer_holder>(dev, renderpass, width, height, std::move(views));
auto ret = value.get();
queue.push_back(std::move(value));
return ret;
}
void remove_unused_framebuffers()
{
// Remove stale framebuffers. Ref counted to prevent use-after-free
for (auto It = g_framebuffers_cache.begin(); It != g_framebuffers_cache.end();)
{
It->second.erase(
std::remove_if(It->second.begin(), It->second.end(), [](const auto& fbo)
{
return (fbo->unused_check_count() >= 2);
}),
It->second.end()
);
if (It->second.empty())
{
It = g_framebuffers_cache.erase(It);
}
else
{
++It;
}
}
}
void clear_framebuffer_cache()
{
g_framebuffers_cache.clear();
}
}