Uniswap · hensha256 · May 26, 2024 · Jun 21, 2023 · Apr 4, 2024 · Apr 4, 2024
@@ -1 +1 @@
-1947
+1662
@@ -1 +1 @@
-2390
+2152
@@ -1 +1 @@
-1686
+1413
@@ -1 +1 @@
-2831
+2120
@@ -1 +1 @@
-2059
+1753
@@ -1 +1 @@
-2941
+2671
@@ -1 +1 @@
-1798
+1504
@@ -1 +1 @@
-2458
+1850
@@ -1 +1 @@
-19986
+19871
@@ -1 +1 @@
-114924
+114348
@@ -1 +1 @@
-130089
+129513
@@ -1 +1 @@
-179527
+178951
@@ -1 +1 @@
-110279
+110009
@@ -1 +1 @@
-121626
+121356
@@ -1 +1 @@
-133455
+132744
@@ -1 +1 @@
-122743
+122135
@@ -1 +1 @@
-144531
+144293
@@ -1 +1 @@
-161161
+160585
@@ -1 +1 @@
-217571
+216725
@@ -1 +1 @@
-145362
+144786
@@ -1 +1 @@
-138626
+138356
@@ -1 +1 @@
-176624
+175722
@@ -1 +1 @@
-152398
+151822
@@ -1 +1 @@
-155040
+154464
@@ -7,6 +7,8 @@ import {SqrtPriceMath} from "./SqrtPriceMath.sol";
 /// @title Computes the result of a swap within ticks
 /// @notice Contains methods for computing the result of a swap within a single tick price range, i.e., a single tick.
 library SwapMath {
+    uint256 internal constant MAX_FEE_PIPS = 1e6;
+
     /// @notice Computes the sqrt price target for the next swap step
     /// @param zeroForOne The direction of the swap, true for currency0 to currency1, false for currency1 to currency0
     /// @param sqrtPriceNextX96 The Q64.96 sqrt price for the next initialized tick
@@ -49,63 +51,53 @@ library SwapMath {
         uint24 feePips
     ) internal pure returns (uint160 sqrtPriceNextX96, uint256 amountIn, uint256 amountOut, uint256 feeAmount) {
         unchecked {
+            uint256 _feePips = feePips; // upcast once and cache
             bool zeroForOne = sqrtPriceCurrentX96 >= sqrtPriceTargetX96;
             bool exactIn = amountRemaining < 0;
 
             if (exactIn) {
-                uint256 amountRemainingLessFee = FullMath.mulDiv(uint256(-amountRemaining), 1e6 - feePips, 1e6);
+                uint256 amountRemainingLessFee =
+                    FullMath.mulDiv(uint256(-amountRemaining), MAX_FEE_PIPS - _feePips, MAX_FEE_PIPS);
                 amountIn = zeroForOne
                     ? SqrtPriceMath.getAmount0Delta(sqrtPriceTargetX96, sqrtPriceCurrentX96, liquidity, true)
                     : SqrtPriceMath.getAmount1Delta(sqrtPriceCurrentX96, sqrtPriceTargetX96, liquidity, true);
                 if (amountRemainingLessFee >= amountIn) {
+                    // `amountIn` is capped by the target price
                     sqrtPriceNextX96 = sqrtPriceTargetX96;
+                    feeAmount = _feePips == MAX_FEE_PIPS
+                        ? amountIn
+                        : FullMath.mulDivRoundingUp(amountIn, _feePips, MAX_FEE_PIPS - _feePips);
                 } else {
                     sqrtPriceNextX96 = SqrtPriceMath.getNextSqrtPriceFromInput(
                         sqrtPriceCurrentX96, liquidity, amountRemainingLessFee, zeroForOne
                     );
+                    amountIn = zeroForOne
+                        ? SqrtPriceMath.getAmount0Delta(sqrtPriceNextX96, sqrtPriceCurrentX96, liquidity, true)
+                        : SqrtPriceMath.getAmount1Delta(sqrtPriceCurrentX96, sqrtPriceNextX96, liquidity, true);
+                    // we didn't reach the target, so take the remainder of the maximum input as fee
+                    feeAmount = uint256(-amountRemaining) - amountIn;
                 }
+                amountOut = zeroForOne
+                    ? SqrtPriceMath.getAmount1Delta(sqrtPriceNextX96, sqrtPriceCurrentX96, liquidity, false)
+                    : SqrtPriceMath.getAmount0Delta(sqrtPriceCurrentX96, sqrtPriceNextX96, liquidity, false);
             } else {
                 amountOut = zeroForOne
                     ? SqrtPriceMath.getAmount1Delta(sqrtPriceTargetX96, sqrtPriceCurrentX96, liquidity, false)
                     : SqrtPriceMath.getAmount0Delta(sqrtPriceCurrentX96, sqrtPriceTargetX96, liquidity, false);
                 if (uint256(amountRemaining) >= amountOut) {
+                    // `amountOut` is capped by the target price
                     sqrtPriceNextX96 = sqrtPriceTargetX96;
                 } else {
-                    sqrtPriceNextX96 = SqrtPriceMath.getNextSqrtPriceFromOutput(
-                        sqrtPriceCurrentX96, liquidity, uint256(amountRemaining), zeroForOne
-                    );
+                    // cap the output amount to not exceed the remaining output amount
+                    amountOut = uint256(amountRemaining);
+                    sqrtPriceNextX96 =
+                        SqrtPriceMath.getNextSqrtPriceFromOutput(sqrtPriceCurrentX96, liquidity, amountOut, zeroForOne);
                 }
-            }
-
-            bool max = sqrtPriceTargetX96 == sqrtPriceNextX96;
-
-            // get the input/output amounts
-            if (zeroForOne) {
-                amountIn = max && exactIn
-                    ? amountIn
-                    : SqrtPriceMath.getAmount0Delta(sqrtPriceNextX96, sqrtPriceCurrentX96, liquidity, true);
-                amountOut = max && !exactIn
-                    ? amountOut
-                    : SqrtPriceMath.getAmount1Delta(sqrtPriceNextX96, sqrtPriceCurrentX96, liquidity, false);
-            } else {
-                amountIn = max && exactIn
-                    ? amountIn
+                amountIn = zeroForOne
+                    ? SqrtPriceMath.getAmount0Delta(sqrtPriceNextX96, sqrtPriceCurrentX96, liquidity, true)
                     : SqrtPriceMath.getAmount1Delta(sqrtPriceCurrentX96, sqrtPriceNextX96, liquidity, true);
-                amountOut = max && !exactIn
-                    ? amountOut
-                    : SqrtPriceMath.getAmount0Delta(sqrtPriceCurrentX96, sqrtPriceNextX96, liquidity, false);
-            }
-
-            // cap the output amount to not exceed the remaining output amount
-            if (!exactIn && amountOut > uint256(amountRemaining)) {
-                amountOut = uint256(amountRemaining);
-            }
-
-            if (exactIn && sqrtPriceNextX96 != sqrtPriceTargetX96) {
-                // we didn't reach the target, so take the remainder of the maximum input as fee
-                feeAmount = uint256(-amountRemaining) - amountIn;
-            } else {
-                feeAmount = feePips == 1e6 ? amountIn : FullMath.mulDivRoundingUp(amountIn, feePips, 1e6 - feePips);
+                // `feePips` cannot be `MAX_FEE_PIPS` for exact out
+                feeAmount = FullMath.mulDivRoundingUp(amountIn, _feePips, MAX_FEE_PIPS - _feePips);
             }
         }
     }