expose more internals

egraphs-good · Nov 18, 2024 · bdf7a46 · bdf7a46
1 parent 6bd6b17
commit bdf7a46
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/function/mod.rs b/src/function/mod.rs
@@ -15,7 +15,7 @@ pub struct Function {
     pub(crate) decl: ResolvedFunctionDecl,
     pub schema: ResolvedSchema,
     pub merge: MergeAction,
-    pub(crate) nodes: table::Table,
+    pub nodes: table::Table,
     sorts: HashSet<Symbol>,
     pub(crate) indexes: Vec<Rc<ColumnIndex>>,
     pub(crate) rebuild_indexes: Vec<Option<CompositeColumnIndex>>,
@@ -491,4 +491,11 @@ impl Function {
     pub fn is_extractable(&self) -> bool {
         !self.decl.unextractable
     }
+
+    pub fn iter(
+        &self,
+        include_subsumed: bool,
+    ) -> impl Iterator<Item = (&[Value], &TupleOutput)> + '_ {
+        self.nodes.iter(include_subsumed)
+    }
 }
diff --git a/src/function/table.rs b/src/function/table.rs
@@ -48,7 +48,7 @@ struct TableOffset {
 }
 
 #[derive(Default, Clone)]
-pub(crate) struct Table {
+pub struct Table {
     max_ts: u32,
     n_stale: usize,
     table: RawTable<TableOffset>,
@@ -262,7 +262,7 @@ impl Table {
     }
 
     /// Iterate over the live entries in the table, in insertion order.
-    pub(crate) fn iter(
+    pub fn iter(
         &self,
         include_subsumed: bool,
     ) -> impl Iterator<Item = (&[Value], &TupleOutput)> + '_ {