art_dice 0.1.0

Contains structures for building dice with arbitrary symbols and functions for determining roll probabilities
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

use std::collections::HashMap;
use std::hash::{Hash, Hasher};

#[derive(PartialEq, Eq, Clone)]
pub struct ItemCounter<T: Hash + PartialEq + Eq + PartialOrd + Ord + Clone> {
    items: HashMap<T, usize>
}

impl<T: Hash + PartialEq + Eq + PartialOrd + Ord + Clone> Hash for ItemCounter<T> {
    fn hash<H: Hasher>(&self, state: &mut H) {
        let mut items: Vec<T> = self.items.keys().cloned().collect();
        items.sort();
        for item in items {
            for _ in 0..(self.items[&item]) {
                item.hash(state);
            }
        }
    }
}

impl<T: Hash + PartialEq + Eq + PartialOrd + Ord + Clone> ItemCounter<T> {
    pub fn new() -> ItemCounter<T> {
        ItemCounter {
            items: HashMap::new()
        }
    }

    pub fn add(&mut self, item: &T) {
        self.add_amount(item, 1)
    }

    pub fn add_amount(&mut self, item: &T, amount: usize) {
        if self.items.contains_key(item) {
            self.items.get_mut(item).map(|x| *x += amount);
        } else {
            self.items.insert(item.clone(), amount);
        }
    }

    pub fn get_count(&self, item: &T) -> usize {
        match self.items.get(item) {
            Some(x) => *x,
            None => 0
        }
    }
}