Edit comment

Vectorized · Jan 9, 2024 · efb6b2a · efb6b2a
1 parent f980cc9
commit efb6b2a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/.gas-snapshot b/.gas-snapshot
@@ -914,9 +914,9 @@ ReceiverTest:test__codesize() (gas: 3310)
 RedBlackTreeLibTest:testRedBlackTreeBenchUint160() (gas: 3438464)
 RedBlackTreeLibTest:testRedBlackTreeBenchUint256() (gas: 5850774)
 RedBlackTreeLibTest:testRedBlackTreeClear() (gas: 57543)
-RedBlackTreeLibTest:testRedBlackTreeClear(uint256) (runs: 256, μ: 294230, ~: 228526)
-RedBlackTreeLibTest:testRedBlackTreeInsertAndRemove(uint256) (runs: 256, μ: 690480, ~: 525248)
-RedBlackTreeLibTest:testRedBlackTreeInsertAndRemove2(uint256) (runs: 256, μ: 420356, ~: 378696)
+RedBlackTreeLibTest:testRedBlackTreeClear(uint256) (runs: 256, μ: 262319, ~: 214235)
+RedBlackTreeLibTest:testRedBlackTreeInsertAndRemove(uint256) (runs: 256, μ: 617283, ~: 516591)
+RedBlackTreeLibTest:testRedBlackTreeInsertAndRemove2(uint256) (runs: 256, μ: 415017, ~: 386681)
 RedBlackTreeLibTest:testRedBlackTreeInsertAndRemove3() (gas: 21537965)
 RedBlackTreeLibTest:testRedBlackTreeInsertBenchStep() (gas: 3711341)
 RedBlackTreeLibTest:testRedBlackTreeInsertBenchUint160() (gas: 3476959)
@@ -925,16 +925,16 @@ RedBlackTreeLibTest:testRedBlackTreeInsertOneGas() (gas: 45574)
 RedBlackTreeLibTest:testRedBlackTreeInsertTenGas() (gas: 283398)
 RedBlackTreeLibTest:testRedBlackTreeInsertThreeGas() (gas: 96286)
 RedBlackTreeLibTest:testRedBlackTreeInsertTwoGas() (gas: 69823)
-RedBlackTreeLibTest:testRedBlackTreeNearest(uint256) (runs: 256, μ: 225523, ~: 222642)
-RedBlackTreeLibTest:testRedBlackTreeNearestAfter(uint256) (runs: 256, μ: 252716, ~: 246502)
-RedBlackTreeLibTest:testRedBlackTreeNearestBefore(uint256) (runs: 256, μ: 227962, ~: 192926)
+RedBlackTreeLibTest:testRedBlackTreeNearest(uint256) (runs: 256, μ: 229059, ~: 222642)
+RedBlackTreeLibTest:testRedBlackTreeNearestAfter(uint256) (runs: 256, μ: 251339, ~: 246502)
+RedBlackTreeLibTest:testRedBlackTreeNearestBefore(uint256) (runs: 256, μ: 227590, ~: 192926)
 RedBlackTreeLibTest:testRedBlackTreePointers() (gas: 91923)
 RedBlackTreeLibTest:testRedBlackTreeRejectsEmptyValue() (gas: 3238)
 RedBlackTreeLibTest:testRedBlackTreeRemoveViaPointer() (gas: 58216)
 RedBlackTreeLibTest:testRedBlackTreeTreeFullReverts() (gas: 50359)
 RedBlackTreeLibTest:testRedBlackTreeTryInsertAndRemove() (gas: 56162)
 RedBlackTreeLibTest:testRedBlackTreeValues() (gas: 175206)
-RedBlackTreeLibTest:testRedBlackTreeValues(uint256) (runs: 256, μ: 266988, ~: 218036)
+RedBlackTreeLibTest:testRedBlackTreeValues(uint256) (runs: 256, μ: 264454, ~: 218626)
 RedBlackTreeLibTest:test__codesize() (gas: 14795)
 ReentrancyGuardTest:testRecursiveDirectUnguardedCall() (gas: 34256)
 ReentrancyGuardTest:testRecursiveIndirectUnguardedCall() (gas: 47773)

diff --git a/src/utils/RedBlackTreeLib.sol b/src/utils/RedBlackTreeLib.sol
@@ -102,14 +102,14 @@ library RedBlackTreeLib {
         assembly {
             function visit(current_) {
                 if iszero(current_) { leave }
-                current_ := or(mload(0x00), current_)
+                current_ := or(mload(0x00), current_) // Cursor.
                 let packed_ := sload(current_)
-                visit(and(packed_, _BITMASK_KEY))
-                let value_ := shr(_BITPOS_PACKED_VALUE, packed_)
+                visit(and(packed_, _BITMASK_KEY)) // Visit left child.
+                let value_ := shr(_BITPOS_PACKED_VALUE, packed_) // Current value.
                 if iszero(value_) { value_ := sload(or(current_, _BIT_FULL_VALUE_SLOT)) }
                 mstore(mload(0x20), value_)
                 mstore(0x20, add(0x20, mload(0x20)))
-                visit(and(shr(_BITPOS_RIGHT, packed_), _BITMASK_KEY))
+                visit(and(shr(_BITPOS_RIGHT, packed_), _BITMASK_KEY)) // Visit right child.
             }
             result := mload(0x40)
             let rootPacked := sload(nodes)