Implementing GetRenderStats and returning percentage done. (#538)

Fixes #537
Autodesk · Sep 17, 2020 · 19e9f55 · 19e9f55
1 parent ecef07c
commit 19e9f55
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/render_delegate/constant_strings.h b/render_delegate/constant_strings.h
@@ -283,6 +283,7 @@ ASTR(tflip);
 ASTR(translation);
 ASTR(thread_priority);
 ASTR(threads);
+ASTR(total_progress);
 ASTR(useSpecularWorkflow);
 ASTR(user_data_float);
 ASTR(user_data_int);

diff --git a/render_delegate/render_delegate.cpp b/render_delegate/render_delegate.cpp
@@ -58,6 +58,7 @@ TF_DEFINE_PRIVATE_TOKENS(_tokens,
     (arnold)
     (openvdbAsset)
     ((arnoldGlobal, "arnold:global:"))
+    (percentDone)
 );
 // clang-format on
 
@@ -513,6 +514,16 @@ HdRenderSettingDescriptorList HdArnoldRenderDelegate::GetRenderSettingDescriptor
     return ret;
 }
 
+VtDictionary HdArnoldRenderDelegate::GetRenderStats() const
+{
+    VtDictionary stats;
+
+    float total_progress = 100.0f;
+    AiRenderGetHintFlt(str::total_progress, total_progress);
+    stats[_tokens->percentDone] = total_progress;
+    return stats;
+}
+
 HdResourceRegistrySharedPtr HdArnoldRenderDelegate::GetResourceRegistry() const { return _resourceRegistry; }
 
 HdRenderPassSharedPtr HdArnoldRenderDelegate::CreateRenderPass(

diff --git a/render_delegate/render_delegate.h b/render_delegate/render_delegate.h
@@ -98,6 +98,11 @@ class HdArnoldRenderDelegate final : public HdRenderDelegate {
     ///  possible Render Settings.
     HDARNOLD_API
     HdRenderSettingDescriptorList GetRenderSettingDescriptors() const override;
+    /// Returns an open-format dictionary of render statistics
+    ///
+    /// @return VtDictionary holding the render stats.
+    HDARNOLD_API
+    VtDictionary GetRenderStats() const override;
     /// Gets the Resource Registry.
     ///
     /// @return Pointer to the shared HdResourceRegistry.