powerio 0.5.0

Fast case parsing and conversion: "pandoc for power systems"
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

//! Typed views over aux object types the transmission core does not model.
//!
//! Contingencies, limit sets, and rating set names matter to transmission
//! studies even though [`crate::network::Network`] does not model them. They
//! are retained losslessly by the generic layer ([`super::AuxFile`]); the
//! views here give them names and structure. Everything stays read only: the
//! data round trips through the retained source, untouched.

use super::auxiliary::AuxFile;

/// One contingency from a `Contingency` DATA section, with the actions of its
/// `CTGElement` SUBDATA.
#[derive(Debug, Clone, PartialEq)]
pub struct Contingency {
    /// `CTGLabel`, the contingency's unique name.
    pub label: String,
    /// One entry per CTGElement line: the action string PowerWorld calls
    /// "WhoAmI + action", e.g. `BRANCH 2 1 1 OPEN`.
    pub actions: Vec<String>,
}

/// The contingencies of a parsed aux file, in file order.
///
/// Empty when the file carries no `Contingency` sections. Rows with no
/// `CTGLabel` field are skipped (a contingency without a name is not
/// addressable).
#[must_use]
pub fn contingencies(aux: &AuxFile) -> Vec<Contingency> {
    let mut out = Vec::new();
    for blk in aux.data_of("Contingency") {
        let Some(label_at) = blk.field_index("CTGLabel") else {
            continue;
        };
        for row in &blk.rows {
            let Some(label) = row.values.get(label_at) else {
                continue;
            };
            let actions = row
                .subdata
                .iter()
                .filter(|s| s.name.eq_ignore_ascii_case("CTGElement"))
                .flat_map(|s| s.lines.iter())
                .filter_map(|line| first_quoted(line))
                .map(str::to_string)
                .collect();
            out.push(Contingency {
                label: label.clone(),
                actions,
            });
        }
    }
    out
}

/// Name → row lookup for the per object-type rating set names
/// (`RatingSetNameBus`, `RatingSetNameBranch`, `RatingSetNameInterface`).
/// Returns `(set_number, name)` pairs in file order.
#[must_use]
pub fn rating_set_names(aux: &AuxFile, object_type: &str) -> Vec<(usize, String)> {
    let mut out = Vec::new();
    for blk in aux.data_of(object_type) {
        let (Some(num_at), Some(name_at)) = (
            blk.field_index("RatingSetNum"),
            blk.field_index("RatingSetName"),
        ) else {
            continue;
        };
        for row in &blk.rows {
            if let (Some(num), Some(name)) = (row.values.get(num_at), row.values.get(name_at)) {
                if let Ok(n) = num.trim().parse() {
                    out.push((n, name.clone()));
                }
            }
        }
    }
    out
}

/// The interior of the first `"..."` on a CTGElement line: its action string.
fn first_quoted(line: &str) -> Option<&str> {
    let start = line.find('"')? + 1;
    let end = start + line[start..].find('"')?;
    Some(line[start..end].trim())
}