use crate::{AnyGraph, Edge, Key, Value, Vertex};
use std::collections::HashMap;
pub trait Kinship<K, V>: AnyGraph<K, V>
where
K: Key,
V: Value,
{
fn successors(&self) -> HashMap<Vertex<K, V>, Vec<Edge<K>>>;
fn predecessors(&self) -> HashMap<Vertex<K, V>, Vec<Edge<K>>>;
fn successors_as_key_and_edges(&self) -> HashMap<K, Vec<Edge<K>>> {
self.successors()
.iter()
.fold(HashMap::new(), |mut acc, (vertex, edges)| {
acc.insert(vertex.key().clone(), edges.iter().cloned().collect());
acc
})
}
fn predecessors_as_key_and_edges(&self) -> HashMap<K, Vec<Edge<K>>> {
self.predecessors()
.iter()
.fold(HashMap::new(), |mut acc, (vertex, edges)| {
acc.insert(vertex.key().clone(), edges.iter().cloned().collect());
acc
})
}
fn key_vertex_map(&self) -> HashMap<K, Vertex<K, V>> {
self.vertices()
.iter()
.fold(HashMap::new(), |mut acc, vertex| {
acc.insert(vertex.key().clone(), vertex.clone());
acc
})
}
}