👯 Add more uint128x2 methods

Author: KimlikDAO-bot

kimlikdao/IDistroStage.sol

@@ -6,9 +6,9 @@ enum DistroStage {
     Presale1,
     Presale2,
     ProtocolSaleStart,
-    ProtocolSaleEnd,
+    ProtocolSaleEnd, // Unlocks Presale1
     ProtocolAMMStart,
-    Presale2Unlock,
+    Presale2Unlock, // Unlocks Presale2
     FinalMint,
     FinalUnlock
 }

types/uint128x2.sol

@@ -34,8 +34,22 @@ function dec(uint128x2 self, uint256 delta) pure returns (uint128x2) {
     }
 }
 
+function sum(uint128x2 self) pure returns (uint256) {
+    unchecked {
+        return (uint128x2.unwrap(self) >> 128) + uint128(uint128x2.unwrap(self));
+    }
+}
+
+function clearLo(uint128x2 self) pure returns (uint128x2) {
+    return uint128x2.wrap(uint128x2.unwrap(self) & ~type(uint128).max);
+}
+
+function clearHi(uint128x2 self) pure returns (uint128x2) {
+    return uint128x2.wrap(uint128x2.unwrap(self) & type(uint128).max);
+}
+
 function equal(uint128x2 self, uint128x2 other) pure returns (bool) {
     return uint128x2.unwrap(self) == uint128x2.unwrap(other);
 }
 
-using {hi, lo, decLo, inc, dec, equal as ==} for uint128x2 global;
+using {hi, lo, decLo, inc, dec, sum, clearLo, clearHi, equal as ==} for uint128x2 global;

types/uint128x2.t.sol

@@ -31,4 +31,28 @@ contract uint128x2Test is Test {
 
         assert(pair2 == pair3);
     }
+
+    function tesetClear() public pure {
+        uint128x2 i = uint128x2From(2, 1);
+
+        i = i.clearHi();
+        assertEq(i.hi(), 0);
+        assertEq(i.lo(), 1);
+
+        uint128x2 j = uint128x2From(3, 2);
+
+        j = j.clearLo();
+        j = j.clearLo();
+        assertEq(j.hi(), 3);
+        assertEq(j.lo(), 0);
+    }
+
+    function testSum() public pure {
+        uint128x2 i = uint128x2From(type(uint128).max - 1, 1);
+
+        assertEq(i.sum(), type(uint128).max);
+        assertEq(uint128x2From(1, 2).sum(), 3);
+        assertEq(uint128x2From(2, 3).sum(), 5);
+        assertEq(uint128x2From(1337, 1338).sum(), 2 * 1337 + 1);
+    }
 }