Skip to main content

weavatrix_graph/payload/stable/
view.rs

1use super::StablePayloadGraph;
2use crate::{EdgeEndpoints, GraphView, IndexGraphView, StableEdgeKey, StableNodeKey};
3
4impl<NodePayload, EdgePayload> GraphView for StablePayloadGraph<NodePayload, EdgePayload> {
5    type Node = StableNodeKey;
6    type Edge = StableEdgeKey;
7
8    fn node_count(&self) -> usize {
9        self.node_count()
10    }
11
12    fn edge_count(&self) -> usize {
13        self.edge_count()
14    }
15
16    fn contains_node(&self, node: Self::Node) -> bool {
17        self.node(node).is_some()
18    }
19
20    fn contains_edge(&self, edge: Self::Edge) -> bool {
21        self.edge(edge).is_some()
22    }
23
24    fn node_indices(&self) -> impl Iterator<Item = Self::Node> + '_ {
25        self.nodes().map(|(key, _)| key)
26    }
27
28    fn edge_indices(&self) -> impl Iterator<Item = Self::Edge> + '_ {
29        self.edges().map(|(key, _)| key)
30    }
31
32    fn edge_endpoints(&self, edge: Self::Edge) -> Option<EdgeEndpoints<Self::Node>> {
33        StablePayloadGraph::edge_endpoints(self, edge)
34    }
35
36    fn outgoing_edges(&self, node: Self::Node) -> impl Iterator<Item = Self::Edge> + '_ {
37        StablePayloadGraph::outgoing_edges(self, node)
38    }
39
40    fn incoming_edges(&self, node: Self::Node) -> impl Iterator<Item = Self::Edge> + '_ {
41        StablePayloadGraph::incoming_edges(self, node)
42    }
43}
44
45impl<NodePayload, EdgePayload> IndexGraphView for StablePayloadGraph<NodePayload, EdgePayload> {
46    fn node_bound(&self) -> usize {
47        self.nodes.len()
48    }
49
50    fn edge_bound(&self) -> usize {
51        self.edges.len()
52    }
53
54    fn node_slot(node: Self::Node) -> usize {
55        node.index()
56    }
57
58    fn edge_slot(edge: Self::Edge) -> usize {
59        edge.index()
60    }
61}