Support for (de)serializing node state (facebook#1570)

Summary:

X-link: facebook/react-native#42751

tsia. Need node state

Differential Revision: D53206323

benchmark/Benchmark.cpp

@@ -194,6 +194,18 @@ void setStylesFromJson(json& j, YGNodeShared node) {
   }
 }
 
+YGNodeRef buildNodeFromJson(json& j, YGConfigRef config) {
+  const YGNodeRef node = YGNodeNewWithConfig(config);
+  json nodeState = j["node"];
+  for (json::iterator it = nodeState.begin(); it != nodeState.end(); it++) {
+    if (it.key() == "always-forms-containing-block") {
+      YGNodeSetAlwaysFormsContainingBlock(node, it.value());
+    }
+  }
+
+  return node;
+}
+
 YogaNodeAndConfig
 buildTreeFromJson(json& j, YogaNodeAndConfig* parent, size_t index) {
   YGConfigShared config = buildConfigFromJson(j);

capture/CaptureTree.cpp

@@ -17,7 +17,8 @@ void captureTree(YGNodeRef node, const std::filesystem::path& path) {
   nodeToString(
       str,
       node,
-      PrintOptions::Style | PrintOptions::Children | PrintOptions::Config);
+      PrintOptions::Style | PrintOptions::Children | PrintOptions::Config |
+          PrintOptions::Node);
   std::ofstream file(path);
   file << str;
 }

capture/NodeToString.cpp

@@ -126,9 +126,9 @@ static void nodeToStringImpl(json& j, YGNodeRef node, PrintOptions options) {
     j["layout"]["top"] = YGNodeStyleGetPosition(node, YGEdgeTop).value;
     j["layout"]["left"] = YGNodeStyleGetPosition(node, YGEdgeLeft).value;
   }
+  const YGNodeRef defaultNode = YGNodeNew();
 
   if ((options & PrintOptions::Style) == PrintOptions::Style) {
-    const YGNodeRef defaultNode = YGNodeNew();
     appendEnumValueIfNotDefault(
         j["style"],
         "flex-direction",
@@ -247,10 +247,6 @@ static void nodeToStringImpl(json& j, YGNodeRef node, PrintOptions options) {
         "min-height",
         YGNodeStyleGetMinHeight(node),
         YGNodeStyleGetMinHeight(defaultNode));
-
-    if (YGNodeHasMeasureFunc(node)) {
-      j["style"]["has-custom-measure"] = true;
-    }
   }
 
   if ((options & PrintOptions::Config) == PrintOptions::Config) {
@@ -286,6 +282,19 @@ static void nodeToStringImpl(json& j, YGNodeRef node, PrintOptions options) {
     }
   }
 
+  if ((options & PrintOptions::Node) == PrintOptions::Node) {
+    appendBoolIfNotDefault(
+        j["node"],
+        "always-forms-containing-block",
+        YGNodeGetAlwaysFormsContainingBlock(node),
+        YGNodeGetAlwaysFormsContainingBlock(defaultNode));
+    if (YGNodeHasMeasureFunc(node)) {
+      j["node"]["has-custom-measure"] = true;
+    }
+  }
+
+  YGNodeFree(defaultNode);
+
   const size_t childCount = YGNodeGetChildCount(node);
   if ((options & PrintOptions::Children) == PrintOptions::Children &&
       childCount > 0) {

capture/NodeToString.h

@@ -18,6 +18,7 @@ enum class PrintOptions : uint8_t {
   Children = 1 << 1,
   Style = 1 << 2,
   Config = 1 << 3,
+  Node = 1 << 4,
 };
 YG_DEFINE_ENUM_FLAG_OPERATORS(PrintOptions);
 

yoga/YGNode.cpp

@@ -329,6 +329,10 @@ void YGNodeSetAlwaysFormsContainingBlock(
   resolveRef(node)->setAlwaysFormsContainingBlock(alwaysFormsContainingBlock);
 }
 
+bool YGNodeGetAlwaysFormsContainingBlock(YGNodeConstRef node) {
+  return resolveRef(node)->alwaysFormsContainingBlock();
+}
+
 // TODO: This leaks internal details to the public API. Remove after removing
 // ComponentKit usage of it.
 bool YGNodeCanUseCachedMeasurement(

yoga/YGNode.h

@@ -273,6 +273,14 @@ YG_EXPORT void YGNodeSetAlwaysFormsContainingBlock(
     YGNodeRef node,
     bool alwaysFormsContainingBlock);
 
+/**
+ * Whether the node will always form a containing block for any descendant. This
+ * can happen in situation where the client implements something like a
+ * transform that can affect containing blocks but is not handled by Yoga
+ * directly.
+ */
+YG_EXPORT bool YGNodeGetAlwaysFormsContainingBlock(YGNodeConstRef node);
+
 /**
  * @deprecated
  */