Method updateTree

frontend/layout/lib/layoutModel.ts:724–776

* Recursively walks the tree to find leaf nodes, update the resize handles, and compute additional properties for each node. * @param balanceTree Whether the tree should also be balanced as it is walked. This should be done if the tree state has just been updated. Defaults to true.

(balanceTree = true)

     * @param balanceTree Whether the tree should also be balanced as it is walked. This should be done if the tree state has just been updated. Defaults to true.
     */
    updateTree(balanceTree = true) {
        if (this.displayContainerRef.current) {
            const newLeafs: LayoutNode[] = [];
            const newAdditionalProps = {};

            const pendingAction = this.getter(this.pendingTreeAction.currentValueAtom);
            const resizeAction =
                pendingAction?.type === LayoutTreeActionType.ResizeNode
                    ? (pendingAction as LayoutTreeResizeNodeAction)
                    : null;
            const resizeHandleSizePx = this.getter(this.resizeHandleSizePx);

            const boundingRect = this.getBoundingRect();

            const magnifiedNodeSize = this.getter(this.magnifiedNodeSizeAtom);

            const callback = (node: LayoutNode) =>
                this.updateTreeHelper(
                    node,
                    newAdditionalProps,
                    newLeafs,
                    resizeHandleSizePx,
                    magnifiedNodeSize,
                    boundingRect,
                    resizeAction
                );
            if (balanceTree) this.treeState.rootNode = balanceNode(this.treeState.rootNode, callback);
            else walkNodes(this.treeState.rootNode, callback);

            // Process ephemeral node, if present.
            const ephemeralNode = this.getter(this.ephemeralNode);
            if (ephemeralNode) {
                this.updateEphemeralNodeProps(
                    ephemeralNode,
                    newAdditionalProps,
                    newLeafs,
                    magnifiedNodeSize,
                    boundingRect
                );
            }

            this.treeState.leafOrder = getLeafOrder(newLeafs, newAdditionalProps);
            this.validateFocusedNode(this.treeState.leafOrder);
            this.validateMagnifiedNode(this.treeState.leafOrder, newAdditionalProps);
            this.cleanupNodeModels(this.treeState.leafOrder);
            this.setter(
                this.leafs,
                newLeafs.sort((a, b) => a.id.localeCompare(b.id))
            );
            this.setter(this.leafOrder, this.treeState.leafOrder);
            this.setter(this.additionalProps, newAdditionalProps);
        }
    }

    /**
     * Per-node callback that is invoked recursively to find leaf nodes, update the resize handles, and compute additional properties associated with the given node.