1 files changed, 14 insertions, 0 deletions
diff --git a/src/theory/bags/solver_state.h b/src/theory/bags/solver_state.h
index 9a4bfdae7..8d70ee8f7 100644
--- a/src/theory/bags/solver_state.h
+++ b/src/theory/bags/solver_state.h
@@ -31,10 +31,24 @@ class SolverState : public TheoryState
  public:
   SolverState(context::Context* c, context::UserContext* u, Valuation val);
 
+  void registerClass(TNode n);
+
+  Node registerBagElement(TNode n);
+
+  std::set<Node>& getBags();
+
+  std::set<Node>& getElements(TypeNode t);
+
+  std::map<Node, Node>& getBagElements(Node B);
+
  private:
   /** constants */
   Node d_true;
   Node d_false;
+  std::set<Node> d_bags;
+  std::map<TypeNode, std::set<Node>> d_elements;
+  /** bag -> element -> multiplicity */
+  std::map<Node, std::map<Node, Node>> d_count;
 }; /* class SolverState */
 
 }  // namespace bags