solverforge-scoring 0.11.1

Incremental constraint scoring for SolverForge
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

pub trait CrossBiWeight<S, A, B, Sc>: Send + Sync {
    fn score(
        &self,
        solution: &S,
        entities_a: &[A],
        entities_b: &[B],
        a_idx: usize,
        b_idx: usize,
    ) -> Sc;
}

pub struct IndexWeight<W>(W);

impl<W> IndexWeight<W> {
    #[inline]
    pub(crate) fn new(weight: W) -> Self {
        Self(weight)
    }
}

impl<S, A, B, W, Sc> CrossBiWeight<S, A, B, Sc> for IndexWeight<W>
where
    W: Fn(&S, usize, usize) -> Sc + Send + Sync,
{
    #[inline]
    fn score(
        &self,
        solution: &S,
        _entities_a: &[A],
        _entities_b: &[B],
        a_idx: usize,
        b_idx: usize,
    ) -> Sc {
        (self.0)(solution, a_idx, b_idx)
    }
}

pub struct PairWeight<W>(W);

impl<W> PairWeight<W> {
    #[inline]
    pub(crate) fn new(weight: W) -> Self {
        Self(weight)
    }
}

impl<S, A, B, W, Sc> CrossBiWeight<S, A, B, Sc> for PairWeight<W>
where
    W: Fn(&A, &B) -> Sc + Send + Sync,
{
    #[inline]
    fn score(
        &self,
        _solution: &S,
        entities_a: &[A],
        entities_b: &[B],
        a_idx: usize,
        b_idx: usize,
    ) -> Sc {
        (self.0)(&entities_a[a_idx], &entities_b[b_idx])
    }
}