otspot-core 0.3.1

Core implementation for otspot (LP/QP/MIP solver) — published as a dependency of the otspot facade
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

//! Spatial B&B のノード (Phase 3)。
//!
//! ノード = 1 つの box subproblem。`var_bounds` は親 problem の bound を
//! 分枝により狭めた領域。`lower_bound` は親から継承 or 自前 interval bound の
//! 最大値。`warm` は親解を子に引き継ぐためのスナップショット。

use crate::options::QpWarmStart;

#[derive(Debug, Clone)]
pub(crate) struct BBNode {
    pub var_bounds: Vec<(f64, f64)>,
    pub lower_bound: f64,
    pub depth: usize,
    pub warm: Option<QpWarmStart>,
}

impl BBNode {
    pub fn root(var_bounds: Vec<(f64, f64)>, lower_bound: f64) -> Self {
        Self {
            var_bounds,
            lower_bound,
            depth: 0,
            warm: None,
        }
    }

    pub fn child(
        &self,
        new_bounds: Vec<(f64, f64)>,
        lower_bound: f64,
        warm: Option<QpWarmStart>,
    ) -> Self {
        Self {
            var_bounds: new_bounds,
            lower_bound,
            depth: self.depth + 1,
            warm,
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn root_node_depth_zero_no_warm() {
        let r = BBNode::root(vec![(0.0, 1.0), (-1.0, 1.0)], -5.0);
        assert_eq!(r.depth, 0);
        assert!(r.warm.is_none());
        assert_eq!(r.var_bounds.len(), 2);
        assert_eq!(r.lower_bound, -5.0);
    }

    #[test]
    fn child_increments_depth_and_propagates_warm() {
        let p = BBNode::root(vec![(0.0, 1.0)], -3.0);
        let warm = QpWarmStart {
            x: vec![0.5],
            y: vec![],
            mu: 1e-6,
        };
        let c = p.child(vec![(0.0, 0.5)], -2.5, Some(warm.clone()));
        assert_eq!(c.depth, 1);
        let w = c.warm.expect("warm propagated");
        assert_eq!(w.x, vec![0.5]);
        assert_eq!(c.lower_bound, -2.5);
    }
}