survex-dist 0.3.0

A tool to calculate the distance between two stations in a Survex 3D file.
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

use petgraph::algo::astar;
use petgraph::graph::NodeIndex;
use survex_rs::data::SurveyData;

#[allow(clippy::unnecessary_unwrap)]
pub fn pathfind_route(data: &SurveyData, route: Vec<NodeIndex>) -> Option<Vec<NodeIndex>> {
    let mut path = Vec::new();
    let mut i = 0;

    while i < route.len() - 1 {
        let start = route[i];
        let end = route[i + 1];
        let result = pathfind(data, start, end);

        if result.is_some() {
            let (_, sub_path) = result.unwrap();
            // Remove the last element of the previous path, as it will be duplicated.
            path.pop();
            path.extend(sub_path);
            i += 1;
        } else {
            return None;
        }
    }

    Some(path)
}

fn pathfind(data: &SurveyData, start: NodeIndex, end: NodeIndex) -> Option<(f64, Vec<NodeIndex>)> {
    let end_coords = data.get_by_index(end).unwrap().borrow().coords;
    astar(
        &data.graph,
        start,
        |stn| stn == end,
        |e| *e.weight(),
        |stn| {
            data.get_by_index(stn)
                .unwrap()
                .borrow()
                .coords
                .distance(&end_coords)
        },
    )
}