surge-io 0.1.0

Surge I/O — Parser/writer for MATPOWER, PSS/E RAW, IEEE CDF, XIIDM, UCTE, and JSON case formats
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

// SPDX-License-Identifier: LicenseRef-PolyForm-Noncommercial-1.0.0
//! Parser-local disjoint-set utilities.

use std::collections::HashMap;

#[derive(Debug, Clone, Default)]
pub(crate) struct UnionFindStr {
    parent: HashMap<String, String>,
}

impl UnionFindStr {
    pub(crate) fn new() -> Self {
        Self::default()
    }

    pub(crate) fn ensure(&mut self, id: &str) {
        self.parent
            .entry(id.to_string())
            .or_insert_with(|| id.to_string());
    }

    pub(crate) fn find(&mut self, id: &str) -> String {
        self.ensure(id);

        let mut root = id.to_string();
        loop {
            let parent = self
                .parent
                .get(&root)
                .expect("union-find parent exists")
                .clone();
            if parent == root {
                break;
            }
            root = parent;
        }

        let mut cur = id.to_string();
        while cur != root {
            let next = self
                .parent
                .get(&cur)
                .expect("union-find parent exists")
                .clone();
            self.parent.insert(cur.clone(), root.clone());
            cur = next;
        }

        root
    }

    pub(crate) fn union(&mut self, a: &str, b: &str) {
        let ra = self.find(a);
        let rb = self.find(b);
        if ra != rb {
            self.parent.insert(rb, ra);
        }
    }
}