enhance 0.0.1

Estimate resource requirements for enhance equipments
Documentation 
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

use super::*;


/// Basic reinforcement model parameters
#[derive(Clone, Debug, Ord, PartialOrd, Eq, PartialEq, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct EnhanceLevel<T, R> {
    /// The level increased when the reinforcement is successful
    #[cfg_attr(feature = "serde", serde(default))]
    pub relative_rate: BTreeMap<i16, R>,
    /// Jump to this level when strengthening fails
    #[cfg_attr(feature = "serde", serde(default))]
    pub absolute_rate: BTreeMap<u16, R>,
    /// Weight of equipment broken
    #[cfg_attr(feature = "serde", serde(default))]
    pub broken_rate: R,
    /// The amount of resources that need to be consumed for strengthening
    #[cfg_attr(feature = "serde", serde(default))]
    pub enhance_cost: BTreeMap<T, u128>,
    /// The number of resources that need to be consumed in order to prevent failure penalties
    #[cfg_attr(feature = "serde", serde(default))]
    pub relegate_cost: BTreeMap<T, u128>,
    /// The number of resources that need to be consumed in order to prevent broken penalties
    #[cfg_attr(feature = "serde", serde(default))]
    pub protect_cost: BTreeMap<T, u128>,
}

impl<T, R> EnhanceLevel<T, R> {
    /// Create a simple reinforcement model
    ///
    /// # Arguments
    ///
    /// * `success`: The success rate of the first level of reinforcement
    /// * `failure`: The failure rate of the first level of reinforcement
    /// * `change`: Change level when reinforcement fails
    ///
    /// # Examples
    ///
    /// ```
    ///
    /// ```
    pub fn simple(success: R, failure: R, change: i16) -> Self where R:Default {
        let mut relative_rate = BTreeMap::new();
        relative_rate.insert(1, success);
        relative_rate.insert(change, failure);
        Self {
            relative_rate,
            absolute_rate: BTreeMap::new(),
            broken_rate: Default::default(),
            enhance_cost: BTreeMap::new(),
            relegate_cost: BTreeMap::new(),
            protect_cost: BTreeMap::new(),
        }
    }

    fn total_rate(&self) -> R
    where
        R: Clone + Add<R, Output = R>,
    {
        let mut out = self.broken_rate.clone();
        for weight in self.relative_rate.values() {
            out = out.add(weight.clone());
        }
        for weight in self.absolute_rate.values() {
            out = out.add(weight.clone());
        }
        out
    }
}