guppy/petgraph_support/
mod.rs

Help
1// Copyright (c) The cargo-guppy Contributors
2// SPDX-License-Identifier: MIT OR Apache-2.0
3
4//! Support for petgraph.
5//!
6//! The code in here is generic over petgraph's traits, and could be upstreamed into petgraph if
7//! desirable.
8
9use fixedbitset::FixedBitSet;
10use petgraph::{graph::IndexType, prelude::*};
11use std::iter::FromIterator;
12
13pub mod dfs;
14pub mod dot;
15pub mod edge_ref;
16pub mod scc;
17pub mod topo;
18pub mod walk;
19
20pub fn edge_triple<ER: EdgeRef>(edge_ref: ER) -> (ER::NodeId, ER::NodeId, ER::EdgeId) {
21    (edge_ref.source(), edge_ref.target(), edge_ref.id())
22}
23
24#[derive(Clone, Debug, Default)]
25pub struct IxBitSet(pub FixedBitSet);
26
27impl IxBitSet {
28    #[inline]
29    pub(crate) fn with_capacity(bits: usize) -> Self {
30        Self(FixedBitSet::with_capacity(bits))
31    }
32
33    #[inline]
34    pub(crate) fn insert_node_ix<Ix: IndexType>(&mut self, bit: NodeIndex<Ix>) {
35        self.0.insert(bit.index());
36    }
37}
38
39impl From<IxBitSet> for FixedBitSet {
40    fn from(ix_set: IxBitSet) -> Self {
41        ix_set.0
42    }
43}
44
45impl<Ix: IndexType> FromIterator<NodeIndex<Ix>> for IxBitSet {
46    fn from_iter<T: IntoIterator<Item = NodeIndex<Ix>>>(iter: T) -> Self {
47        IxBitSet(iter.into_iter().map(|node_ix| node_ix.index()).collect())
48    }
49}
50
51impl<Ix: IndexType> FromIterator<EdgeIndex<Ix>> for IxBitSet {
52    fn from_iter<T: IntoIterator<Item = EdgeIndex<Ix>>>(iter: T) -> Self {
53        IxBitSet(iter.into_iter().map(|edge_ix| edge_ix.index()).collect())
54    }
55}
56
57impl<Ix: IndexType> Extend<NodeIndex<Ix>> for IxBitSet {
58    fn extend<T: IntoIterator<Item = NodeIndex<Ix>>>(&mut self, iter: T) {
59        self.0
60            .extend(iter.into_iter().map(|node_ix| node_ix.index()));
61    }
62}
guppy/petgraph_support/mod.rs

guppy/petgraph_support/
mod.rs
