bambam-omf 0.2.3

OvertureMaps Foundation support for The Behavior and Advanced Mobility Big Access Model
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

use itertools::Itertools;
use routee_compass_core::model::network::Vertex;
use std::collections::HashMap;

use crate::{
    collection::{
        OvertureMapsCollectionError, TransportationConnectorRecord, TransportationSegmentRecord,
    },
    graph::segment_split::SegmentSplit,
};

pub fn get_connectors_mapping(
    connectors: &[TransportationConnectorRecord],
) -> Result<(Vec<Vertex>, HashMap<String, usize>), OvertureMapsCollectionError> {
    let vertices = connectors
        .iter()
        .enumerate()
        .map(|(idx, c)| c.try_to_vertex(idx))
        .collect::<Result<Vec<Vertex>, OvertureMapsCollectionError>>()?;

    let mapping: HashMap<String, usize> = connectors
        .iter()
        .enumerate()
        .map(|(idx, c)| (c.id.clone(), idx))
        .collect();

    Ok((vertices, mapping))
}

pub fn get_connector_splits(
    segment: &TransportationSegmentRecord,
) -> Result<Vec<SegmentSplit>, OvertureMapsCollectionError> {
    Ok(segment
        .connectors
        .as_ref()
        .ok_or(OvertureMapsCollectionError::InvalidSegmentConnectors(
            String::from("connectors is None"),
        ))?
        .iter()
        .tuple_windows()
        .map(|(src, dst)| SegmentSplit::ConnectorSplit {
            connector_id_src: src.connector_id.clone(),
            at_src: src.at,
            connector_id_dst: dst.connector_id.clone(),
            at_dst: dst.at,
        })
        .collect::<Vec<SegmentSplit>>())
}