Cryptomatte: Session from Existing Render Result.

Utility to construct a cryptomatte session from a render result or
openexr file. This will allow D3959 to be more aware of the context
it is working on and would also support external render engines in
the cryptomatte color picker.

1 files changed, 97 insertions, 3 deletions

diff --git a/source/blender/blenkernel/intern/cryptomatte_test.cc b/source/blender/blenkernel/intern/cryptomatte_test.cc
index 4a25f5f7d87..d9be252d654 100644
--- a/source/blender/blenkernel/intern/cryptomatte_test.cc
+++ b/source/blender/blenkernel/intern/cryptomatte_test.cc
@@ -17,7 +17,13 @@
  */
 #include "testing/testing.h"
 
+#include "BKE_cryptomatte.h"
 #include "BKE_cryptomatte.hh"
+#include "BKE_image.h"
+
+#include "RE_pipeline.h"
+
+#include "MEM_guardedalloc.h"
 
 namespace blender::bke::cryptomatte::tests {
 
@@ -41,7 +47,7 @@ TEST(cryptomatte, extract_layer_name)
   ASSERT_EQ("", BKE_cryptomatte_extract_layer_name(""));
 }
 
-TEST(cryptomatte, cryptomatte_layer)
+TEST(cryptomatte, layer)
 {
   blender::bke::cryptomatte::CryptomatteLayer layer;
   ASSERT_EQ("{}", layer.manifest());
@@ -53,7 +59,7 @@ TEST(cryptomatte, cryptomatte_layer)
   ASSERT_EQ("{\"Object\":\"0000007b\",\"Object2\":\"0758946e\"}", layer.manifest());
 }
 
-TEST(cryptomatte, cryptomatte_layer_quoted)
+TEST(cryptomatte, layer_quoted)
 {
   blender::bke::cryptomatte::CryptomatteLayer layer;
   layer.add_hash("\"Object\"", 123);
@@ -66,7 +72,7 @@ static void test_cryptomatte_manifest(std::string expected, std::string manifest
             blender::bke::cryptomatte::CryptomatteLayer::read_from_manifest(manifest)->manifest());
 }
 
-TEST(cryptomatte, cryptomatte_layer_from_manifest)
+TEST(cryptomatte, layer_from_manifest)
 {
   test_cryptomatte_manifest("{}", "{}");
   test_cryptomatte_manifest("{\"Object\":\"12345678\"}", "{\"Object\": \"12345678\"}");
@@ -82,4 +88,92 @@ TEST(cryptomatte, cryptomatte_layer_from_manifest)
       "{\"Object\\\"01\\\"\":\"12345678\",\"Object\":\"12345678\", \"Object2\":\"87654321\"}");
 }
 
+TEST(cryptomatte, extract_layer_hash_from_metadata_key)
+{
+  EXPECT_EQ("eb4c67b",
+            blender::bke::cryptomatte::CryptomatteStampDataCallbackData::extract_layer_hash(
+                "cryptomatte/eb4c67b/conversion"));
+  EXPECT_EQ("qwerty",
+            blender::bke::cryptomatte::CryptomatteStampDataCallbackData::extract_layer_hash(
+                "cryptomatte/qwerty/name"));
+  /* Check if undefined behaviors are handled. */
+  EXPECT_EQ("",
+            blender::bke::cryptomatte::CryptomatteStampDataCallbackData::extract_layer_hash(
+                "cryptomatte/name"));
+  EXPECT_EQ("",
+            blender::bke::cryptomatte::CryptomatteStampDataCallbackData::extract_layer_hash(
+                "cryptomatte/"));
+}
+
+static void validate_cryptomatte_session_from_stamp_data(void *UNUSED(data),
+                                                         const char *propname,
+                                                         char *propvalue,
+                                                         int UNUSED(len))
+{
+  blender::StringRefNull prop_name(propname);
+  if (!prop_name.startswith("cryptomatte/")) {
+    return;
+  }
+
+  if (prop_name == "cryptomatte/87f095e/name") {
+    EXPECT_STREQ("viewlayername.layer1", propvalue);
+  }
+  else if (prop_name == "cryptomatte/87f095e/hash") {
+    EXPECT_STREQ("MurmurHash3_32", propvalue);
+  }
+  else if (prop_name == "cryptomatte/87f095e/conversion") {
+    EXPECT_STREQ("uint32_to_float32", propvalue);
+  }
+  else if (prop_name == "cryptomatte/87f095e/manifest") {
+    EXPECT_STREQ("{\"Object\":\"12345678\"}", propvalue);
+  }
+
+  else if (prop_name == "cryptomatte/c42daa7/name") {
+    EXPECT_STREQ("viewlayername.layer2", propvalue);
+  }
+  else if (prop_name == "cryptomatte/c42daa7/hash") {
+    EXPECT_STREQ("MurmurHash3_32", propvalue);
+  }
+  else if (prop_name == "cryptomatte/c42daa7/conversion") {
+    EXPECT_STREQ("uint32_to_float32", propvalue);
+  }
+  else if (prop_name == "cryptomatte/c42daa7/manifest") {
+    EXPECT_STREQ("{\"Object2\":\"87654321\"}", propvalue);
+  }
+
+  else {
+    EXPECT_EQ("Unhandled", std::string(propname) + ": " + propvalue);
+  }
+}
+
+TEST(cryptomatte, session_from_stamp_data)
+{
+  /* Create CryptomatteSession from stamp data. */
+  RenderResult *render_result = static_cast<RenderResult *>(
+      MEM_callocN(sizeof(RenderResult), __func__));
+  BKE_render_result_stamp_data(render_result, "cryptomatte/qwerty/name", "layer1");
+  BKE_render_result_stamp_data(
+      render_result, "cryptomatte/qwerty/manifest", "{\"Object\":\"12345678\"}");
+  BKE_render_result_stamp_data(render_result, "cryptomatte/uiop/name", "layer2");
+  BKE_render_result_stamp_data(
+      render_result, "cryptomatte/uiop/manifest", "{\"Object2\":\"87654321\"}");
+  CryptomatteSession *session = BKE_cryptomatte_init_from_render_result(render_result);
+  EXPECT_NE(session, nullptr);
+  RE_FreeRenderResult(render_result);
+
+  /* Create StampData from CryptomatteSession. */
+  ViewLayer view_layer;
+  BLI_strncpy(view_layer.name, "viewlayername", sizeof(view_layer.name));
+  RenderResult *render_result2 = static_cast<RenderResult *>(
+      MEM_callocN(sizeof(RenderResult), __func__));
+  BKE_cryptomatte_store_metadata(session, render_result2, &view_layer);
+
+  /* Validate StampData. */
+  BKE_stamp_info_callback(
+      nullptr, render_result2->stamp_data, validate_cryptomatte_session_from_stamp_data, false);
+
+  RE_FreeRenderResult(render_result2);
+  BKE_cryptomatte_free(session);
+}
+
 }  // namespace blender::bke::cryptomatte::tests