feat: update computeWeights

primitivefinance · Dec 11, 2023 · 0f9c8c2 · 0f9c8c2
1 parent 0609815
commit 0f9c8c2
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/contracts/strategies/G3MStrategy.sol b/contracts/strategies/G3MStrategy.sol
@@ -256,11 +256,22 @@ contract G3MStrategy is IG3MStrategy {
                 configs[poolId].endUpdate - configs[poolId].startUpdate;
             uint256 timeElapsed = block.timestamp - configs[poolId].startUpdate;
             uint256 t = timeElapsed * WAD / duration;
-            uint256 fw0 = G3MStrategyLib.computeISFunction(config.startWeightX);
-            uint256 fw1 = G3MStrategyLib.computeISFunction(config.endWeightX);
 
-            weightX = G3MStrategyLib.computeSFunction(t, fw1 - fw0, fw0);
-            weightY = WAD - weightX;
+            if (config.startWeightX < config.endWeightX) {
+                uint256 fw0 =
+                    G3MStrategyLib.computeISFunction(config.startWeightX);
+                uint256 fw1 =
+                    G3MStrategyLib.computeISFunction(config.endWeightX);
+                weightX = G3MStrategyLib.computeSFunction(t, fw1 - fw0, fw0);
+                weightY = WAD - weightX;
+            } else {
+                uint256 fw0 =
+                    G3MStrategyLib.computeISFunction(WAD - config.startWeightX);
+                uint256 fw1 =
+                    G3MStrategyLib.computeISFunction(WAD - config.endWeightX);
+                weightY = G3MStrategyLib.computeSFunction(t, fw1 - fw0, fw0);
+                weightX = WAD - weightY;
+            }
         }
     }
 }