Moves the graph isomorphism implementation out of rdf_test_casespull/10/head
parent
f90cfbc39a
commit
723bb22b18
@ -0,0 +1,154 @@ |
||||
use model::data::*; |
||||
use std::collections::BTreeSet; |
||||
use std::collections::HashMap; |
||||
use std::collections::HashSet; |
||||
use std::collections::hash_map::DefaultHasher; |
||||
use std::hash::Hash; |
||||
use std::hash::Hasher; |
||||
use store::memory::MemoryGraph; |
||||
|
||||
#[derive(Eq, PartialEq, Hash, Ord, PartialOrd)] |
||||
struct SubjectPredicate<'a> { |
||||
subject: &'a NamedOrBlankNode, |
||||
predicate: &'a NamedNode, |
||||
} |
||||
|
||||
impl<'a> SubjectPredicate<'a> { |
||||
fn new(subject: &'a NamedOrBlankNode, predicate: &'a NamedNode) -> Self { |
||||
Self { subject, predicate } |
||||
} |
||||
} |
||||
|
||||
#[derive(Eq, PartialEq, Hash, Ord, PartialOrd)] |
||||
struct PredicateObject<'a> { |
||||
predicate: &'a NamedNode, |
||||
object: &'a Term, |
||||
} |
||||
|
||||
impl<'a> PredicateObject<'a> { |
||||
fn new(predicate: &'a NamedNode, object: &'a Term) -> Self { |
||||
Self { predicate, object } |
||||
} |
||||
} |
||||
|
||||
fn subject_predicates_for_object<'a>( |
||||
graph: &'a MemoryGraph, |
||||
object: &'a Term, |
||||
) -> impl Iterator<Item = SubjectPredicate<'a>> { |
||||
graph |
||||
.triples_for_object(object) |
||||
.map(|t| SubjectPredicate::new(t.subject(), t.predicate())) |
||||
} |
||||
|
||||
fn predicate_objects_for_subject<'a>( |
||||
graph: &'a MemoryGraph, |
||||
subject: &'a NamedOrBlankNode, |
||||
) -> impl Iterator<Item = PredicateObject<'a>> { |
||||
graph |
||||
.triples_for_subject(subject) |
||||
.map(|t| PredicateObject::new(t.predicate(), t.object())) |
||||
} |
||||
|
||||
fn hash_blank_nodes<'a>( |
||||
bnodes: HashSet<&'a BlankNode>, |
||||
graph: &'a MemoryGraph, |
||||
) -> HashMap<u64, Vec<&'a BlankNode>> { |
||||
let mut bnodes_by_hash: HashMap<u64, Vec<&BlankNode>> = HashMap::default(); |
||||
|
||||
// NB: we need to sort the triples to have the same hash
|
||||
for bnode in bnodes.into_iter() { |
||||
let mut hasher = DefaultHasher::new(); |
||||
|
||||
{ |
||||
let subject = NamedOrBlankNode::from(bnode.clone()); |
||||
let mut po_set: BTreeSet<PredicateObject> = BTreeSet::default(); |
||||
for po in predicate_objects_for_subject(&graph, &subject) { |
||||
if !po.object.is_blank_node() { |
||||
po_set.insert(po); |
||||
} |
||||
} |
||||
for po in po_set { |
||||
po.hash(&mut hasher); |
||||
} |
||||
} |
||||
|
||||
{ |
||||
let object = Term::from(bnode.clone()); |
||||
let mut sp_set: BTreeSet<SubjectPredicate> = BTreeSet::default(); |
||||
for sp in subject_predicates_for_object(&graph, &object) { |
||||
if !sp.subject.is_blank_node() { |
||||
sp_set.insert(sp); |
||||
} |
||||
} |
||||
for sp in sp_set { |
||||
sp.hash(&mut hasher); |
||||
} |
||||
} |
||||
|
||||
bnodes_by_hash |
||||
.entry(hasher.finish()) |
||||
.or_insert_with(Vec::default) |
||||
.push(bnode); |
||||
} |
||||
|
||||
bnodes_by_hash |
||||
} |
||||
|
||||
pub trait GraphIsomorphism { |
||||
/// Checks if two graphs are [isomorphic](https://www.w3.org/TR/rdf11-concepts/#dfn-graph-isomorphism)
|
||||
fn is_isomorphic(&self, other: &Self) -> bool; |
||||
} |
||||
|
||||
impl GraphIsomorphism for MemoryGraph { |
||||
//TODO: proper isomorphism building
|
||||
fn is_isomorphic(&self, other: &Self) -> bool { |
||||
if self.len() != other.len() { |
||||
return false; |
||||
} |
||||
|
||||
let mut self_bnodes: HashSet<&BlankNode> = HashSet::default(); |
||||
let mut other_bnodes: HashSet<&BlankNode> = HashSet::default(); |
||||
|
||||
for t in self { |
||||
if let NamedOrBlankNode::BlankNode(subject) = t.subject() { |
||||
self_bnodes.insert(subject); |
||||
if let Term::BlankNode(object) = t.object() { |
||||
self_bnodes.insert(object); |
||||
} |
||||
} else if let Term::BlankNode(object) = t.object() { |
||||
self_bnodes.insert(object); |
||||
} else if !other.contains(t) { |
||||
return false; |
||||
} |
||||
} |
||||
for t in other { |
||||
if let NamedOrBlankNode::BlankNode(subject) = t.subject() { |
||||
other_bnodes.insert(subject); |
||||
if let Term::BlankNode(object) = t.object() { |
||||
other_bnodes.insert(object); |
||||
} |
||||
} else if let Term::BlankNode(object) = t.object() { |
||||
other_bnodes.insert(object); |
||||
} else if !self.contains(t) { |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
let self_bnodes_by_hash = hash_blank_nodes(self_bnodes, &self); |
||||
let other_bnodes_by_hash = hash_blank_nodes(other_bnodes, &other); |
||||
|
||||
if self_bnodes_by_hash.len() != other_bnodes_by_hash.len() { |
||||
return false; |
||||
} |
||||
|
||||
for hash in self_bnodes_by_hash.keys() { |
||||
if self_bnodes_by_hash.get(hash).map(|l| l.len()) |
||||
!= other_bnodes_by_hash.get(hash).map(|l| l.len()) |
||||
{ |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
true |
||||
} |
||||
} |
@ -0,0 +1,148 @@ |
||||
use model::data::*; |
||||
use std::collections::HashSet; |
||||
use std::fmt; |
||||
use std::iter::FromIterator; |
||||
|
||||
#[derive(Debug, Clone, Default)] |
||||
pub struct MemoryGraph { |
||||
triples: HashSet<Triple>, |
||||
} |
||||
|
||||
impl MemoryGraph { |
||||
pub fn iter(&self) -> impl Iterator<Item = &Triple> { |
||||
self.triples.iter() |
||||
} |
||||
|
||||
pub fn triples_for_subject<'a>( |
||||
&'a self, |
||||
subject: &'a NamedOrBlankNode, |
||||
) -> impl Iterator<Item = &'a Triple> { |
||||
self.iter().filter(move |t| t.subject() == subject) |
||||
} |
||||
|
||||
pub fn triples_for_predicate<'a>( |
||||
&'a self, |
||||
predicate: &'a NamedNode, |
||||
) -> impl Iterator<Item = &'a Triple> { |
||||
self.iter().filter(move |t| t.predicate() == predicate) |
||||
} |
||||
|
||||
pub fn triples_for_object<'a>(&'a self, object: &'a Term) -> impl Iterator<Item = &'a Triple> { |
||||
self.iter().filter(move |t| t.object() == object) |
||||
} |
||||
|
||||
pub fn triples_for_subject_predicate<'a>( |
||||
&'a self, |
||||
subject: &'a NamedOrBlankNode, |
||||
predicate: &'a NamedNode, |
||||
) -> impl Iterator<Item = &'a Triple> { |
||||
self.iter() |
||||
.filter(move |t| t.subject() == subject && t.predicate() == predicate) |
||||
} |
||||
|
||||
pub fn objects_for_subject_predicate<'a>( |
||||
&'a self, |
||||
subject: &'a NamedOrBlankNode, |
||||
predicate: &'a NamedNode, |
||||
) -> impl Iterator<Item = &'a Term> { |
||||
self.triples_for_subject_predicate(subject, predicate) |
||||
.map(|t| t.object()) |
||||
} |
||||
|
||||
pub fn object_for_subject_predicate<'a>( |
||||
&'a self, |
||||
subject: &'a NamedOrBlankNode, |
||||
predicate: &'a NamedNode, |
||||
) -> Option<&'a Term> { |
||||
self.objects_for_subject_predicate(subject, predicate) |
||||
.nth(0) |
||||
} |
||||
|
||||
pub fn triples_for_predicate_object<'a>( |
||||
&'a self, |
||||
predicate: &'a NamedNode, |
||||
object: &'a Term, |
||||
) -> impl Iterator<Item = &'a Triple> { |
||||
self.iter() |
||||
.filter(move |t| t.predicate() == predicate && t.object() == object) |
||||
} |
||||
|
||||
pub fn subjects_for_predicate_object<'a>( |
||||
&'a self, |
||||
predicate: &'a NamedNode, |
||||
object: &'a Term, |
||||
) -> impl Iterator<Item = &'a NamedOrBlankNode> { |
||||
self.triples_for_predicate_object(predicate, object) |
||||
.map(|t| t.subject()) |
||||
} |
||||
|
||||
pub fn subject_for_predicate_object<'a>( |
||||
&'a self, |
||||
predicate: &'a NamedNode, |
||||
object: &'a Term, |
||||
) -> Option<&'a NamedOrBlankNode> { |
||||
self.subjects_for_predicate_object(predicate, object).nth(0) |
||||
} |
||||
|
||||
pub fn len(&self) -> usize { |
||||
self.triples.len() |
||||
} |
||||
|
||||
pub fn is_empty(&self) -> bool { |
||||
self.triples.is_empty() |
||||
} |
||||
|
||||
pub fn contains(&self, value: &Triple) -> bool { |
||||
self.triples.contains(value) |
||||
} |
||||
|
||||
pub fn insert(&mut self, value: Triple) -> bool { |
||||
self.triples.insert(value) |
||||
} |
||||
} |
||||
|
||||
impl fmt::Display for MemoryGraph { |
||||
fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result { |
||||
for triple in &self.triples { |
||||
write!(fmt, "{}\n", triple)?; |
||||
} |
||||
Ok(()) |
||||
} |
||||
} |
||||
|
||||
impl IntoIterator for MemoryGraph { |
||||
type Item = Triple; |
||||
type IntoIter = <HashSet<Triple> as IntoIterator>::IntoIter; |
||||
|
||||
fn into_iter(self) -> <Self as IntoIterator>::IntoIter { |
||||
self.triples.into_iter() |
||||
} |
||||
} |
||||
|
||||
impl<'a> IntoIterator for &'a MemoryGraph { |
||||
type Item = &'a Triple; |
||||
type IntoIter = <&'a HashSet<Triple> as IntoIterator>::IntoIter; |
||||
|
||||
fn into_iter(self) -> <Self as IntoIterator>::IntoIter { |
||||
self.triples.iter() |
||||
} |
||||
} |
||||
|
||||
impl FromIterator<Triple> for MemoryGraph { |
||||
fn from_iter<I: IntoIterator<Item = Triple>>(iter: I) -> Self { |
||||
let triples = HashSet::from_iter(iter); |
||||
Self { triples } |
||||
} |
||||
} |
||||
|
||||
impl Extend<Triple> for MemoryGraph { |
||||
fn extend<I: IntoIterator<Item = Triple>>(&mut self, iter: I) { |
||||
self.triples.extend(iter) |
||||
} |
||||
} |
||||
|
||||
impl<'a> Extend<&'a Triple> for MemoryGraph { |
||||
fn extend<I: IntoIterator<Item = &'a Triple>>(&mut self, iter: I) { |
||||
self.triples.extend(iter.into_iter().cloned()) |
||||
} |
||||
} |
@ -0,0 +1,2 @@ |
||||
pub mod isomorphism; |
||||
pub mod memory; |
Loading…
Reference in new issue