From 974423d51f04356376e2fb295383bd4319b09acc Mon Sep 17 00:00:00 2001
From: Khalid <KhalidMustafin@users.noreply.github.com>
Date: Tue, 7 Oct 2025 15:03:16 +0500
Subject: [PATCH] =?UTF-8?q?Pure=20integer,=20fixed-point=20arithmetic=20?=
 =?UTF-8?q?=E2=86=92=20reproducible=20across=20environments=20and=20tests.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Fewer operations on hot paths → faster and easier to read.
---
 src/driftpy/math/amm.py  |  8 ++---
 src/driftpy/math/fuel.py | 64 ++++++++++++++++------------------------
 2 files changed, 28 insertions(+), 44 deletions(-)

diff --git a/src/driftpy/math/amm.py b/src/driftpy/math/amm.py
index 589c7d96..6e3b4682 100644
--- a/src/driftpy/math/amm.py
+++ b/src/driftpy/math/amm.py
@@ -111,12 +111,10 @@ def calculate_effective_leverage(
 
     effective_gap = max(0, local_base_asset_value - net_base_asset_value)
 
-    effective_leverage = (
-        effective_gap / (max(0, total_fee_minus_distributions) + 1)
-        + 1 / QUOTE_PRECISION
-    )
+    denom = max(1, total_fee_minus_distributions)
+    effective_leverage_qp = (effective_gap * QUOTE_PRECISION) // denom
 
-    return effective_leverage
+    return effective_leverage_qp
 
 
 def calculate_reference_price_offset(
diff --git a/src/driftpy/math/fuel.py b/src/driftpy/math/fuel.py
index 48304d86..7b406c40 100644
--- a/src/driftpy/math/fuel.py
+++ b/src/driftpy/math/fuel.py
@@ -1,56 +1,42 @@
+from typing import Final
 from driftpy.types import SpotMarketAccount, PerpMarketAccount
 from driftpy.constants.numeric_constants import QUOTE_PRECISION, FUEL_WINDOW
 
+_SCALE_10: Final[int] = QUOTE_PRECISION // 10
+_DENOM: Final[int] = FUEL_WINDOW * _SCALE_10
+
+def _scaled_fuel(value_abs: int, fuel_bonus_numerator: int, boost: int) -> int:
+    if value_abs == 0 or fuel_bonus_numerator == 0 or boost == 0:
+        return 0
+
+    return (value_abs * fuel_bonus_numerator * boost) // _DENOM
+
 
 def calculate_insurance_fuel_bonus(
     spot_market: SpotMarketAccount, token_stake_amount: int, fuel_bonus_numerator: int
 ) -> int:
-    insurance_fund_fuel = (
-        abs(token_stake_amount) * fuel_bonus_numerator
-    ) * spot_market.fuel_boost_insurance
-    insurace_fund_fuel_per_day = insurance_fund_fuel // FUEL_WINDOW
-    insurance_fund_fuel_scaled = insurace_fund_fuel_per_day // (QUOTE_PRECISION // 10)
-
-    return insurance_fund_fuel_scaled
+    value_abs = abs(token_stake_amount)
+    return _scaled_fuel(value_abs, fuel_bonus_numerator, spot_market.fuel_boost_insurance)
 
 
 def calculate_spot_fuel_bonus(
     spot_market: SpotMarketAccount, signed_token_value: int, fuel_bonus_numerator: int
 ) -> int:
-    spot_fuel_scaled: int
-
-    # dust
-    if abs(signed_token_value) <= QUOTE_PRECISION:
-        spot_fuel_scaled = 0
-    elif signed_token_value > 0:
-        deposit_fuel = (
-            abs(signed_token_value) * fuel_bonus_numerator
-        ) * spot_market.fuel_boost_deposits
-        deposit_fuel_per_day = deposit_fuel // FUEL_WINDOW
-        spot_fuel_scaled = deposit_fuel_per_day // (QUOTE_PRECISION // 10)
-    else:
-        borrow_fuel = (
-            abs(signed_token_value) * fuel_bonus_numerator
-        ) * spot_market.fuel_boost_borrows
-        borrow_fuel_per_day = borrow_fuel // FUEL_WINDOW
-        spot_fuel_scaled = borrow_fuel_per_day // (QUOTE_PRECISION // 10)
-
-    return spot_fuel_scaled
+    value_abs = abs(signed_token_value)
+
+    if value_abs <= QUOTE_PRECISION:
+        return 0
+
+    boost = spot_market.fuel_boost_deposits if signed_token_value > 0 else spot_market.fuel_boost_borrows
+    return _scaled_fuel(value_abs, fuel_bonus_numerator, boost)
 
 
 def calculate_perp_fuel_bonus(
     perp_market: PerpMarketAccount, base_asset_value: int, fuel_bonus_numerator: int
 ) -> int:
-    perp_fuel_scaled: int
-
-    # dust
-    if abs(base_asset_value) <= QUOTE_PRECISION:
-        perp_fuel_scaled = 0
-    else:
-        perp_fuel = (
-            abs(base_asset_value) * fuel_bonus_numerator
-        ) * perp_market.fuel_boost_position
-        perp_fuel_per_day = perp_fuel // FUEL_WINDOW
-        perp_fuel_scaled = perp_fuel_per_day // (QUOTE_PRECISION // 10)
-
-    return perp_fuel_scaled
+    value_abs = abs(base_asset_value)
+
+    if value_abs <= QUOTE_PRECISION:
+        return 0
+
+    return _scaled_fuel(value_abs, fuel_bonus_numerator, perp_market.fuel_boost_position)
\ No newline at end of file