Add delegate for Vulkan dump resources

Move code about writing data from vulkan_replay_dump_resources_draw_calls and vulkan_replay_dump_resources_compute_ray_tracing to DefaultVulkanDumpResourcesDelegate on vulkan_replay_dump_resources_json.
LunarG · Jan 4, 2025 · 68d39bf · 68d39bf
1 parent 599920e
commit 68d39bf
Show file tree

Hide file tree

Showing 9 changed files with 3,281 additions and 3,108 deletions.
diff --git a/framework/decode/dx12_browse_consumer.h b/framework/decode/dx12_browse_consumer.h
@@ -63,7 +63,7 @@ struct TrackRootParameter
     // The other parameter types have no resources or descriptors info, so no track.
 };
 
-enum DumpDrawCallType
+enum class DumpDrawCallType
 {
     kUnknown,
     kDraw,

diff --git a/framework/decode/vulkan_replay_dump_resources.cpp b/framework/decode/vulkan_replay_dump_resources.cpp
@@ -46,9 +46,10 @@ GFXRECON_BEGIN_NAMESPACE(decode)
 
 VulkanReplayDumpResourcesBase::VulkanReplayDumpResourcesBase(const VulkanReplayOptions& options,
                                                              CommonObjectInfoTable*     object_info_table) :
-    QueueSubmit_indices_(options.QueueSubmit_Indices),
-    recording_(false), dump_resources_before_(options.dump_resources_before), object_info_table_(object_info_table),
-    output_json_per_command(options.dump_resources_json_per_command), dump_json_(options)
+    QueueSubmit_indices_(options.QueueSubmit_Indices), recording_(false),
+    dump_resources_before_(options.dump_resources_before), object_info_table_(object_info_table),
+    output_json_per_command(options.dump_resources_json_per_command), user_delegate_(nullptr),
+    active_delegate_(nullptr), default_delegate_(nullptr)
 {
     capture_filename = std::filesystem::path(options.capture_filename).stem().string();
 
@@ -57,9 +58,20 @@ VulkanReplayDumpResourcesBase::VulkanReplayDumpResourcesBase(const VulkanReplayO
         return;
     }
 
+    if (user_delegate_ != nullptr)
+    {
+        active_delegate_ = user_delegate_;
+    }
+    else
+    {
+        // Use a default delegate if none was provided.
+        default_delegate_ = std::make_unique<DefaultVulkanDumpResourcesDelegate>(options, capture_filename);
+        active_delegate_  = default_delegate_.get();
+    }
+
     if (!options.dump_resources_json_per_command)
     {
-        dump_json_.Open(options.capture_filename, options.dump_resources_output_dir);
+        active_delegate_->Open();
     }
 
     for (size_t i = 0; i < options.BeginCommandBuffer_Indices.size(); ++i)
@@ -75,8 +87,7 @@ VulkanReplayDumpResourcesBase::VulkanReplayDumpResourcesBase(const VulkanReplayO
                                                                options.RenderPass_Indices[i],
                                                                *object_info_table,
                                                                options,
-                                                               dump_json_,
-                                                               capture_filename));
+                                                               *active_delegate_));
         }
 
         if ((i < options.Dispatch_Indices.size() && options.Dispatch_Indices[i].size()) ||
@@ -92,8 +103,7 @@ VulkanReplayDumpResourcesBase::VulkanReplayDumpResourcesBase(const VulkanReplayO
                                                   : std::vector<uint64_t>(),
                                               *object_info_table_,
                                               options,
-                                              dump_json_,
-                                              capture_filename));
+                                              *active_delegate_));
         }
     }
 }
@@ -105,7 +115,7 @@ VulkanReplayDumpResourcesBase::~VulkanReplayDumpResourcesBase()
 
 void VulkanReplayDumpResourcesBase::Release()
 {
-    dump_json_.Close();
+    active_delegate_->Close();
     draw_call_contexts.clear();
     dispatch_ray_contexts.clear();
     cmd_buf_begin_map_.clear();
@@ -1847,7 +1857,7 @@ VkResult VulkanReplayDumpResourcesBase::QueueSubmit(const std::vector<VkSubmitIn
 
     if (!output_json_per_command)
     {
-        dump_json_.BlockStart();
+        active_delegate_->DumpStart();
     }
 
     for (size_t s = 0; s < submit_infos.size(); s++)
@@ -1906,7 +1916,7 @@ VkResult VulkanReplayDumpResourcesBase::QueueSubmit(const std::vector<VkSubmitIn
 
     if (!output_json_per_command)
     {
-        dump_json_.BlockEnd();
+        active_delegate_->DumpEnd();
     }
 
     // Looks like we didn't submit anything. Do the submission as it would have been done

diff --git a/framework/decode/vulkan_replay_dump_resources.h b/framework/decode/vulkan_replay_dump_resources.h
@@ -348,11 +348,14 @@ class VulkanReplayDumpResourcesBase
     std::unordered_map<uint64_t, DrawCallsDumpingContext>         draw_call_contexts;
     std::unordered_map<uint64_t, DispatchTraceRaysDumpingContext> dispatch_ray_contexts;
 
-    bool                          recording_;
-    bool                          dump_resources_before_;
-    CommonObjectInfoTable*        object_info_table_;
-    VulkanReplayDumpResourcesJson dump_json_;
-    bool                          output_json_per_command;
+    bool                   recording_;
+    bool                   dump_resources_before_;
+    CommonObjectInfoTable* object_info_table_;
+    bool                   output_json_per_command;
+
+    std::unique_ptr<DefaultVulkanDumpResourcesDelegate> default_delegate_;
+    VulkanDumpResourcesDelegate*                        user_delegate_;
+    VulkanDumpResourcesDelegate*                        active_delegate_;
 
     std::string capture_filename;