commerce-theory 0.1.1

Runtime Rust mirror of the CommerceTheory Lean package
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

use std::collections::BTreeMap;

use crate::foundation::*;
use crate::inventory::*;

pub type AllocationKey = (Nat, Nat);

pub fn allocation_key_for_total(allocation: &Allocation) -> AllocationKey {
    allocation_key(allocation)
}

pub fn allocation_key_support(allocations: &[Allocation]) -> Vec<AllocationKey> {
    let mut keys: Vec<_> = allocations.iter().map(allocation_key_for_total).collect();
    keys.sort_unstable();
    keys.dedup();
    keys
}

pub fn allocation_quantity_for_key(
    allocations: &[Allocation],
    key: AllocationKey,
) -> DomainResult<Quantity> {
    checked_sum(
        allocations
            .iter()
            .filter(|allocation| allocation_key_for_total(allocation) == key)
            .map(|allocation| allocation.quantity()),
        "allocation_quantity_for_key",
    )
}

pub fn allocation_quantity_by_key(
    allocations: &[Allocation],
) -> DomainResult<BTreeMap<AllocationKey, Quantity>> {
    let mut totals = BTreeMap::new();
    for allocation in allocations {
        let key = allocation_key_for_total(allocation);
        let current = totals.get(&key).copied().unwrap_or(0);
        totals.insert(
            key,
            checked_add(current, allocation.quantity(), "allocation_quantity_by_key")?,
        );
    }
    Ok(totals)
}