updated formula and tests (#242)

Co-authored-by: Aodhgan <aodhgan@pooltogether.com>
pooltogether · Oct 14, 2021 · b0cec3f · b0cec3f
1 parent 1f26a90
commit b0cec3f
Show file tree

Hide file tree

Showing 2 changed files with 83 additions and 147 deletions.
diff --git a/contracts/DrawCalculator.sol b/contracts/DrawCalculator.sol
@@ -427,14 +427,10 @@ contract DrawCalculator is IDrawCalculator, Ownable {
         pure
         returns (uint256)
     {
-        uint256 bitRangeDecimal = 2**uint256(_bitRangeSize);
-        uint256 numberOfPrizesForIndex = bitRangeDecimal**_prizeTierIndex;
-
-        while (_prizeTierIndex > 0) {
-            numberOfPrizesForIndex -= bitRangeDecimal**(_prizeTierIndex - 1);
-            _prizeTierIndex--;
+        if (_prizeTierIndex > 0) {
+            return ( 1 << _bitRangeSize * _prizeTierIndex ) - ( 1 << _bitRangeSize * (_prizeTierIndex - 1) );
+        } else {
+            return 1;
         }
-
-        return numberOfPrizesForIndex;
     }
 }