From 212eb0671e40cc696283816e55e5dceed9c8a030 Mon Sep 17 00:00:00 2001
From: Zbandut <zbandut@ads.uni-magdeburg.de>
Date: Thu, 17 Apr 2025 15:17:06 +0200
Subject: [PATCH 1/2] Price impact calculator script

---
 python/test/price_impact_calculator.py | 59 ++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 python/test/price_impact_calculator.py

diff --git a/python/test/price_impact_calculator.py b/python/test/price_impact_calculator.py
new file mode 100644
index 0000000..a629ab9
--- /dev/null
+++ b/python/test/price_impact_calculator.py
@@ -0,0 +1,59 @@
+from src.pools.stable_math import compute_invariant, compute_out_given_exact_in
+
+DECIMALS = 10**18
+
+def simulate_custom_slippage(
+    amp: int,
+    total_tvl_usd: float,
+    composition_pct: list[float],
+    swap_amount_usd: float,
+    token_index_in: int = 0,
+    token_index_out: int = 1,
+    fee_decimal: float = 0.0001
+):
+    total_pct = sum(composition_pct)
+    normalized_weights = [p / total_pct for p in composition_pct]
+
+    balances = [int(total_tvl_usd * w * DECIMALS) for w in normalized_weights]
+
+    amount_in = int(swap_amount_usd)
+
+    invariant = compute_invariant(amp, balances.copy())
+
+    amount_out = compute_out_given_exact_in(
+        amplification_parameter=amp,
+        balances=balances.copy(),
+        token_index_in=token_index_in,
+        token_index_out=token_index_out,
+        token_amount_in=amount_in,
+        invariant=invariant,
+    )
+
+    amount_out_after_fee = amount_out * (DECIMALS - int(fee_decimal * DECIMALS)) / DECIMALS
+
+    slippage_bps = (amount_out_after_fee / amount_in - 1) * 10_000
+
+    return {
+        "AMP": amp,
+        "TVL (USD)": total_tvl_usd,
+        "Swap Amount (USD)": swap_amount_usd,
+        "From Token Index": token_index_in,
+        "To Token Index": token_index_out,
+        "Out Amount (USD)": amount_out_after_fee,
+        "Price Impact (bps)": slippage_bps,
+        "Balances (USD)": [round(b / DECIMALS, 2) for b in balances],
+        "Composition (%)": [round(w * 100, 2) for w in normalized_weights],
+        "Pool Size": len(composition_pct),
+    }
+
+if __name__ == "__main__":
+    result = simulate_custom_slippage(
+        amp=8000,
+        total_tvl_usd=40_000_000,
+        composition_pct=[50, 50],
+        swap_amount_usd=10_000_000,
+        token_index_in=0,
+        token_index_out=1,
+        fee_decimal=0.0001,
+    )
+    print(result)

From 1c2d9ce666c12f8e4ffb131897fa3601cbc7a32e Mon Sep 17 00:00:00 2001
From: Mathy Lysandra Zrevk <mathy.lysandra.zrevk@gmail.com>
Date: Thu, 17 Apr 2025 15:51:45 +0200
Subject: [PATCH 2/2] Fix import path for compute_invariant and
 compute_out_given_exact_in

---
 python/test/price_impact_calculator.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/python/test/price_impact_calculator.py b/python/test/price_impact_calculator.py
index a629ab9..0018eaf 100644
--- a/python/test/price_impact_calculator.py
+++ b/python/test/price_impact_calculator.py
@@ -1,4 +1,9 @@
-from src.pools.stable_math import compute_invariant, compute_out_given_exact_in
+import sys
+import os
+
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'src')))
+
+from pools.stable_math import compute_invariant, compute_out_given_exact_in
 
 DECIMALS = 10**18