872 lines
24 KiB
Rust
872 lines
24 KiB
Rust
use crate::arena::*;
|
|
use crate::atom_table::*;
|
|
use crate::instructions::*;
|
|
use crate::machine::heap::*;
|
|
use crate::machine::loader::PredicateQueue;
|
|
use crate::machine::machine_errors::*;
|
|
use crate::machine::machine_indices::*;
|
|
use crate::parser::ast::*;
|
|
use crate::parser::parser::CompositeOpDesc;
|
|
use crate::parser::rug::{Integer, Rational};
|
|
use crate::types::*;
|
|
|
|
use fxhash::FxBuildHasher;
|
|
|
|
use indexmap::{IndexMap, IndexSet};
|
|
use ordered_float::OrderedFloat;
|
|
|
|
use slice_deque::*;
|
|
|
|
use std::cell::Cell;
|
|
use std::convert::TryFrom;
|
|
use std::fmt;
|
|
use std::ops::AddAssign;
|
|
use std::path::PathBuf;
|
|
use std::rc::Rc;
|
|
|
|
use crate::{is_infix, is_postfix};
|
|
|
|
pub type PredicateKey = (Atom, usize); // name, arity.
|
|
|
|
pub type Predicate = Vec<PredicateClause>;
|
|
|
|
// vars of predicate, toplevel offset. Vec<Term> is always a vector
|
|
// of vars (we get their adjoining cells this way).
|
|
pub type JumpStub = Vec<Term>;
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub enum TopLevel {
|
|
Fact(Term), // Term, line_num, col_num
|
|
Predicate(Predicate),
|
|
Query(Vec<QueryTerm>),
|
|
Rule(Rule), // Rule, line_num, col_num
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy)]
|
|
pub enum AppendOrPrepend {
|
|
Append,
|
|
Prepend,
|
|
}
|
|
|
|
impl AppendOrPrepend {
|
|
#[inline]
|
|
pub(crate) fn is_append(self) -> bool {
|
|
match self {
|
|
AppendOrPrepend::Append => true,
|
|
AppendOrPrepend::Prepend => false,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy)]
|
|
pub enum Level {
|
|
Deep,
|
|
Root,
|
|
Shallow,
|
|
}
|
|
|
|
impl Level {
|
|
pub(crate) fn child_level(self) -> Level {
|
|
match self {
|
|
Level::Root => Level::Shallow,
|
|
_ => Level::Deep,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub enum QueryTerm {
|
|
// register, clause type, subterms, use default call policy.
|
|
Clause(Cell<RegType>, ClauseType, Vec<Term>, bool),
|
|
BlockedCut, // a cut which is 'blocked by letters', like the P term in P -> Q.
|
|
UnblockedCut(Cell<VarReg>),
|
|
GetLevelAndUnify(Cell<VarReg>, Rc<String>),
|
|
Jump(JumpStub),
|
|
}
|
|
|
|
impl QueryTerm {
|
|
pub(crate) fn set_default_caller(&mut self) {
|
|
match self {
|
|
&mut QueryTerm::Clause(_, _, _, ref mut use_default_cp) => *use_default_cp = true,
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
pub(crate) fn arity(&self) -> usize {
|
|
match self {
|
|
&QueryTerm::Clause(_, _, ref subterms, ..) => subterms.len(),
|
|
&QueryTerm::BlockedCut | &QueryTerm::UnblockedCut(..) => 0,
|
|
&QueryTerm::Jump(ref vars) => vars.len(),
|
|
&QueryTerm::GetLevelAndUnify(..) => 1,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct Rule {
|
|
pub(crate) head: (Atom, Vec<Term>, QueryTerm),
|
|
pub(crate) clauses: Vec<QueryTerm>,
|
|
}
|
|
|
|
#[derive(Clone, Debug, Hash)]
|
|
pub enum ListingSource {
|
|
DynamicallyGenerated,
|
|
File(Atom, PathBuf), // filename, path
|
|
User,
|
|
}
|
|
|
|
impl ListingSource {
|
|
pub(crate) fn from_file_and_path(filename: Atom, path_buf: PathBuf) -> Self {
|
|
ListingSource::File(filename, path_buf)
|
|
}
|
|
}
|
|
|
|
pub trait ClauseInfo {
|
|
fn is_consistent(&self, clauses: &PredicateQueue) -> bool {
|
|
match clauses.first() {
|
|
Some(cl) => {
|
|
self.name() == ClauseInfo::name(cl) && self.arity() == ClauseInfo::arity(cl)
|
|
}
|
|
None => true,
|
|
}
|
|
}
|
|
|
|
fn name(&self) -> Option<Atom>;
|
|
fn arity(&self) -> usize;
|
|
}
|
|
|
|
impl ClauseInfo for PredicateKey {
|
|
#[inline]
|
|
fn name(&self) -> Option<Atom> {
|
|
Some(self.0)
|
|
}
|
|
|
|
#[inline]
|
|
fn arity(&self) -> usize {
|
|
self.1
|
|
}
|
|
}
|
|
|
|
impl ClauseInfo for Term {
|
|
fn name(&self) -> Option<Atom> {
|
|
//, atom_tbl: &AtomTable) -> Option<StringBuffer> {
|
|
match self {
|
|
Term::Clause(_, name, terms) => {
|
|
// let str_buf = StringBuffer::from(*name, atom_tbl);
|
|
|
|
match name.as_str() {
|
|
// str_buf.as_str() {
|
|
":-" => {
|
|
match terms.len() {
|
|
1 => None, // a declaration.
|
|
2 => terms[0].name(), //.map(|name| StringBuffer::from(name, atom_tbl)),
|
|
_ => Some(*name),
|
|
}
|
|
}
|
|
_ => Some(*name), //str_buf),
|
|
}
|
|
}
|
|
Term::Literal(_, Literal::Atom(name)) => Some(*name), //Some(StringBuffer::from(*name, atom_tbl)),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
fn arity(&self) -> usize {
|
|
match self {
|
|
Term::Clause(_, name, terms) => match name.as_str() {
|
|
":-" => match terms.len() {
|
|
1 => 0,
|
|
2 => terms[0].arity(),
|
|
_ => terms.len(),
|
|
},
|
|
_ => terms.len(),
|
|
},
|
|
_ => 0,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ClauseInfo for Rule {
|
|
fn name(&self) -> Option<Atom> {
|
|
Some(self.head.0)
|
|
}
|
|
|
|
fn arity(&self) -> usize {
|
|
self.head.1.len()
|
|
}
|
|
}
|
|
|
|
impl ClauseInfo for PredicateClause {
|
|
fn name(&self) -> Option<Atom> {
|
|
match self {
|
|
&PredicateClause::Fact(ref term, ..) => term.name(),
|
|
&PredicateClause::Rule(ref rule, ..) => rule.name(),
|
|
}
|
|
}
|
|
|
|
fn arity(&self) -> usize {
|
|
match self {
|
|
&PredicateClause::Fact(ref term, ..) => term.arity(),
|
|
&PredicateClause::Rule(ref rule, ..) => rule.arity(),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub enum PredicateClause {
|
|
Fact(Term),
|
|
Rule(Rule),
|
|
}
|
|
|
|
impl PredicateClause {
|
|
// TODO: add this to `Term` in `crate::parser` like `first_arg`.
|
|
pub(crate) fn args(&self) -> Option<&[Term]> {
|
|
match self {
|
|
PredicateClause::Fact(term, ..) => match term {
|
|
Term::Clause(_, _, args) => Some(&args),
|
|
_ => None,
|
|
},
|
|
PredicateClause::Rule(rule, ..) => {
|
|
if rule.head.1.is_empty() {
|
|
None
|
|
} else {
|
|
Some(&rule.head.1)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub enum ModuleSource {
|
|
Library(Atom),
|
|
File(Atom),
|
|
}
|
|
|
|
impl ModuleSource {
|
|
pub(crate) fn as_functor_stub(&self) -> MachineStub {
|
|
match self {
|
|
ModuleSource::Library(name) => {
|
|
functor!(atom!("library"), [atom(name)])
|
|
}
|
|
ModuleSource::File(name) => {
|
|
functor!(name)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Copy, Hash, Debug)]
|
|
pub enum MetaSpec {
|
|
Minus,
|
|
Plus,
|
|
Either,
|
|
RequiresExpansionWithArgument(usize),
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub enum Declaration {
|
|
Dynamic(Atom, usize),
|
|
MetaPredicate(Atom, Atom, Vec<MetaSpec>), // module name, name, meta-specs
|
|
Module(ModuleDecl),
|
|
NonCountedBacktracking(Atom, usize), // name, arity
|
|
Op(OpDecl),
|
|
UseModule(ModuleSource),
|
|
UseQualifiedModule(ModuleSource, IndexSet<ModuleExport>),
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy, Eq, Hash, PartialEq, Ord, PartialOrd)]
|
|
pub struct OpDecl {
|
|
pub(crate) op_desc: OpDesc,
|
|
pub(crate) name: Atom,
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn fixity(spec: u32) -> Fixity {
|
|
match spec {
|
|
XFY | XFX | YFX => Fixity::In,
|
|
XF | YF => Fixity::Post,
|
|
FX | FY => Fixity::Pre,
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
|
|
impl OpDecl {
|
|
#[inline]
|
|
pub(crate) fn new(op_desc: OpDesc, name: Atom) -> Self {
|
|
Self { op_desc, name }
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn remove(&mut self, op_dir: &mut OpDir) {
|
|
let prec = self.op_desc.get_prec();
|
|
self.op_desc.set(0, self.op_desc.get_spec());
|
|
|
|
self.insert_into_op_dir(op_dir);
|
|
self.op_desc.set(prec, self.op_desc.get_spec());
|
|
}
|
|
|
|
pub(crate) fn insert_into_op_dir(&self, op_dir: &mut OpDir) -> Option<OpDesc> {
|
|
let key = (self.name, fixity(self.op_desc.get_spec() as u32));
|
|
|
|
match op_dir.get_mut(&key) {
|
|
Some(cell) => {
|
|
let (old_prec, old_spec) = cell.get();
|
|
cell.set(self.op_desc.get_prec(), self.op_desc.get_spec());
|
|
return Some(OpDesc::build_with(old_prec, old_spec));
|
|
}
|
|
None => {}
|
|
}
|
|
|
|
op_dir.insert(key, self.op_desc)
|
|
}
|
|
|
|
pub(crate) fn submit(
|
|
&self,
|
|
existing_desc: Option<CompositeOpDesc>,
|
|
op_dir: &mut OpDir,
|
|
) -> Result<(), SessionError> {
|
|
let (spec, name) = (self.op_desc.get_spec(), self.name.clone());
|
|
|
|
if is_infix!(spec as u32) {
|
|
if let Some(desc) = existing_desc {
|
|
if desc.post > 0 {
|
|
return Err(SessionError::OpIsInfixAndPostFix(name));
|
|
}
|
|
}
|
|
}
|
|
|
|
if is_postfix!(spec as u32) {
|
|
if let Some(desc) = existing_desc {
|
|
if desc.inf > 0 {
|
|
return Err(SessionError::OpIsInfixAndPostFix(name));
|
|
}
|
|
}
|
|
}
|
|
|
|
self.insert_into_op_dir(op_dir);
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub enum AtomOrString {
|
|
Atom(Atom),
|
|
String(String),
|
|
}
|
|
|
|
impl AtomOrString {
|
|
#[inline]
|
|
pub fn as_str(&self) -> &str {
|
|
match self {
|
|
AtomOrString::Atom(atom) if atom == &atom!("[]") => "",
|
|
AtomOrString::Atom(atom) => atom.as_str(),
|
|
AtomOrString::String(string) => string.as_str(),
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub fn to_string(self) -> String {
|
|
match self {
|
|
AtomOrString::Atom(atom) => {
|
|
atom.as_str().to_owned()
|
|
}
|
|
AtomOrString::String(string) => {
|
|
string
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub(crate) fn fetch_atom_op_spec(
|
|
name: Atom,
|
|
spec: Option<OpDesc>,
|
|
op_dir: &OpDir,
|
|
) -> Option<OpDesc> {
|
|
fetch_op_spec_from_existing(name, 2, spec, op_dir)
|
|
.or_else(|| fetch_op_spec_from_existing(name, 1, spec, op_dir))
|
|
}
|
|
|
|
pub(crate) fn fetch_op_spec_from_existing(
|
|
name: Atom,
|
|
arity: usize,
|
|
op_desc: Option<OpDesc>,
|
|
op_dir: &OpDir,
|
|
) -> Option<OpDesc> {
|
|
if let Some(ref op_desc) = &op_desc {
|
|
if op_desc.arity() != arity {
|
|
/* it's possible to extend operator functors with
|
|
* additional terms. When that happens,
|
|
* void the op_spec by returning None. */
|
|
return None;
|
|
}
|
|
}
|
|
|
|
op_desc.or_else(|| fetch_op_spec(name, arity, op_dir))
|
|
}
|
|
|
|
pub(crate) fn fetch_op_spec(name: Atom, arity: usize, op_dir: &OpDir) -> Option<OpDesc> {
|
|
match arity {
|
|
2 => op_dir.get(&(name, Fixity::In)).and_then(|op_desc| {
|
|
if op_desc.get_prec() > 0 {
|
|
Some(*op_desc)
|
|
} else {
|
|
None
|
|
}
|
|
}),
|
|
1 => {
|
|
if let Some(op_desc) = op_dir.get(&(name.clone(), Fixity::Pre)) {
|
|
if op_desc.get_prec() > 0 {
|
|
return Some(*op_desc);
|
|
}
|
|
}
|
|
|
|
op_dir.get(&(name, Fixity::Post)).and_then(|op_desc| {
|
|
if op_desc.get_prec() > 0 {
|
|
Some(*op_desc)
|
|
} else {
|
|
None
|
|
}
|
|
})
|
|
}
|
|
0 => {
|
|
fetch_atom_op_spec(name, None, op_dir)
|
|
}
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub(crate) type ModuleDir = IndexMap<Atom, Module, FxBuildHasher>;
|
|
|
|
#[derive(Debug, Clone, Eq, Hash, PartialEq)]
|
|
pub enum ModuleExport {
|
|
OpDecl(OpDecl),
|
|
PredicateKey(PredicateKey),
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct ModuleDecl {
|
|
pub(crate) name: Atom,
|
|
pub(crate) exports: Vec<ModuleExport>,
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub struct Module {
|
|
pub(crate) module_decl: ModuleDecl,
|
|
pub(crate) code_dir: CodeDir,
|
|
pub(crate) op_dir: OpDir,
|
|
pub(crate) meta_predicates: MetaPredicateDir,
|
|
pub(crate) extensible_predicates: ExtensiblePredicates,
|
|
pub(crate) local_extensible_predicates: LocalExtensiblePredicates,
|
|
pub(crate) listing_src: ListingSource,
|
|
}
|
|
|
|
// Module's and related types are defined in forms.
|
|
impl Module {
|
|
pub(crate) fn new(
|
|
module_decl: ModuleDecl,
|
|
listing_src: ListingSource,
|
|
) -> Self {
|
|
Module {
|
|
module_decl,
|
|
code_dir: CodeDir::with_hasher(FxBuildHasher::default()),
|
|
op_dir: default_op_dir(),
|
|
meta_predicates: MetaPredicateDir::with_hasher(FxBuildHasher::default()),
|
|
extensible_predicates: ExtensiblePredicates::with_hasher(FxBuildHasher::default()),
|
|
local_extensible_predicates: LocalExtensiblePredicates::with_hasher(
|
|
FxBuildHasher::default(),
|
|
),
|
|
listing_src,
|
|
}
|
|
}
|
|
|
|
pub(crate) fn new_in_situ(module_decl: ModuleDecl) -> Self {
|
|
Module {
|
|
module_decl,
|
|
code_dir: CodeDir::with_hasher(FxBuildHasher::default()),
|
|
op_dir: OpDir::with_hasher(FxBuildHasher::default()),
|
|
meta_predicates: MetaPredicateDir::with_hasher(FxBuildHasher::default()),
|
|
extensible_predicates: ExtensiblePredicates::with_hasher(FxBuildHasher::default()),
|
|
local_extensible_predicates: LocalExtensiblePredicates::with_hasher(
|
|
FxBuildHasher::default()
|
|
),
|
|
listing_src: ListingSource::DynamicallyGenerated,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Copy, Clone)]
|
|
pub enum Number {
|
|
Float(OrderedFloat<f64>),
|
|
Integer(TypedArenaPtr<Integer>),
|
|
Rational(TypedArenaPtr<Rational>),
|
|
Fixnum(Fixnum),
|
|
}
|
|
|
|
impl Default for Number {
|
|
fn default() -> Self {
|
|
Number::Fixnum(Fixnum::build_with(0))
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for Number {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
match self {
|
|
Number::Float(fl) => write!(f, "{}", fl),
|
|
Number::Integer(n) => write!(f, "{}", n),
|
|
Number::Rational(r) => write!(f, "{}", r),
|
|
Number::Fixnum(n) => write!(f, "{}", n.get_num()),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub trait ArenaFrom<T> {
|
|
fn arena_from(value: T, arena: &mut Arena) -> Self;
|
|
}
|
|
|
|
impl ArenaFrom<Integer> for Number {
|
|
#[inline]
|
|
fn arena_from(value: Integer, arena: &mut Arena) -> Number {
|
|
Number::Integer(arena_alloc!(value, arena))
|
|
}
|
|
}
|
|
|
|
impl ArenaFrom<Rational> for Number {
|
|
#[inline]
|
|
fn arena_from(value: Rational, arena: &mut Arena) -> Number {
|
|
Number::Rational(arena_alloc!(value, arena))
|
|
}
|
|
}
|
|
|
|
impl ArenaFrom<usize> for Number {
|
|
#[inline]
|
|
fn arena_from(value: usize, arena: &mut Arena) -> Number {
|
|
match i64::try_from(value) {
|
|
Ok(value) => Fixnum::build_with_checked(value)
|
|
.map(Number::Fixnum)
|
|
.unwrap_or_else(|_| Number::Integer(arena_alloc!(Integer::from(value), arena))),
|
|
Err(_) => Number::Integer(arena_alloc!(Integer::from(value), arena)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ArenaFrom<u64> for Number {
|
|
#[inline]
|
|
fn arena_from(value: u64, arena: &mut Arena) -> Number {
|
|
match i64::try_from(value) {
|
|
Ok(value) => Fixnum::build_with_checked(value)
|
|
.map(Number::Fixnum)
|
|
.unwrap_or_else(|_| Number::Integer(arena_alloc!(Integer::from(value), arena))),
|
|
Err(_) => Number::Integer(arena_alloc!(Integer::from(value), arena)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ArenaFrom<i64> for Number {
|
|
#[inline]
|
|
fn arena_from(value: i64, arena: &mut Arena) -> Number {
|
|
Fixnum::build_with_checked(value)
|
|
.map(Number::Fixnum)
|
|
.unwrap_or_else(|_| Number::Integer(arena_alloc!(Integer::from(value), arena)))
|
|
}
|
|
}
|
|
|
|
impl ArenaFrom<isize> for Number {
|
|
#[inline]
|
|
fn arena_from(value: isize, arena: &mut Arena) -> Number {
|
|
Fixnum::build_with_checked(value as i64)
|
|
.map(Number::Fixnum)
|
|
.unwrap_or_else(|_| Number::Integer(arena_alloc!(Integer::from(value), arena)))
|
|
}
|
|
}
|
|
|
|
impl ArenaFrom<u32> for Number {
|
|
#[inline]
|
|
fn arena_from(value: u32, _arena: &mut Arena) -> Number {
|
|
Number::Fixnum(Fixnum::build_with(value as i64))
|
|
}
|
|
}
|
|
|
|
impl ArenaFrom<i32> for Number {
|
|
#[inline]
|
|
fn arena_from(value: i32, _arena: &mut Arena) -> Number {
|
|
Number::Fixnum(Fixnum::build_with(value as i64))
|
|
}
|
|
}
|
|
|
|
impl ArenaFrom<Number> for Literal {
|
|
#[inline]
|
|
fn arena_from(value: Number, arena: &mut Arena) -> Literal {
|
|
match value {
|
|
Number::Fixnum(n) => Literal::Fixnum(n),
|
|
Number::Integer(n) => Literal::Integer(n),
|
|
Number::Float(f) => Literal::Float(arena_alloc!(f, arena)),
|
|
Number::Rational(r) => Literal::Rational(r),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ArenaFrom<Number> for HeapCellValue {
|
|
#[inline]
|
|
fn arena_from(value: Number, arena: &mut Arena) -> HeapCellValue {
|
|
match value {
|
|
Number::Fixnum(n) => fixnum_as_cell!(n),
|
|
Number::Integer(n) => typed_arena_ptr_as_cell!(n),
|
|
Number::Float(n) => typed_arena_ptr_as_cell!(arena_alloc!(n, arena)),
|
|
Number::Rational(n) => typed_arena_ptr_as_cell!(n),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Number {
|
|
#[inline]
|
|
pub(crate) fn is_positive(&self) -> bool {
|
|
match self {
|
|
&Number::Fixnum(n) => n.get_num() > 0,
|
|
&Number::Integer(ref n) => &**n > &0,
|
|
&Number::Float(f) => f.is_sign_positive(),
|
|
&Number::Rational(ref r) => &**r > &0,
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn is_negative(&self) -> bool {
|
|
match self {
|
|
&Number::Fixnum(n) => n.get_num() < 0,
|
|
&Number::Integer(ref n) => &**n < &0,
|
|
&Number::Float(OrderedFloat(f)) => f.is_sign_negative(),
|
|
&Number::Rational(ref r) => &**r < &0,
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn is_zero(&self) -> bool {
|
|
match self {
|
|
&Number::Fixnum(n) => n.get_num() == 0,
|
|
&Number::Integer(ref n) => &**n == &0,
|
|
&Number::Float(f) => f == OrderedFloat(0f64),
|
|
&Number::Rational(ref r) => &**r == &0,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub(crate) enum OptArgIndexKey {
|
|
Literal(usize, usize, Literal, Vec<Literal>), // index, IndexingCode location, opt arg, alternatives
|
|
List(usize, usize), // index, IndexingCode location
|
|
None,
|
|
Structure(usize, usize, Atom, usize), // index, IndexingCode location, name, arity
|
|
}
|
|
|
|
impl OptArgIndexKey {
|
|
#[inline]
|
|
pub(crate) fn take(&mut self) -> OptArgIndexKey {
|
|
std::mem::replace(self, OptArgIndexKey::None)
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn arg_num(&self) -> usize {
|
|
match &self {
|
|
OptArgIndexKey::Literal(arg_num, ..)
|
|
| OptArgIndexKey::Structure(arg_num, ..)
|
|
| OptArgIndexKey::List(arg_num, _) => {
|
|
// these are always at least 1.
|
|
*arg_num
|
|
}
|
|
OptArgIndexKey::None => 0,
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn is_some(&self) -> bool {
|
|
self.switch_on_term_loc().is_some()
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn switch_on_term_loc(&self) -> Option<usize> {
|
|
match &self {
|
|
OptArgIndexKey::Literal(_, loc, ..)
|
|
| OptArgIndexKey::Structure(_, loc, ..)
|
|
| OptArgIndexKey::List(_, loc) => Some(*loc),
|
|
OptArgIndexKey::None => None,
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn set_switch_on_term_loc(&mut self, value: usize) {
|
|
match self {
|
|
OptArgIndexKey::Literal(_, ref mut loc, ..)
|
|
| OptArgIndexKey::Structure(_, ref mut loc, ..)
|
|
| OptArgIndexKey::List(_, ref mut loc) => {
|
|
*loc = value;
|
|
}
|
|
OptArgIndexKey::None => {}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl AddAssign<usize> for OptArgIndexKey {
|
|
#[inline]
|
|
fn add_assign(&mut self, n: usize) {
|
|
match self {
|
|
OptArgIndexKey::Literal(_, ref mut o, ..)
|
|
| OptArgIndexKey::List(_, ref mut o)
|
|
| OptArgIndexKey::Structure(_, ref mut o, ..) => {
|
|
*o += n;
|
|
}
|
|
OptArgIndexKey::None => {}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub(crate) struct ClauseIndexInfo {
|
|
pub(crate) clause_start: usize,
|
|
pub(crate) opt_arg_index_key: OptArgIndexKey,
|
|
}
|
|
|
|
impl ClauseIndexInfo {
|
|
#[inline]
|
|
pub(crate) fn new(clause_start: usize) -> Self {
|
|
Self {
|
|
clause_start,
|
|
opt_arg_index_key: OptArgIndexKey::None,
|
|
// index_locs: vec![],
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Copy, Debug)]
|
|
pub(crate) struct PredicateInfo {
|
|
pub(crate) is_extensible: bool,
|
|
pub(crate) is_discontiguous: bool,
|
|
pub(crate) is_dynamic: bool,
|
|
pub(crate) is_multifile: bool,
|
|
pub(crate) has_clauses: bool,
|
|
}
|
|
|
|
impl Default for PredicateInfo {
|
|
#[inline]
|
|
fn default() -> Self {
|
|
PredicateInfo {
|
|
is_extensible: false,
|
|
is_discontiguous: false,
|
|
is_dynamic: false,
|
|
is_multifile: false,
|
|
has_clauses: false,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl PredicateInfo {
|
|
#[inline]
|
|
pub(crate) fn compile_incrementally(&self) -> bool {
|
|
let base = self.is_extensible && self.has_clauses;
|
|
base && (self.is_discontiguous || self.is_multifile)
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn must_retract_local_clauses(&self) -> bool {
|
|
self.is_extensible && self.has_clauses && !self.is_discontiguous
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub(crate) struct LocalPredicateSkeleton {
|
|
pub(crate) is_discontiguous: bool,
|
|
pub(crate) is_dynamic: bool,
|
|
pub(crate) is_multifile: bool,
|
|
pub(crate) clause_clause_locs: SliceDeque<usize>,
|
|
pub(crate) clause_assert_margin: usize,
|
|
pub(crate) retracted_dynamic_clauses: Option<Vec<ClauseIndexInfo>>, // always None if non-dynamic.
|
|
}
|
|
|
|
impl LocalPredicateSkeleton {
|
|
#[inline]
|
|
pub(crate) fn new() -> Self {
|
|
Self {
|
|
is_discontiguous: false,
|
|
is_dynamic: false,
|
|
is_multifile: false,
|
|
clause_clause_locs: sdeq![],
|
|
clause_assert_margin: 0,
|
|
retracted_dynamic_clauses: Some(vec![]),
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn predicate_info(&self) -> PredicateInfo {
|
|
PredicateInfo {
|
|
is_extensible: true,
|
|
is_discontiguous: self.is_discontiguous,
|
|
is_dynamic: self.is_dynamic,
|
|
is_multifile: self.is_multifile,
|
|
has_clauses: !self.clause_clause_locs.is_empty(),
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn reset(&mut self) {
|
|
self.clause_clause_locs.clear();
|
|
self.clause_assert_margin = 0;
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn add_retracted_dynamic_clause_info(&mut self, clause_info: ClauseIndexInfo) {
|
|
debug_assert_eq!(self.is_dynamic, true);
|
|
|
|
if self.retracted_dynamic_clauses.is_none() {
|
|
self.retracted_dynamic_clauses = Some(vec![]);
|
|
}
|
|
|
|
self.retracted_dynamic_clauses
|
|
.as_mut()
|
|
.unwrap()
|
|
.push(clause_info);
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub(crate) struct PredicateSkeleton {
|
|
pub(crate) core: LocalPredicateSkeleton,
|
|
pub(crate) clauses: SliceDeque<ClauseIndexInfo>,
|
|
}
|
|
|
|
impl PredicateSkeleton {
|
|
#[inline]
|
|
pub(crate) fn new() -> Self {
|
|
PredicateSkeleton {
|
|
core: LocalPredicateSkeleton::new(),
|
|
clauses: sdeq![],
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub(crate) fn predicate_info(&self) -> PredicateInfo {
|
|
PredicateInfo {
|
|
is_extensible: true,
|
|
is_discontiguous: self.core.is_discontiguous,
|
|
is_dynamic: self.core.is_dynamic,
|
|
is_multifile: self.core.is_multifile,
|
|
has_clauses: !self.clauses.is_empty(),
|
|
}
|
|
}
|
|
|
|
pub(crate) fn target_pos_of_clause_clause_loc(
|
|
&self,
|
|
clause_clause_loc: usize,
|
|
) -> Option<usize> {
|
|
let search_result = self.core.clause_clause_locs[0..self.core.clause_assert_margin]
|
|
.binary_search_by(|loc| clause_clause_loc.cmp(&loc));
|
|
|
|
match search_result {
|
|
Ok(loc) => Some(loc),
|
|
Err(_) => self.core.clause_clause_locs[self.core.clause_assert_margin..]
|
|
.binary_search_by(|loc| loc.cmp(&clause_clause_loc))
|
|
.map(|loc| loc + self.core.clause_assert_margin)
|
|
.ok(),
|
|
}
|
|
}
|
|
}
|