simply post order iterators.

This commit is contained in:
Mark Thom
2017-11-12 01:36:19 -07:00
parent e5a235cf80
commit e0f2ca80b5
2 changed files with 6 additions and 28 deletions

View File

@@ -3,7 +3,6 @@ use prolog::fixtures::*;
use prolog::num::{BigInt, Zero}; use prolog::num::{BigInt, Zero};
use prolog::ordered_float::{OrderedFloat}; use prolog::ordered_float::{OrderedFloat};
use std::cell::Cell;
use std::cmp::{min, max}; use std::cmp::{min, max};
use std::vec::Vec; use std::vec::Vec;
@@ -12,20 +11,10 @@ pub struct ArithExprIterator<'a> {
} }
impl<'a> ArithExprIterator<'a> { impl<'a> ArithExprIterator<'a> {
fn push_clause(&mut self, child_num: usize, ct: ClauseType<'a>, child_terms: &'a Vec<Box<Term>>)
{
self.state_stack.push(IteratorState::Clause(child_num, ct, child_terms));
}
fn push_subterm(&mut self, lvl: Level, term: &'a Term) { fn push_subterm(&mut self, lvl: Level, term: &'a Term) {
self.state_stack.push(IteratorState::to_state(lvl, term)); self.state_stack.push(IteratorState::to_state(lvl, term));
} }
fn push_final_cons(&mut self, lvl: Level, cell: &'a Cell<RegType>, head: &'a Term, tail: &'a Term)
{
self.state_stack.push(IteratorState::FinalCons(lvl, cell, head, tail));
}
fn new(term: &'a Term) -> Result<Self, ArithmeticError> { fn new(term: &'a Term) -> Result<Self, ArithmeticError> {
let state = match term { let state = match term {
&Term::AnonVar => &Term::AnonVar =>
@@ -62,12 +51,12 @@ impl<'a> Iterator for ArithExprIterator<'a> {
if child_num == child_terms.len() { if child_num == child_terms.len() {
return Some(TermRef::Clause(ct, child_terms)); return Some(TermRef::Clause(ct, child_terms));
} else { } else {
self.push_clause(child_num + 1, ct, child_terms); self.state_stack.push(IteratorState::Clause(child_num + 1, ct, child_terms));
self.push_subterm(ct.level_of_subterms(), child_terms[child_num].as_ref()); self.push_subterm(ct.level_of_subterms(), child_terms[child_num].as_ref());
} }
}, },
IteratorState::InitialCons(lvl, cell, head, tail) => { IteratorState::InitialCons(lvl, cell, head, tail) => {
self.push_final_cons(lvl, cell, head, tail); self.state_stack.push(IteratorState::FinalCons(lvl, cell, head, tail));
self.push_subterm(Level::Deep, tail); self.push_subterm(Level::Deep, tail);
self.push_subterm(Level::Deep, head); self.push_subterm(Level::Deep, head);
}, },

View File

@@ -1,6 +1,5 @@
use prolog::ast::*; use prolog::ast::*;
use std::cell::Cell;
use std::collections::VecDeque; use std::collections::VecDeque;
use std::iter::*; use std::iter::*;
use std::vec::Vec; use std::vec::Vec;
@@ -10,20 +9,10 @@ pub struct QueryIterator<'a> {
} }
impl<'a> QueryIterator<'a> { impl<'a> QueryIterator<'a> {
fn push_clause(&mut self, child_num: usize, ct: ClauseType<'a>, child_terms: &'a Vec<Box<Term>>)
{
self.state_stack.push(IteratorState::Clause(child_num, ct, child_terms));
}
fn push_subterm(&mut self, lvl: Level, term: &'a Term) { fn push_subterm(&mut self, lvl: Level, term: &'a Term) {
self.state_stack.push(IteratorState::to_state(lvl, term)); self.state_stack.push(IteratorState::to_state(lvl, term));
} }
fn push_final_cons(&mut self, lvl: Level, cell: &'a Cell<RegType>, head: &'a Term, tail: &'a Term)
{
self.state_stack.push(IteratorState::FinalCons(lvl, cell, head, tail));
}
fn from_term(term: &'a Term) -> Self { fn from_term(term: &'a Term) -> Self {
let state = match term { let state = match term {
&Term::AnonVar => &Term::AnonVar =>
@@ -76,7 +65,7 @@ impl QueryTerm {
impl<'a> Iterator for QueryIterator<'a> { impl<'a> Iterator for QueryIterator<'a> {
type Item = TermRef<'a>; type Item = TermRef<'a>;
fn next(&mut self) -> Option<Self::Item> { fn next(&mut self) -> Option<Self::Item> {
while let Some(iter_state) = self.state_stack.pop() { while let Some(iter_state) = self.state_stack.pop() {
match iter_state { match iter_state {
@@ -93,12 +82,12 @@ impl<'a> Iterator for QueryIterator<'a> {
return None return None
}; };
} else { } else {
self.push_clause(child_num + 1, ct, child_terms); self.state_stack.push(IteratorState::Clause(child_num + 1, ct, child_terms));
self.push_subterm(ct.level_of_subterms(), child_terms[child_num].as_ref()); self.push_subterm(ct.level_of_subterms(), child_terms[child_num].as_ref());
} }
}, },
IteratorState::InitialCons(lvl, cell, head, tail) => { IteratorState::InitialCons(lvl, cell, head, tail) => {
self.push_final_cons(lvl, cell, head, tail); self.state_stack.push(IteratorState::FinalCons(lvl, cell, head, tail));
self.push_subterm(Level::Deep, tail); self.push_subterm(Level::Deep, tail);
self.push_subterm(Level::Deep, head); self.push_subterm(Level::Deep, head);
}, },
@@ -111,7 +100,7 @@ impl<'a> Iterator for QueryIterator<'a> {
}; };
} }
None None
} }
} }