feat: update computeWeights function

primitivefinance · Dec 8, 2023 · 38151aa · 38151aa
1 parent 3863d0e
commit 38151aa
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/contracts/strategies/G3MStrategy.sol b/contracts/strategies/G3MStrategy.sol
@@ -248,14 +248,19 @@ contract G3MStrategy is IG3MStrategy {
     {
         Config memory config = configs[poolId];
 
-        uint256 duration =
-            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 (block.timestamp >= config.endUpdate) {
+            weightX = config.endWeightX;
+            weightY = WAD - config.endWeightX;
+        } else {
+            uint256 duration =
+                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;
+        }
     }
 }