1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
//! # HydraDX Math
//!
//! A collection of utilities to make performing liquidity pool
//! calculations more convenient.

#![cfg_attr(not(feature = "std"), no_std)]

#[cfg(all(not(feature = "std"), test))]
extern crate std;

#[cfg(test)]
#[macro_use]
extern crate approx;

#[cfg(test)]
mod test_pow_accuracy;

pub mod ema;
pub mod fee;
pub mod fraction;
pub mod lbp;
pub mod liquidity_mining;
pub mod omnipool;
pub mod omnipool_subpools;
pub mod ratio;
pub mod stableswap;
pub mod support;
#[cfg(test)]
pub mod test_utils;
pub mod transcendental;
pub mod types;
pub mod xyk;

#[macro_export]
macro_rules! ensure {
    ($e:expr, $f:expr) => {
        match $e {
            true => (),
            false => {
                return Err($f);
            }
        }
    };
}

#[macro_export]
macro_rules! round_up {
    ($e:expr) => {
        $e.checked_add(FIXED_ROUND_UP).ok_or(Overflow)
    };
}

#[macro_export]
macro_rules! to_u256 {
    ($($x:expr),+) => (
        {($(U256::from($x)),+)}
    );
}

#[macro_export]
macro_rules! to_u128_wrapper {
    ($($x:expr),+) => (
        {($(U128::from($x)),+)}
    );
}

#[macro_export]
macro_rules! to_balance {
    ($x:expr) => {
        Balance::try_from($x).map_err(|_| Overflow)
    };
}

#[macro_export]
macro_rules! to_lbp_weight {
    ($x:expr) => {
        LBPWeight::try_from($x).map_err(|_| Overflow)
    };
}

#[derive(Eq, PartialEq, Debug)]
pub enum MathError {
    Overflow,
    InsufficientOutReserve,
    ZeroWeight,
    ZeroReserve,
    ZeroDuration,
    DivisionByZero,
}

#[cfg(test)]
mod conversion_tests {
    use super::MathError::Overflow;
    use crate::types::Balance;
    use crate::types::LBPWeight;
    use core::convert::TryFrom;

    const FIXED_ROUND_UP: Balance = 1;

    #[test]
    fn test_conversion() {
        let one: u32 = 1;
        assert_eq!(to_balance!(one), Ok(1u128));
        assert_eq!(to_lbp_weight!(one), Ok(1u32));
        assert_eq!(round_up!(Balance::from(one)), Ok(2u128));
    }
}