Add config version, and invalidate layout on config change (#1674)

Summary:
X-link: https://github.com/facebook/react-native/pull/45259

This is a continuation of the previous PR: https://github.com/facebook/react-native/pull/45047

I made the change more generic for allowing any kind of config change to invalidate layout.

Changelog: [Internal]

Pull Request resolved: https://github.com/facebook/yoga/pull/1674

Reviewed By: rozele

Differential Revision: D59286992

Pulled By: NickGerleman

fbshipit-source-id: f46f35b03d5d9a743b798844ee3e1a02c271ccde
This commit is contained in:
Andrew Coates
2024-07-03 12:46:18 -07:00
committed by Facebook GitHub Bot
parent a1e9abb9b3
commit e4fe14ab3e
7 changed files with 212 additions and 5 deletions

176
tests/YGScaleChangeTest.cpp Normal file
View File

@@ -0,0 +1,176 @@
/*
* Copyright (c) Meta Platforms, Inc. and affiliates.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*/
#include <gtest/gtest.h>
#include <yoga/Yoga.h>
TEST(YogaTest, scale_change_invalidates_layout) {
YGConfigRef config = YGConfigNew();
YGNodeRef root = YGNodeNewWithConfig(config);
YGConfigSetPointScaleFactor(config, 1.0f);
YGNodeStyleSetFlexDirection(root, YGFlexDirectionRow);
YGNodeStyleSetWidth(root, 50);
YGNodeStyleSetHeight(root, 50);
YGNodeRef root_child0 = YGNodeNewWithConfig(config);
YGNodeStyleSetFlexGrow(root_child0, 1);
YGNodeInsertChild(root, root_child0, 0);
YGNodeRef root_child1 = YGNodeNewWithConfig(config);
YGNodeStyleSetFlexGrow(root_child1, 1);
YGNodeInsertChild(root, root_child1, 1);
YGNodeCalculateLayout(root, YGUndefined, YGUndefined, YGDirectionLTR);
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root_child0));
ASSERT_FLOAT_EQ(25, YGNodeLayoutGetLeft(root_child1));
YGConfigSetPointScaleFactor(config, 1.5f);
YGNodeCalculateLayout(root, YGUndefined, YGUndefined, YGDirectionLTR);
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root_child0));
// Left should change due to pixel alignment of new scale factor
ASSERT_FLOAT_EQ(25.333334f, YGNodeLayoutGetLeft(root_child1));
YGNodeFreeRecursive(root);
YGConfigFree(config);
}
TEST(YogaTest, errata_config_change_relayout) {
YGConfig* config = YGConfigNew();
YGConfigSetErrata(config, YGErrataStretchFlexBasis);
YGNodeRef root = YGNodeNewWithConfig(config);
YGNodeStyleSetWidth(root, 500);
YGNodeStyleSetHeight(root, 500);
YGNodeRef root_child0 = YGNodeNewWithConfig(config);
YGNodeStyleSetAlignItems(root_child0, YGAlignFlexStart);
YGNodeInsertChild(root, root_child0, 0);
YGNodeRef root_child0_child0 = YGNodeNewWithConfig(config);
YGNodeStyleSetFlexGrow(root_child0_child0, 1);
YGNodeStyleSetFlexShrink(root_child0_child0, 1);
YGNodeInsertChild(root_child0, root_child0_child0, 0);
YGNodeRef root_child0_child0_child0 = YGNodeNewWithConfig(config);
YGNodeStyleSetFlexGrow(root_child0_child0_child0, 1);
YGNodeStyleSetFlexShrink(root_child0_child0_child0, 1);
YGNodeInsertChild(root_child0_child0, root_child0_child0_child0, 0);
YGNodeCalculateLayout(root, YGUndefined, YGUndefined, YGDirectionLTR);
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetTop(root));
ASSERT_FLOAT_EQ(500, YGNodeLayoutGetWidth(root));
ASSERT_FLOAT_EQ(500, YGNodeLayoutGetHeight(root));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetTop(root_child0));
ASSERT_FLOAT_EQ(500, YGNodeLayoutGetWidth(root_child0));
ASSERT_FLOAT_EQ(500, YGNodeLayoutGetHeight(root_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root_child0_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetTop(root_child0_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetWidth(root_child0_child0));
ASSERT_FLOAT_EQ(500, YGNodeLayoutGetHeight(root_child0_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root_child0_child0_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetTop(root_child0_child0_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetWidth(root_child0_child0_child0));
ASSERT_FLOAT_EQ(500, YGNodeLayoutGetHeight(root_child0_child0_child0));
YGConfigSetErrata(config, YGErrataNone);
YGNodeCalculateLayout(root, YGUndefined, YGUndefined, YGDirectionLTR);
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetTop(root));
ASSERT_FLOAT_EQ(500, YGNodeLayoutGetWidth(root));
ASSERT_FLOAT_EQ(500, YGNodeLayoutGetHeight(root));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetTop(root_child0));
ASSERT_FLOAT_EQ(500, YGNodeLayoutGetWidth(root_child0));
// This should be modified by the lack of the errata
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetHeight(root_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root_child0_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetTop(root_child0_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetWidth(root_child0_child0));
// This should be modified by the lack of the errata
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetHeight(root_child0_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root_child0_child0_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetTop(root_child0_child0_child0));
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetWidth(root_child0_child0_child0));
// This should be modified by the lack of the errata
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetHeight(root_child0_child0_child0));
YGNodeFreeRecursive(root);
YGConfigFree(config);
}
TEST(YogaTest, setting_compatible_config_maintains_layout_cache) {
static uint32_t measureCallCount = 0;
auto measureCustom = [](YGNodeConstRef /*node*/,
float /*width*/,
YGMeasureMode /*widthMode*/,
float /*height*/,
YGMeasureMode /*heightMode*/) {
measureCallCount++;
return YGSize{
.width = 25.0f,
.height = 25.0f,
};
};
YGConfigRef config = YGConfigNew();
YGNodeRef root = YGNodeNewWithConfig(config);
YGConfigSetPointScaleFactor(config, 1.0f);
YGNodeStyleSetFlexDirection(root, YGFlexDirectionRow);
YGNodeStyleSetWidth(root, 50);
YGNodeStyleSetHeight(root, 50);
YGNodeRef root_child0 = YGNodeNewWithConfig(config);
EXPECT_EQ(0, measureCallCount);
YGNodeSetMeasureFunc(root_child0, measureCustom);
YGNodeInsertChild(root, root_child0, 0);
YGNodeRef root_child1 = YGNodeNewWithConfig(config);
YGNodeStyleSetFlexGrow(root_child1, 1);
YGNodeInsertChild(root, root_child1, 1);
YGNodeCalculateLayout(root, YGUndefined, YGUndefined, YGDirectionLTR);
EXPECT_EQ(1, measureCallCount);
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root_child0));
ASSERT_FLOAT_EQ(25, YGNodeLayoutGetLeft(root_child1));
YGConfigRef config2 = YGConfigNew();
// Calling YGConfigSetPointScaleFactor multiple times, ensures that config2
// gets a different config version that config1
YGConfigSetPointScaleFactor(config2, 1.0f);
YGConfigSetPointScaleFactor(config2, 1.5f);
YGConfigSetPointScaleFactor(config2, 1.0f);
YGNodeSetConfig(root, config2);
YGNodeSetConfig(root_child0, config2);
YGNodeSetConfig(root_child1, config2);
YGNodeCalculateLayout(root, YGUndefined, YGUndefined, YGDirectionLTR);
// Measure should not be called again, as layout should have been cached since
// config is functionally the same as before
EXPECT_EQ(1, measureCallCount);
ASSERT_FLOAT_EQ(0, YGNodeLayoutGetLeft(root_child0));
ASSERT_FLOAT_EQ(25, YGNodeLayoutGetLeft(root_child1));
YGNodeFreeRecursive(root);
YGConfigFree(config);
YGConfigFree(config2);
}

View File

@@ -2140,6 +2140,7 @@ bool calculateLayoutInternal(
const bool needToVisitNode =
(node->isDirty() && layout->generationCount != generationCount) ||
layout->configVersion != node->getConfig()->getVersion() ||
layout->lastOwnerDirection != ownerDirection;
if (needToVisitNode) {
@@ -2255,6 +2256,7 @@ bool calculateLayoutInternal(
reason);
layout->lastOwnerDirection = ownerDirection;
layout->configVersion = node->getConfig()->getVersion();
if (cachedResults == nullptr) {
layoutMarkerData.maxMeasureCache = std::max(

View File

@@ -31,7 +31,10 @@ bool Config::useWebDefaults() const {
void Config::setExperimentalFeatureEnabled(
ExperimentalFeature feature,
bool enabled) {
if (isExperimentalFeatureEnabled(feature) != enabled) {
experimentalFeatures_.set(static_cast<size_t>(feature), enabled);
version_++;
}
}
bool Config::isExperimentalFeatureEnabled(ExperimentalFeature feature) const {
@@ -43,15 +46,24 @@ ExperimentalFeatureSet Config::getEnabledExperiments() const {
}
void Config::setErrata(Errata errata) {
if (errata_ != errata) {
errata_ = errata;
version_++;
}
}
void Config::addErrata(Errata errata) {
if (!hasErrata(errata)) {
errata_ |= errata;
version_++;
}
}
void Config::removeErrata(Errata errata) {
if (hasErrata(errata)) {
errata_ &= (~errata);
version_++;
}
}
Errata Config::getErrata() const {
@@ -63,7 +75,10 @@ bool Config::hasErrata(Errata errata) const {
}
void Config::setPointScaleFactor(float pointScaleFactor) {
if (pointScaleFactor_ != pointScaleFactor) {
pointScaleFactor_ = pointScaleFactor;
version_++;
}
}
float Config::getPointScaleFactor() const {
@@ -78,6 +93,10 @@ void* Config::getContext() const {
return context_;
}
uint32_t Config::getVersion() const noexcept {
return version_;
}
void Config::setLogger(YGLogger logger) {
logger_ = logger;
}

View File

@@ -53,6 +53,8 @@ class YG_EXPORT Config : public ::YGConfig {
void setContext(void* context);
void* getContext() const;
uint32_t getVersion() const noexcept;
void setLogger(YGLogger logger);
void log(
const yoga::Node* node,
@@ -72,6 +74,7 @@ class YG_EXPORT Config : public ::YGConfig {
bool useWebDefaults_ : 1 = false;
uint32_t version_ = 0;
ExperimentalFeatureSet experimentalFeatures_{};
Errata errata_ = Errata::None;
float pointScaleFactor_ = 1.0f;

View File

@@ -21,6 +21,7 @@ bool LayoutResults::operator==(LayoutResults layout) const {
direction() == layout.direction() &&
hadOverflow() == layout.hadOverflow() &&
lastOwnerDirection == layout.lastOwnerDirection &&
configVersion == layout.configVersion &&
nextCachedMeasurementsIndex == layout.nextCachedMeasurementsIndex &&
cachedLayout == layout.cachedLayout &&
computedFlexBasis == layout.computedFlexBasis;

View File

@@ -30,6 +30,7 @@ struct LayoutResults {
// Instead of recomputing the entire layout every single time, we cache some
// information to break early when nothing changed
uint32_t generationCount = 0;
uint32_t configVersion = 0;
Direction lastOwnerDirection = Direction::Inherit;
uint32_t nextCachedMeasurementsIndex = 0;

View File

@@ -136,6 +136,11 @@ void Node::setConfig(yoga::Config* config) {
if (yoga::configUpdateInvalidatesLayout(*config_, *config)) {
markDirtyAndPropagate();
layout_.configVersion = 0;
} else {
// If the config is functionally the same, then align the configVersion so
// that we can reuse the layout cache
layout_.configVersion = config->getVersion();
}
config_ = config;