Files
mev-beta/pkg/uniswap/lookup/optimized.go
2025-10-04 09:31:02 -05:00

70 lines
1.8 KiB
Go

package lookup
import (
"math"
"math/big"
)
// SqrtPriceX96ToPriceWithLookup converts sqrtPriceX96 to a price using lookup tables
func SqrtPriceX96ToPriceWithLookup(sqrtPriceX96 *big.Int) *big.Float {
// price = (sqrtPriceX96 / 2^96)^2
// price = sqrtPriceX96^2 / 2^192
// Calculate sqrtPriceX96^2
sqrtPriceSquared := new(big.Int).Mul(sqrtPriceX96, sqrtPriceX96)
// Convert to big.Float for division
price := new(big.Float).SetInt(sqrtPriceSquared)
// Divide by 2^192 using lookup table
q192 := GetQ192()
q192Float := new(big.Float).SetInt(q192)
price.Quo(price, q192Float)
return price
}
// PriceToSqrtPriceX96WithLookup converts a price to sqrtPriceX96 using lookup tables
func PriceToSqrtPriceX96WithLookup(price *big.Float) *big.Int {
// sqrtPriceX96 = sqrt(price) * 2^96
// Calculate sqrt(price)
sqrtPrice := new(big.Float).Sqrt(price)
// Multiply by 2^96 using lookup table
q96Int := GetQ96()
q96 := new(big.Float).SetInt(q96Int)
sqrtPrice.Mul(sqrtPrice, q96)
// Convert to big.Int
sqrtPriceX96 := new(big.Int)
sqrtPrice.Int(sqrtPriceX96)
return sqrtPriceX96
}
// TickToSqrtPriceX96WithLookup converts a tick to sqrtPriceX96 using lookup tables
func TickToSqrtPriceX96WithLookup(tick int) *big.Int {
// sqrtPriceX96 = 1.0001^(tick/2) * 2^96
// For better performance with large tick values, we'll use logarithms
// but with cached base values
lnBase := math.Log(1.0001) // ln(1.0001) ≈ 9.999500016666e-05
logResult := lnBase * float64(tick) / 2.0
result := math.Exp(logResult)
// Convert to big.Float
price := new(big.Float).SetFloat64(result)
// Multiply by 2^96 using lookup table
q96Int := GetQ96()
q96 := new(big.Float).SetInt(q96Int)
price.Mul(price, q96)
// Convert to big.Int
sqrtPriceX96 := new(big.Int)
price.Int(sqrtPriceX96)
return sqrtPriceX96
}