multiversx_sc_modules/bonding_curve/curves/
linear_function.rs

1multiversx_sc::imports!();
2multiversx_sc::derive_imports!();
3
4use crate::bonding_curve::{curves::curve_function::CurveFunction, utils::structs::CurveArguments};
5
6#[type_abi]
7#[derive(TopEncode, TopDecode, NestedEncode, NestedDecode, PartialEq, Eq, Clone)]
8pub struct LinearFunction<M: ManagedTypeApi> {
9    pub initial_price: BigUint<M>,
10    pub linear_coefficient: BigUint<M>,
11}
12
13impl<M: ManagedTypeApi> CurveFunction<M> for LinearFunction<M> {
14    fn calculate_price(
15        &self,
16        token_start: &BigUint<M>,
17        amount: &BigUint<M>,
18        _arguments: &CurveArguments<M>,
19    ) -> BigUint<M> {
20        &self.linear_coefficient * &sum_interval(amount, token_start) + &self.initial_price * amount
21    }
22}
23
24fn sum_interval<M: ManagedTypeApi>(n: &BigUint<M>, x: &BigUint<M>) -> BigUint<M> {
25    x * n + &(n - 1u32) * n / 2u32
26}