query_psos

2 files changed, 18 insertions, 0 deletions

diff --git a/video.cpp b/video.cpp
index 7287985..f108319 100644
--- a/video.cpp
+++ b/video.cpp
@@ -4481,3 +4481,20 @@ void Device_Debug_Hooks::on_vram_alloc(size_t size, size_t align) {
 	(void)size;
 	(void)align;
 }
+
+int Device_Debug_Hooks::query_psos(Pipeline* psos) {
+	Device_Vk* dv = (Device_Vk*)dev;
+	int count = 0;
+	if (psos) {
+		for (auto i : dv->pso_cache) {
+			auto& psok = i.first;
+			psos[count++] = psok.pso;
+		}
+	} else {
+		for (auto i : dv->pso_cache) {
+			(void)i;
+			count++;
+		}
+	}
+	return count;
+}
diff --git a/video.hpp b/video.hpp
index 6d70536..c93093e 100644
--- a/video.hpp
+++ b/video.hpp
@@ -439,6 +439,7 @@ struct Device_Debug_Hooks {
 	virtual void on_present(Context& ctx);
 	virtual void on_page_alloc(size_t size);
 	virtual void on_vram_alloc(size_t size, size_t align);
+	int query_psos(Pipeline* psos);
 };
 
 struct Context;