1 files changed, 11 insertions, 1 deletions

diff --git a/include/vk-renderer.hpp b/include/vk-renderer.hpp
index b478b25..775ea89 100644
--- a/include/vk-renderer.hpp
+++ b/include/vk-renderer.hpp
@@ -11,7 +11,7 @@ class window;
 
 class vkrenderer {
     public:
-        vkrenderer(window& wnd, size_t scene_buffer_size);
+        vkrenderer(window& wnd, size_t scene_buffer_size, size_t geometry_buffer_size, size_t bvh_buffer_size);
 
         ~vkrenderer();
 
@@ -31,6 +31,14 @@ class vkrenderer {
             return scene_buffer.alloc_info.pMappedData;
         }
 
+        inline void* geometry_buffer_ptr() {
+            return geometry_buffer.alloc_info.pMappedData;
+        }
+
+        inline void* bvh_buffer_ptr() {
+            return bvh_buffer.alloc_info.pMappedData;
+        }
+
     private:
 
         void handle_swapchain_result(VkResult function_result);
@@ -52,6 +60,8 @@ class vkrenderer {
         Pipeline                        compute_pipeline;
 
         Buffer                          scene_buffer;
+        Buffer                          geometry_buffer;
+        Buffer                          bvh_buffer;
 
 
         std::vector<VkDescriptorSetLayoutBinding> tonemapping_sets_bindings;